def test_tcp_client_recv(self):
""" Test the tcp client receive method"""
client = ModbusTcpClient()
self.assertRaises(ConnectionException, lambda: client.receive(1024))
client.socket = MockSocket()
self.assertEqual('', client.receive(0))
self.assertEqual('\x00'*4, client.receive(4))
def test_tcp_client_send(self):
""" Test the tcp client send method"""
client = ModbusTcpClient()
self.assertRaises(ConnectionException, lambda: client.send(None))
client.socket = MockSocket()
self.assertEqual(0, client.send(None))
self.assertEqual(4, client.send('1234'))
def test_tcp_client_recv(self):
""" Test the tcp client receive method"""
client = ModbusTcpClient()
self.assertRaises(ConnectionException, lambda: client.receive(1024))
client.socket = MockSocket()
self.assertEqual('', client.receive(0))
self.assertEqual('\x00' * 4, client.receive(4))
def test_tcp_client_send(self):
""" Test the tcp client send method"""
client = ModbusTcpClient()
self.assertRaises(ConnectionException, lambda: client.send(None))
client.socket = MockSocket()
self.assertEqual(0, client.send(None))
self.assertEqual(4, client.send('1234'))
def test_basic_sync_tcp_client(self):
""" Test the basic methods for the tcp sync client"""
# receive/send
client = ModbusTcpClient()
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_tcp_client(self):
""" Test the basic methods for the tcp sync client"""
# receive/send
client = ModbusTcpClient()
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))
