Add package mode

Author: minitriga

infrahub_sdk/ctl/example_repo/copier.yml

@@ -26,3 +26,7 @@ tests:
   type: bool
   help: Would you like to have testing?
   default: false
+package_mode:
+  type: bool
+  help: Would you like to have a python library initialized?
+  default: false

infrahub_sdk/ctl/example_repo/poetry.lock

@@ -4,13 +4,23 @@ version = "0.1.0"
 description = "Infrahub Repository"
 authors = []
 readme = "README.md"
+{% if package_mode %}
+packages = [
+    { include = "lib" }
+]
+{% else %}
+package-mode = false
+{% endif %}
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.13"
-
+infrahub-sdk = {extras = ["ctl"], version = "^1.7.2"}
 
 [tool.poetry.group.dev.dependencies]
-ruff = "^0.9.6"
+ruff = "^0.9.10"
+pytest = "^8.3.5"
+infrahub-testcontainers = "^1.1.8"
+pytest-asyncio = "^0.25.3"
 
 [build-system]
 requires = ["poetry-core"]

infrahub_sdk/ctl/example_repo/pyproject.toml.jinja

@@ -0,0 +1,32 @@
+# yaml-language-server: $schema=https://schema.infrahub.app/infrahub/schema/latest.json
+---
+version: "1.0"
+nodes:
+  - name: Device
+    namespace: Network
+    human_friendly_id: ['hostname__value']
+    attributes:
+      - name: hostname
+        kind: Text
+        unique: true
+      - name: model
+        kind: Text
+    relationships:
+      - name: interfaces
+        cardinality: many
+        peer: NetworkInterface
+        kind: Component
+  - name: Interface
+    namespace: Network
+    attributes:
+      - name: name
+        kind: Text
+      - name: description
+        kind: Text
+        optional: true
+    relationships:
+      - name: device
+        cardinality: one
+        peer: NetworkDevice
+        optional: false
+        kind: Parent

infrahub_sdk/ctl/example_repo/schemas/example.yml

@@ -0,0 +1,26 @@
+import pytest
+from pathlib import Path
+
+from typing import Any
+from infrahub_sdk.yaml import SchemaFile
+
+CURRENT_DIRECTORY = Path(__file__).parent.resolve()
+
+
+@pytest.fixture
+def root_directory() -> Path:
+    """
+    Return the path of the root directory of the repository.
+    """
+    return CURRENT_DIRECTORY.parent.parent
+
+
+@pytest.fixture
+def schemas_directory(root_directory: Path) -> Path:
+    return root_directory / "schema"
+
+
+@pytest.fixture
+def schemas(schemas_directory: Path) -> list[dict[str, Any]]:
+    schema_files = SchemaFile.load_from_disk(paths=[schemas_directory])
+    return [item.content for item in schema_files if item.content]

infrahub_sdk/ctl/example_repo/{% if package_mode %}lib{% endif %}/__init__.py

@@ -0,0 +1,24 @@
+import pytest
+from pathlib import Path
+from infrahub_sdk import InfrahubClient
+from infrahub_sdk.testing.docker import TestInfrahubDockerClient
+from infrahub_sdk.protocols import CoreGenericRepository
+from infrahub_sdk.testing.repository import GitRepo
+
+
+class TestInfrahub(TestInfrahubDockerClient):
+    @pytest.fixture(scope="class")
+    def infrahub_version(self) -> str:
+        """Required to define the version of infrahub to use."""
+        return "1.1.8"
+
+    @pytest.mark.asyncio
+    async def test_load_schema(
+        self, default_branch: str, client: InfrahubClient, schemas
+    ):
+        await client.schema.wait_until_converged(branch=default_branch)
+
+        resp = await client.schema.load(
+            schemas=schemas, branch=default_branch, wait_until_converged=True
+        )
+        assert resp.errors == {}

infrahub_sdk/ctl/example_repo/{% if package_mode %}lib{% endif %}/example.py

@@ -307,12 +307,12 @@ def test_repo_init(self) -> None:
                 "scripts": True,
                 "tests": True,
                 "transforms": True,
+                "package_mode": False,
             }
 
             yaml.safe_dump(answers, temp_yaml)
             temp_yaml.close()
             runner.invoke(app, ["repository", "init", str(dst), "--data", str(yaml_path)])
-
             coppied_answers = yaml.safe_load((dst / ".copier-answers.yml").read_text())
             coppied_answers.pop("_src_path")