infrahub-testcontainers 1.5.0b2__tar.gz → 1.5.1__tar.gz

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: infrahub-testcontainers
-Version: 1.5.0b2
+Version: 1.5.1
 Summary: Testcontainers instance for Infrahub to easily build integration tests
 License: Apache-2.0
 Author: OpsMill

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/infrahub_testcontainers/container.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import shutil
 import time
@@ -72,7 +74,10 @@ class InfrahubDockerCompose(DockerCompose):
 
     @classmethod
     def init(
-        cls, directory: Path | None = None, version: str | None = None, deployment_type: str | None = None
+        cls,
+        directory: Path | None = None,
+        version: str | None = None,
+        deployment_type: str | None = None,
     ) -> Self:
         if not directory:
             directory = Path.cwd()
@@ -84,7 +89,11 @@ class InfrahubDockerCompose(DockerCompose):
         if version == "local" and infrahub_image_version:
             version = infrahub_image_version
 
-        compose = cls(project_name=cls.generate_project_name(), context=directory, deployment_type=deployment_type)
+        compose = cls(
+            project_name=cls.generate_project_name(),
+            context=directory,
+            deployment_type=deployment_type,
+        )
         compose.create_docker_file(directory=directory)
         compose.create_env_file(directory=directory, version=version)
 
@@ -240,7 +249,10 @@ class InfrahubDockerCompose(DockerCompose):
         }
 
     def database_create_backup(
-        self, backup_name: str = "neo4j_database.backup", dest_dir: Path | None = None, compress: bool = False
+        self,
+        backup_name: str = "neo4j_database.backup",
+        dest_dir: Path | None = None,
+        compress: bool = False,
     ) -> None:
         assert self.use_neo4j_enterprise
 
@@ -283,7 +295,14 @@ class InfrahubDockerCompose(DockerCompose):
                 self.start_container(service_name=service_name)
 
             self.exec_in_container(
-                command=["cypher-shell", "-u", "neo4j", "-p", "admin", "STOP DATABASE neo4j;"],
+                command=[
+                    "cypher-shell",
+                    "-u",
+                    "neo4j",
+                    "-p",
+                    "admin",
+                    "STOP DATABASE neo4j;",
+                ],
                 service_name=service_name,
             )
 
@@ -368,7 +387,14 @@ class InfrahubDockerCompose(DockerCompose):
             time.sleep(10)
 
             self.exec_in_container(
-                command=["cypher-shell", "-u", "neo4j", "-p", "admin", "DROP DATABASE neo4j;"],
+                command=[
+                    "cypher-shell",
+                    "-u",
+                    "neo4j",
+                    "-p",
+                    "admin",
+                    "DROP DATABASE neo4j;",
+                ],
                 service_name=service_name,
             )
 

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/infrahub_testcontainers/helpers.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import subprocess  # noqa: S404
 import uuid
@@ -24,13 +26,21 @@ class TestInfrahubDocker:
         env["INFRAHUB_API_TOKEN"] = PROJECT_ENV_VARIABLES["INFRAHUB_TESTING_INITIAL_ADMIN_TOKEN"]
         env["INFRAHUB_MAX_CONCURRENT_EXECUTION"] = "1"
         result = subprocess.run(  # noqa: S602
-            f"infrahubctl run {script}", shell=True, capture_output=True, text=True, env=env, check=False
+            f"infrahubctl run {script}",
+            shell=True,
+            capture_output=True,
+            text=True,
+            env=env,
+            check=False,
         )
         return result.stdout
 
     @staticmethod
     def execute_command(
-        command: str, address: str, concurrent_execution: int = 10, pagination_size: int = 50
+        command: str,
+        address: str,
+        concurrent_execution: int = 10,
+        pagination_size: int = 50,
     ) -> subprocess.CompletedProcess[str]:
         env = os.environ.copy()
         env["INFRAHUB_ADDRESS"] = address
@@ -79,7 +89,9 @@ class TestInfrahubDocker:
         deployment_type: str | None,
     ) -> InfrahubDockerCompose:
         return InfrahubDockerCompose.init(
-            directory=tmp_directory, version=infrahub_version, deployment_type=deployment_type
+            directory=tmp_directory,
+            version=infrahub_version,
+            deployment_type=deployment_type,
         )
 
     @pytest.fixture(scope="class")
@@ -90,7 +102,10 @@ class TestInfrahubDocker:
             tests_failed_during_class = request.session.testsfailed - tests_failed_before_class
             if tests_failed_during_class > 0:
                 stdout, stderr = infrahub_compose.get_logs("infrahub-server", "task-worker")
-                warnings.warn(f"Container logs:\nStdout:\n{stdout}\nStderr:\n{stderr}", stacklevel=2)
+                warnings.warn(
+                    f"Container logs:\nStdout:\n{stdout}\nStderr:\n{stderr}",
+                    stacklevel=2,
+                )
             infrahub_compose.stop()
 
         request.addfinalizer(cleanup)

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/infrahub_testcontainers/models.py

@@ -1,6 +1,8 @@
-from datetime import UTC, datetime
+from __future__ import annotations
+
+from datetime import datetime, timezone
 from enum import Enum
-from typing import Any
+from typing import Any, Union
 
 from pydantic import BaseModel, Field
 
@@ -15,24 +17,24 @@ class ContextUnit(str, Enum):
 class MeasurementDefinition(BaseModel):
     name: str
     description: str
-    dimensions: list[str] = Field(default_factory=dict)
+    dimensions: list[str] = Field(default_factory=list)
     unit: ContextUnit
 
 
 class InfrahubResultContext(BaseModel):
     name: str
-    value: int | float | str
+    value: Union[int, float, str]
     unit: ContextUnit
 
 
 class InfrahubActiveMeasurementItem(BaseModel):
     definition: MeasurementDefinition
-    start_time: datetime = Field(default_factory=lambda: datetime.now(UTC))
+    start_time: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
     context: dict[str, Any] = Field(default_factory=dict)
 
 
 class InfrahubMeasurementItem(BaseModel):
     name: str
-    value: int | float | str
+    value: Union[int, float, str]
     unit: ContextUnit
     context: dict[str, Any] = Field(default_factory=dict)

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/infrahub_testcontainers/performance_test.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import hashlib
 import json
-from datetime import UTC, datetime
+from datetime import datetime, timezone
 from types import TracebackType
 from typing import Any
 
@@ -35,7 +37,7 @@ class InfrahubPerformanceTest:
         self.env_vars = {}
         self.project_name = ""
         self.test_info = {}
-        self.start_time = datetime.now(UTC)
+        self.start_time = datetime.now(timezone.utc)
         self.end_time: datetime | None = None
         self.results_url = results_url
         self.scraper_endpoint = ""
@@ -57,7 +59,7 @@ class InfrahubPerformanceTest:
 
     def finalize(self, session: pytest.Session) -> None:
         if self.initialized:
-            self.end_time = datetime.now(UTC)
+            self.end_time = datetime.now(timezone.utc)
             self.extract_test_session_information(session)
             self.send_results()
 
@@ -129,7 +131,7 @@ class InfrahubPerformanceTest:
         if not exc_type and self.active_measurements:
             self.add_measurement(
                 definition=self.active_measurements.definition,
-                value=(datetime.now(UTC) - self.active_measurements.start_time).total_seconds() * 1000,
+                value=(datetime.now(timezone.utc) - self.active_measurements.start_time).total_seconds() * 1000,
                 context=self.active_measurements.context,
             )
 

{infrahub_testcontainers-1.5.0b2 → infrahub_testcontainers-1.5.1}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "infrahub-testcontainers"
-version = "1.5.0b2"
+version = "1.5.1"
 requires-python = ">=3.9"
 
 [tool.poetry]
 name = "infrahub-testcontainers"
-version = "1.5.0b2"
+version = "1.5.1"
 description = "Testcontainers instance for Infrahub to easily build integration tests"
 authors = ["OpsMill <info@opsmill.com>"]
 readme = "README.md"
@@ -43,3 +43,9 @@ prefect-client = "3.4.23"
 
 [tool.poetry.group.dev.dependencies]
 rich = "^13.9.4"
+pytest-cov = "7.0.0"
+pytest-xdist = "~3.4"
+
+[tool.ruff]
+line-length = 120
+target-version = "py39"