48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from data.encryption import FieldEncrypter, _VERSIONS, DecryptionFailureException
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('test_data', [
|
||
|
|
'',
|
||
|
|
'hello world',
|
||
|
|
'wassup?!',
|
||
|
|
'IGZ2Y8KUN3EFWAZZXR3D7U4V5NXDVYZI5VGU6STPB6KM83PAB8WRGM32RD9FW0C0',
|
||
|
|
'JLRFBYS1EHKUE73S99HWOQWNPGLUZTBRF5HQEFUJS5BK3XVB54RNXYV4AUMJXCMC',
|
||
|
|
'a' * 3,
|
||
|
|
'a' * 4,
|
||
|
|
'a' * 5,
|
||
|
|
'a' * 31,
|
||
|
|
'a' * 32,
|
||
|
|
'a' * 33,
|
||
|
|
'a' * 150,
|
||
|
|
u'😇',
|
||
|
|
])
|
||
|
|
@pytest.mark.parametrize('version', _VERSIONS.keys())
|
||
|
|
@pytest.mark.parametrize('secret_key', [
|
||
|
|
u'test1234',
|
||
|
|
'test1234',
|
||
|
|
'thisisanothercoolsecretkeyhere',
|
||
|
|
'107383705745765174750346070528443780244192102846031525796571939503548634055845',
|
||
|
|
])
|
||
|
|
@pytest.mark.parametrize('use_valid_key', [
|
||
|
|
True,
|
||
|
|
False,
|
||
|
|
])
|
||
|
|
def test_encryption(test_data, version, secret_key, use_valid_key):
|
||
|
|
encrypter = FieldEncrypter(secret_key, version)
|
||
|
|
encrypted = encrypter.encrypt_value(test_data, field_max_length=255)
|
||
|
|
assert encrypted != test_data
|
||
|
|
|
||
|
|
if use_valid_key:
|
||
|
|
decrypted = encrypter.decrypt_value(encrypted)
|
||
|
|
assert decrypted == test_data
|
||
|
|
|
||
|
|
with pytest.raises(DecryptionFailureException):
|
||
|
|
encrypter.decrypt_value('somerandomvalue')
|
||
|
|
else:
|
||
|
|
decrypter = FieldEncrypter('some other key', version)
|
||
|
|
with pytest.raises(DecryptionFailureException):
|
||
|
|
decrypter.decrypt_value(encrypted)
|