p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Tests: Reduce code duplication in test_data_store_encryptor.py

This commit is contained in:
Mike Salvatore 2021-10-06 12:45:54 -04:00
parent 8310204e66
commit 9ee00c3044
1 changed files with 14 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
monkey/tests/unit_tests/monkey_island/cc/server_utils/encryption/test_data_store_encryptor.py
View File

@ -14,8 +14,6 @@ pytestmark = pytest.mark.slow
PLAINTEXT = "Hello, Monkey!" PLAINTEXT = "Hello, Monkey!"
MOCK_SECRET = "53CR31" MOCK_SECRET = "53CR31"
KEY_FILENAME = "test_key.bin"
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def cleanup_encryptor(): def cleanup_encryptor():
@ -25,11 +23,11 @@ def cleanup_encryptor():
@pytest.fixture @pytest.fixture
def key_file(tmp_path): def key_file(tmp_path):
return tmp_path / KEY_FILENAME return tmp_path / "test_key.bin"
def test_encryption(tmp_path): def test_encryption(tmp_path):
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(tmp_path, MOCK_SECRET)
encrypted_data = get_datastore_encryptor().encrypt(PLAINTEXT) encrypted_data = get_datastore_encryptor().encrypt(PLAINTEXT)
assert encrypted_data != PLAINTEXT assert encrypted_data != PLAINTEXT
@ -38,35 +36,35 @@ def test_encryption(tmp_path):
assert decrypted_data == PLAINTEXT assert decrypted_data == PLAINTEXT
def test_key_creation(key_file, tmp_path): def test_key_creation(key_file):
assert not key_file.is_file() assert not key_file.is_file()
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
assert key_file.is_file() assert key_file.is_file()
def test_existing_key_reused(key_file, tmp_path): def test_existing_key_reused(key_file):
assert not key_file.is_file() assert not key_file.is_file()
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
key_file_hash_1 = get_file_sha256_hash(key_file) key_file_hash_1 = get_file_sha256_hash(key_file)
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
key_file_hash_2 = get_file_sha256_hash(key_file) key_file_hash_2 = get_file_sha256_hash(key_file)
assert key_file_hash_1 == key_file_hash_2 assert key_file_hash_1 == key_file_hash_2
def test_key_removal(key_file, tmp_path): def test_key_removal(key_file):
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
assert key_file.is_file() assert key_file.is_file()
get_datastore_encryptor().erase_key() get_datastore_encryptor().erase_key()
assert not key_file.is_file() assert not key_file.is_file()
def test_key_removal__no_key(key_file, tmp_path): def test_key_removal__no_key(key_file):
assert not key_file.is_file() assert not key_file.is_file()
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
assert key_file.is_file() assert key_file.is_file()
get_datastore_encryptor().erase_key() get_datastore_encryptor().erase_key()
@ -76,11 +74,11 @@ def test_key_removal__no_key(key_file, tmp_path):
get_datastore_encryptor().erase_key() get_datastore_encryptor().erase_key()
def test_reinitialize_datastore_encryptor(key_file, tmp_path): def test_reinitialize_datastore_encryptor(key_file):
initialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) initialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
key_file_hash_1 = get_file_sha256_hash(key_file) key_file_hash_1 = get_file_sha256_hash(key_file)
reinitialize_datastore_encryptor(tmp_path, MOCK_SECRET, KEY_FILENAME) reinitialize_datastore_encryptor(key_file.parent, MOCK_SECRET, key_file.name)
key_file_hash_2 = get_file_sha256_hash(key_file) key_file_hash_2 = get_file_sha256_hash(key_file)
assert key_file_hash_1 != key_file_hash_2 assert key_file_hash_1 != key_file_hash_2