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

UT: Fix tests for event encryption

This commit is contained in:
Kekoa Kaaikala 2022-09-16 13:43:10 +00:00
parent 54fe2a6dca
commit c19e50b7f1
2 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/tests/unit_tests/monkey_island/cc/repository/test_agent_event_encryption.py
View File

@ -30,7 +30,7 @@ def encryptor(key_file):
@pytest.fixture @pytest.fixture
def serializer(): def serializer():
return PydanticAgentEventSerializer() return PydanticAgentEventSerializer(FakeAgentEvent)
def test_agent_event_encryption__encrypts(encryptor, serializer): def test_agent_event_encryption__encrypts(encryptor, serializer):

21
monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_event_repository.py
View File

@ -18,6 +18,7 @@ from monkey_island.cc.repository import (
RetrievalError, RetrievalError,
StorageError, StorageError,
) )
from monkey_island.cc.server_utils.encryption import RepositoryEncryptor
class FakeAgentEvent(AbstractAgentEvent): class FakeAgentEvent(AbstractAgentEvent):
@ -54,8 +55,20 @@ def mongo_client(event_serializer_registry):
@pytest.fixture @pytest.fixture
def mongo_repository(mongo_client, event_serializer_registry) -> IAgentEventRepository: def key_file(tmp_path):
return MongoEventRepository(mongo_client, event_serializer_registry) return tmp_path / "test_key.bin"
@pytest.fixture
def encryptor(key_file):
encryptor = RepositoryEncryptor(key_file)
encryptor.unlock(b"password")
return encryptor
@pytest.fixture
def mongo_repository(mongo_client, event_serializer_registry, encryptor) -> IAgentEventRepository:
return MongoEventRepository(mongo_client, event_serializer_registry, encryptor)
@pytest.fixture @pytest.fixture
@ -75,9 +88,9 @@ def error_raising_mongo_client(mongo_client) -> mongomock.MongoClient:
@pytest.fixture @pytest.fixture
def error_raising_mongo_repository( def error_raising_mongo_repository(
error_raising_mongo_client, event_serializer_registry error_raising_mongo_client, event_serializer_registry, encryptor
) -> IAgentEventRepository: ) -> IAgentEventRepository:
return MongoEventRepository(error_raising_mongo_client, event_serializer_registry) return MongoEventRepository(error_raising_mongo_client, event_serializer_registry, encryptor)
def assert_same_contents(a, b): def assert_same_contents(a, b):