From cbfc98a9f43b5a0dae95337057c18c9dc2a298e3 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Tue, 27 Aug 2024 16:00:29 +0100 Subject: wip: update TestLambdaHandler & lambda_handler function --- tests/test_load_lambda.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'tests/test_load_lambda.py') diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py index a29b75a..9286e48 100644 --- a/tests/test_load_lambda.py +++ b/tests/test_load_lambda.py @@ -35,7 +35,7 @@ class TestLambdaHandler: def test_lambda_handler_returns_success(self, mocker): mocker.patch( "src.load_lambda.upload_dfs_to_database", - return_value={"uploaded": ["table_one", "table_two"]}, + return_value={"uploaded": ["table_one", "table_two"], "not_uploaded": []}, ) result = lambda_handler(None, None) assert result["statusCode"] == 200 @@ -45,14 +45,20 @@ class TestLambdaHandler: def test_lambda_handler_does_not_upload_anything(self, mocker): mocker.patch( "src.load_lambda.upload_dfs_to_database", - return_value={"uploaded": []}, + return_value={"uploaded": [], "not_uploaded": []}, ) 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): - pass + mocker.patch( + "src.load_lambda.upload_dfs_to_database", + return_value={"test": []}, + ) + + with pytest.raises(Exception): + lambda_handler(None, None) class TestRetrieveSecrets: -- cgit v1.2.3