def testTcpClientSend(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 _send_cipher(self, request): self._fifo = bytes() length = len(request) padding = 16 - (length % 16) self._transactionID = request[:2] request = HEADER + bytes(request[2:]) + bytes([0xff for i in range(0, padding)]) crypto_header = bytes([1, 0, length, padding]) encrypted_request = crypto_header + self._aes_ecb.encrypt(request) return ModbusTcpClient._send(self, encrypted_request) - len(crypto_header) - padding
def testBasicSyncTcpClient(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._recv(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))