#2257: add firewall via config + fix router hop ip address + shuffling around tests

tests/integration_tests/configuration_file_parsing/nodes/network/test_firewall_config.py

@@ -0,0 +1,45 @@
+import pytest
+
+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.firewall import Firewall
+from tests.integration_tests.configuration_file_parsing import DMZ_NETWORK, load_config
+
+
+@pytest.fixture(scope="function")
+def dmz_config() -> Network:
+    game = load_config(DMZ_NETWORK)
+    return game.simulation.network
+
+
+def test_firewall_is_in_configuration(dmz_config):
+    """Test that the firewall exists in the configuration file."""
+    network: Network = dmz_config
+
+    assert network.get_node_by_hostname("firewall")
+
+
+def test_firewall_routes_are_correctly_added(dmz_config):
+    """Test that the firewall routes have been correctly added to and configured in the network."""
+    network: Network = dmz_config
+
+    firewall: Firewall = network.get_node_by_hostname("firewall")
+    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")
+
+    # there should be a route to client_1
+    assert firewall.route_table.find_best_route(client_1.network_interface[1].ip_address)
+    assert dmz_server.ping(client_1.network_interface[1].ip_address)
+    assert external_computer.ping(client_1.network_interface[1].ip_address)
+    assert external_server.ping(client_1.network_interface[1].ip_address)
+
+
+def test_firewall_acl_rules_correctly_added():
+    """
+    Test that makes sure that the firewall ACLs have been configured onto the firewall
+    node via configuration file.
+    """
+    pass

tests/integration_tests/configuration_file_parsing/nodes/network/test_router_config.py

@@ -0,0 +1,54 @@
+import pytest
+
+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.integration_tests.configuration_file_parsing import DMZ_NETWORK, load_config
+
+
+@pytest.fixture(scope="function")
+def dmz_config() -> Network:
+    game = load_config(DMZ_NETWORK)
+    return game.simulation.network
+
+
+def test_router_is_in_configuration(dmz_config):
+    """Test that the router exists in the configuration file."""
+    network: Network = dmz_config
+
+    assert network.get_node_by_hostname("router_1")
+
+
+def test_router_routes_are_correctly_added(dmz_config):
+    """Test that makes sure that router routes have been added from the configuration file."""
+    network: Network = dmz_config
+
+    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")
+
+    # there should be a route to dmz_server
+    assert router_1.route_table.find_best_route(dmz_server.network_interface[1].ip_address)
+    assert client_1.ping(dmz_server.network_interface[1].ip_address)
+    assert external_computer.ping(dmz_server.network_interface[1].ip_address)
+    assert external_server.ping(dmz_server.network_interface[1].ip_address)
+
+    # there should be a route to external_computer
+    assert router_1.route_table.find_best_route(external_computer.network_interface[1].ip_address)
+    assert client_1.ping(external_computer.network_interface[1].ip_address)
+    assert dmz_server.ping(external_computer.network_interface[1].ip_address)
+    assert external_server.ping(external_computer.network_interface[1].ip_address)
+
+    # there should be a route to external_server
+    assert router_1.route_table.find_best_route(external_server.network_interface[1].ip_address)
+    assert client_1.ping(external_server.network_interface[1].ip_address)
+    assert dmz_server.ping(external_server.network_interface[1].ip_address)
+    assert external_computer.ping(external_server.network_interface[1].ip_address)
+
+
+def test_router_acl_rules_correctly_added():
+    """Test that makes sure that the router ACLs have been configured onto the router node via configuration file."""
+    pass

tests/integration_tests/configuration_file_parsing/nodes/test_node_config.py

@@ -0,0 +1,26 @@
+from primaite.config.load import example_config_path
+from primaite.simulator.network.container import Network
+from tests.integration_tests.configuration_file_parsing import DMZ_NETWORK, load_config
+
+
+def test_example_config():
+    """Test that the example config can be parsed properly."""
+    game = load_config(example_config_path())
+    network: Network = game.simulation.network
+
+    assert len(network.nodes) == 10  # 10 nodes in example network
+    assert len(network.routers) == 1  # 1 router in network
+    assert len(network.switches) == 2  # 2 switches in network
+    assert len(network.servers) == 5  # 5 servers in network
+
+
+def test_dmz_config():
+    """Test that the DMZ network config can be parsed properly."""
+    game = load_config(DMZ_NETWORK)
+
+    network: Network = game.simulation.network
+
+    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) == 2  # 2 servers in network