diff options
| -rw-r--r-- | tests/test_extract_lambda.py | 20 |
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() |