From 5675dfd20b24f1e0eef06b11d9ec41a894d3a5bf Mon Sep 17 00:00:00 2001 From: salmonik Date: Tue, 28 Apr 2026 20:05:36 +0200 Subject: multitude of changes. but primarly, use db for all data. Signed-off-by: salmonik --- tests/conftest.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'tests/conftest.py') 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 -- cgit v1.2.3