def assert_response_matches_request(self, request_bytes, response_bytes): host = "127.0.0.1" port = 22322 server = Server(io_loop=self.io_loop) server.set_handler("text") server.listen(port, address=host) s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) stream = iostream.IOStream(s, io_loop=self.io_loop) def start_test(): stream.connect((host, port), send_request) def send_request(): stream.write(request_bytes, write_finished) def write_finished(*args, **kwargs): stream.read_bytes(len(response_bytes), receive_response) def receive_response(data): self.assertEqual(data, response_bytes) stream.close() self.stop() self.io_loop.add_callback(start_test) self.wait()
def assert_response_matches_request(self, request_bytes, response_bytes): host = '127.0.0.1' port = 22322 server = Server(io_loop=self.io_loop) server.set_handler('text') server.listen(port, address=host) s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) stream = iostream.IOStream(s, io_loop=self.io_loop) def start_test(): stream.connect((host, port), send_request) def send_request(): stream.write(request_bytes, write_finished) def write_finished(*args, **kwargs): stream.read_bytes(len(response_bytes), receive_response) def receive_response(data): self.assertEqual(data, response_bytes) stream.close() self.stop() self.io_loop.add_callback(start_test) self.wait()
def passes_io_loop_to_new_handler(self): server = Server(io_loop=self.io_loop) server.set_handler('text') self.assertIs(server.io_loop, server.handler.io_loop)
def sets_a_binary_handler(self): server = Server(io_loop=self.io_loop) server.set_handler('binary') self.assertIsInstance(server.handler, BinaryProtocolHandler)
def sets_a_text_handler(self): server = Server(io_loop=self.io_loop) server.set_handler('text') self.assertIsInstance(server.handler, TextProtocolHandler)