arkitekt-next 0.24.0__tar.gz → 0.25.0__tar.gz

{arkitekt_next-0.24.0 → arkitekt_next-0.25.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arkitekt-next
-Version: 0.24.0
+Version: 0.25.0
 Summary: client for the arkitekt_next platform
 Author-email: jhnnsrs <jhnnsrs@gmail.com>
 License-Expression: MIT

arkitekt_next-0.25.0/arkitekt_next/pytest/__init__.py

@@ -0,0 +1 @@
+from .fixtures import *

arkitekt_next-0.25.0/arkitekt_next/pytest/fixtures.py

@@ -0,0 +1,86 @@
+from typing import Generator
+import pytest
+from arkitekt_next.cli.main import cli
+from click.testing import CliRunner
+from arkitekt_server.create import temp_server, ArkitektServerConfig
+from dokker import Deployment
+from dataclasses import dataclass
+from arkitekt_next.app import App
+from fakts_next.grants.remote import FaktsEndpoint
+from arkitekt_next import easy
+from dokker import local
+
+from arkitekt_next.service_registry import get_default_service_registry
+import pytest
+
+def pytest_addoption(parser: pytest.Parser) -> None:
+    """Register custom command line options for pytest."""
+    parser.addoption(
+        "--test-against",
+        action="store",
+        default="temp",
+        help="Base URL of the Arkitekt server to use in tests",
+    )
+
+
+@pytest.fixture(scope="session")
+def running_server(request: pytest.FixtureRequest) -> Generator[Deployment, None, None]:
+    """ Generates a local Arkitekt server deployment for testing purposes. """
+    
+    if not request.config.getoption("--test-against") == "temp":
+        raise ValueError("Invalid test environment specified. Currently only 'temp' is supported.")
+
+    config = ArkitektServerConfig()
+
+    with temp_server() as temp_path:
+        
+        setup = local(temp_path / "docker-compose.yaml")
+        
+        setup.add_health_check(
+            url=lambda spec: f"http://localhost:{spec.find_service('gateway').get_port_for_internal(80).published}/lok/ht",
+            service="lok",
+            timeout=5,
+            max_retries=10,
+        )
+        with setup as setup:
+            setup.down()
+            
+            
+            setup.up()
+            
+            setup.check_health()
+            yield setup
+            setup.down()
+ 
+ 
+ 
+@dataclass
+class EasyApp:
+    """Dataclass to hold the Arkitekt server deployment."""
+    deployment: Deployment
+    app: App         
+            
+
+@pytest.fixture(scope="session")
+def test_app(running_server: Deployment) -> Generator[EasyApp, None, None]:
+    """Fixture to ensure the Arkitekt server is running."""
+    
+    async def device_code_hook(endpoint: FaktsEndpoint, device_code: str):
+        
+        await running_server.arun(
+            "lok", f"uv run python manage.py validatecode {device_code} --user demo --org arkitektio"
+        )
+        
+        
+    registry = get_default_service_registry()
+        
+    assert registry, "Service registry must be initialized"    
+        
+
+
+    with easy(url=f"http://localhost:{running_server.spec.find_service('gateway').get_port_for_internal(80).published}", device_code_hook=device_code_hook) as app:
+        
+        yield EasyApp(
+            deployment=running_server,
+            app=app
+        )

{arkitekt_next-0.24.0 → arkitekt_next-0.25.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arkitekt-next"
-version = "0.24.0"
+version = "0.25.0"
 description = "client for the arkitekt_next platform"
 authors = [{ name = "jhnnsrs", email = "jhnnsrs@gmail.com" }]
 requires-python = ">=3.11,<4"
@@ -56,6 +56,9 @@ blok = [
 [project.scripts]
 arkitekt-next = "arkitekt_next.cli.main:cli"
 
+[project.entry-points.pytest11]
+arkitekt_next = "arkitekt_next.pytest"
+
 [tool.uv]
 dev-dependencies = [
     "autoflake>=1.7.7,<2",