1bf51c7741
- Added .show() methods to new router classes to enable inspection of the components as you would a real router. - Removed gateway from the NIC and added default_gateway to Node so that Node has a single default gateway. - Added some routing tests to check that ping can be performed when router between subnets.
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from typing import Tuple
|
|
|
|
import pytest
|
|
|
|
from primaite.simulator.network.hardware.base import Link, NIC, Node
|
|
from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
|
|
from primaite.simulator.network.transmission.network_layer import IPProtocol
|
|
from primaite.simulator.network.transmission.transport_layer import Port
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def pc_a_pc_b_router_1() -> Tuple[Node, Node, Router]:
|
|
pc_a = Node(hostname="pc_a", default_gateway="192.168.0.1")
|
|
nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0")
|
|
pc_a.connect_nic(nic_a)
|
|
pc_a.power_on()
|
|
|
|
pc_b = Node(hostname="pc_b", default_gateway="192.168.1.1")
|
|
nic_b = NIC(ip_address="192.168.1.10", subnet_mask="255.255.255.0")
|
|
pc_b.connect_nic(nic_b)
|
|
pc_b.power_on()
|
|
|
|
router_1 = Router(hostname="router_1")
|
|
router_1.power_on()
|
|
|
|
router_1.configure_port(1, "192.168.0.1", "255.255.255.0")
|
|
router_1.configure_port(2, "192.168.1.1", "255.255.255.0")
|
|
|
|
Link(endpoint_a=nic_a, endpoint_b=router_1.ethernet_ports[1])
|
|
Link(endpoint_a=nic_b, endpoint_b=router_1.ethernet_ports[2])
|
|
router_1.enable_port(1)
|
|
router_1.enable_port(2)
|
|
|
|
router_1.acl.add_rule(action=ACLAction.PERMIT, src_port=Port.ARP, dst_port=Port.ARP, position=22)
|
|
|
|
router_1.acl.add_rule(action=ACLAction.PERMIT, protocol=IPProtocol.ICMP, position=23)
|
|
return pc_a, pc_b, router_1
|
|
|
|
|
|
def test_ping_default_gateway(pc_a_pc_b_router_1):
|
|
pc_a, pc_b, router_1 = pc_a_pc_b_router_1
|
|
|
|
assert pc_a.ping(pc_a.default_gateway)
|
|
|
|
|
|
def test_ping_other_router_port(pc_a_pc_b_router_1):
|
|
pc_a, pc_b, router_1 = pc_a_pc_b_router_1
|
|
|
|
assert pc_a.ping(pc_b.default_gateway)
|
|
|
|
|
|
def test_host_on_other_subnet(pc_a_pc_b_router_1):
|
|
pc_a, pc_b, router_1 = pc_a_pc_b_router_1
|
|
|
|
assert pc_a.ping("192.168.1.10")
|