Fix stacklevel for deprecated pytest fixtures

aio-libs · Jan 24, 2020 · 434a9f3 · 434a9f3
1 parent 7bdee2c
commit 434a9f3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/aiohttp/pytest_plugin.py b/aiohttp/pytest_plugin.py
@@ -234,7 +234,8 @@ def proactor_loop():  # type: ignore
 @pytest.fixture
 def unused_port(aiohttp_unused_port):  # type: ignore # pragma: no cover
     warnings.warn("Deprecated, use aiohttp_unused_port fixture instead",
-                  DeprecationWarning)
+                  DeprecationWarning,
+                  stacklevel=2)
     return aiohttp_unused_port
 
 
@@ -270,7 +271,8 @@ async def finalize():  # type: ignore
 @pytest.fixture
 def test_server(aiohttp_server):  # type: ignore  # pragma: no cover
     warnings.warn("Deprecated, use aiohttp_server fixture instead",
-                  DeprecationWarning)
+                  DeprecationWarning,
+                  stacklevel=2)
     return aiohttp_server
 
 
@@ -300,7 +302,8 @@ async def finalize():  # type: ignore
 @pytest.fixture
 def raw_test_server(aiohttp_raw_server):  # type: ignore  # pragma: no cover
     warnings.warn("Deprecated, use aiohttp_raw_server fixture instead",
-                  DeprecationWarning)
+                  DeprecationWarning,
+                  stacklevel=2)
     return aiohttp_raw_server
 
 
@@ -348,5 +351,6 @@ async def finalize():  # type: ignore
 @pytest.fixture
 def test_client(aiohttp_client):  # type: ignore  # pragma: no cover
     warnings.warn("Deprecated, use aiohttp_client fixture instead",
-                  DeprecationWarning)
+                  DeprecationWarning,
+                  stacklevel=2)
     return aiohttp_client