Stash changes

pkgs/clan-cli/tests/test_db_api.py

@@ -1,16 +1,81 @@
 from openapi_client import ApiClient
-from openapi_client.api import default_api
-from openapi_client.models import Machine, Status
+from openapi_client.api import DefaultApi
+from openapi_client.api.entities_api import EntitiesApi
+from openapi_client.api.services_api import ServicesApi
+from openapi_client.models import (
+    Entity,
+    EntityCreate,
+    Machine,
+    ServiceCreate,
+    Status,
+)
 
-default_entity_did_url = "entity_did=did%3Asov%3Atest%3A1234"
-default_entity_did = "did:sov:test:1234"
-default_entity_did2 = "did:sov:test:1235"
-default_entity_did3 = "did:sov:test:1236"
-default_entity_did4 = "did:sov:test:1237"
-default_entity_did5 = "did:sov:test:1238"
+uuids = [
+    "e95bb72f-b1b3-4452-8065-c7acf09068fc",
+    "411d772e-1ad0-4d99-8da0-133ab2972322",
+    "8cfdf359-c3b9-4951-9e51-08dce797725a",
+    "24b5b4de-9f5f-4e60-878e-cc5be085fd0d",
+    "d45f9687-c413-43b9-8e0d-cb610b39fcaf",
+    "083e09a0-1d71-4819-83e2-ce2a6d831713",
+    "e6f74e55-c163-4368-98c0-a2b04c99d6e3",
+    "1b577ba7-c9dd-4e66-b695-9350e9db0b6c",
+    "bfd9e653-98a4-4451-9d97-bcc2908f213d",
+    "0e481624-b886-437c-89a0-b9e73651cc72",
+]
 
 
 def test_health(api_client: ApiClient) -> None:
-    default = default_api.DefaultApi(api_client=api_client)
+    default = DefaultApi(api_client=api_client)
     res: Machine = default.health()
     assert res.status == Status.ONLINE
+
+
+def test_entities_empty(api_client: ApiClient) -> None:
+    entity = EntitiesApi(api_client=api_client)
+    res = entity.get_all_entities()
+    assert res == []
+
+
+def create_entities(num: int = 10) -> list[EntityCreate]:
+    res = []
+    for i in range(num):
+        en = EntityCreate(
+            did=f"did:sov:test:12{i}",
+            name=f"C{i}",
+            ip=f"127.0.0.1:{7000+i}",
+            visible=True,
+            other={},
+        )
+        res.append(en)
+    return res
+
+
+def create_service(idx: int, entity: Entity) -> ServiceCreate:
+    se = ServiceCreate(
+        uuid=uuids[idx],
+        service_name=f"Carlos Printing{idx}",
+        service_type="3D Printing",
+        endpoint_url=f"{entity.ip}/v1/print_daemon{idx}",
+        status="unknown",
+        other={"action": ["register", "deregister", "delete", "create"]},
+        entity_did=entity.did,
+    )
+
+    return se
+
+
+def test_create_entities(api_client: ApiClient) -> None:
+    api = EntitiesApi(api_client=api_client)
+    for own_entity in create_entities():
+        res: Entity = api.create_entity(own_entity)
+        assert res.did == own_entity.did
+        assert res.attached is False
+
+
+def test_create_services(api_client: ApiClient) -> None:
+    sapi = ServicesApi(api_client=api_client)
+    eapi = EntitiesApi(api_client=api_client)
+    for idx, entity in enumerate(eapi.get_all_entities()):
+        service_obj = create_service(idx, entity)
+        service = sapi.create_service(service_obj)
+        assert service.uuid == service_obj.uuid