1 files changed, 22 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
index e9f0f6f..6b7c3bc 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,10 +1,32 @@
 import asyncio
+import os
+import sys
+import tempfile
 from unittest.mock import AsyncMock, MagicMock
 import pytest
 
 import discord
 from discord.ext import commands
 
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+
+
+@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"):