infrahub-testcontainers 1.2.3__py3-none-any.whl → 1.2.5__py3-none-any.whl

infrahub_testcontainers/container.py

@@ -57,6 +57,7 @@ PROJECT_ENV_VARIABLES: dict[str, str] = {
     "INFRAHUB_TESTING_API_SERVER_COUNT": "2",
     "INFRAHUB_TESTING_TASK_WORKER_COUNT": "2",
     "INFRAHUB_TESTING_PREFECT_UI_ENABLED": "true",
+    "INFRAHUB_TESTING_DOCKER_PULL": "true",
 }
 
 
@@ -173,6 +174,9 @@ class InfrahubDockerCompose(DockerCompose):
 
         up_cmd = [*base_cmd, "up"]
 
+        if self.get_env_var("INFRAHUB_TESTING_DOCKER_PULL") == "false":
+            up_cmd.extend(["--pull", "never"])
+
         # build means modifying the up command
         if self.wait:
             up_cmd.append("--wait")

infrahub_testcontainers/helpers.py

@@ -12,7 +12,7 @@ from .container import PROJECT_ENV_VARIABLES, InfrahubDockerCompose
 class TestInfrahubDocker:
     @pytest.fixture(scope="class")
     def infrahub_version(self) -> str:
-        return infrahub_version
+        return os.getenv("INFRAHUB_TESTING_IMAGE_VER") or infrahub_version
 
     @staticmethod
     def execute_ctl_run(address: str, script: str) -> str:

infrahub_testcontainers/models.py

@@ -1,4 +1,4 @@
-from datetime import UTC, datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Any
 
@@ -27,7 +27,7 @@ class InfrahubResultContext(BaseModel):
 
 class InfrahubActiveMeasurementItem(BaseModel):
     definition: MeasurementDefinition
-    start_time: datetime = datetime.now(UTC)
+    start_time: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
     context: dict[str, Any] = Field(default_factory=dict)
 
 

infrahub_testcontainers/performance_test.py

@@ -1,6 +1,6 @@
 import hashlib
 import json
-from datetime import UTC, datetime
+from datetime import datetime, timezone
 from types import TracebackType
 from typing import Any
 
@@ -35,7 +35,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 +57,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 +129,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.2.3.dist-info → infrahub_testcontainers-1.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: infrahub-testcontainers
-Version: 1.2.3
+Version: 1.2.5
 Summary: Testcontainers instance for Infrahub to easily build integration tests
 Home-page: https://opsmill.com
 License: Apache-2.0

{infrahub_testcontainers-1.2.3.dist-info → infrahub_testcontainers-1.2.5.dist-info}/RECORD

@@ -1,16 +1,16 @@
 infrahub_testcontainers/__init__.py,sha256=oPpmesGgYBSdKTg1L37FGwYBeao1EHury5SJGul-CT8,216
 infrahub_testcontainers/constants.py,sha256=mZ4hLvcf4rKk9wC7EId4MQxAY0sk4V99deB04N0J2bg,85
-infrahub_testcontainers/container.py,sha256=jQ0du9E_xg5oNk8YSsDzbJqBI1W7_J6jJpuo1OeoG1M,10488
+infrahub_testcontainers/container.py,sha256=FUkEVkPwUtYGfjHd6brLkI_3D4-qYmH5BGAOlyJrFU4,10652
 infrahub_testcontainers/docker-compose.test.yml,sha256=MqN8TPIAWNOZg93m_P2HH5etyYvtgYmwR92SyHR_GDQ,7017
 infrahub_testcontainers/haproxy.cfg,sha256=QUkG2Xu-hKoknPOeYKAkBT_xJH6U9CfIS0DTMFZJsnk,1305
-infrahub_testcontainers/helpers.py,sha256=RdKnBP7Mg-EVnXlSelj6INmoDLDRWabt9bOZcGjH7wM,3535
+infrahub_testcontainers/helpers.py,sha256=zsvBOql5qM2OX1ybPcklqF-nzWYHkZI3Gk3KZhxWOtU,3578
 infrahub_testcontainers/host.py,sha256=Z4_gGoGKKeM_HGVS7SdYL1FTNGyLBk8wzicdSKHpfmM,1486
 infrahub_testcontainers/measurements.py,sha256=gR-uTasSIFCXrwvnNpIpfsQIopKftT7pBiarCgIShaQ,2214
-infrahub_testcontainers/models.py,sha256=R735sO9i6D1TTtwlB0rweN3rWmZMYoSFfk1zt5XgT-Y,909
-infrahub_testcontainers/performance_test.py,sha256=82g4hfDuEesV7T8U12UjMV7ujZQMy_q30CSNQCdDVlQ,5993
+infrahub_testcontainers/models.py,sha256=ASYyvl7d_WQz_i7y8-3iab9hwwmCl3OCJavqVbe8nXU,954
+infrahub_testcontainers/performance_test.py,sha256=PSE03jevzv63n-v5rxNIzuQsbtCR-pVqLVpJddjCHhs,6025
 infrahub_testcontainers/plugin.py,sha256=vk33oG44MA2zxZwqMsS8_CkScm5LwuwwFmSOtmeAdMU,5357
 infrahub_testcontainers/prometheus.yml,sha256=610xQEyj3xuVJMzPkC4m1fRnCrjGpiRBrXA2ytCLa54,599
-infrahub_testcontainers-1.2.3.dist-info/METADATA,sha256=apUf37wqQqIRI6_8JJcNWCmvUcit1s_DtwNLVuyR2iA,981
-infrahub_testcontainers-1.2.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-infrahub_testcontainers-1.2.3.dist-info/entry_points.txt,sha256=gHOERdtVE0P8dYz6FHkn2KplpbXvCDJQnuWg_IP0-qQ,76
-infrahub_testcontainers-1.2.3.dist-info/RECORD,,
+infrahub_testcontainers-1.2.5.dist-info/METADATA,sha256=XRcGdOcQvTzGYZOegFvLY_BO-QPqzhby22uf7hO5E5A,981
+infrahub_testcontainers-1.2.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+infrahub_testcontainers-1.2.5.dist-info/entry_points.txt,sha256=gHOERdtVE0P8dYz6FHkn2KplpbXvCDJQnuWg_IP0-qQ,76
+infrahub_testcontainers-1.2.5.dist-info/RECORD,,