chores

tests/conftest.py

@@ -0,0 +1,18 @@
+import pytest
+import sqlite3
+
+from blossomtune_gradio import database
+
+
+@pytest.fixture
+def in_memory_db(mocker):
+    """
+    Fixture to set up and tear down an in-memory SQLite database for tests.
+    It ensures that the same connection object is used for both schema
+    initialization and the test execution.
+    """
+    con = sqlite3.connect(":memory:")
+    mocker.patch("sqlite3.connect", return_value=con)
+    database.init()
+    yield con
+    con.close()