From 5ae0d01c0429833e61e36c73d2fad4aef565ddd2 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 28 Aug 2024 12:02:21 +0100 Subject: wip: add tests for upload_dfs_to_database --- tests/test_load_lambda.py | 35 +++++++++++++++++++++++++++++++---- 1 file changed, 31 insertions(+), 4 deletions(-) diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py index 25a176d..cb27178 100644 --- a/tests/test_load_lambda.py +++ b/tests/test_load_lambda.py @@ -199,7 +199,34 @@ class TestConvertParquetToDfs: class TestUploadDfsToDatabase: - # Full success test - # Partial success test - # Failure test - pass + def test_function_returns_dictionary_with_uploaded_and_not_uploaded_keys( + self, mocker + ): + mocker.patch( + "src.load_lambda.convert_parquet_files_to_dfs", + return_value={"dim_counterparty.parquet": pd.DataFrame()}, + ) + mocker.patch( + "src.load_lambda.connect_to_db_and_return_engine", + return_value="test_engine", + ) + + result = upload_dfs_to_database() + + assert "uploaded" in result + assert "not_uploaded" in result + + def test_function_returns_uploaded_and_not_uploaded_tables(self, mocker): + mocker.patch( + "src.load_lambda.convert_parquet_files_to_dfs", + return_value={"dim_counterparty.parquet": pd.DataFrame()}, + ) + mocker.patch( + "src.load_lambda.connect_to_db_and_return_engine", + return_value="test_engine", + ) + + result = upload_dfs_to_database() + + assert result["uploaded"] == ["dim_counterparty"] + assert result["not_uploaded"] == [] -- cgit v1.2.3