supported engine update (#131)

Co-authored-by: Caitlin Russell <caitarus@amazon.com>

Author: crus-umich

SecretsManagerRDSMySQLRotationSingleUser/lambda_function.py

@@ -418,7 +418,8 @@ def get_secret_dict(service_client, arn, stage, token=None):
     secret_dict = json.loads(plaintext)
 
     # Run validations against the secret
-    if 'engine' not in secret_dict or secret_dict['engine'] != 'mysql':
+    supported_engines = ["mysql", "aurora-mysql"]
+    if 'engine' not in secret_dict or secret_dict['engine'] not in supported_engines:
         raise KeyError("Database engine must be set to 'mysql' in order to use this rotation lambda")
     for field in required_fields:
         if field not in secret_dict:

SecretsManagerRDSPostgreSQLRotationSingleUser/lambda_function.py

@@ -429,7 +429,8 @@ def get_secret_dict(service_client, arn, stage, token=None):
     secret_dict = json.loads(plaintext)
 
     # Run validations against the secret
-    if 'engine' not in secret_dict or secret_dict['engine'] != 'postgres':
+    supported_engines = ["postgres", "aurora-postgresql"]
+    if 'engine' not in secret_dict or secret_dict['engine'] not in supported_engines:
         raise KeyError("Database engine must be set to 'postgres' in order to use this rotation lambda")
     for field in required_fields:
         if field not in secret_dict: