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