diff options
| -rw-r--r-- | tests/test_load_lambda.py | 23 |
1 files changed, 18 insertions, 5 deletions
diff --git a/tests/test_load_lambda.py b/tests/test_load_lambda.py index 3df94e4..9b0a271 100644 --- a/tests/test_load_lambda.py +++ b/tests/test_load_lambda.py @@ -3,6 +3,7 @@ import pyarrow.parquet as pq from io import BytesIO from moto import mock_aws import boto3 +import botocore.exceptions import os import pytest from src.load_lambda import * @@ -29,10 +30,6 @@ def mock_sm_client(aws_credentials): yield boto3.client("secretsmanager") -@pytest.fixture -def mock_parquet_file(mocker): - return mocker.patch("src.load_lambda.convert_parquet_files_to_dfs") - class TestLambdaHandler: pass @@ -58,6 +55,19 @@ class TestRetrieveSecrets: assert isinstance(result, dict) + def test_retrieve_secrets_returns_correct_keys_and_values(self, mock_sm_client): + secret_name = "test_secret" + + result = retrieve_secrets(mock_sm_client, secret_name) + + assert result["user"] == "test_user_id" + assert result["password"] == "test_password" + + def test_retrieve_secrets_returns_client_error_if_no_secret(self, mock_sm_client): + secret_name = "another_test_secret" + + with pytest.raises(botocore.exceptions.ClientError) as error: + retrieve_secrets(mock_sm_client, secret_name) class TestConnectToDBAndReturnEngine: @@ -112,4 +122,7 @@ class TestConvertParquetToDfs: def mock_connect_db(mocker): - return mocker.patch("src.load_lambda.connect_to_db_and_return_engine")
\ No newline at end of file + return mocker.patch("src.load_lambda.connect_to_db_and_return_engine") + +class TestUploadDfsToDatabase: + pass
\ No newline at end of file |