Add tests

viren-nadkarni · viren-nadkarni · commit 475fa6e84a2a · 2025-02-26T18:04:14.000+05:30
diff --git a/tests/unit/test_tagging.py b/tests/unit/test_tagging.py
@@ -1,30 +1,47 @@
-import unittest
+import pytest
 
 from localstack.utils.tagging import TaggingService
 
 
-class TestTaggingService(unittest.TestCase):
-    svc = TaggingService()
+class TestTaggingService:
+    @pytest.fixture
+    def tagging_service(self):
+        def _factory(**kwargs):
+            return TaggingService(**kwargs)
 
-    def test_list_empty(self):
-        result = self.svc.list_tags_for_resource("test")
-        self.assertEqual({"Tags": []}, result)
+        return _factory
 
-    def test_create_tag(self):
+    def test_list_empty(self, tagging_service):
+        svc = tagging_service()
+        result = svc.list_tags_for_resource("test")
+        assert result == {"Tags": []}
+
+    def test_create_tag(self, tagging_service):
+        svc = tagging_service()
         tags = [{"Key": "key_key", "Value": "value_value"}]
-        self.svc.tag_resource("arn", tags)
-        actual = self.svc.list_tags_for_resource("arn")
+        svc.tag_resource("arn", tags)
+        actual = svc.list_tags_for_resource("arn")
         expected = {"Tags": [{"Key": "key_key", "Value": "value_value"}]}
-        self.assertDictEqual(expected, actual)
+        assert actual == expected
 
-    def test_delete_tag(self):
+    def test_delete_tag(self, tagging_service):
+        svc = tagging_service()
         tags = [{"Key": "key_key", "Value": "value_value"}]
-        self.svc.tag_resource("arn", tags)
-        self.svc.untag_resource("arn", ["key_key"])
-        result = self.svc.list_tags_for_resource("arn")
-        self.assertEqual({"Tags": []}, result)
-
-    def test_list_empty_delete(self):
-        self.svc.untag_resource("arn", ["key_key"])
-        result = self.svc.list_tags_for_resource("arn")
-        self.assertEqual({"Tags": []}, result)
+        svc.tag_resource("arn", tags)
+        svc.untag_resource("arn", ["key_key"])
+        result = svc.list_tags_for_resource("arn")
+        assert result == {"Tags": []}
+
+    def test_list_empty_delete(self, tagging_service):
+        svc = tagging_service()
+        svc.untag_resource("arn", ["key_key"])
+        result = svc.list_tags_for_resource("arn")
+        assert result == {"Tags": []}
+
+    def test_field_name_override(self, tagging_service):
+        svc = tagging_service(key_field="keY", value_field="valuE")
+        tags = [{"keY": "my", "valuE": "congratulations"}]
+        svc.tag_resource("arn", tags)
+        assert svc.list_tags_for_resource("arn") == {
+            "Tags": [{"keY": "my", "valuE": "congratulations"}]
+        }
