Skip to content

Commit d20adbd

Browse files
committed
Island: Move OTP repository into authentication_service
Issue #3078
1 parent 92e25cf commit d20adbd

File tree

4 files changed

+11
-7
lines changed

4 files changed

+11
-7
lines changed

monkey/monkey_island/cc/repositories/__init__.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from .errors import RemovalError, RepositoryError, RetrievalError, StorageError, UnknownRecordError
2+
from .consts import MONGO_OBJECT_ID_KEY
23

34

45
from .i_file_repository import FileNotFoundError, IFileRepository
@@ -11,7 +12,6 @@
1112
from .i_agent_event_repository import IAgentEventRepository
1213
from .i_agent_log_repository import IAgentLogRepository
1314
from .i_agent_plugin_repository import IAgentPluginRepository
14-
from .i_otp_repository import IOTPRepository
1515

1616

1717
from .local_storage_file_repository import LocalStorageFileRepository
@@ -30,7 +30,6 @@
3030
from .mongo_agent_repository import MongoAgentRepository
3131
from .mongo_node_repository import MongoNodeRepository
3232
from .mongo_agent_event_repository import MongoAgentEventRepository
33-
from .mongo_otp_repository import MongoOTPRepository
3433
from .file_agent_log_repository import FileAgentLogRepository
3534
from .file_agent_plugin_repository import FileAgentPluginRepository
3635

monkey/monkey_island/cc/repositories/mongo_otp_repository.py monkey/monkey_island/cc/services/authentication_service/mongo_otp_repository.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,15 @@
11
from pymongo import MongoClient
22

3+
from monkey_island.cc.repositories import (
4+
MONGO_OBJECT_ID_KEY,
5+
RemovalError,
6+
RetrievalError,
7+
StorageError,
8+
UnknownRecordError,
9+
)
310
from monkey_island.cc.server_utils.encryption import ILockableEncryptor
411

5-
from . import IOTPRepository
6-
from .consts import MONGO_OBJECT_ID_KEY
7-
from .errors import RemovalError, RetrievalError, StorageError, UnknownRecordError
12+
from .i_otp_repository import IOTPRepository
813

914

1015
class MongoOTPRepository(IOTPRepository):

monkey/tests/unit_tests/monkey_island/cc/repositories/test_mongo_otp_repository.py monkey/tests/unit_tests/monkey_island/cc/services/authentication_service/test_mongo_otp_repository.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@
55
import pytest
66

77
from monkey_island.cc.repositories import (
8-
IOTPRepository,
9-
MongoOTPRepository,
108
RemovalError,
119
RetrievalError,
1210
StorageError,
1311
UnknownRecordError,
1412
)
13+
from monkey_island.cc.services.authentication_service.i_otp_repository import IOTPRepository
14+
from monkey_island.cc.services.authentication_service.mongo_otp_repository import MongoOTPRepository
1515

1616

1717
@dataclass

0 commit comments

Comments
 (0)