def test_addNIO(vpcs_device):
with patch('gns3.node.Node.httpPost') as mock:
port = Port("Port 1")
port.setPortNumber(0)
port.setAdapterNumber(0)
nio = NIOUDP(4242, "127.0.0.1", 4243)
vpcs_device.addNIO(port, nio)
assert mock.called
args, kwargs = mock.call_args
assert args[0] == "/vpcs/vms/{vm_id}/adapters/0/ports/0/nio".format(
vm_id=vpcs_device.vm_id())
# Connect the signal
signal_mock = Mock()
vpcs_device.nio_signal.connect(signal_mock)
# Callback
args[1]({}, context=kwargs["context"])
# Check the signal
assert signal_mock.called
args, kwargs = signal_mock.call_args
assert args[0] == vpcs_device.id()
assert args[1] == port.id()
def _createNIOUDP(self, nio):
"""
Creates a NIO UDP.
:param nio: nio string
"""
match = re.search(r"""^nio_udp:(\d+):(.+):(\d+)$""", nio)
if match:
lport = int(match.group(1))
rhost = match.group(2)
rport = int(match.group(3))
return NIOUDP(lport, rhost, rport)
return None
def test_deleteNIO(vpcs_device):
with patch('gns3.node.Node.httpPost') as mock_post:
with patch('gns3.node.Node.httpDelete') as mock_delete:
port = Port("Port 1")
port.setPortNumber(0)
port.setAdapterNumber(0)
nio = NIOUDP(4242, "127.0.0.1", 4243)
vpcs_device.addNIO(port, nio)
vpcs_device.deleteNIO(port)
assert mock_delete.called
args, kwargs = mock_delete.call_args
assert args[0] == "/vpcs/vms/{vm_id}/adapters/0/ports/0/nio".format(vm_id=vpcs_device.vm_id())