Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

close daemon connector on server close to avoid errors on tests

Browse source
This commit is contained in:
Victor Shyba 2019-07-14 00:46:22 -03:00 committed by Lex Berezhny
parent 510d88fff1
commit 8749c78622
2 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
torba/torba/server/daemon.py
View file

@ -57,6 +57,11 @@ class Daemon:
self.available_rpcs = {} self.available_rpcs = {}
self.connector = aiohttp.TCPConnector() self.connector = aiohttp.TCPConnector()
async def close(self):
if self.connector:
await self.connector.close()
self.connector = None
def set_url(self, url): def set_url(self, url):
"""Set the URLS to the given list, and switch to the first one.""" """Set the URLS to the given list, and switch to the first one."""
urls = url.split(',') urls = url.split(',')

1
torba/torba/server/server.py
View file

@ -146,6 +146,7 @@ class Server:
task.cancel() task.cancel()
await asyncio.wait(self.cancellable_tasks) await asyncio.wait(self.cancellable_tasks)
self.shutdown_event.set() self.shutdown_event.set()
await self.daemon.close()
def run(self): def run(self):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()