diff --git a/monkey/tests/unit_tests/common/agent_events/test_exploitation_event.py b/monkey/tests/unit_tests/common/agent_events/test_exploitation_event.py new file mode 100644 index 000000000..54a7c67ad --- /dev/null +++ b/monkey/tests/unit_tests/common/agent_events/test_exploitation_event.py @@ -0,0 +1,82 @@ +from ipaddress import IPv4Address +from uuid import UUID + +import pytest + +from common.agent_events import ExploitationEvent + +TARGET_IP_STR = "192.168.1.10" +AGENT_ID = UUID("012e7238-7b81-4108-8c7f-0787bc3f3c10") +TIMESTAMP = 1664371327.4067292 + +EXPLOITATION_EVENT = ExploitationEvent( + source=AGENT_ID, + timestamp=TIMESTAMP, + target=IPv4Address(TARGET_IP_STR), + success=True, + exploiter_name="SSHExploiter", +) + +EXPLOITATION_OBJECT_DICT = { + "source": AGENT_ID, + "timestamp": TIMESTAMP, + "target": IPv4Address(TARGET_IP_STR), + "success": True, + "exploiter_name": "SSHExploiter", + "error_message": "", +} + +EXPLOITATION_SIMPLE_DICT = { + "source": str(AGENT_ID), + "timestamp": TIMESTAMP, + "target": TARGET_IP_STR, + "success": "true", + "exploiter_name": "SSHExploiter", + "error_message": "", +} + + +@pytest.mark.parametrize( + "EXPLOITATION_event_dict", [EXPLOITATION_OBJECT_DICT, EXPLOITATION_SIMPLE_DICT] +) +def test_constructor(EXPLOITATION_event_dict): + assert ExploitationEvent(**EXPLOITATION_event_dict) == EXPLOITATION_EVENT + + +@pytest.mark.parametrize( + "key, value", + [ + ("target", None), + ("success", "not-a-bool"), + ("exploiter_name", None), + ("error_message", None), + ], +) +def test_construct_invalid_field__type_error(key, value): + invalid_type_dict = EXPLOITATION_SIMPLE_DICT.copy() + invalid_type_dict[key] = value + + with pytest.raises(TypeError): + ExploitationEvent(**invalid_type_dict) + + +@pytest.mark.parametrize( + "key, value", + [ + ("target", "not-an-ip"), + ], +) +def test_construct_invalid_field__value_error(key, value): + invalid_type_dict = EXPLOITATION_SIMPLE_DICT.copy() + invalid_type_dict[key] = value + + with pytest.raises(ValueError): + ExploitationEvent(**invalid_type_dict) + + +def test_construct__extra_fields_forbidden(): + extra_field_dict = EXPLOITATION_SIMPLE_DICT.copy() + extra_field_dict["extra_field"] = 99 # red balloons + + with pytest.raises(ValueError): + ExploitationEvent(**extra_field_dict)