|
| 1 | +from aws_lambda_powertools.utilities.parser import envelopes, event_parser |
| 2 | +from aws_lambda_powertools.utilities.parser.models import APIGatewayProxyEventModel |
| 3 | +from aws_lambda_powertools.utilities.typing import LambdaContext |
| 4 | +from tests.functional.parser.schemas import MyApiGatewayBusiness |
| 5 | +from tests.functional.parser.utils import load_event |
| 6 | + |
| 7 | + |
| 8 | +@event_parser(model=MyApiGatewayBusiness, envelope=envelopes.ApiGatewayEnvelope) |
| 9 | +def handle_apigw_with_envelope(event: MyApiGatewayBusiness, _: LambdaContext): |
| 10 | + assert event.message == "Hello" |
| 11 | + assert event.username == "Ran" |
| 12 | + |
| 13 | + |
| 14 | +@event_parser(model=APIGatewayProxyEventModel) |
| 15 | +def handle_apigw_event(event: APIGatewayProxyEventModel, _: LambdaContext): |
| 16 | + assert event.body == "Hello from Lambda!" |
| 17 | + return event |
| 18 | + |
| 19 | + |
| 20 | +def test_apigw_event_with_envelope(): |
| 21 | + event = load_event("apiGatewayProxyEvent.json") |
| 22 | + event["body"] = '{"message": "Hello", "username": "Ran"}' |
| 23 | + handle_apigw_with_envelope(event, LambdaContext()) |
| 24 | + |
| 25 | + |
| 26 | +def test_apigw_event(): |
| 27 | + event = load_event("apiGatewayProxyEvent.json") |
| 28 | + parsed_event: APIGatewayProxyEventModel = handle_apigw_event(event, LambdaContext()) |
| 29 | + assert parsed_event.version == event["version"] |
| 30 | + assert parsed_event.resource == event["resource"] |
| 31 | + assert parsed_event.path == event["path"] |
| 32 | + assert parsed_event.headers == event["headers"] |
| 33 | + assert parsed_event.multiValueHeaders == event["multiValueHeaders"] |
| 34 | + assert parsed_event.queryStringParameters == event["queryStringParameters"] |
| 35 | + assert parsed_event.multiValueQueryStringParameters == event["multiValueQueryStringParameters"] |
| 36 | + |
| 37 | + request_context = parsed_event.requestContext |
| 38 | + assert request_context.accountId == event["requestContext"]["accountId"] |
| 39 | + assert request_context.apiId == event["requestContext"]["apiId"] |
| 40 | + |
| 41 | + authorizer = request_context.authorizer |
| 42 | + assert authorizer.claims is None |
| 43 | + assert authorizer.scopes is None |
| 44 | + |
| 45 | + assert request_context.domainName == event["requestContext"]["domainName"] |
| 46 | + assert request_context.domainPrefix == event["requestContext"]["domainPrefix"] |
| 47 | + assert request_context.extendedRequestId == event["requestContext"]["extendedRequestId"] |
| 48 | + assert request_context.httpMethod == event["requestContext"]["httpMethod"] |
| 49 | + |
| 50 | + identity = request_context.identity |
| 51 | + assert identity.accessKey == event["requestContext"]["identity"]["accessKey"] |
| 52 | + assert identity.accountId == event["requestContext"]["identity"]["accountId"] |
| 53 | + assert identity.caller == event["requestContext"]["identity"]["caller"] |
| 54 | + assert ( |
| 55 | + identity.cognitoAuthenticationProvider == event["requestContext"]["identity"]["cognitoAuthenticationProvider"] |
| 56 | + ) |
| 57 | + assert identity.cognitoAuthenticationType == event["requestContext"]["identity"]["cognitoAuthenticationType"] |
| 58 | + assert identity.cognitoIdentityId == event["requestContext"]["identity"]["cognitoIdentityId"] |
| 59 | + assert identity.cognitoIdentityPoolId == event["requestContext"]["identity"]["cognitoIdentityPoolId"] |
| 60 | + assert identity.principalOrgId == event["requestContext"]["identity"]["principalOrgId"] |
| 61 | + assert str(identity.sourceIp) == event["requestContext"]["identity"]["sourceIp"] |
| 62 | + assert identity.user == event["requestContext"]["identity"]["user"] |
| 63 | + assert identity.userAgent == event["requestContext"]["identity"]["userAgent"] |
| 64 | + assert identity.userArn == event["requestContext"]["identity"]["userArn"] |
| 65 | + assert identity.clientCert is not None |
| 66 | + assert identity.clientCert.clientCertPem == event["requestContext"]["identity"]["clientCert"]["clientCertPem"] |
| 67 | + assert identity.clientCert.subjectDN == event["requestContext"]["identity"]["clientCert"]["subjectDN"] |
| 68 | + assert identity.clientCert.issuerDN == event["requestContext"]["identity"]["clientCert"]["issuerDN"] |
| 69 | + assert identity.clientCert.serialNumber == event["requestContext"]["identity"]["clientCert"]["serialNumber"] |
| 70 | + assert ( |
| 71 | + identity.clientCert.validity.notBefore |
| 72 | + == event["requestContext"]["identity"]["clientCert"]["validity"]["notBefore"] |
| 73 | + ) |
| 74 | + assert ( |
| 75 | + identity.clientCert.validity.notAfter |
| 76 | + == event["requestContext"]["identity"]["clientCert"]["validity"]["notAfter"] |
| 77 | + ) |
| 78 | + |
| 79 | + assert request_context.path == event["requestContext"]["path"] |
| 80 | + assert request_context.protocol == event["requestContext"]["protocol"] |
| 81 | + assert request_context.requestId == event["requestContext"]["requestId"] |
| 82 | + assert request_context.requestTime == event["requestContext"]["requestTime"] |
| 83 | + convert_time = int(round(request_context.requestTimeEpoch.timestamp() * 1000)) |
| 84 | + assert convert_time == 1583349317135 |
| 85 | + assert request_context.resourceId == event["requestContext"]["resourceId"] |
| 86 | + assert request_context.resourcePath == event["requestContext"]["resourcePath"] |
| 87 | + assert request_context.stage == event["requestContext"]["stage"] |
| 88 | + |
| 89 | + assert parsed_event.pathParameters == event["pathParameters"] |
| 90 | + assert parsed_event.stageVariables == event["stageVariables"] |
| 91 | + assert parsed_event.body == event["body"] |
| 92 | + assert parsed_event.isBase64Encoded == event["isBase64Encoded"] |
| 93 | + |
| 94 | + assert request_context.connectedAt is None |
| 95 | + assert request_context.connectionId is None |
| 96 | + assert request_context.eventType is None |
| 97 | + assert request_context.messageDirection is None |
| 98 | + assert request_context.messageId is None |
| 99 | + assert request_context.routeKey is None |
| 100 | + assert request_context.operationName is None |
| 101 | + assert identity.apiKey is None |
| 102 | + assert identity.apiKeyId is None |
0 commit comments