fix: also allow psycopg2-binary as a valid driver (#208)

schireson · Apr 15, 2024 · d3f87de · d3f87de
1 parent f013626
commit d3f87de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytest_mock_resources/container/postgres.py b/src/pytest_mock_resources/container/postgres.py
@@ -135,7 +135,7 @@ def detect_driver(drivername: Optional[str] = None, async_: bool = False) -> str
         if any(Distribution.discover(name="asyncpg")):
             return "postgresql+asyncpg"
     else:
-        if any(Distribution.discover(name="psycopg2")):
+        if any(Distribution.discover(name="psycopg2")) or any(Distribution.discover(name="psycopg2-binary")):
             return "postgresql+psycopg2"
 
     raise ValueError(  # pragma: no cover