Merge pull request 'massive changes' (#3) from dev into main

Reviewed-on: https://codeberg.org/salmonik/senator-sperma/pulls/3
stable enough

1 files changed, 83 insertions, 0 deletions

diff --git a/tests/cogs/test_ai.py b/tests/cogs/test_ai.py
new file mode 100644
index 0000000..a1c3fd4
--- /dev/null
+++ b/tests/cogs/test_ai.py
@@ -0,0 +1,83 @@
+import pytest
+from unittest.mock import AsyncMock, patch, MagicMock
+import discord
+import httpx
+from tests.conftest import MockApplicationContext
+
+
+@pytest.fixture
+def ai_cog(bot):
+    from cogs.ai import AI
+    cog = AI(bot)
+    return cog
+
+
+@pytest.fixture
+def mock_ai_response():
+    return {
+        "choices": [
+            {
+                "message": {
+                    "content": "Test response from AI"
+                }
+            }
+        ]
+    }
+
+
+@pytest.mark.asyncio
+async def test_senator_command_success(ai_cog, mock_channel, mock_guild, bot, mock_ai_response):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+
+    mock_response = MagicMock()
+    mock_response.json = MagicMock(return_value=mock_ai_response)
+
+    mock_client = MagicMock()
+    mock_client.post = AsyncMock(return_value=mock_response)
+    mock_client.__aenter__ = AsyncMock(return_value=mock_response)
+    mock_client.__aexit__ = AsyncMock(return_value=None)
+
+    with patch('cogs.ai.httpx.AsyncClient') as mock_httpx:
+        instance = mock_httpx.return_value.__aenter__.return_value
+        instance.post = AsyncMock(return_value=mock_response)
+        with patch.object(ctx, 'send_followup', new_callable=AsyncMock) as mock_followup:
+            await ai_cog.senator.callback(ai_cog, ctx, text="Hello", tts=False)
+            mock_followup.assert_called_once_with("Test response from AI")
+
+
+@pytest.mark.asyncio
+async def test_senator_command_empty_choices(ai_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+
+    mock_response = MagicMock()
+    mock_response.json = MagicMock(return_value={"choices": []})
+
+    mock_client = MagicMock()
+    mock_client.post = AsyncMock(return_value=mock_response)
+    mock_client.__aenter__ = AsyncMock(return_value=mock_response)
+    mock_client.__aexit__ = AsyncMock(return_value=None)
+
+    with patch('cogs.ai.httpx.AsyncClient') as mock_httpx:
+        instance = mock_httpx.return_value.__aenter__.return_value
+        instance.post = AsyncMock(return_value=mock_response)
+        with patch.object(ctx, 'send_followup', new_callable=AsyncMock) as mock_followup:
+            await ai_cog.senator.callback(ai_cog, ctx, text="Hello", tts=False)
+            mock_followup.assert_called_once_with("bot siem zesroł")
+
+
+@pytest.mark.asyncio
+async def test_senator_command_no_choices_key(ai_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+
+    mock_response = MagicMock()
+    mock_response.json = MagicMock(return_value={})
+
+    with patch('cogs.ai.httpx.AsyncClient') as mock_httpx:
+        instance = mock_httpx.return_value.__aenter__.return_value
+        instance.post = AsyncMock(return_value=mock_response)
+        with patch('cogs.ai.logger') as mock_logger:
+            try:
+                await ai_cog.senator.callback(ai_cog, ctx, text="Hello", tts=False)
+            except KeyError:
+                pass
+            assert mock_logger.info.called