Question

使用asyncio.start_server()测试服务器程序时，此方法似乎没有调用第一个client_connected_cb方法。当程序在测试之外运行时，它按预期工作。我怀疑这与正确处理事件循环无关。

为了测试，我有一些非异步代码连接到套接字服务器，它在收听套接字时无限期挂起。

为什么事件循环不处理单元测试中的服务器代码？

测试

@pytest.mark.asyncio
async def test_connect(server, connection):
    await server.start()
    connection.connect()
    connection.join_lobby('John')
    assert server.lobby.user_count() == 1

服务器

async def start(self):
    loop = asyncio.get_event_loop()
    coro = asyncio.start_server(self.new_client, self.HOST, self.PORT, loop = loop)
    self.server_coro = await loop.create_task(coro)
    print('Serving on {}'.format(self.server_coro.sockets[0].getsockname()))

连接代码：

def connect(self):
    self.s.connect((self.HOST, self.PORT))
    print('Just connected')

def join_lobby(self, name):
    self.s.send(b'Some data')
    print('Just tried to join lobby')
    msg = self.s.recv(BUFFER_SIZE)
    print(msg)

当s.recv行被注释掉时，会显示以下输出。当没有注释s.recv时，测试挂起（因为服务器永远不会发送数据）：

Serving on ('0.0.0.0', 8888)
Just connected
Just tried to join lobby

Answer 1

你没有像这样混合阻塞和非阻塞代码。您的recv（）调用阻止整个线程并使您的异步服务器无法运行。您还必须使客户端异步。

asyncio.start_server没有在pytest-asyncio测试中运行

1 个答案: