From 368ddde20f0ce43d3846ab6f1052b62fde203540 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 3 Oct 2022 13:15:12 -0400 Subject: [PATCH] Common: Register serializers for {Exploitation,Propagation}Event --- monkey/common/agent_event_serializers/register.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/monkey/common/agent_event_serializers/register.py b/monkey/common/agent_event_serializers/register.py index 1ccf6cadf..fbedb2f51 100644 --- a/monkey/common/agent_event_serializers/register.py +++ b/monkey/common/agent_event_serializers/register.py @@ -1,4 +1,10 @@ -from common.agent_events import CredentialsStolenEvent, PingScanEvent, TCPScanEvent +from common.agent_events import ( + CredentialsStolenEvent, + ExploitationEvent, + PingScanEvent, + PropagationEvent, + TCPScanEvent, +) from . import AgentEventSerializerRegistry, PydanticAgentEventSerializer @@ -11,3 +17,5 @@ def register_common_agent_event_serializers( ) event_serializer_registry[PingScanEvent] = PydanticAgentEventSerializer(PingScanEvent) event_serializer_registry[TCPScanEvent] = PydanticAgentEventSerializer(TCPScanEvent) + event_serializer_registry[PropagationEvent] = PydanticAgentEventSerializer(PropagationEvent) + event_serializer_registry[ExploitationEvent] = PydanticAgentEventSerializer(ExploitationEvent)