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

UT: Use new Credentials object in test_credential_telem_send()

This commit is contained in:
Mike Salvatore 2022-07-14 13:11:45 -04:00
parent 2b245b34cb
commit 04d72c0d36
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
monkey/tests/unit_tests/infection_monkey/telemetry/test_credentials_telem.py
View File

@ -2,7 +2,7 @@ import json
import pytest import pytest
from common.credentials import Credentials, Password, SSHKeypair, Username from common.credentials import Credentials, Password, Username
from infection_monkey.telemetry.credentials_telem import CredentialsTelem from infection_monkey.telemetry.credentials_telem import CredentialsTelem
USERNAME = "m0nkey" USERNAME = "m0nkey"
@ -14,24 +14,15 @@ PRIVATE_KEY = "priv_key"
@pytest.fixture @pytest.fixture
def credentials_for_test(): def credentials_for_test():
return Credentials( return Credentials(Username(USERNAME), Password(PASSWORD))
[Username(USERNAME)], [Password(PASSWORD), SSHKeypair(PRIVATE_KEY, PUBLIC_KEY)]
)
def test_credential_telem_send(spy_send_telemetry, credentials_for_test): def test_credential_telem_send(spy_send_telemetry, credentials_for_test):
expected_data = [ expected_data = [
{ {
"identities": [{"username": USERNAME, "credential_type": "USERNAME"}], "identity": {"username": USERNAME, "credential_type": "USERNAME"},
"secrets": [ "secret": {"password": PASSWORD, "credential_type": "PASSWORD"},
{"password": PASSWORD, "credential_type": "PASSWORD"},
{
"private_key": PRIVATE_KEY,
"public_key": PUBLIC_KEY,
"credential_type": "SSH_KEYPAIR",
},
],
} }
] ]