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