improv: test to cover custom jmespath opts

Author: heitorlessa

aws_lambda_powertools/utilities/validation/validator.py

@@ -24,7 +24,6 @@ def validator(
 
     **Validate incoming event**
 
-        import json
         from aws_lambda_powertools.utilities.validation import validator
 
         @validator(inbound_schema=json_schema_dict)
@@ -33,7 +32,6 @@ def handler(event, context):
 
     **Validate incoming and outgoing event**
 
-        import json
         from aws_lambda_powertools.utilities.validation import validator
 
         @validator(inbound_schema=json_schema_dict, outbound_schema=response_json_schema_dict)
@@ -42,7 +40,6 @@ def handler(event, context):
 
     **Unwrap event before validating against actual payload - using built-in envelopes**
 
-        import json
         from aws_lambda_powertools.utilities.validation import validator, envelopes
 
         @validator(inbound_schema=json_schema_dict, envelope=envelopes.API_GATEWAY_REST)
@@ -51,7 +48,6 @@ def handler(event, context):
 
     **Unwrap event before validating against actual payload - using custom JMESPath expression**
 
-        import json
         from aws_lambda_powertools.utilities.validation import validator
 
         @validator(inbound_schema=json_schema_dict, envelope="payload[*].my_data")
@@ -60,7 +56,6 @@ def handler(event, context):
 
     **Unwrap and deserialize JSON string event before validating against actual payload - using built-in functions**
 
-        import json
         from aws_lambda_powertools.utilities.validation import validator
 
         @validator(inbound_schema=json_schema_dict, envelope="Records[*].powertools_json(body)")

tests/functional/validator/test_validator.py

@@ -1,4 +1,6 @@
+import jmespath
 import pytest
+from jmespath import functions
 
 from aws_lambda_powertools.utilities.validation import envelopes, exceptions, validate, validator
 
@@ -32,7 +34,7 @@ def test_validate_json_string_no_envelope(schema, wrapped_event_json_string):
 
 def test_validate_invalid_schema_format(raw_event):
     with pytest.raises(exceptions.InvalidSchemaFormatError):
-        validate(event=raw_event, schema="")
+        validate(event=raw_event, schema="schema.json")
 
 
 def test_validate_invalid_envelope_expression(schema, wrapped_event):
@@ -103,3 +105,19 @@ def lambda_handler(evt, context):
 
     with pytest.raises(ValueError):
         lambda_handler(raw_event, {})
+
+
+def test_custom_jmespath_function_overrides_builtin_functions(schema, wrapped_event_json_string):
+    class CustomFunctions(functions.Functions):
+        @functions.signature({"types": ["string"]})
+        def _func_echo_decoder(self, value):
+            return value
+
+    jmespath_opts = {"custom_functions": CustomFunctions()}
+    with pytest.raises(jmespath.exceptions.UnknownFunctionError, match="Unknown function: powertools_json()"):
+        validate(
+            event=wrapped_event_json_string,
+            schema=schema,
+            envelope="powertools_json(data).payload",
+            jmespath_options=jmespath_opts,
+        )