#2257: temporarily commit changes - added startup and shut down durations to node config + adding routes

tests/integration_tests/configuration_file_parsing/router_game_configuration.py

@@ -5,6 +5,9 @@ import yaml
 
 from primaite.game.game import PrimaiteGame
 from primaite.simulator.network.container import Network
+from primaite.simulator.network.hardware.nodes.host.computer import Computer
+from primaite.simulator.network.hardware.nodes.host.server import Server
+from primaite.simulator.network.hardware.nodes.network.router import Router
 from tests import TEST_ASSETS_ROOT
 
 DMZ_NETWORK = TEST_ASSETS_ROOT / "configs/dmz_network.yaml"
@@ -27,12 +30,27 @@ def test_dmz_config():
     assert len(network.nodes) == 9  # 9 nodes in network
     assert len(network.routers) == 2  # 2 routers in network
     assert len(network.switches) == 3  # 3 switches in network
-    assert len(network.servers) == 1  # 1 server in network
+    assert len(network.servers) == 2  # 2 servers in network
 
 
 def test_router_routes_are_correctly_added():
     """Test that makes sure that router routes have been added from the configuration file."""
-    pass
+    game = load_config(DMZ_NETWORK)
+
+    network: Network = game.simulation.network
+
+    router_1: Router = network.get_node_by_hostname("router_1")
+    client_1: Computer = network.get_node_by_hostname("client_1")
+    dmz_server: Server = network.get_node_by_hostname("dmz_server")
+    external_computer: Computer = network.get_node_by_hostname("external_computer")
+    external_server: Server = network.get_node_by_hostname("external_server")
+
+    # test that client_1 has a route to the DMZ and external nodes - they are on a second router
+
+    # there should be a route to the dmz server
+    assert router_1.route_table.find_best_route(dmz_server.network_interface[1].ip_address)
+    # ping DMZ server
+    # assert client_1.ping(dmz_server.network_interface[1].ip_address)
 
 
 def test_firewall_node_added_to_network():