119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
"""Unit tests for cloud.sdk.client.CloudClient (task 8.2)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from cloud.config import CloudConfig
|
|
from cloud.plugins import PluginRegistry
|
|
from cloud.pool import DevicePool
|
|
from cloud.sdk.api import create_cloud_router
|
|
from cloud.sdk.client import CloudClient
|
|
from cloud.scheduler import TaskScheduler
|
|
from cloud.store import CloudStore
|
|
from core.models import Device
|
|
|
|
|
|
pytest.importorskip("fastapi")
|
|
from fastapi import FastAPI # noqa: E402
|
|
from fastapi.testclient import TestClient # noqa: E402
|
|
|
|
|
|
def _config() -> CloudConfig:
|
|
return CloudConfig(
|
|
sync_interval_seconds=30,
|
|
stale_after_seconds=60,
|
|
max_queue_depth=100,
|
|
default_assignment_strategy="fifo_match",
|
|
api_version_prefix="/v1",
|
|
db_path="cloud/cloud.sqlite3",
|
|
)
|
|
|
|
|
|
def _client_and_pool(tmp_path):
|
|
store = CloudStore(tmp_path / "cloud.sqlite3")
|
|
pool = DevicePool(store, _config())
|
|
scheduler = TaskScheduler(pool, store, _config())
|
|
plugin_registry = PluginRegistry(store)
|
|
app = FastAPI()
|
|
app.include_router(
|
|
create_cloud_router(
|
|
pool=pool,
|
|
scheduler=scheduler,
|
|
plugin_registry=plugin_registry,
|
|
)
|
|
)
|
|
test_client = TestClient(app)
|
|
cloud_client = CloudClient("http://testserver", http_client=test_client)
|
|
return cloud_client, pool
|
|
|
|
|
|
def test_client_submit_and_get_status_round_trip(tmp_path) -> None:
|
|
client, _ = _client_and_pool(tmp_path)
|
|
|
|
submission = client.submit_task(goal="open settings")
|
|
assert "task_id" in submission
|
|
task_id = submission["task_id"]
|
|
|
|
status = client.get_task_status(task_id)
|
|
assert status["id"] == task_id
|
|
assert status["status"] == "queued"
|
|
assert status["goal"] == "open settings"
|
|
|
|
|
|
def test_client_list_devices_and_hosts(tmp_path) -> None:
|
|
client, pool = _client_and_pool(tmp_path)
|
|
pool.sync_host_devices(
|
|
"host-a",
|
|
[Device(id="dev-1", driver_type="wda", status="idle")], # type: ignore[arg-type]
|
|
address="a:8000",
|
|
)
|
|
|
|
devices = client.list_devices()
|
|
assert [d["device_id"] for d in devices] == ["dev-1"]
|
|
assert devices[0]["host_id"] == "host-a"
|
|
|
|
hosts = client.list_hosts()
|
|
assert [h["host_id"] for h in hosts] == ["host-a"]
|
|
|
|
|
|
def test_client_plugin_listing_and_registration(tmp_path) -> None:
|
|
client, _ = _client_and_pool(tmp_path)
|
|
|
|
assert client.list_plugins() == []
|
|
registered = client.register_plugin(
|
|
name="demo",
|
|
version="1.0.0",
|
|
entry_point_kind="tool",
|
|
target="cloud.store:CloudStore",
|
|
)
|
|
assert registered["name"] == "demo"
|
|
assert registered["wired"] is False
|
|
|
|
listed = client.list_plugins()
|
|
assert [p["name"] for p in listed] == ["demo"]
|
|
|
|
|
|
def test_client_submit_with_constraints(tmp_path) -> None:
|
|
client, pool = _client_and_pool(tmp_path)
|
|
pool.sync_host_devices(
|
|
"host-a",
|
|
[Device(id="dev-1", driver_type="wda", status="idle")], # type: ignore[arg-type]
|
|
)
|
|
submission = client.submit_task(
|
|
goal="x",
|
|
driver_type="wda",
|
|
capability_tags=[],
|
|
)
|
|
task_id = submission["task_id"]
|
|
status = client.get_task_status(task_id)
|
|
assert status["status"] == "queued"
|
|
|
|
|
|
def test_client_unknown_task_raises(tmp_path) -> None:
|
|
import httpx
|
|
|
|
client, _ = _client_and_pool(tmp_path)
|
|
with pytest.raises(httpx.HTTPStatusError):
|
|
client.get_task_status("does-not-exist")
|