From 68fe2edd08901162b40f88770d25caa75016e81d Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Thu, 7 Jul 2022 17:34:49 +0200
Subject: [PATCH] Island: Init MongoCredentialsRepository

---
 .../mongo_credentials_repository.py           | 66 +++++++++++++++++++
 .../test_mongo_credentials_repository.py      | 31 +++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 monkey/monkey_island/cc/repository/mongo_credentials_repository.py
 create mode 100644 monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_credentials_repository.py

diff --git a/monkey/monkey_island/cc/repository/mongo_credentials_repository.py b/monkey/monkey_island/cc/repository/mongo_credentials_repository.py
new file mode 100644
index 000000000..c124490a4
--- /dev/null
+++ b/monkey/monkey_island/cc/repository/mongo_credentials_repository.py
@@ -0,0 +1,66 @@
+from typing import Sequence
+
+from common.credentials import Credentials
+from monkey_island.cc.database import mongo
+from monkey_island.cc.repository import RemovalError, RetrievalError, StorageError
+from monkey_island.cc.repository.i_credentials_repository import ICredentialsRepository
+
+
+class MongoCredentialsRepository(ICredentialsRepository):
+    """
+    Store credentials in a mongo database that can be used to propagate around the network.
+    """
+
+    def get_configured_credentials(self) -> Sequence[Credentials]:
+        try:
+            configured_credentials = list(mongo.db.configured_credentials.find({}))
+            return configured_credentials
+        except Exception as err:
+            raise RetrievalError(err)
+
+    def get_stolen_credentials(self) -> Sequence[Credentials]:
+        try:
+            stolen_credentials = list(mongo.db.stolen_credentials.find({}))
+            return stolen_credentials
+        except Exception as err:
+            raise RetrievalError(err)
+
+    def get_all_credentials(self) -> Sequence[Credentials]:
+        try:
+            configured_credentials = self.get_configured_credentials()
+            stolen_credentials = self.get_stolen_credentials()
+
+            return [*configured_credentials, *stolen_credentials]
+        except RetrievalError as err:
+            raise err
+
+    def save_configured_credentials(self, credentials: Credentials):
+        try:
+            mongo.db.configured_credentials.insert_one(credentials)
+        except Exception as err:
+            raise StorageError(err)
+
+    def save_stolen_credentials(self, credentials: Credentials):
+        try:
+            mongo.db.stolen_credentials.insert_one(credentials)
+        except Exception as err:
+            raise StorageError(err)
+
+    def remove_configured_credentials(self):
+        try:
+            mongo.db.configured_credentials.remove({})
+        except Exception as err:
+            raise RemovalError(err)
+
+    def remove_stolen_credentials(self):
+        try:
+            mongo.db.stolen_credentials.remove({})
+        except Exception as err:
+            raise RemovalError(err)
+
+    def remove_all_credentials(self):
+        try:
+            self.remove_configured_credentials()
+            self.remove_stolen_credentials()
+        except RemovalError as err:
+            raise err
diff --git a/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_credentials_repository.py b/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_credentials_repository.py
new file mode 100644
index 000000000..28b1c2eab
--- /dev/null
+++ b/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_credentials_repository.py
@@ -0,0 +1,31 @@
+import mongoengine
+import pytest
+
+from monkey_island.cc.repository import MongoCredentialsRepository
+
+
+@pytest.fixture
+def fake_mongo(monkeypatch):
+    mongo = mongoengine.connection.get_connection()
+    monkeypatch.setattr("monkey_island.cc.repository.mongo_credentials_repository.mongo", mongo)
+
+
+def test_mongo_repository_get_configured(fake_mongo):
+
+    actual_configured_credentials = MongoCredentialsRepository().get_configured_credentials()
+
+    assert actual_configured_credentials == []
+
+
+def test_mongo_repository_get_stolen(fake_mongo):
+
+    actual_stolen_credentials = MongoCredentialsRepository().get_stolen_credentials()
+
+    assert actual_stolen_credentials == []
+
+
+def test_mongo_repository_get_all(fake_mongo):
+
+    actual_credentials = MongoCredentialsRepository().get_all_credentials()
+
+    assert actual_credentials == []