cloudflare test 4

Author: petercrocker

.gitignore

@@ -15,4 +15,5 @@ __pycache__
 **/.pytest_cache
 **/.ruff_cache
 docker-compose.override.yml
-docker-compose.yml
+docker-compose.yml
+infrahub_demo.egg-info

pyproject.toml

@@ -1,3 +1,7 @@
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
 [project]
 authors = [
     {name = "Tomek Zajac", email = "tomek@mastuazi.com"},
@@ -20,6 +24,9 @@ version = "0.1.5"
 description = "Design driven automation with Infrahub"
 readme = "README.md"
 
+[tool.setuptools]
+packages = []
+
 [tool.mypy]
 pretty = true
 ignore_missing_imports = true

tests/integration/test_workflow.py

@@ -1,3 +1,4 @@
+# pyright: reportAttributeAccessIssue=false
 """Integration test for the DC-2 demo workflow.
 
 This test validates the complete workflow from the README:
@@ -9,10 +10,14 @@
 6. Run DC generator
 7. Create proposed change
 8. Validate and merge
+
+Note: Pylance warnings about .value attributes are suppressed at file level.
+The Infrahub SDK uses dynamic attribute generation at runtime.
 """
 
 import logging
 import time
+from pathlib import Path
 
 import pytest
 from infrahub_sdk import InfrahubClient, InfrahubClientSync
@@ -48,9 +53,9 @@ def test_01_schema_load(self, client_main: InfrahubClientSync) -> None:
         logging.info("Schema load stderr: %s", load_schemas.stderr)
 
         # Check that schemas loaded successfully (even if returncode is non-zero due to warnings)
-        assert "loaded successfully" in load_schemas.stdout or load_schemas.returncode == 0, (
-            f"Schema load failed: {load_schemas.stdout}\n{load_schemas.stderr}"
-        )
+        assert (
+            "loaded successfully" in load_schemas.stdout or load_schemas.returncode == 0
+        ), f"Schema load failed: {load_schemas.stdout}\n{load_schemas.stderr}"
 
     def test_02_load_menu(self, client_main: InfrahubClientSync) -> None:
         """Load menu definitions."""
@@ -102,11 +107,10 @@ async def test_05_add_repository(
 
         client = async_client_main
         src_directory = PROJECT_DIRECTORY
-
         git_repository = GitRepo(
             name="demo_repo",
             src_directory=src_directory,
-            dst_directory=remote_repos_dir,
+            dst_directory=Path(remote_repos_dir),
         )
 
         response = await git_repository.add_to_infrahub(client=client)
@@ -147,9 +151,13 @@ async def test_05_add_repository(
         )
         logging.info("Repository synchronized successfully")
 
-    def test_06_create_branch(self, client_main: InfrahubClientSync, default_branch: str) -> None:
+    def test_06_create_branch(
+        self, client_main: InfrahubClientSync, default_branch: str
+    ) -> None:
         """Create a new branch for the DC-2 deployment."""
-        logging.info("Starting test: test_06_create_branch - branch: %s", default_branch)
+        logging.info(
+            "Starting test: test_06_create_branch - branch: %s", default_branch
+        )
 
         # Check if branch already exists
         existing_branches = client_main.branch.all()
@@ -159,7 +167,9 @@ def test_06_create_branch(self, client_main: InfrahubClientSync, default_branch:
             client_main.branch.create(default_branch)
             logging.info("Created branch: %s", default_branch)
 
-    def test_07_load_dc2_design(self, client_main: InfrahubClientSync, default_branch: str) -> None:
+    def test_07_load_dc2_design(
+        self, client_main: InfrahubClientSync, default_branch: str
+    ) -> None:
         """Load DC-2 design data onto the branch."""
         logging.info("Starting test: test_07_load_dc2_design")
 
@@ -236,7 +246,9 @@ async def test_09_run_generator(
             time.sleep(10)
 
         if not definition:
-            pytest.skip("Generator definition 'create_dc' not available - skipping generator test")
+            pytest.skip(
+                "Generator definition 'create_dc' not available - skipping generator test"
+            )
 
         logging.info("Found generator: %s", definition.name.value)
 
@@ -262,7 +274,7 @@ async def test_09_run_generator(
         logging.info("Generator task started: %s", task_id)
 
         # Wait for generator to complete (can take a while for DC generation)
-        task = client.task.wait_for_completion(id=task_id, timeout=1800)
+        task = await client.task.wait_for_completion(id=task_id, timeout=1800)
 
         assert task.state == TaskState.COMPLETED, (
             f"Task {task.id} - generator create_dc did not complete successfully. "
@@ -283,15 +295,19 @@ async def test_10_verify_devices_created(
         devices = await client.all(kind="DcimGenericDevice")
 
         if not devices:
-            pytest.skip("No devices found - generator may not have run (test_09 may have been skipped)")
+            pytest.skip(
+                "No devices found - generator may not have run (test_09 may have been skipped)"
+            )
 
         logging.info("Found %d devices after generator run", len(devices))
 
         # Check for specific device types (spine, leaf, etc.)
         device_names = [device.name.value for device in devices]
         logging.info("Created devices: %s", ", ".join(device_names))
 
-    def test_11_create_diff(self, client_main: InfrahubClientSync, default_branch: str) -> None:
+    def test_11_create_diff(
+        self, client_main: InfrahubClientSync, default_branch: str
+    ) -> None:
         """Create a diff for the branch."""
         logging.info("Starting test: test_11_create_diff")
 
@@ -392,7 +408,11 @@ def test_12_create_proposed_change(
         if failed_validations:
             for result in failed_validations:
                 name = result.name.value if hasattr(result, "name") else str(result.id)
-                conclusion = result.conclusion.value if hasattr(result, "conclusion") else "unknown"
+                conclusion = (
+                    result.conclusion.value
+                    if hasattr(result, "conclusion")
+                    else "unknown"
+                )
                 logging.error(
                     "Validation failed: %s - %s",
                     name,
@@ -404,10 +424,10 @@ def test_12_create_proposed_change(
         logging.info("Validations completed. Results:")
         for result in validation_results:
             name = result.name.value if hasattr(result, "name") else str(result.id)
-            conclusion = result.conclusion.value if hasattr(result, "conclusion") else "unknown"
-            logging.info(
-                "  - %s: %s", name, conclusion
+            conclusion = (
+                result.conclusion.value if hasattr(result, "conclusion") else "unknown"
             )
+            logging.info("  - %s: %s", name, conclusion)
 
     def test_13_merge_proposed_change(
         self, client_main: InfrahubClientSync, default_branch: str
@@ -444,7 +464,9 @@ def test_13_merge_proposed_change(
         )
         logging.info("Proposed change merged successfully")
 
-    async def test_14_verify_merge_to_main(self, async_client_main: InfrahubClient) -> None:
+    async def test_14_verify_merge_to_main(
+        self, async_client_main: InfrahubClient
+    ) -> None:
         """Verify that DC-2 and devices exist in main branch."""
         logging.info("Starting test: test_14_verify_merge_to_main")
 

tests/unit/test_j2_transforms.yml

@@ -13,14 +13,3 @@ infrahub_tests:
         spec:
           kind: "jinja2-transform-unit-render"
           directory: simulators
-  # - resource: Jinja2Transform
-  #   resource_name: topology_clab
-  #   tests:
-  #     - name: syntax_check
-  #       spec:
-  #         kind: jinja2-transform-smoke
-  #     - name: baseline
-  #       expect: PASS
-  #       spec:
-  #         kind: "jinja2-transform-unit-render"
-  #         directory: tests/simulatorsy