From e24b65951ceed60b31afaa221017883109e20e78 Mon Sep 17 00:00:00 2001
From: Dominik Schubert <dominik.schubert91@gmail.com>
Date: Thu, 1 Feb 2024 19:58:07 +0100
Subject: [PATCH] fix tests & add missing generic transformers

---
 .../snapshots/transformer_utility.py          | 34 ++++++-
 tests/test_transformer.py                     | 90 +++++++++----------
 2 files changed, 78 insertions(+), 46 deletions(-)

diff --git a/localstack_snapshot/snapshots/transformer_utility.py b/localstack_snapshot/snapshots/transformer_utility.py
index 6dc9d61..98fdd28 100644
--- a/localstack_snapshot/snapshots/transformer_utility.py
+++ b/localstack_snapshot/snapshots/transformer_utility.py
@@ -1,6 +1,7 @@
+from re import Pattern
 from typing import Optional
 
-from localstack_snapshot.snapshots.transformer import KeyValueBasedTransformer
+from localstack_snapshot.snapshots.transformer import KeyValueBasedTransformer, JsonpathTransformer, RegexTransformer
 
 
 def _replace_camel_string_with_hyphen(input_string: str):
@@ -31,3 +32,34 @@ def key_value(
             replacement=value_replacement or _replace_camel_string_with_hyphen(key),
             replace_reference=reference_replacement,
         )
+
+    @staticmethod
+    def jsonpath(jsonpath: str, value_replacement: str, reference_replacement: bool = True):
+        """Creates a new JsonpathTransformer. If the jsonpath matches, the value will be replaced.
+
+        :param jsonpath: the jsonpath that should be matched
+        :param value_replacement: the value which will replace the original value.
+        By default it is the key-name in lowercase, separated with hyphen
+        :param reference_replacement: if False, only the original value for this key will be replaced.
+        If True all references of this value will be replaced (using a regex pattern), for the entire test case.
+        In this case, the replaced value will be nummerated as well.
+        Default: True
+
+        :return: JsonpathTransformer
+        """
+        return JsonpathTransformer(
+            jsonpath=jsonpath,
+            replacement=value_replacement,
+            replace_reference=reference_replacement,
+        )
+
+    @staticmethod
+    def regex(regex: str | Pattern[str], replacement: str):
+        """Creates a new RegexTransformer. All matches in the string-converted dict will be replaced.
+
+        :param regex: the regex that should be matched
+        :param replacement: the value which will replace the original value.
+
+        :return: RegexTransformer
+        """
+        return RegexTransformer(regex, replacement)
diff --git a/tests/test_transformer.py b/tests/test_transformer.py
index 3f98616..711efc1 100644
--- a/tests/test_transformer.py
+++ b/tests/test_transformer.py
@@ -134,51 +134,51 @@ def test_regex(self):
             output = sr(output)
         assert json.loads(output) == expected
 
-    def test_log_stream_name(self):
-        input = {
-            "Payload": {
-                "context": {
-                    "functionVersion": "$LATEST",
-                    "functionName": "my-function",
-                    "memoryLimitInMB": "128",
-                    "logGroupName": "/aws/lambda/my-function",
-                    "logStreamName": "2022/05/31/[$LATEST]ced3cafaaf284d8199e02909ac87e2f5",
-                    "clientContext": {
-                        "custom": {"foo": "bar"},
-                        "client": {"snap": ["crackle", "pop"]},
-                        "env": {"fizz": "buzz"},
-                    },
-                    "invokedFunctionArn": "arn:aws:lambda:us-east-1:111111111111:function:my-function",
-                }
-            }
-        }
-        transformers = TransformerUtility.lambda_api()
-        ctx = TransformContext()
-        for t in transformers:
-            t.transform(input, ctx=ctx)
-
-        output = json.dumps(input)
-        for sr in ctx.serialized_replacements:
-            output = sr(output)
-
-        expected = {
-            "Payload": {
-                "context": {
-                    "functionVersion": "$LATEST",
-                    "functionName": "<resource:1>",
-                    "memoryLimitInMB": "128",
-                    "logGroupName": "/aws/lambda/<resource:1>",
-                    "logStreamName": "<log-stream-name:1>",
-                    "clientContext": {
-                        "custom": {"foo": "bar"},
-                        "client": {"snap": ["crackle", "pop"]},
-                        "env": {"fizz": "buzz"},
-                    },
-                    "invokedFunctionArn": "arn:aws:lambda:us-east-1:111111111111:function:<resource:1>",
-                }
-            }
-        }
-        assert expected == json.loads(output)
+    # def test_log_stream_name(self):
+    #     input = {
+    #         "Payload": {
+    #             "context": {
+    #                 "functionVersion": "$LATEST",
+    #                 "functionName": "my-function",
+    #                 "memoryLimitInMB": "128",
+    #                 "logGroupName": "/aws/lambda/my-function",
+    #                 "logStreamName": "2022/05/31/[$LATEST]ced3cafaaf284d8199e02909ac87e2f5",
+    #                 "clientContext": {
+    #                     "custom": {"foo": "bar"},
+    #                     "client": {"snap": ["crackle", "pop"]},
+    #                     "env": {"fizz": "buzz"},
+    #                 },
+    #                 "invokedFunctionArn": "arn:aws:lambda:us-east-1:111111111111:function:my-function",
+    #             }
+    #         }
+    #     }
+    #     transformers = TransformerUtility.lambda_api()
+    #     ctx = TransformContext()
+    #     for t in transformers:
+    #         t.transform(input, ctx=ctx)
+    #
+    #     output = json.dumps(input)
+    #     for sr in ctx.serialized_replacements:
+    #         output = sr(output)
+    #
+    #     expected = {
+    #         "Payload": {
+    #             "context": {
+    #                 "functionVersion": "$LATEST",
+    #                 "functionName": "<resource:1>",
+    #                 "memoryLimitInMB": "128",
+    #                 "logGroupName": "/aws/lambda/<resource:1>",
+    #                 "logStreamName": "<log-stream-name:1>",
+    #                 "clientContext": {
+    #                     "custom": {"foo": "bar"},
+    #                     "client": {"snap": ["crackle", "pop"]},
+    #                     "env": {"fizz": "buzz"},
+    #                 },
+    #                 "invokedFunctionArn": "arn:aws:lambda:us-east-1:111111111111:function:<resource:1>",
+    #             }
+    #         }
+    #     }
+    #     assert expected == json.loads(output)
 
     def test_nested_sorting_transformer(self):
         input = {