wip: add tests for upload_dfs_to_database

1 files 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"] == []