changes to support better type checking for retrieving an arbitrary node from a session, get_node now requires an expected class that the node would be an instance of, if the returned node is not an instance a CoreError is thrown, this also helps editors pick up expected types to account for variable/function usage better as well
This commit is contained in:
parent
19ee367dc5
commit
d5254e6a91
21 changed files with 149 additions and 126 deletions
|
@ -3,6 +3,7 @@ import pytest
|
|||
from core.emulator.emudata import NodeOptions
|
||||
from core.emulator.enumerations import NodeTypes
|
||||
from core.errors import CoreError
|
||||
from core.nodes.base import CoreNode
|
||||
|
||||
MODELS = ["router", "host", "PC", "mdr"]
|
||||
NET_TYPES = [NodeTypes.SWITCH, NodeTypes.HUB, NodeTypes.WIRELESS_LAN]
|
||||
|
@ -45,7 +46,7 @@ class TestNodes:
|
|||
|
||||
# then
|
||||
with pytest.raises(CoreError):
|
||||
session.get_node(node.id)
|
||||
session.get_node(node.id, CoreNode)
|
||||
|
||||
def test_node_sethwaddr(self, session):
|
||||
# given
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue