def test_delete(): ses = botocore.session.get_session().create_client("ses", region_name="eu-west-1") stubber = Stubber(ses) provider._ses = ses addStubberDeleteResponse(stubber, attributes) stubber.activate() request = Request("Delete", attributes) response = handler(request, ()) assert response["Status"] == "SUCCESS", response["Reason"] stubber.assert_no_pending_responses()
def test_create_with_override(): ses = botocore.session.get_session().create_client("ses", region_name="eu-west-1") stubber = Stubber(ses) provider._ses = ses addStubberCreateResponse(stubber, attributes, override=True) stubber.activate() request = Request("Create", attributes) request["ResourceProperties"]["ForceOverride"] = True response = handler(request, ()) assert response["Status"] == "SUCCESS", response["Reason"] stubber.assert_no_pending_responses()
def handler(request, context): logging.basicConfig(level=os.getenv("LOG_LEVEL", "INFO")) if request["ResourceType"] == "Custom::DkimTokens": return dkim_tokens_provider.handler(request, context) elif request["ResourceType"] == "Custom::DomainIdentity": return domain_identity_provider.handler(request, context) elif request["ResourceType"] in [ "Custom::SESActiveReceiptRuleSet", "Custom::ActiveReceiptRuleSet", ]: return active_rule_set_provider.handler(request, context) elif request["ResourceType"] == "Custom::IdentityNotifications": return identity_notifications_provider.handler(request, context) elif request["ResourceType"] == "Custom::VerifiedIdentity": return verified_identity_provider.handler(request, context) else: return cfn_dkim_provider.handler(request, context)
def test_refusal_to_overwrite_settings(): ses = botocore.session.get_session().create_client("ses", region_name="eu-west-1") stubber = Stubber(ses) provider._ses = ses stubber.add_response( "get_identity_notification_attributes", GetIdentityNotificationAttributesResponse(attributes), {"Identities": ["lists.binx.io"]}, ) stubber.activate() provider._ses = ses request = Request("Create", attributes) response = handler(request, ()) assert response["Status"] == "FAILED" stubber.assert_no_pending_responses()