39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
import pytest
|
|
|
|
from primaite.simulator.network.container import NetworkContainer
|
|
from primaite.simulator.network.hardware.base import Node
|
|
|
|
|
|
def test_adding_removing_nodes():
|
|
"""Check that we can create and add a node to a network."""
|
|
net = NetworkContainer()
|
|
n1 = Node(hostname="computer")
|
|
net.add_node(n1)
|
|
assert n1.parent is net
|
|
assert n1 in net
|
|
|
|
net.remove_node(n1)
|
|
assert n1.parent is None
|
|
assert n1 not in net
|
|
|
|
|
|
def test_readding_node():
|
|
"""Check that warning is raised when readding a node."""
|
|
net = NetworkContainer()
|
|
n1 = Node(hostname="computer")
|
|
net.add_node(n1)
|
|
with pytest.raises(RuntimeWarning):
|
|
net.add_node(n1)
|
|
assert n1.parent is net
|
|
assert n1 in net
|
|
|
|
|
|
def test_removing_nonexistent_node():
|
|
"""Check that warning is raised when trying to remove a node that is not in the network."""
|
|
net = NetworkContainer()
|
|
n1 = Node(hostname="computer")
|
|
with pytest.raises(RuntimeWarning):
|
|
net.remove_node(n1)
|
|
assert n1.parent is None
|
|
assert n1 not in net
|