1 files changed, 19 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
index e9f0f6f..01ece10 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,4 +1,6 @@
 import asyncio
+import os
+import tempfile
 from unittest.mock import AsyncMock, MagicMock
 import pytest
 
@@ -6,6 +8,23 @@ import discord
 from discord.ext import commands
 
 
+@pytest.fixture(autouse=True, scope="function")
+def setup_test_env(monkeypatch):
+    test_db = os.path.join(tempfile.gettempdir(), "test_senator.db")
+    if os.path.exists(test_db):
+        os.remove(test_db)
+    monkeypatch.setenv("SENATOR_DB_PATH", test_db)
+    reload_database()
+    yield
+    if os.path.exists(test_db):
+        os.remove(test_db)
+
+
+def reload_database():
+    import cogs.utils.database as db_module
+    db_module._initialized = False
+
+
 class MockUser:
     def __init__(self, id=12345, name="testuser", discriminator="0001"):
         self.id = id