forked from p34709852/monkey
Add unit tests for string list encryptor
This commit is contained in:
parent
a1c0af4257
commit
88f3a2b9ca
|
@ -0,0 +1,24 @@
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from monkey_island.cc.models.utils.field_encryptors.string_list_encryptor import StringListEncryptor
|
||||||
|
from monkey_island.cc.server_utils.encryptor import initialize_encryptor
|
||||||
|
|
||||||
|
MOCK_STRING_LIST = ["test_1", "test_2"]
|
||||||
|
EMPTY_LIST = []
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def uses_encryptor(data_for_tests_dir):
|
||||||
|
initialize_encryptor(data_for_tests_dir)
|
||||||
|
|
||||||
|
|
||||||
|
def test_string_list_encryptor(uses_encryptor):
|
||||||
|
encrypted_list = StringListEncryptor.encrypt(MOCK_STRING_LIST)
|
||||||
|
assert not encrypted_list == MOCK_STRING_LIST
|
||||||
|
decrypted_list = StringListEncryptor.decrypt(encrypted_list)
|
||||||
|
assert decrypted_list == MOCK_STRING_LIST
|
||||||
|
|
||||||
|
|
||||||
|
def test_string_list_encryptor__empty_list(uses_encryptor):
|
||||||
|
encrypted_list = StringListEncryptor.encrypt(EMPTY_LIST)
|
||||||
|
StringListEncryptor.decrypt(encrypted_list)
|
Loading…
Reference in New Issue