Add compat for old pytest in type hints

dalibo · Oct 6, 2023 · 0075e45 · 0075e45
1 parent 5932782
commit 0075e45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -45,7 +45,7 @@ def get_version() -> str:
     ],
     extras_require={
         "test": [
-            "pytest",
+            "pytest >= 6.0.2",
         ],
     },
     entry_points={

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -7,6 +7,11 @@
 
 from . import PatroniAPI
 
+if tuple(int(v) for v in pytest.__version__.split(".", 2)) >= (6, 2):
+    TempPathFactory = pytest.TempPathFactory
+else:
+    from _pytest.tmpdir import TempPathFactory  # type: ignore[no-redef]
+
 
 @pytest.fixture(
     params=[False, True],
@@ -23,7 +28,7 @@ def datadir() -> Path:
 
 @pytest.fixture(scope="session")
 def patroni_api(
-    tmp_path_factory: pytest.TempPathFactory, datadir: Path
+    tmp_path_factory: TempPathFactory, datadir: Path
 ) -> Iterator[PatroniAPI]:
     """A fake HTTP server for the Patroni API serving files from a temporary
     directory.