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