test: update_tags

jorwoods · jorwoods · commit 01b171efcb68 · 2024-08-24T12:19:07.000-05:00
diff --git a/test/test_tagging.py b/test/test_tagging.py
@@ -1,3 +1,4 @@
+from contextlib import ExitStack
 import re
 from typing import Iterable
 import uuid
@@ -6,7 +7,6 @@
 import pytest
 import requests_mock
 import tableauserverclient as TSC
-from tableauserverclient.server.endpoint.resource_tagger import content
 
 
 @pytest.fixture
@@ -153,6 +153,42 @@ def test_delete_tags(get_server, endpoint_type, item, tags) -> None:
     urls = {r.url.split("/")[-1] for r in history}
     assert urls == tag_set
 
+@pytest.mark.parametrize("endpoint_type, item", *sample_taggable_items)
+@pytest.mark.parametrize("tags", sample_tags)
+def test_update_tags(get_server, endpoint_type, item, tags) -> None:
+    if isinstance(item, str):
+        return
+    endpoint = getattr(get_server, endpoint_type)
+    id_ = getattr(item, "id", item)
+    tags = set([tags] if isinstance(tags, str) else tags)
+    with ExitStack() as stack:
+        if hasattr(item, "_initial_tags"):
+            initial_tags = set(['x','y','z'])
+            item._initial_tags = initial_tags
+            add_tags_xml = add_tag_xml_response_factory(tags - initial_tags)
+            delete_tags_xml = add_tag_xml_response_factory(initial_tags - tags)
+            m = stack.enter_context(requests_mock.mock())
+            m.put(
+                f"{endpoint.baseurl}/{id_}/tags",
+                status_code=200,
+                text=add_tags_xml,
+            )
+
+            tag_paths = "|".join(initial_tags - tags)
+            tag_paths = f"({tag_paths})"
+            matcher = re.compile(rf"{endpoint.baseurl}\/{id_}\/tags\/{tag_paths}")
+            m.delete(
+                matcher,
+                status_code=200,
+                text=delete_tags_xml,
+            )
+
+        else:
+            stack.enter_context(pytest.raises(NotImplementedError))
+
+    
+        endpoint.update_tags(item)
+
 
 def test_tags_batch_add(get_server) -> None:
     server = get_server
