From 0ea88c0216d9e5eca9e4aca4f2fa427d38184648 Mon Sep 17 00:00:00 2001 From: Ellie Date: Tue, 27 Aug 2024 16:40:21 +0100 Subject: add passing tests for lambda handler --- tests/test_load_lambda.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'tests/test_load_lambda.py') 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: -- cgit v1.2.3