diff --git a/hubstorage/collectionsrt.py b/hubstorage/collectionsrt.py index 9071543..1fc0969 100644 --- a/hubstorage/collectionsrt.py +++ b/hubstorage/collectionsrt.py @@ -28,6 +28,9 @@ def set(self, _type, _name, _values): else: raise + def list(self): + return self.apiget('list') + def delete(self, _type, _name, _keys): return self.apipost((_type, _name, 'deleted'), jl=_keys) diff --git a/tests/test_collections.py b/tests/test_collections.py index 22d26e9..99c2d58 100644 --- a/tests/test_collections.py +++ b/tests/test_collections.py @@ -27,10 +27,14 @@ def post_get_delete_test(self): self.project.collections.new_versioned_cached_store(self.test_collection_name), ] + self.assertEqual(len(self.project.collections.list()), 4) + test_collections.extend( self.project.collections.new_collection(t, self.test_collection_name + 'b') for t in ('s', 'vs', 'cs', 'vcs')) + self.assertEqual(len(self.project.collections.list()), 9) + for col in test_collections: col.set(item_to_send) returned_item = col.get(test_key) @@ -38,6 +42,8 @@ def post_get_delete_test(self): col.delete(test_key) self.assertRaises(KeyError, col.get, test_key) + self.assertEqual(len(self.project.collections.list()), 0) + def post_scan_test(self): col = self.project.collections.new_store(self.test_collection_name)