Fix TestToolshed.test_categories_client test for TS 2.0

nsoranzo · Dec 11, 2023 · 20ae9d0 · 20ae9d0
1 parent 1044c33
commit 20ae9d0
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/bioblend/_tests/TestToolshed.py b/bioblend/_tests/TestToolshed.py
@@ -26,8 +26,18 @@ def test_categories_client(self):
         # get_repositories
         repositories = self.ts.categories.get_repositories(visualization_category_id)
         repositories_reversed = self.ts.categories.get_repositories(visualization_category_id, sort_order="desc")
-        assert repositories["repositories"][0]["model_class"] == "Repository"
         assert len(repositories["repositories"]) > 200
+        assert {
+            "deprecated",
+            "description",
+            "homepage_url",
+            "id",
+            "name",
+            "owner",
+            "remote_repository_url",
+            "type",
+            "update_time",
+        } <= set(repositories["repositories"][0].keys())
         assert repositories["repositories"][0] == repositories_reversed["repositories"][-1]
 
     def test_repositories_client(self):