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

Add tests for network node adding/removal

Browse Source
This commit is contained in:
Marek Wolan
2023-08-24 10:26:17 +01:00
parent a82ffb9747
commit 4077eb3a5c
3 changed files with 60 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/primaite/simulator/network/container.py
View File

@@ -52,7 +52,9 @@ class NetworkContainer(SimComponent):
:type node: Node
"""
if node in self:
_LOGGER.warning(f"Can't add node {node}. It is already in the network.")
msg = f"Can't add node {node}. It is already in the network."
_LOGGER.warning(msg)
raise RuntimeWarning(msg)
self.nodes[node.uuid] = node
node.parent = self
@@ -64,7 +66,9 @@ class NetworkContainer(SimComponent):
:type node: Node
"""
if node not in self:
_LOGGER.warning(f"Can't remove node {node}. It's not in the network.")
msg = f"Can't remove node {node}. It's not in the network."
_LOGGER.warning(msg)
raise RuntimeWarning(msg)
del self.nodes[node.uuid]
del node.parent # misleading?