Common: Register serializers for {Exploitation,Propagation}Event

guardicore · Oct 3, 2022 · 368ddde · 368ddde
1 parent eb16969
commit 368ddde
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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)