def run_test_server(loop, consumer_factory, address=None):
address = address or ('127.0.0.1', 0)
server = TcpServer(loop, '127.0.0.1', 0,
consumer_factory=consumer_factory)
try:
yield server
finally:
server.stop_serving()
def test_echo_serve(self):
loop = get_event_loop()
server = TcpServer(loop, '127.0.0.1', 0, EchoServerProtocol)
yield server.start_serving()
sock = server.sock
fn = sock.fileno()
self.assertFalse(is_socket_closed(sock))
client = Echo()
address = sock.getsockname()
result = yield client.request(address, b'Hello!')
self.assertEqual(result, b'Hello!')
self.assertEqual(server.concurrent_connections, 1)
result = yield client.request(address, b'ciao')
self.assertEqual(result, b'ciao')
self.assertEqual(server.concurrent_connections, 1)
yield server.stop_serving()
yield async_while(3, lambda: not is_socket_closed(sock))
self.assertTrue(is_socket_closed(sock))
