fix: mock aws services before importing src functions

1 files 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()