def test_generate(self, raw_sock): state = ConnectionState() conn = Connect('localhost', 4433) conn.process(state) self.assertFalse(state.msg_sock.sock.buffer_writes) node = TCPBufferingEnable() node.process(state) self.assertTrue(state.msg_sock.sock.buffer_writes)
def test_generate(self, raw_sock): state = ConnectionState() conn = Connect('localhost', 4433) conn.process(state) node = TCPBufferingEnable() node.process(state) node = RawMessageGenerator(12, bytearray(b'\xff')) msg = node.generate(state) state.msg_sock.sendMessageBlocking(msg) raw_sock.return_value.send.assert_not_called() raw_sock.return_value.sendall.assert_not_called() flush = TCPBufferingFlush() flush.process(state) raw_sock.return_value.sendall.assert_called_once_with( bytearray(b'\x0c\x03\x00\x00\x01\xff'))