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_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_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_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_sync_udp_client_instantiation(self):
client = ModbusUdpClient()
self.assertNotEqual(client, None)
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'))
