def send_receive_message(server, client): """ Common test utility to test sending a request/response exchange between client and server. Sends: Client ---"Hello"---> Server ---"World"---> Client """ msg_data = "Hello" msg = Message(data=msg_data, codec=Codec()) client.send(msg.to_buffer()) server.accept_connection() recv = Message.from_buffer( data=server.receive(len(msg.data)), codec=Codec() ) # Server received data assert recv.data == msg_data msg_data = "World" resp = Message(data=msg_data, codec=Codec()) server.send(resp.to_buffer()) recv = Message.from_buffer( data=client.receive(len(resp.data)), codec=Codec() ) # Client received response assert recv.data == msg_data
def test_basic_message(): codec = Codec() buff = b'Hello world' assert isinstance(buff, bytes) message = Message.from_buffer(buff, codec) assert isinstance(message, Message) assert isinstance(message.data, str) assert isinstance(message.codec, Codec) assert b'Hello world' == message.to_buffer()
def send_receive_message(server, client): """ Client ---"Hello"---> Server ---"World"---> Client """ msg_data = 'Hello' msg = Message(data=msg_data, codec=Codec()) client.send(msg.to_buffer()) server.accept_connection() recv = Message.from_buffer(data=server.receive(len(msg.data)), codec=Codec()) # Server received data assert recv.data == msg_data msg_data = 'World' resp = Message(data=msg_data, codec=Codec()) server.send(resp.to_buffer()) recv = Message.from_buffer(data=client.receive(len(resp.data)), codec=Codec()) # Client received response assert recv.data == msg_data