add a test suite

Signed-off-by: salmonik <raisersoft@proton.me>

1 files changed, 198 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..c471b6b
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,198 @@
+import asyncio
+from unittest.mock import AsyncMock, MagicMock
+import pytest
+
+import discord
+from discord.ext import commands
+
+
+class MockUser:
+    def __init__(self, id=12345, name="testuser", discriminator="0001"):
+        self.id = id
+        self.name = name
+        self.discriminator = discriminator
+        self.mention = f"@{name}"
+        self.display_name = name
+        self.avatar = None
+        self.bot = False
+        self.guild = None
+        self._state = MagicMock()
+
+    def __str__(self):
+        return f"{self.name}#{self.discriminator}"
+
+
+class MockChannel:
+    def __init__(self, id=11111, name="test-channel", type=discord.ChannelType.text):
+        self.id = id
+        self.name = name
+        self.type = type
+        self.mention = f"#{name}"
+        self.guild = None
+        self.category = None
+        self._state = MagicMock()
+
+    async def send(self, *args, **kwargs):
+        return MagicMock()
+
+
+class MockTextChannel(MockChannel):
+    def __init__(self, id=11111, name="test-channel"):
+        super().__init__(id, name, discord.ChannelType.text)
+
+
+class MockVoiceChannel(MockChannel):
+    def __init__(self, id=11112, name="test-voice"):
+        super().__init__(id, name, discord.ChannelType.voice)
+        self.voice_states = {}
+
+
+class MockGuild:
+    def __init__(self, id=99999, name="Test Server"):
+        self.id = id
+        self.name = name
+        self.channels = []
+        self.roles = []
+        self.emojis = []
+        self.me = MockUser(id=111111, name="Sen Bot")
+        self._state = MagicMock()
+        self._default_role = None
+
+    def get_channel(self, channel_id):
+        for ch in self.channels:
+            if ch.id == channel_id:
+                return ch
+        return None
+
+    def get_role(self, role_id):
+        for role in self.roles:
+            if role.id == role_id:
+                return role
+        return None
+
+    def get_member(self, member_id):
+        return None
+
+
+class MockMessage:
+    def __init__(self, id=22222, content="test message", author=None, channel=None, guild=None):
+        self.id = id
+        self.content = content
+        self.author = author or MockUser()
+        self.channel = channel or MockChannel()
+        self.guild = guild
+        self.attachments = []
+        self.embeds = []
+        self.jump_url = "https://discord.com/channels/..."
+        self.created_at = None
+
+    async def reply(self, *args, **kwargs):
+        return MagicMock()
+
+
+class MockInteraction:
+    def __init__(self, user=None, channel=None, guild=None):
+        self.user = user or MockUser()
+        self.channel = channel or MockChannel()
+        self.guild = guild or MockGuild()
+        self.message = None
+        self.response = AsyncMock()
+        self.followup = AsyncMock()
+        self.client = None
+        self.id = 33333
+        self.token = "test_token"
+        self._state = MagicMock()
+    
+    async def defer(self, *args, **kwargs):
+        pass
+    
+    async def respond(self, *args, **kwargs):
+        pass
+    
+    async def delete_original_response(self, *args, **kwargs):
+        pass
+    
+    async def send(self, *args, **kwargs):
+        return MagicMock()
+
+
+class MockApplicationContext:
+    def __init__(self, bot=None, interaction=None, user=None, channel=None, guild=None):
+        self.bot = bot
+        self.interaction = interaction or MockInteraction(user=user, channel=channel, guild=guild)
+        self.user = user or MockUser()
+        self.channel = channel or MockChannel()
+        self.guild = guild or MockGuild()
+        self.message = None
+        self.response = AsyncMock()
+        self.followup = AsyncMock()
+        self.client = self.bot
+        self.id = 33333
+        self.token = "test_token"
+        self._state = self.interaction._state
+        self.author = self.user
+
+    async def defer(self, *args, **kwargs):
+        pass
+
+    async def respond(self, *args, **kwargs):
+        pass
+
+    async def delete_original_response(self, *args, **kwargs):
+        pass
+
+    async def send(self, *args, **kwargs):
+        return MagicMock()
+
+    async def reply(self, *args, **kwargs):
+        return MagicMock()
+
+    async def send_followup(self, *args, **kwargs):
+        return MagicMock()
+
+
+@pytest.fixture
+def mock_guild():
+    return MockGuild()
+
+
+@pytest.fixture
+def mock_channel(mock_guild):
+    channel = MockTextChannel()
+    channel.guild = mock_guild
+    return channel
+
+
+@pytest.fixture
+def mock_voice_channel(mock_guild):
+    channel = MockVoiceChannel()
+    channel.guild = mock_guild
+    return channel
+
+
+@pytest.fixture
+def mock_user():
+    return MockUser()
+
+
+@pytest.fixture
+def mock_member(mock_guild):
+    member = MockUser()
+    member.guild = mock_guild
+    return member
+
+
+@pytest.fixture
+def mock_message(mock_channel, mock_user, mock_guild):
+    return MockMessage(channel=mock_channel, author=mock_user, guild=mock_guild)
+
+
+@pytest.fixture
+def mock_interaction(mock_channel, mock_user, mock_guild):
+    return MockInteraction(user=mock_user, channel=mock_channel, guild=mock_guild)
+
+
+@pytest.fixture
+def bot():
+    intents = discord.Intents.all()
+    return commands.Bot(intents=intents, command_prefix="!", loop=asyncio.new_event_loop())
\ No newline at end of file