p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/tests/unit_tests/monkey_island/cc/island_event_handlers/test_handle_agent_registration.py
View File

@ -27,12 +27,14 @@ MACHINE = Machine(
network_interfaces=[IPv4Interface("192.168.2.2/24")], network_interfaces=[IPv4Interface("192.168.2.2/24")],
) )
IP = "192.168.1.1:5000"
AGENT_REGISTRATION_DATA = AgentRegistrationData( AGENT_REGISTRATION_DATA = AgentRegistrationData(
id=AGENT_ID, id=AGENT_ID,
machine_hardware_id=MACHINE.hardware_id, machine_hardware_id=MACHINE.hardware_id,
start_time=0, start_time=0,
parent_id=None, 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")], network_interfaces=[IPv4Interface("192.168.1.2/24")],
) )