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_process_with_SSLv2(self, mock_sock):
state = ConnectionState()
connect = Connect(1, 2, (0, 2))
connect.process(state)
self.assertEqual(state.msg_sock.version, (0, 2))
mock_sock.assert_called_once_with(socket.AF_INET, socket.SOCK_STREAM)
instance = mock_sock.return_value
instance.connect.assert_called_once_with((1, 2))
self.assertIs(state.msg_sock.sock.socket, instance)
def test_process(self, raw_sock):
state = ConnectionState()
self.assertIsNone(state.msg_sock)
node = Connect('localhost', 4433)
node.process(state)
raw_sock.assert_called_once_with(socket.AF_INET, socket.SOCK_STREAM)
raw_sock.return_value.connect.assert_called_once_with(('localhost',
4433))
self.assertIsNotNone(state.msg_sock)
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'))
