def test_validate_json_string_no_envelope(schema, wrapped_event_json_string):
# WHEN data key contains a JSON String
with pytest.raises(exceptions.SchemaValidationError,
match=".*data must be object"):
validate(event=wrapped_event_json_string,
schema=schema,
envelope="data.payload")
def test_apigateway_envelope(schema, apigateway_event):
# Payload v1 and v2 remains consistent where the payload is (body)
validate(event=apigateway_event,
schema=schema,
envelope=envelopes.API_GATEWAY_REST)
validate(event=apigateway_event,
schema=schema,
envelope=envelopes.API_GATEWAY_HTTP)
def test_validate_accept_schema_custom_format(
eventbridge_schema_registry_cloudtrail_v2_s3,
eventbridge_cloudtrail_s3_head_object_event):
validate(
event=eventbridge_cloudtrail_s3_head_object_event,
schema=eventbridge_schema_registry_cloudtrail_v2_s3,
formats={"int64": lambda v: True},
)
def test_validate_date_time_format(schema_datetime_format):
raw_event = {"message": "2021-06-29T14:46:06.804Z"}
validate(event=raw_event, schema=schema_datetime_format)
invalid_datetime = {"message": "2021-06-29T14"}
with pytest.raises(exceptions.SchemaValidationError,
match="data.message must be date-time"):
validate(event=invalid_datetime, schema=schema_datetime_format)
def test_validate_invalid_custom_format(
eventbridge_schema_registry_cloudtrail_v2_s3,
eventbridge_cloudtrail_s3_head_object_event, invalid_format):
with pytest.raises(exceptions.InvalidSchemaFormatError):
validate(
event=eventbridge_cloudtrail_s3_head_object_event,
schema=eventbridge_schema_registry_cloudtrail_v2_s3,
formats=invalid_format,
)
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,
)
def test_sns_envelope(schema, sns_event):
validate(event=sns_event, schema=schema, envelope=envelopes.SNS)
def test_validate_raw_event(schema, raw_event):
validate(event=raw_event, schema=schema)
def test_sqs_envelope(sqs_event, schema_array):
validate(event=sqs_event, schema=schema_array, envelope=envelopes.SQS)
def test_validate_invalid_event(schema):
b64_event = "eyJtZXNzYWdlIjogImhlbGxvIGhlbGxvIiwgInVzZXJuYW1lIjogImJsYWggYmxhaCJ9="
with pytest.raises(exceptions.SchemaValidationError):
validate(event=b64_event, schema=schema)
def test_cloudwatch_logs_envelope(cloudwatch_logs_schema,
cloudwatch_logs_event):
validate(event=cloudwatch_logs_event,
schema=cloudwatch_logs_schema,
envelope=envelopes.CLOUDWATCH_LOGS)
def test_validate_event_does_not_conform_with_schema(schema):
with pytest.raises(exceptions.SchemaValidationError):
validate(event={"message": "hello_world"}, schema=schema)
def test_validate_base64_string_envelope(schema,
wrapped_event_base64_json_string):
validate(event=wrapped_event_base64_json_string,
schema=schema,
envelope="powertools_json(powertools_base64(data))")
def test_validate_json_string_envelope(schema, wrapped_event_json_string):
validate(event=wrapped_event_json_string,
schema=schema,
envelope="powertools_json(data).payload")
def test_validate_wrapped_event_raw_envelope(schema, wrapped_event):
validate(event=wrapped_event, schema=schema, envelope="data.payload")
def test_eventbridge_envelope(schema, eventbridge_event):
validate(event=eventbridge_event,
schema=schema,
envelope=envelopes.EVENTBRIDGE)
def test_kinesis_data_stream_envelope(schema_array, kinesis_event):
validate(event=kinesis_event,
schema=schema_array,
envelope=envelopes.KINESIS_DATA_STREAM)
def test_validate_invalid_schema_format(raw_event):
with pytest.raises(exceptions.InvalidSchemaFormatError):
validate(event=raw_event, schema="schema.json")
def lambda_handler(event, context):
try:
validate(event=event, schema=schemas.INPUT, envelope="queryStringParameters")
except SchemaValidationError:
client.capture_exception()
return {
"statusCode": 400,
"body": json.dumps({"message": "Bad Request"}),
}
except:
client.capture_exception()
return {
"statusCode": 500,
"body": json.dumps({"message": "Internal Request failed"})
}
try:
conn = pymysql.connect(host=rds_host, user=name, passwd=password, db=db_name, connect_timeout=5)
except pymysql.MySQLError as e:
client.capture_exception()
return {
"statusCode": 500,
"body": json.dumps({"message": "Internal Request failed"})
}
sys.exit()
try:
conn.cursor().execute(
"""insert into feedback (name, country, subject) values({0},{1},{2})"""
.format(event["queryStringParameters"]['name'], event["queryStringParameters"]['country'], event["queryStringParameters"]['subject'])
)
try:
SENDER = sender_email
RECIPIENT = recepient_email
AWS_REGION = "us-west-2"
SUBJECT = "new Feedback"
BODY_HTML = """<html><body>
new feeback from user:\n
name: {0} \n
country: {1} \n
subject: {2}
</body></html>"""
.format(event["queryStringParameters"]['name'], event["queryStringParameters"]['country'], event["queryStringParameters"]['subject'])
client = boto3.client('SENDER',region_name=AWS_REGION)
response = client.send_email(
Destination={'ToAddresses': [RECIPIENT]},
Message={
'Body': {'Html': {'Data': BODY_HTML}},
'Subject': {'Data': SUBJECT},
},
Source=SENDER
)
except ClientError as e:
print(e.response['Error']['Message'])
else:
print("Email sent! Message ID:")
print(response['MessageId'])
return {
"statusCode": 200,
"body": json.dumps({"message": "form submitted"})
}
except:
client.capture_exception()
return {
"statusCode": 500,
"body": json.dumps({"message": "Internal Request failed"})
}
def test_validate_invalid_envelope_expression(schema, wrapped_event):
with pytest.raises(exceptions.InvalidEnvelopeExpressionError):
validate(event=wrapped_event, schema=schema, envelope=True)
