diff --git a/tests/integration_tests/system/red_applications/test_dos_bot_and_server.py b/tests/integration_tests/system/red_applications/test_dos_bot_and_server.py index 7ab7d104..e42862bf 100644 --- a/tests/integration_tests/system/red_applications/test_dos_bot_and_server.py +++ b/tests/integration_tests/system/red_applications/test_dos_bot_and_server.py @@ -24,7 +24,7 @@ def dos_bot_and_db_server(client_server) -> Tuple[DoSBot, Computer, DatabaseServ dos_bot: DoSBot = computer.software_manager.software.get("DoSBot") dos_bot.configure( - target_ip_address=IPv4Address(server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address), + target_ip_address=IPv4Address(server.network_interface[1].ip_address), target_port=Port.POSTGRES_SERVER, ) @@ -54,7 +54,7 @@ def dos_bot_db_server_green_client(example_network) -> Network: dos_bot: DoSBot = client_1.software_manager.software.get("DoSBot") dos_bot.configure( - target_ip_address=IPv4Address(server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address), + target_ip_address=IPv4Address(server.network_interface[1].ip_address), target_port=Port.POSTGRES_SERVER, ) diff --git a/tests/integration_tests/system/test_dns_client_server.py b/tests/integration_tests/system/test_dns_client_server.py index 78d2035c..e6275459 100644 --- a/tests/integration_tests/system/test_dns_client_server.py +++ b/tests/integration_tests/system/test_dns_client_server.py @@ -29,7 +29,7 @@ def dns_client_and_dns_server(client_server) -> Tuple[DNSClient, Computer, DNSSe # register arcd.com as a domain dns_server.dns_register( domain_name="arcd.com", - domain_ip_address=IPv4Address(server.network_interfaces.get(next(iter(server.network_interfaces))).ip_address), + domain_ip_address=IPv4Address(server.network_interface[1].ip_address), ) return dns_client, computer, dns_server, server