kopia lustrzana https://github.com/Yakifo/amqtt
enhancing the 'test_init_handler' case so that it checks to make sure that the ProtocolHandler defaults to the correct loop.
rodzic
5c5305baa6
commit
ec31def92c
|
@ -38,11 +38,23 @@ class ProtocolHandlerTest(unittest.TestCase):
|
||||||
self.loop.close()
|
self.loop.close()
|
||||||
|
|
||||||
def test_init_handler(self):
|
def test_init_handler(self):
|
||||||
Session()
|
|
||||||
handler = ProtocolHandler(self.plugin_manager, loop=self.loop)
|
async def test_coro() -> None:
|
||||||
assert handler.session is None
|
try:
|
||||||
assert handler._loop is self.loop
|
Session()
|
||||||
self.check_empty_waiters(handler)
|
handler = ProtocolHandler(self.plugin_manager)
|
||||||
|
assert handler.session is None
|
||||||
|
assert handler._loop is self.loop
|
||||||
|
self.check_empty_waiters(handler)
|
||||||
|
future.set_result(True)
|
||||||
|
except Exception as ae:
|
||||||
|
future.set_exception(ae)
|
||||||
|
|
||||||
|
future: asyncio.Future[Any] = asyncio.Future()
|
||||||
|
self.loop.run_until_complete(test_coro())
|
||||||
|
exception = future.exception()
|
||||||
|
if exception:
|
||||||
|
raise exception
|
||||||
|
|
||||||
def test_start_stop(self):
|
def test_start_stop(self):
|
||||||
async def server_mock(reader, writer) -> None:
|
async def server_mock(reader, writer) -> None:
|
||||||
|
|
Ładowanie…
Reference in New Issue