Remove encrypted-at-rest catalog from tests.

Author: igorborgest

cloudformation/databases.yaml

@@ -369,9 +369,7 @@ Resources:
             Fn::ImportValue: aws-data-wrangler-base-KmsKeyArn
           ReturnConnectionPasswordEncrypted: true
         EncryptionAtRest:
-          CatalogEncryptionMode: SSE-KMS
-          SseAwsKmsKeyId:
-            Fn::ImportValue: aws-data-wrangler-base-KmsKeyArn
+          CatalogEncryptionMode: DISABLED
 Outputs:
   DatabasesPassword:
     Value:

tests/_utils.py

@@ -2,6 +2,7 @@
 import time
 from datetime import datetime
 from decimal import Decimal
+from typing import Iterator
 
 import boto3
 import botocore.exceptions
@@ -507,12 +508,12 @@ def ensure_athena_query_metadata(df, ctas_approach=True, encrypted=False):
         assert df.query_metadata["Statistics"]["DataManifestLocation"] is not None
 
 
-def get_time_str_with_random_suffix():
+def get_time_str_with_random_suffix() -> str:
     time_str = datetime.utcnow().strftime("%Y%m%d%H%M%S%f")
     return f"{time_str}_{random.randrange(16**6):06x}"
 
 
-def path_generator(bucket):
+def path_generator(bucket: str) -> Iterator[str]:
     s3_path = f"s3://{bucket}/{get_time_str_with_random_suffix()}/"
     print(f"S3 Path: {s3_path}")
     time.sleep(1)

tests/conftest.py

@@ -165,7 +165,7 @@ def redshift_external_schema(cloudformation_outputs, databases_parameters, glue_
 
 
 @pytest.fixture(scope="function")
-def glue_table(glue_database):
+def glue_table(glue_database: str) -> None:
     name = f"tbl_{get_time_str_with_random_suffix()}"
     print(f"Table name: {name}")
     wr.catalog.delete_table_if_exists(database=glue_database, table=name)

tests/test_athena.py

@@ -173,7 +173,7 @@ def test_athena(path, glue_database, glue_table, kms_key, workgroup0, workgroup1
     wr.catalog.delete_table_if_exists(database=glue_database, table=table)
 
 
-def test_catalog(path, glue_database, glue_table):
+def test_catalog(path: str, glue_database: str, glue_table: str) -> None:
     account_id = boto3.client("sts").get_caller_identity().get("Account")
     assert wr.catalog.does_table_exist(database=glue_database, table=glue_table) is False
     wr.catalog.create_parquet_table(