oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add more tests for office lan creation

Browse Source
This commit is contained in:
Marek Wolan
2024-10-04 11:36:48 +01:00
parent b4cc1b4379
commit 07c4860059
1 changed files with 46 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
tests/unit_tests/_primaite/_simulator/_network/test_creation.py
View File

@@ -1,33 +1,20 @@
# © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK # © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK
from ipaddress import IPv4Address
import pytest import pytest
from primaite.simulator.network.container import Network from primaite.simulator.network.container import Network
from primaite.simulator.network.creation import OfficeLANAdder from primaite.simulator.network.creation import NetworkNodeAdder, OfficeLANAdder
param_names = ("lan_name", "subnet_base", "pcs_ip_block_start", "num_pcs", "include_router", "bandwidth")
@pytest.mark.parametrize( param_vals = (
("lan_name", "subnet_base", "pcs_ip_block_start", "num_pcs", "include_router", "bandwidth"), ("CORP-NETWORK", 3, 10, 6, True, 45),
( ("OTHER-NETWORK", 10, 25, 26, True, 100),
("CORP-NETWORK", 3, 10, 6, True, 45), ("OTHER-NETWORK", 10, 25, 55, False, 100),
("OTHER-NETWORK", 10, 25, 26, True, 100),
("OTHER-NETWORK", 10, 25, 55, False, 100),
),
) )
def test_office_lan_adder(lan_name, subnet_base, pcs_ip_block_start, num_pcs, include_router, bandwidth): param_dicts = [dict(zip(param_names, vals)) for vals in param_vals]
net = Network()
office_lan_config = OfficeLANAdder.ConfigSchema(
lan_name=lan_name,
subnet_base=subnet_base,
pcs_ip_block_start=pcs_ip_block_start,
num_pcs=num_pcs,
include_router=include_router,
bandwidth=bandwidth,
)
OfficeLANAdder.add_nodes_to_net(config=office_lan_config, network=net)
def _assert_valid_creation(net: Network, lan_name, subnet_base, pcs_ip_block_start, num_pcs, include_router, bandwidth):
"""Assert that the network contains the correct nodes as described by config items"""
num_switches = 1 if num_pcs <= 23 else num_pcs // 23 + 2 num_switches = 1 if num_pcs <= 23 else num_pcs // 23 + 2
num_routers = 1 if include_router else 0 num_routers = 1 if include_router else 0
total_nodes = num_pcs + num_switches + num_routers total_nodes = num_pcs + num_switches + num_routers
@@ -43,3 +30,40 @@ def test_office_lan_adder(lan_name, subnet_base, pcs_ip_block_start, num_pcs, in
# check that computers occupy address range 192.168.3.10 - 192.168.3.16 # check that computers occupy address range 192.168.3.10 - 192.168.3.16
computer_ip_last_octets = {str(n.network_interface[1].ip_address).split(".")[-1] for n in net.computer_nodes} computer_ip_last_octets = {str(n.network_interface[1].ip_address).split(".")[-1] for n in net.computer_nodes}
assert computer_ip_last_octets == {str(i) for i in range(pcs_ip_block_start, pcs_ip_block_start + num_pcs)} assert computer_ip_last_octets == {str(i) for i in range(pcs_ip_block_start, pcs_ip_block_start + num_pcs)}
@pytest.mark.parametrize("kwargs", param_dicts)
def test_office_lan_adder(kwargs):
"""Assert that adding an office lan via the python API works correctly."""
net = Network()
office_lan_config = OfficeLANAdder.ConfigSchema(
lan_name=kwargs["lan_name"],
subnet_base=kwargs["subnet_base"],
pcs_ip_block_start=kwargs["pcs_ip_block_start"],
num_pcs=kwargs["num_pcs"],
include_router=kwargs["include_router"],
bandwidth=kwargs["bandwidth"],
)
OfficeLANAdder.add_nodes_to_net(config=office_lan_config, network=net)
_assert_valid_creation(net=net, **kwargs)
@pytest.mark.parametrize("kwargs", param_dicts)
def test_office_lan_from_config(kwargs):
"""Assert that the base class can add an office lan given a config dict."""
net = Network()
config = dict(
type="office_lan",
lan_name=kwargs["lan_name"],
subnet_base=kwargs["subnet_base"],
pcs_ip_block_start=kwargs["pcs_ip_block_start"],
num_pcs=kwargs["num_pcs"],
include_router=kwargs["include_router"],
bandwidth=kwargs["bandwidth"],
)
NetworkNodeAdder.from_config(config=config, network=net)
_assert_valid_creation(net=net, **kwargs)