Enhance key value function test (#11)

tiurin · web-flow · commit 87e8d98c1a89 · 2025-03-13T14:54:53.000+01:00
diff --git a/tests/test_transformer.py b/tests/test_transformer.py
@@ -52,9 +52,15 @@ def test_key_value_replacement(self):
 
     def test_key_value_replacement_custom_function(self):
         input = {
-            "hello": "world",
+            "hello": "12characters",
             "hello2": "again",
-            "path": {"to": {"anotherkey": "hi", "inside": {"hello": "inside"}}},
+            "path": {
+                "to": {
+                    "anotherkey": "hi",
+                    "twelvesymbol": {"hello": "twelvesymbol"},
+                    "fifteen_symbols": {"hello": "fifteen_symbols"},
+                }
+            },
         }
 
         key_value = TransformerUtility.key_value_replacement_function(
@@ -64,34 +70,56 @@ def test_key_value_replacement_custom_function(self):
         )
 
         expected_key_value = {
-            "hello": "placeholder(5)",
+            "hello": "placeholder(12)",
             "hello2": "again",
-            "path": {"to": {"anotherkey": "hi", "inside": {"hello": "placeholder(6)"}}},
+            "path": {
+                "to": {
+                    "anotherkey": "hi",
+                    "twelvesymbol": {"hello": "placeholder(12)"},
+                    "fifteen_symbols": {"hello": "placeholder(15)"},
+                }
+            },
         }
 
-        copied = copy.deepcopy(input)
         ctx = TransformContext()
-        assert key_value.transform(copied, ctx=ctx) == expected_key_value
+        assert key_value.transform(input, ctx=ctx) == expected_key_value
         assert ctx.serialized_replacements == []
 
-        copied = copy.deepcopy(input)
+    def test_key_value_replacement_custom_function_reference_replacement(self):
+        input = {
+            "hello": "12characters",
+            "hello2": "again",
+            "path": {
+                "to": {
+                    "anotherkey": "hi",
+                    "twelvesymbol": {"hello": "twelvesymbol"},
+                    "fifteen_symbols": {"hello": "fifteen_symbols"},
+                }
+            },
+        }
+
         key_value = TransformerUtility.key_value_replacement_function(
             "hello",
             replacement_function=lambda k, v: f"placeholder({len(v)})",
             reference_replacement=True,
         )
         # replacement counters are per replacement key, so it will start from 1 again.
         expected_key_value_reference = {
-            "hello": "<placeholder(5):1>",
+            "hello": "<placeholder(12):1>",
             "hello2": "again",
             "path": {
-                "to": {"anotherkey": "hi", "<placeholder(6):1>": {"hello": "<placeholder(6):1>"}}
+                "to": {
+                    "anotherkey": "hi",
+                    "<placeholder(12):2>": {"hello": "<placeholder(12):2>"},
+                    "<placeholder(15):1>": {"hello": "<placeholder(15):1>"},
+                }
             },
         }
-        assert key_value.transform(copied, ctx=ctx) == copied
-        assert len(ctx.serialized_replacements) == 2
+        ctx = TransformContext()
+        assert key_value.transform(input, ctx=ctx) == input
+        assert len(ctx.serialized_replacements) == 3
 
-        tmp = json.dumps(copied, default=str)
+        tmp = json.dumps(input, default=str)
         for sr in ctx.serialized_replacements:
             tmp = sr(tmp)
 
