UT: Use SocketAddress.from_string() in test_handle_agent_registration.py

This commit is contained in:
Shreya Malviya 2022-09-27 13:11:13 +05:30
parent d1199fdab2
commit 9b4de6bab8
1 changed files with 3 additions and 1 deletions

View File

@ -27,12 +27,14 @@ MACHINE = Machine(
network_interfaces=[IPv4Interface("192.168.2.2/24")],
)
IP = "192.168.1.1:5000"
AGENT_REGISTRATION_DATA = AgentRegistrationData(
id=AGENT_ID,
machine_hardware_id=MACHINE.hardware_id,
start_time=0,
parent_id=None,
cc_server=SocketAddress(ip="192.168.1.1", port="5000"),
cc_server=SocketAddress.from_string(IP),
network_interfaces=[IPv4Interface("192.168.1.2/24")],
)