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())