From 01d48158121472229bad675fa0596cc09efca746 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Thu, 22 Aug 2024 11:34:19 +0100
Subject: fix: create two mocked buckets and check if extract_bucket is
 returned

---
 tests/test_extract_lambda.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/test_extract_lambda.py b/tests/test_extract_lambda.py
index 9cf5684..92f53aa 100644
--- a/tests/test_extract_lambda.py
+++ b/tests/test_extract_lambda.py
@@ -153,7 +153,14 @@ class TestExtractBucket:
         assert result == "extract_bucket"
 
     def test_bucket_returns_first_bucket(self, s3_client):
-        bucket1 = s3_client.create_bucket(
+        # Redefine what the test does
+        # Create two buckets and check that only extract_bucket is returned
+
+        s3_client.create_bucket(
+            Bucket="extract_bucket",
+            CreateBucketConfiguration={"LocationConstraint": "eu-west-2"},
+        )
+        s3_client.create_bucket(
             Bucket="bucket1",
             CreateBucketConfiguration={"LocationConstraint": "eu-west-2"},
         )
-- 
cgit v1.2.3