Add tests for network node adding/removal

tests/integration_tests/network/test_network_creation.py

@@ -0,0 +1,38 @@
+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