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()
