30 lines
987 B
Python
30 lines
987 B
Python
import pytest
|
|
from cryptography_s_des_exploring.sdes.encrypt import encrypt, encrypt_block
|
|
|
|
|
|
@pytest.mark.parametrize("key, message, expected_ciphertext", [
|
|
('1111001110', '11110011', '00100001'),
|
|
('1111001111', '00010011', '11110111'),
|
|
# 0b1101000 = 104 ('h')
|
|
('1111001110', '01101000', '10100011'),
|
|
# 0b1100101 = 101 ('e')
|
|
('1111001110', '01100101', '10000111'),
|
|
# 0b1101100 = 108 ('l')
|
|
('1111001110', '01101100', '11100110'),
|
|
# 0b1101111 = 111 ('o')
|
|
('1111001110', '01101111', '11001110'),
|
|
])
|
|
def test_encrypt_block(key, message, expected_ciphertext):
|
|
result = encrypt_block(key, message)
|
|
assert expected_ciphertext == result
|
|
|
|
|
|
@pytest.mark.parametrize("key, message, expected_ciphertext", [
|
|
('1111001110', 'hello', b'wkwaR1xMpoE='),
|
|
('1111001110', 'привіт', b'nEp+f+C3CRmcSkp/5s1MCQ=='),
|
|
])
|
|
def test_encrypt(key, message, expected_ciphertext):
|
|
result = encrypt(key, message)
|
|
assert expected_ciphertext == result
|
|
|