From 499996329f08fbaee626ea9c2641fba24213bdbf Mon Sep 17 00:00:00 2001
From: brian <schnitzelen@gmail.com>
Date: Thu, 19 Dec 2024 15:33:25 +0000
Subject: [PATCH] added fixtures

---
 .../datastore/tests/integration/conftest.py   | 69 ++++++++++++++++++-
 1 file changed, 68 insertions(+), 1 deletion(-)

diff --git a/shared/datastore/tests/integration/conftest.py b/shared/datastore/tests/integration/conftest.py
index 98a22ac..0ad380a 100644
--- a/shared/datastore/tests/integration/conftest.py
+++ b/shared/datastore/tests/integration/conftest.py
@@ -1,8 +1,12 @@
 """Integration test configurations."""
 
 import os
+import random
+from collections.abc import Iterator
+from io import BytesIO
 
 import pytest
+from minio import Minio
 from PIL import Image
 from testcontainers.minio import MinioContainer
 
@@ -11,6 +15,7 @@ env_var_map = {
     'MINIO_ACCESS_KEY': 'test-access-key',
     'MINIO_SECRET_KEY': 'test-secret-key',
     'MINIO_BUCKET_NAME': 'test-bucket',
+    'MINIO_OBJECT_NAME': '46KXJMFIAPVLM0TKRFZR5YPPTVJ6PJNX',
 }
 container_map = {
     'minio': MinioContainer(
@@ -58,8 +63,70 @@ def populate_env(
     request.addfinalizer(cleanup_env)
 
 
+@pytest.fixture(scope='session')
+def minio_client(
+    setup_infrastructure,
+    populate_env,
+) -> Iterator[Minio]:
+    # prepare arguments
+    minio_endpoint = str(os.getenv('MINIO_ENDPOINT'))
+    minio_access_key = str(os.getenv('MINIO_ACCESS_KEY'))
+    minio_secret_key = str(os.getenv('MINIO_SECRET_KEY'))
+    minio_bucket_name = str(os.getenv('MINIO_BUCKET_NAME'))
+    # connect to minio
+    client = Minio(
+        endpoint=minio_endpoint,
+        access_key=minio_access_key,
+        secret_key=minio_secret_key,
+        secure=False,
+    )
+    # ensure bucket exists
+    if not client.bucket_exists(bucket_name=minio_bucket_name):
+        client.make_bucket(bucket_name=minio_bucket_name)
+    # expose client
+    yield client
+
+
 @pytest.fixture
-def image() -> Image.Image:
+def data() -> Iterator[bytes]:
+    # generate random data
+    num_bytes = 2**21  # 2 MB
+    data = random.randbytes(n=num_bytes)
+    # expose data
+    yield data
+
+
+@pytest.fixture
+def data_in_minio(
+    minio_client,
+    data,
+) -> Iterator[tuple[BytesIO, str, str]]:
+    # prepare arguments
+    minio_bucket_name = str(os.getenv('MINIO_BUCKET_NAME'))
+    minio_object_name = str(os.getenv('MINIO_OBJECT_NAME'))
+    # convert data
+    buffer = BytesIO(data)
+    # prepare for saving
+    num_bytes = buffer.tell()
+    buffer.seek(0)
+    # send data to bucket
+    minio_client.put_object(
+        bucket_name=minio_bucket_name,
+        object_name=minio_object_name,
+        length=num_bytes,
+        data=buffer,
+    )
+    # expose data
+    yield buffer, minio_bucket_name, minio_object_name
+    # clean up
+    minio_client.remove_object(
+        bucket_name=minio_bucket_name,
+        object_name=minio_object_name,
+    )
+
+
+@pytest.fixture
+def image() -> Iterator[Image.Image]:
     # generate image
     image = Image.new(mode='RGB', size=(480, 480))
     # expose image