asyncio.timeout doesn't support sync context manager protocol (#183)

* asyncio.timeout doesn't support sync context manager protocol

* Fix tests

* Fix tests

Makefile

@@ -21,7 +21,7 @@ test:
 	@python3 -m pytest -vv --rootdir tests .
 
 pyenv:
-	echo aiopg-listen > .python-version && pyenv install -s 3.11.0 && pyenv virtualenv -f 3.11.0 aiopg-listen
+	echo aiopg-listen > .python-version && pyenv install -s 3.11.2 && pyenv virtualenv -f 3.11.2 aiopg-listen
 
 pyenv-delete:
 	pyenv virtualenv-delete -f aiopg-listen

aiopg_listen/listener.py

@@ -107,16 +107,14 @@ async def _process_notifications(
             notification: NotificationOrTimeout
 
             if notifications.empty():
-                try:
-                    timeout_ctx = (
-                        NO_TIMEOUT_CTX
-                        if notification_timeout == NO_TIMEOUT
-                        else timeout(notification_timeout)
-                    )
-                    with timeout_ctx:  # type: ignore
-                        notification = await notifications.get()
-                except asyncio.TimeoutError:
-                    notification = Timeout(channel)
+                if notification_timeout == NO_TIMEOUT:
+                    notification = await notifications.get()
+                else:
+                    try:
+                        async with timeout(notification_timeout):
+                            notification = await notifications.get()
+                    except asyncio.TimeoutError:
+                        notification = Timeout(channel)
             else:
                 while not notifications.empty():
                     notification = notifications.get_nowait()

requirements-dev.txt

@@ -1,4 +1,5 @@
 pytest==7.2.2
+pytest-aiohttp==1.0.4
 pytest-runner==6.0.0
 isort==5.12.0
 flake8==6.0.0

setup.cfg

@@ -0,0 +1,4 @@
+[tool:pytest]
+norecursedirs = .git .venv
+python_files = tests.py test_*.py *_tests.py
+asyncio_mode = auto