diff --git a/pkgs/clan-cli/tests/test_db_api.py b/pkgs/clan-cli/tests/test_db_api.py index a2431ca..8c28a81 100644 --- a/pkgs/clan-cli/tests/test_db_api.py +++ b/pkgs/clan-cli/tests/test_db_api.py @@ -114,7 +114,7 @@ def test_create_entities(api_client: ApiClient) -> None: def test_create_services(api_client: ApiClient) -> None: sapi = ServicesApi(api_client=api_client) eapi = EntitiesApi(api_client=api_client) - for midx, entity in enumerate(eapi.get_all_entities()): + for midx, entity in enumerate(eapi.get_entity_by_roles([Role("service_prosumer")])): service_obj = create_service(midx, entity) service = sapi.create_service(service_obj) assert service.uuid == service_obj.uuid