p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

UT: Add test_all_keys_if_credentials_empty() test

This commit is contained in:
Mike Salvatore 2022-04-21 12:21:33 -04:00
parent e06e855a48
commit a6603c64d6
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
monkey/tests/unit_tests/infection_monkey/credential_store/test_aggregating_credentials_store.py
View File

@ -21,6 +21,14 @@ CONTROL_CHANNEL_CREDENTIALS = {
], ],
} }
EMPTY_CHANNEL_CREDENTIALS = {
"exploit_user_list": [],
"exploit_password_list": [],
"exploit_lm_hash_list": [],
"exploit_ntlm_hash_list": [],
"exploit_ssh_keys": [],
}
TEST_CREDENTIALS = [ TEST_CREDENTIALS = [
Credentials( Credentials(
[Username("user1"), Username("user3")], [Username("user1"), Username("user3")],
@ -52,8 +60,6 @@ def aggregating_credentials_store() -> AggregatingCredentialsStore:
def test_get_credentials_from_store(aggregating_credentials_store): def test_get_credentials_from_store(aggregating_credentials_store):
actual_stored_credentials = aggregating_credentials_store.get_credentials() actual_stored_credentials = aggregating_credentials_store.get_credentials()
print(actual_stored_credentials)
assert actual_stored_credentials["exploit_user_list"] == set( assert actual_stored_credentials["exploit_user_list"] == set(
CONTROL_CHANNEL_CREDENTIALS["exploit_user_list"] CONTROL_CHANNEL_CREDENTIALS["exploit_user_list"]
) )
@ -93,3 +99,17 @@ def test_add_credentials_to_store(aggregating_credentials_store):
) )
assert len(actual_stored_credentials["exploit_ssh_keys"]) == 3 assert len(actual_stored_credentials["exploit_ssh_keys"]) == 3
def test_all_keys_if_credentials_empty():
control_channel = MagicMock()
control_channel.get_credentials_for_propagation.return_value = EMPTY_CHANNEL_CREDENTIALS
credentials_store = AggregatingCredentialsStore(control_channel)
actual_stored_credentials = credentials_store.get_credentials()
print(type(actual_stored_credentials))
assert "exploit_user_list" in actual_stored_credentials
assert "exploit_password_list" in actual_stored_credentials
assert "exploit_ntlm_hash_list" in actual_stored_credentials
assert "exploit_ssh_keys" in actual_stored_credentials