python3.5 support

mosquito · May 7, 2020 · 20e662d · 20e662d
1 parent 1ef61af
commit 20e662d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/requirements.dev.txt b/requirements.dev.txt
@@ -1,5 +1,6 @@
 aiohttp<4
 async-timeout
+async-generator
 coverage==4.5.1
 coveralls
 pylava

diff --git a/tests/pytest_plugin/test_tcp_proxy.py b/tests/pytest_plugin/test_tcp_proxy.py
@@ -3,6 +3,7 @@
 import time
 
 import pytest
+from async_generator import async_generator, yield_
 
 import aiomisc
 
@@ -44,11 +45,12 @@ def services(service: HashServer):
 
 
 @pytest.fixture()
+@async_generator
 async def proxy(tcp_proxy, localhost, server_port):
     proxy = tcp_proxy(localhost, server_port)
 
     try:
-        yield proxy
+        await yield_(proxy)
     finally:
         await asyncio.wait_for(proxy.close(), timeout=1)