pytest-databases 0.2.3__tar.gz → 0.2.4__tar.gz

{pytest_databases-0.2.3 → pytest_databases-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytest-databases
-Version: 0.2.3
+Version: 0.2.4
 Summary: Reusable database fixtures for any and all databases.
 Project-URL: Documentation, https://github.com/litestar-org/pytest-databases#readme
 Project-URL: Issues, https://github.com/litestar-org/pytest-databases/issues

{pytest_databases-0.2.3 → pytest_databases-0.2.4}/pyproject.toml

@@ -12,7 +12,7 @@ license = "MIT"
 name = "pytest-databases"
 readme = "README.md"
 requires-python = ">=3.8"
-version = "0.2.3"
+version = "0.2.4"
 #
 authors = [{ name = "Cody Fincher", email = "cody.fincher@gmail.com" }]
 keywords = [

{pytest_databases-0.2.3 → pytest_databases-0.2.4}/src/pytest_databases/docker/__init__.py

@@ -34,7 +34,7 @@ from typing import TYPE_CHECKING, Any, Callable
 
 import pytest
 
-from pytest_databases.helpers import wrap_sync
+from pytest_databases.helpers import simple_string_hash, wrap_sync
 
 if TYPE_CHECKING:
     from collections.abc import Awaitable, Generator
@@ -70,17 +70,18 @@ SKIP_DOCKER_COMPOSE: bool = bool(os.environ.get("SKIP_DOCKER_COMPOSE", False))
 USE_LEGACY_DOCKER_COMPOSE: bool = bool(
     os.environ.get("USE_LEGACY_DOCKER_COMPOSE", os.getenv("GITHUB_ACTIONS") != "true")
 )
+COMPOSE_PROJECT_NAME: str = f"pytest-databases-{simple_string_hash(__file__)}"
 
 
 class DockerServiceRegistry:
-    def __init__(self, worker_id: str) -> None:
+    def __init__(self, worker_id: str, compose_project_name: str = COMPOSE_PROJECT_NAME) -> None:
         self._running_services: set[str] = set()
         self.docker_ip = self._get_docker_ip()
         self._base_command = ["docker-compose"] if USE_LEGACY_DOCKER_COMPOSE else ["docker", "compose"]
         self._base_command.extend(
             [
                 f"--file={Path(__file__).parent / 'docker-compose.yml'}",
-                f"--project-name=pytest-databases-{worker_id}",
+                f"--project-name={compose_project_name}-{worker_id}",
             ],
         )
 
@@ -132,11 +133,16 @@ class DockerServiceRegistry:
 
 
 @pytest.fixture(scope="session")
-def docker_services(worker_id: str = "main") -> Generator[DockerServiceRegistry, None, None]:
+def compose_project_name() -> str:
+    return os.environ.get("COMPOSE_PROJECT_NAME", COMPOSE_PROJECT_NAME)
+
+
+@pytest.fixture(scope="session")
+def docker_services(compose_project_name: str, worker_id: str = "main") -> Generator[DockerServiceRegistry, None, None]:
     if os.getenv("GITHUB_ACTIONS") == "true" and sys.platform != "linux":
         pytest.skip("Docker not available on this platform")
 
-    registry = DockerServiceRegistry(worker_id)
+    registry = DockerServiceRegistry(worker_id, compose_project_name=compose_project_name)
     try:
         yield registry
     finally:

{pytest_databases-0.2.3 → pytest_databases-0.2.4}/src/pytest_databases/helpers.py

@@ -23,6 +23,7 @@
 
 from __future__ import annotations
 
+import hashlib
 import inspect
 from contextlib import AbstractAsyncContextManager, AbstractContextManager
 from functools import partial
@@ -81,3 +82,21 @@ def wrap_sync(fn: Callable[P, T]) -> Callable[P, Awaitable[T]]:
         return await anyio.to_thread.run_sync(partial(fn, *args, **kwargs))
 
     return wrapped
+
+
+def simple_string_hash(string_to_hash: str) -> str:
+    """Generates a short hash based on a string.
+
+    Args:
+      string_to_hash: The string to hash.
+
+    Returns:
+      A short hash string.
+    """
+
+    string_bytes = string_to_hash.encode("utf-8")
+    hasher = hashlib.sha256()
+    hasher.update(string_bytes)
+    digest = hasher.digest()
+    hex_string = digest.hex()
+    return hex_string[:12]