def test_udp_client_recv(self): """ Test the udp client receive method""" client = ModbusUdpClient() self.assertRaises(ConnectionException, lambda: client.receive(1024)) client.socket = MockSocket() self.assertEqual('', client.receive(0)) self.assertEqual('\x00'*4, client.receive(4))
def test_udp_client_send(self): """ Test the udp client send method""" client = ModbusUdpClient() self.assertRaises(ConnectionException, lambda: client.send(None)) client.socket = MockSocket() self.assertEqual(0, client.send(None)) self.assertEqual(4, client.send('1234'))
def test_udp_client_connect(self): """ Test the Udp client connection method""" with patch.object(socket, 'socket') as mock_method: mock_method.return_value = object() client = ModbusUdpClient() self.assertTrue(client.connect()) with patch.object(socket, 'socket') as mock_method: mock_method.side_effect = socket.error() client = ModbusUdpClient() self.assertFalse(client.connect())
def test_udp_client_recv(self): """ Test the udp client receive method""" client = ModbusUdpClient() self.assertRaises(ConnectionException, lambda: client.receive(1024)) client.socket = MockSocket() self.assertEqual('', client.receive(0)) self.assertEqual('\x00' * 4, client.receive(4))
def test_basic_sync_udp_client(self): """ Test the basic methods for the udp sync client""" # receive/send client = ModbusUdpClient() client.socket = MockSocket() self.assertEqual(0, client.send(None)) self.assertEqual(1, client.send('\x00')) self.assertEqual('\x00', client.receive(1)) # connect/disconnect self.assertTrue(client.connect()) client.close() # already closed socket client.socket = False client.close() self.assertEqual("127.0.0.1:502", str(client))
def test_udp_client_address_family(self): """ Test the Udp client get address family method""" client = ModbusUdpClient() self.assertEqual(socket.AF_INET, client._get_address_family('127.0.0.1')) self.assertEqual(socket.AF_INET6, client._get_address_family('::1'))
def test_sync_udp_client_instantiation(self): client = ModbusUdpClient() self.assertNotEqual(client, None)