def test_echo(port): message = 'hello'.encode('utf8') def echo(socket): socket.send(socket.recv(1024)) server = Server(LOCALHOST, port, echo) server.start() with Client(LOCALHOST, port) as client: client.send(message) echoed = client.recv(1024) server.stop() assert echoed == message
def test_connect_twice_in_parallel(port): message_1 = 'hello 1'.encode('utf8') message_2 = 'hello 2'.encode('utf8') def echo(socket): socket.send(socket.recv(1024)) server = Server(LOCALHOST, port, echo) server.start() with Client(LOCALHOST, port) as client_1: client_1.send(message_1) echoed_1 = client_1.recv(1024) with Client(LOCALHOST, port) as client_2: client_2.send(message_2) echoed_2 = client_2.recv(1024) server.stop() assert echoed_1 == message_1 assert echoed_2 == message_2
def test_server_startup(port): def handler(socket): pass server = Server(LOCALHOST, port, handler) server.start() server.stop()