Fix test

koordinates · Oct 24, 2023 · 5d93d6c · 5d93d6c
1 parent ead93c0
commit 5d93d6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/koordinates/api/utils.py b/koordinates/api/utils.py
@@ -96,7 +96,13 @@ def capabilities_from_dataset_response(dataset: dict) -> Set[Capability]:
         if not dataset.get("repository"):
             capabilities.remove(Capability.Clone)
         else:
-            repo_user_capabilities = dataset["repository"].get(
+            repo = dataset["repository"]
+            if repo and not isinstance(repo, dict):
+                from .client import KoordinatesClient
+                repo = KoordinatesClient.instance().get_json(
+                    repo
+                )
+            repo_user_capabilities = repo.get(
                 "user_capabilities", []
             )
             if 'can-clone' not in repo_user_capabilities:

diff --git a/koordinates/test/test_dataset.py b/koordinates/test/test_dataset.py
@@ -59,6 +59,10 @@ def test_point_cloud(self):
         "download",
         "write"
     ],
+    "user_capabilities": [
+      "can-star",
+      "can-clone"
+    ],
     "name": "some repo name",
     "description": ""
 }