From 60459fbd98156849c399747c20635ff92d6718f8 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Thu, 22 Aug 2024 11:42:14 +0100
Subject: fix: add missing mock_conn fixture

---
 tests/test_extract_lambda.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests')

diff --git a/tests/test_extract_lambda.py b/tests/test_extract_lambda.py
index db6e25f..9d4d63c 100644
--- a/tests/test_extract_lambda.py
+++ b/tests/test_extract_lambda.py
@@ -19,6 +19,12 @@ from src.extract_lambda import (
 )
 
 
+@pytest.fixture
+def mock_conn():
+    with patch("src.extract_lambda.Connection") as mock:
+        yield mock
+
+
 @pytest.fixture(scope="function")
 def mock_config():
     env_vars = {
@@ -214,6 +220,7 @@ class TestConnectToDatabase:
 
 
 class TestProcessAndUploadTables:
+    # Added missing mock_conn fixture
     def test_error_process_and_upload_tables(self, mock_conn, s3_client, caplog):
         caplog.set_level(logging.INFO)
 
-- 
cgit v1.2.3