add passing tests for lambda handler

1 files changed, 9 insertions, 7 deletions

diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py
index 9286e48..0b13b54 100644
--- a/tests/test_load_lambda.py
+++ b/tests/test_load_lambda.py
@@ -32,7 +32,7 @@ def mock_sm_client(aws_credentials):
 
 
 class TestLambdaHandler:
-    def test_lambda_handler_returns_success(self, mocker):
+    def test_lambda_handler_returns_200_and_table_name_if_uploaded(self, mocker):
         mocker.patch(
             "src.load_lambda.upload_dfs_to_database",
             return_value={"uploaded": ["table_one", "table_two"], "not_uploaded": []},
@@ -42,23 +42,25 @@ class TestLambdaHandler:
         assert "table_one" in result["body"]
         assert "table_two" in result["body"]
 
-    def test_lambda_handler_does_not_upload_anything(self, mocker):
+    def test_lambda_handler_returns_200_and_table_name_if_not_uploaded(self, mocker):
         mocker.patch(
             "src.load_lambda.upload_dfs_to_database",
-            return_value={"uploaded": [], "not_uploaded": []},
+            return_value={"uploaded": [], "not_uploaded": ["table_one"]},
         )
         result = lambda_handler(None, None)
         assert result["statusCode"] == 200
         assert "No dataframes were uploaded" in result["body"]
 
-    def test_lambda_handler_returns_exception(self, mocker):
+    def test_lambda_handler_returns_error_if_both_lists_empty(self, mocker):
         mocker.patch(
             "src.load_lambda.upload_dfs_to_database",
-            return_value={"test": []},
+            return_value={"uploaded": [], "not_uploaded": []},
         )
 
-        with pytest.raises(Exception):
-            lambda_handler(None, None)
+        result = lambda_handler(None, None)
+
+        assert result == {"error"}
+
 
 
 class TestRetrieveSecrets: