From 6cfe607e1e1d25784a3ca0f54a76647efa9f4bd8 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Thu, 22 Aug 2024 12:05:30 +0100 Subject: fix: mock aws services before importing src functions --- tests/test_extract_lambda.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/tests/test_extract_lambda.py b/tests/test_extract_lambda.py index ee677bd..1266cbb 100644 --- a/tests/test_extract_lambda.py +++ b/tests/test_extract_lambda.py @@ -8,15 +8,6 @@ from unittest import TestCase import os import logging import json -from src.extract_lambda import ( - list_existing_s3_files, - connect_to_database, - DBConnectionException, - lambda_handler, - process_and_upload_tables, - retrieve_secrets, - extract_bucket, -) from pg8000.native import InterfaceError @@ -73,6 +64,17 @@ def s3_mock_bucket(s3_client): return bucket +from src.extract_lambda import ( # noqa: E402 + list_existing_s3_files, + connect_to_database, + DBConnectionException, + lambda_handler, + process_and_upload_tables, + retrieve_secrets, + extract_bucket, +) + + class TestLambdaHandler: def test_files_processed_and_uploaded_successfully(self, mocker): mock_db = MagicMock() -- cgit v1.2.3