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)
def sets_a_text_handler(self):
server = Server(io_loop=self.io_loop)
server.set_handler('text')
self.assertIsInstance(server.handler, TextProtocolHandler)
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)
