Improved tag test

labthings · Jun 29, 2020 · d7889d0 · d7889d0
1 parent b0e17dc
commit d7889d0
Showing 1 changed file with 5 additions and 19 deletions.
diff --git a/tests/test_server_spec_apispec.py b/tests/test_server_spec_apispec.py
@@ -38,32 +38,18 @@ def test_view_to_apispec_tags(spec):
     class Index(View):
         """Index docstring"""
 
-        tags = ["tag1", "tag2"]
+        tags = set(["tag1", "tag2"])
 
         def get(self):
             return "GET"
 
         def post(self):
             return "POST"
 
-    assert apispec.view_to_apispec_operations(Index, spec) == {
-        "get": {
-            "description": "Index docstring",
-            "summary": "Index docstring",
-            "tags": ["tag1", "tag2"],
-            "responses": {
-                200: {"description": "OK", "content": {"application/json": {}}}
-            },
-        },
-        "post": {
-            "description": "Index docstring",
-            "summary": "Index docstring",
-            "tags": ["tag1", "tag2"],
-            "responses": {
-                200: {"description": "OK", "content": {"application/json": {}}}
-            },
-        },
-    }
+    spec_ops = apispec.view_to_apispec_operations(Index, spec)
+
+    assert set(spec_ops["get"]["tags"]) == Index.tags
+    assert set(spec_ops["post"]["tags"]) == Index.tags
 
 
 def test_dict_to_apispec_operations_params(spec):