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

UT: Use Credentials object instead of dict in MongoCredentialsRepository

This commit is contained in:
Ilija Lazoroski 2022-07-11 15:03:37 +02:00
parent 14cbf07a21
commit 4dd4b38fc9
1 changed files with 16 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_credentials_repository.py
View File

@ -1,7 +1,7 @@
import mongomock import mongomock
import pytest import pytest
from common.credentials import Credentials from common.credentials import Credentials, LMHash, NTHash, Password, SSHKeypair, Username
from monkey_island.cc.repository import MongoCredentialsRepository from monkey_island.cc.repository import MongoCredentialsRepository
USER1 = "test_user_1" USER1 = "test_user_1"
@ -15,41 +15,25 @@ NT_HASH = "7A21990FCD3D759941E45C490F143D5F"
PUBLIC_KEY = "MY_PUBLIC_KEY" PUBLIC_KEY = "MY_PUBLIC_KEY"
PRIVATE_KEY = "MY_PRIVATE_KEY" PRIVATE_KEY = "MY_PRIVATE_KEY"
CREDENTIALS_DICT_1 = { IDENTITIES_1 = (Username(USER1), Username(USER2))
"identities": [ SECRETS_1 = (
{"credential_type": "USERNAME", "username": USER1}, Password(PASSWORD),
{"credential_type": "USERNAME", "username": USER2}, LMHash(LM_HASH),
], NTHash(NT_HASH),
"secrets": [ SSHKeypair(PRIVATE_KEY, PUBLIC_KEY),
{"credential_type": "PASSWORD", "password": PASSWORD}, )
{"credential_type": "LM_HASH", "lm_hash": LM_HASH}, CREDENTIALS_OBJECT_1 = Credentials(IDENTITIES_1, SECRETS_1)
{"credential_type": "NT_HASH", "nt_hash": NT_HASH},
{
"credential_type": "SSH_KEYPAIR",
"public_key": PUBLIC_KEY,
"private_key": PRIVATE_KEY,
},
],
}
CREDENTIALS_DICT_2 = { IDENTITIES_2 = (Username(USER3),)
"identities": [ SECRETS_2 = (Password(PASSWORD2), Password(PASSWORD3))
{"credential_type": "USERNAME", "username": USER3}, CREDENTIALS_OBJECT_2 = Credentials(IDENTITIES_2, SECRETS_2)
],
"secrets": [
{"credential_type": "PASSWORD", "password": PASSWORD2},
{"credential_type": "PASSWORD", "password": PASSWORD3},
],
}
CONFIGURED_CREDENTIALS = [Credentials.from_mapping(CREDENTIALS_DICT_1)]
STOLEN_CREDENTIALS = [Credentials.from_mapping(CREDENTIALS_DICT_2)] CONFIGURED_CREDENTIALS = [CREDENTIALS_OBJECT_1]
CREDENTIALS_LIST = [ STOLEN_CREDENTIALS = [CREDENTIALS_OBJECT_2]
Credentials.from_mapping(CREDENTIALS_DICT_1),
Credentials.from_mapping(CREDENTIALS_DICT_2), CREDENTIALS_LIST = [CREDENTIALS_OBJECT_1, CREDENTIALS_OBJECT_2]
]
@pytest.fixture @pytest.fixture