From 6df2c29e309704a88d31b2c9ac3fbf213344aace Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Mon, 26 Sep 2022 17:13:27 +0530
Subject: [PATCH] Common: Update cc_server's type in AgentRegistrationData to
 SocketAddress

---
 monkey/common/agent_registration_data.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/monkey/common/agent_registration_data.py b/monkey/common/agent_registration_data.py
index 6d45df89f..5d281be25 100644
--- a/monkey/common/agent_registration_data.py
+++ b/monkey/common/agent_registration_data.py
@@ -7,7 +7,7 @@ from pydantic import validator
 
 from .base_models import InfectionMonkeyBaseModel
 from .transforms import make_immutable_sequence
-from .types import HardwareID
+from .types import HardwareID, SocketAddress
 
 
 class AgentRegistrationData(InfectionMonkeyBaseModel):
@@ -15,7 +15,7 @@ class AgentRegistrationData(InfectionMonkeyBaseModel):
     machine_hardware_id: HardwareID
     start_time: datetime
     parent_id: Optional[UUID]
-    cc_server: str
+    cc_server: SocketAddress
     network_interfaces: Sequence[IPv4Interface]
 
     _make_immutable_sequence = validator("network_interfaces", pre=True, allow_reuse=True)(