Skip to content

Commit 8487c89

Browse files
committed
tests(validator): add missing tests for validator
Changes: * validator - Add test for the validator decator with an envelope * data_classes - Call DictWrapper constructor with the nested dict
1 parent b654ca6 commit 8487c89

File tree

4 files changed

+23
-15
lines changed

4 files changed

+23
-15
lines changed

aws_lambda_powertools/utilities/data_classes/alb_event.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class ALBEvent(BaseProxyEvent):
1919

2020
@property
2121
def request_context(self) -> ALBEventRequestContext:
22-
return ALBEventRequestContext(self)
22+
return ALBEventRequestContext(self._data)
2323

2424
@property
2525
def http_method(self) -> str:

aws_lambda_powertools/utilities/data_classes/api_gateway_proxy_event.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ def multi_value_query_string_parameters(self) -> Optional[Dict[str, List[str]]]:
231231

232232
@property
233233
def request_context(self) -> APIGatewayEventRequestContext:
234-
return APIGatewayEventRequestContext(self)
234+
return APIGatewayEventRequestContext(self._data)
235235

236236
@property
237237
def path_parameters(self) -> Optional[Dict[str, str]]:
@@ -371,7 +371,7 @@ def cookies(self) -> Optional[List[str]]:
371371

372372
@property
373373
def request_context(self) -> RequestContextV2:
374-
return RequestContextV2(self)
374+
return RequestContextV2(self._data)
375375

376376
@property
377377
def path_parameters(self) -> Optional[Dict[str, str]]:

aws_lambda_powertools/utilities/data_classes/cognito_user_pool_event.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def user_name(self) -> str:
5151
@property
5252
def caller_context(self) -> CallerContext:
5353
"""The caller context"""
54-
return CallerContext(self)
54+
return CallerContext(self._data)
5555

5656

5757
class PreSignUpTriggerEventRequest(DictWrapper):
@@ -119,11 +119,11 @@ class PreSignUpTriggerEvent(BaseTriggerEvent):
119119

120120
@property
121121
def request(self) -> PreSignUpTriggerEventRequest:
122-
return PreSignUpTriggerEventRequest(self)
122+
return PreSignUpTriggerEventRequest(self._data)
123123

124124
@property
125125
def response(self) -> PreSignUpTriggerEventResponse:
126-
return PreSignUpTriggerEventResponse(self)
126+
return PreSignUpTriggerEventResponse(self._data)
127127

128128

129129
class PostConfirmationTriggerEventRequest(DictWrapper):
@@ -156,7 +156,7 @@ class PostConfirmationTriggerEvent(BaseTriggerEvent):
156156

157157
@property
158158
def request(self) -> PostConfirmationTriggerEventRequest:
159-
return PostConfirmationTriggerEventRequest(self)
159+
return PostConfirmationTriggerEventRequest(self._data)
160160

161161

162162
class UserMigrationTriggerEventRequest(DictWrapper):
@@ -257,11 +257,11 @@ class UserMigrationTriggerEvent(BaseTriggerEvent):
257257

258258
@property
259259
def request(self) -> UserMigrationTriggerEventRequest:
260-
return UserMigrationTriggerEventRequest(self)
260+
return UserMigrationTriggerEventRequest(self._data)
261261

262262
@property
263263
def response(self) -> UserMigrationTriggerEventResponse:
264-
return UserMigrationTriggerEventResponse(self)
264+
return UserMigrationTriggerEventResponse(self._data)
265265

266266

267267
class CustomMessageTriggerEventRequest(DictWrapper):
@@ -342,11 +342,11 @@ class CustomMessageTriggerEvent(BaseTriggerEvent):
342342

343343
@property
344344
def request(self) -> CustomMessageTriggerEventRequest:
345-
return CustomMessageTriggerEventRequest(self)
345+
return CustomMessageTriggerEventRequest(self._data)
346346

347347
@property
348348
def response(self) -> CustomMessageTriggerEventResponse:
349-
return CustomMessageTriggerEventResponse(self)
349+
return CustomMessageTriggerEventResponse(self._data)
350350

351351

352352
class PreAuthenticationTriggerEventRequest(DictWrapper):
@@ -386,7 +386,7 @@ class PreAuthenticationTriggerEvent(BaseTriggerEvent):
386386
@property
387387
def request(self) -> PreAuthenticationTriggerEventRequest:
388388
"""Pre Authentication Request Parameters"""
389-
return PreAuthenticationTriggerEventRequest(self)
389+
return PreAuthenticationTriggerEventRequest(self._data)
390390

391391

392392
class PostAuthenticationTriggerEventRequest(DictWrapper):
@@ -428,7 +428,7 @@ class PostAuthenticationTriggerEvent(BaseTriggerEvent):
428428
@property
429429
def request(self) -> PostAuthenticationTriggerEventRequest:
430430
"""Post Authentication Request Parameters"""
431-
return PostAuthenticationTriggerEventRequest(self)
431+
return PostAuthenticationTriggerEventRequest(self._data)
432432

433433

434434
class GroupOverrideDetails(DictWrapper):
@@ -552,9 +552,9 @@ class PreTokenGenerationTriggerEvent(BaseTriggerEvent):
552552
@property
553553
def request(self) -> PreTokenGenerationTriggerEventRequest:
554554
"""Pre Token Generation Request Parameters"""
555-
return PreTokenGenerationTriggerEventRequest(self)
555+
return PreTokenGenerationTriggerEventRequest(self._data)
556556

557557
@property
558558
def response(self) -> PreTokenGenerationTriggerEventResponse:
559559
"""Pre Token Generation Response Parameters"""
560-
return PreTokenGenerationTriggerEventResponse(self)
560+
return PreTokenGenerationTriggerEventResponse(self._data)

tests/functional/validator/test_validator.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,14 @@ def lambda_handler(evt, context):
8282
lambda_handler(raw_event, {})
8383

8484

85+
def test_validator_incoming_envelope(schema, apigateway_event):
86+
@validator(inbound_schema=schema, envelope=envelopes.API_GATEWAY_REST)
87+
def lambda_handler(evt, context):
88+
pass
89+
90+
lambda_handler(apigateway_event, {})
91+
92+
8593
def test_validator_outgoing(schema_response, raw_response):
8694
@validator(outbound_schema=schema_response)
8795
def lambda_handler(evt, context):

0 commit comments

Comments
 (0)