Fix implementation and tests

Author: leandrodamascena

aws_lambda_powertools/utilities/parser/models/__init__.py

@@ -51,40 +51,18 @@
     CloudWatchLogsModel,
 )
 from .cognito import (
-    CognitoCallerContextModel,
-    CognitoChallengeResultModel,
-    CognitoClaimsOverrideDetailsModel,
-    CognitoCreateAuthChallengeRequestModel,
-    CognitoCreateAuthChallengeResponseModel,
     CognitoCreateAuthChallengeTriggerModel,
-    CognitoCustomEmailSenderRequestModel,
     CognitoCustomEmailSenderTriggerModel,
-    CognitoCustomMessageRequestModel,
-    CognitoCustomMessageResponseModel,
     CognitoCustomMessageTriggerModel,
-    CognitoCustomSMSSenderRequestModel,
     CognitoCustomSMSSenderTriggerModel,
-    CognitoDefineAuthChallengeRequestModel,
-    CognitoDefineAuthChallengeResponseModel,
     CognitoDefineAuthChallengeTriggerModel,
-    CognitoGroupConfigurationModel,
-    CognitoMigrateUserRequestModel,
-    CognitoMigrateUserResponseModel,
     CognitoMigrateUserTriggerModel,
-    CognitoPostAuthenticationRequestModel,
     CognitoPostAuthenticationTriggerModel,
-    CognitoPostConfirmationRequestModel,
     CognitoPostConfirmationTriggerModel,
-    CognitoPreAuthenticationRequestModel,
     CognitoPreAuthenticationTriggerModel,
-    CognitoPreSignupRequestModel,
-    CognitoPreSignupResponseModel,
     CognitoPreSignupTriggerModel,
-    CognitoPreTokenGenerationTriggerModel,
-    CognitoPreTokenGenRequestModel,
-    CognitoPreTokenGenResponseModel,
-    CognitoVerifyAuthChallengeRequestModel,
-    CognitoVerifyAuthChallengeResponseModel,
+    CognitoPreTokenGenerationTriggerModelV1,
+    CognitoPreTokenGenerationTriggerModelV2AndV3,
     CognitoVerifyAuthChallengeTriggerModel,
 )
 from .dynamodb import (
@@ -175,41 +153,20 @@
     "CloudWatchLogsDecode",
     "CloudWatchLogsLogEvent",
     "CloudWatchLogsModel",
-    "CognitoCallerContextModel",
-    "CognitoChallengeResultModel",
-    "CognitoPreSignupRequestModel",
-    "CognitoPreSignupResponseModel",
     "CognitoPreSignupTriggerModel",
-    "CognitoPostConfirmationRequestModel",
     "CognitoPostConfirmationTriggerModel",
-    "CognitoPreAuthenticationRequestModel",
     "CognitoPreAuthenticationTriggerModel",
-    "CognitoPostAuthenticationRequestModel",
     "CognitoPostAuthenticationTriggerModel",
-    "CognitoGroupConfigurationModel",
-    "CognitoPreTokenGenRequestModel",
-    "CognitoClaimsOverrideDetailsModel",
-    "CognitoPreTokenGenResponseModel",
     "CognitoPreTokenGenerationTriggerModel",
-    "CognitoMigrateUserRequestModel",
-    "CognitoMigrateUserResponseModel",
     "CognitoMigrateUserTriggerModel",
-    "CognitoCustomMessageRequestModel",
-    "CognitoCustomMessageResponseModel",
     "CognitoCustomMessageTriggerModel",
-    "CognitoCustomEmailSenderRequestModel",
     "CognitoCustomEmailSenderTriggerModel",
-    "CognitoCustomSMSSenderRequestModel",
     "CognitoCustomSMSSenderTriggerModel",
-    "CognitoDefineAuthChallengeRequestModel",
-    "CognitoDefineAuthChallengeResponseModel",
     "CognitoDefineAuthChallengeTriggerModel",
-    "CognitoCreateAuthChallengeRequestModel",
-    "CognitoCreateAuthChallengeResponseModel",
     "CognitoCreateAuthChallengeTriggerModel",
-    "CognitoVerifyAuthChallengeRequestModel",
-    "CognitoVerifyAuthChallengeResponseModel",
     "CognitoVerifyAuthChallengeTriggerModel",
+    "CognitoPreTokenGenerationTriggerModelV1",
+    "CognitoPreTokenGenerationTriggerModelV2AndV3",
     "AlbModel",
     "AlbRequestContext",
     "AlbRequestContextData",