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

Island: Modify ICredentialsRepository to save a Sequence of Credentials

This commit is contained in:
Mike Salvatore 2022-07-07 13:06:38 -04:00
parent 68fe2edd08
commit 7bcf49daf9
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/monkey_island/cc/repository/i_credentials_repository.py
View File

@ -43,7 +43,7 @@ class ICredentialsRepository(ABC):
"""
pass
def save_configured_credentials(self, credentials: Credentials):
def save_configured_credentials(self, credentials: Sequence[Credentials]):
"""
Save credentials that were configured.
@ -52,7 +52,7 @@ class ICredentialsRepository(ABC):
"""
pass
def save_stolen_credentials(self, credentials: Credentials):
def save_stolen_credentials(self, credentials: Sequence[Credentials]):
"""
Save credentials that were stolen during a simulation.

8
monkey/monkey_island/cc/repository/mongo_credentials_repository.py
View File

@ -34,15 +34,15 @@ class MongoCredentialsRepository(ICredentialsRepository):
except RetrievalError as err:
raise err
def save_configured_credentials(self, credentials: Credentials):
def save_configured_credentials(self, credentials: Sequence[Credentials]):
try:
mongo.db.configured_credentials.insert_one(credentials)
mongo.db.configured_credentials.insert_many(credentials)
except Exception as err:
raise StorageError(err)
def save_stolen_credentials(self, credentials: Credentials):
def save_stolen_credentials(self, credentials: Sequence[Credentials]):
try:
mongo.db.stolen_credentials.insert_one(credentials)
mongo.db.stolen_credentials.insert_many(credentials)
except Exception as err:
raise StorageError(err)

4
monkey/tests/monkey_island/stub_propagation_credentials_repository.py
View File

@ -47,10 +47,10 @@ class StubPropagationCredentialsRepository(ICredentialsRepository):
Credentials.from_mapping(PROPAGATION_CREDENTIALS_2, monkey_guid="second_guid"),
]
def save_configured_credentials(self, credentials: Credentials):
def save_configured_credentials(self, credentials: Sequence[Credentials]):
pass
def save_stolen_credentials(self, credentials: Credentials):
def save_stolen_credentials(self, credentials: Sequence[Credentials]):
pass
def remove_configured_credentials(self):