diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_credential_processing.py b/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_credential_processing.py index 94874cd4a..2ad17431d 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_credential_processing.py +++ b/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_credential_processing.py @@ -27,7 +27,7 @@ fake_special_username = "$m0nk3y.user" cred_telem_special_usernames = deepcopy(CREDENTIAL_TELEM_TEMPLATE) cred_telem_special_usernames["data"] = [ { - "identities": [{"username": fake_special_username, "credential_type": "username"}], + "identities": [{"username": fake_special_username, "credential_type": "USERNAME"}], "secrets": [], } ] diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_ssh_key_processing.py b/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_ssh_key_processing.py index b06d7cd3d..2d012c4ee 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_ssh_key_processing.py +++ b/monkey/tests/unit_tests/monkey_island/cc/services/telemetry/processing/credentials/test_ssh_key_processing.py @@ -17,7 +17,7 @@ fake_monkey_guid = "272405690278083" fake_ip_address = "192.168.56.1" fake_private_key = "-----BEGIN OPENSSH PRIVATE KEY-----\nb3BlbnNzaC1rZXktdjEAAAAACmFlczI1N\n" -fake_partial_secret = {"private_key": fake_private_key, "credential_type": "ssh_keypair"} +fake_partial_secret = {"private_key": fake_private_key, "credential_type": "SSH_KEYPAIR"} fake_username = "ubuntu" fake_public_key = ( @@ -27,9 +27,9 @@ fake_public_key = ( fake_secret_full = { "private_key": fake_private_key, "public_key": fake_public_key, - "credential_type": "ssh_keypair", + "credential_type": "SSH_KEYPAIR", } -fake_identity = {"username": fake_username, "credential_type": "username"} +fake_identity = {"username": fake_username, "credential_type": "USERNAME"} ssh_telem = deepcopy(CREDENTIAL_TELEM_TEMPLATE) ssh_telem["data"] = [{"identities": [fake_identity], "secrets": [fake_secret_full]}]