def echo(self, binary): @server.websocket_handler def websocket_handler(ws): tu.check_context() @ws.data_handler def data_handler(buff): tu.check_context() ws.write_buffer(buff) server.listen(8080) if binary: self.buff = TestUtils.gen_buffer(1000) else: self.str_ = TestUtils.random_unicode_string(1000) def connect_handler(ws): tu.check_context() received = Buffer.create() @ws.data_handler def data_handler(buff): tu.check_context() received.append_buffer(buff) if received.length == buff.length: tu.azzert(TestUtils.buffers_equal(buff, received)) tu.test_complete() if binary: ws.write_binary_frame(self.buff) else: ws.write_text_frame(self.str_) client.connect_web_socket("/someurl", connect_handler)
def echo(self, binary): @server.websocket_handler def websocket_handler(ws): tu.check_thread() @ws.data_handler def data_handler(buff): tu.check_thread() ws.write(buff) if binary: self.buff = TestUtils.gen_buffer(1000) else: self.str_ = TestUtils.random_unicode_string(1000) def connect_handler(ws): tu.check_thread() received = Buffer.create() @ws.data_handler def data_handler(buff): tu.check_thread() received.append_buffer(buff) if received.length == buff.length: tu.azzert(TestUtils.buffers_equal(buff, received)) tu.test_complete() if binary: ws.write_binary_frame(self.buff) else: ws.write_text_frame(self.str_) def listen_handler(err, serv): tu.azzert(err == None) tu.azzert(serv == server) client.connect_web_socket("/someurl", connect_handler) server.listen(8080, "0.0.0.0", listen_handler)