From e26b7be8331d89826fbf95e1b1bd4fe88186c307 Mon Sep 17 00:00:00 2001 From: Ellie Date: Fri, 23 Aug 2024 17:04:29 +0100 Subject: add updated tests --- tests/test_load_lambda.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py index f1c2b01..3e42c2a 100644 --- a/tests/test_load_lambda.py +++ b/tests/test_load_lambda.py @@ -25,6 +25,9 @@ def mock_s3_client(aws_credentials): class TestLambdaHandler: pass +class TestRetrieveSecrets: + pass + class TestConnectToDBAndReturnEngine: pass @@ -58,7 +61,18 @@ class TestGetTransformBucket: assert result == "transform_bucket" class TestConvertParquetToDfs: - pass + def test_function_returns_empty_dictionary_if_no_files(self, mock_s3_client): + mock_s3_client.create_bucket( + Bucket="transform_bucket", + CreateBucketConfiguration={"LocationConstraint": "eu-west-2"}, + ) + result = convert_parquet_files_to_dfs(bucket_name="transform_bucket", client=mock_s3_client) + assert result == {} + + def test_function_returns_dictionary_with_table_with_file_key(): + # need to mock parquet file and upload to mock bucket + result = convert_parquet_files_to_dfs(bucket_name="transform_bucket", client=mock_s3_client) + assert "dim_staff" in result class TestUploadDfsToDatabase: pass \ No newline at end of file -- cgit v1.2.3