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

Tests: Test InvalidCiphertextError

This commit is contained in:
Mike Salvatore 2021-10-06 09:51:03 -04:00
parent e673667b34
commit 8310204e66
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/tests/unit_tests/monkey_island/cc/server_utils/encryption/test_password_based_encryption.py
View File

@ -5,39 +5,45 @@ from tests.unit_tests.monkey_island.cc.services.utils.ciphertexts_for_encryption
) )
from monkey_island.cc.server_utils.encryption import ( from monkey_island.cc.server_utils.encryption import (
InvalidCiphertextError,
InvalidCredentialsError, InvalidCredentialsError,
PasswordBasedStringEncryptor, PasswordBasedStringEncryptor,
) )
# Mark all tests in this module as slow
pytestmark = pytest.mark.slow
MONKEY_CONFIGS_DIR_PATH = "monkey_configs" MONKEY_CONFIGS_DIR_PATH = "monkey_configs"
STANDARD_PLAINTEXT_MONKEY_CONFIG_FILENAME = "monkey_config_standard.json" STANDARD_PLAINTEXT_MONKEY_CONFIG_FILENAME = "monkey_config_standard.json"
PASSWORD = "hello123" PASSWORD = "hello123"
INCORRECT_PASSWORD = "goodbye321" INCORRECT_PASSWORD = "goodbye321"
@pytest.mark.slow
def test_encrypt_decrypt_string(monkey_config_json): def test_encrypt_decrypt_string(monkey_config_json):
pb_encryptor = PasswordBasedStringEncryptor(PASSWORD) pb_encryptor = PasswordBasedStringEncryptor(PASSWORD)
encrypted_config = pb_encryptor.encrypt(monkey_config_json) encrypted_config = pb_encryptor.encrypt(monkey_config_json)
assert pb_encryptor.decrypt(encrypted_config) == monkey_config_json assert pb_encryptor.decrypt(encrypted_config) == monkey_config_json
@pytest.mark.slow
def test_decrypt_string__wrong_password(monkey_config_json): def test_decrypt_string__wrong_password(monkey_config_json):
pb_encryptor = PasswordBasedStringEncryptor(INCORRECT_PASSWORD) pb_encryptor = PasswordBasedStringEncryptor(INCORRECT_PASSWORD)
with pytest.raises(InvalidCredentialsError): with pytest.raises(InvalidCredentialsError):
pb_encryptor.decrypt(VALID_CIPHER_TEXT) pb_encryptor.decrypt(VALID_CIPHER_TEXT)
@pytest.mark.slow
def test_decrypt_string__malformed_corrupted(): def test_decrypt_string__malformed_corrupted():
pb_encryptor = PasswordBasedStringEncryptor(PASSWORD) pb_encryptor = PasswordBasedStringEncryptor(PASSWORD)
with pytest.raises(ValueError): with pytest.raises(ValueError):
pb_encryptor.decrypt(MALFORMED_CIPHER_TEXT_CORRUPTED) pb_encryptor.decrypt(MALFORMED_CIPHER_TEXT_CORRUPTED)
@pytest.mark.slow
def test_decrypt_string__no_password(monkey_config_json): def test_decrypt_string__no_password(monkey_config_json):
pb_encryptor = PasswordBasedStringEncryptor("") pb_encryptor = PasswordBasedStringEncryptor("")
with pytest.raises(InvalidCredentialsError): with pytest.raises(InvalidCredentialsError):
pb_encryptor.decrypt(VALID_CIPHER_TEXT) pb_encryptor.decrypt(VALID_CIPHER_TEXT)
def test_decrypt_string__invalid_cyphertext(monkey_config_json):
pb_encryptor = PasswordBasedStringEncryptor("")
with pytest.raises(InvalidCiphertextError):
pb_encryptor.decrypt("")