From 857a7364ddb9b10529593c5e9c3578eb95c1da9b Mon Sep 17 00:00:00 2001
From: salmonik <raisersoft@proton.me>
Date: Sat, 18 Apr 2026 16:24:42 +0200
Subject: add a test suite

Signed-off-by: salmonik <raisersoft@proton.me>
---
 tests/cogs/test_downloader.py | 93 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 tests/cogs/test_downloader.py

(limited to 'tests/cogs/test_downloader.py')

diff --git a/tests/cogs/test_downloader.py b/tests/cogs/test_downloader.py
new file mode 100644
index 0000000..91eef93
--- /dev/null
+++ b/tests/cogs/test_downloader.py
@@ -0,0 +1,93 @@
+import pytest
+from unittest.mock import AsyncMock, patch, MagicMock
+import discord
+from tests.conftest import MockApplicationContext, MockMessage
+
+
+@pytest.fixture
+def downloader_cog(bot):
+    from cogs.downloader import Downloader
+    cog = Downloader(bot)
+    return cog
+
+
+@pytest.mark.asyncio
+async def test_download_command_success_file(downloader_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+    
+    mock_file_path = "/tmp/test_video.mp4"
+    
+    with patch('cogs.downloader.Downloader.download_standalone', return_value={
+        "status": "success",
+        "type": "file",
+        "content": mock_file_path,
+        "filename": "test_video.mp4"
+    }):
+        with patch('discord.File', return_value=MagicMock()) as mock_file:
+            with patch.object(ctx, 'respond', new_callable=AsyncMock) as mock_respond:
+                await downloader_cog.download.callback(downloader_cog, ctx, url="https://example.com/video", catbox=False, fformat="mp4", filename_user=None)
+                mock_respond.assert_called_once()
+                call_args = mock_respond.call_args
+                assert 'file' in str(call_args) or mock_file.called
+
+
+@pytest.mark.asyncio
+async def test_download_command_success_url(downloader_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+    
+    with patch('cogs.downloader.Downloader.download_standalone', return_value={
+        "status": "success",
+        "type": "url",
+        "content": "https://catbox.2137697.xyz/sperma/test.mp4"
+    }):
+        with patch.object(ctx, 'respond', new_callable=AsyncMock) as mock_respond:
+            await downloader_cog.download.callback(downloader_cog, ctx, url="https://example.com/video", catbox=True, fformat="mp4", filename_user=None)
+            mock_respond.assert_called_once()
+            call_args = mock_respond.call_args
+            assert "pobioron" in str(call_args)
+
+
+@pytest.mark.asyncio
+async def test_download_command_error(downloader_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+    
+    with patch('cogs.downloader.Downloader.download_standalone', return_value={
+        "status": "error",
+        "type": "error",
+        "content": "Download failed",
+        "details": "Connection timeout"
+    }):
+        with patch('cogs.downloader.logger') as mock_logger:
+            await downloader_cog.download.callback(downloader_cog, ctx, url="https://badurl.com", catbox=False, fformat="mp4", filename_user=None)
+            assert mock_logger.error.called
+
+
+@pytest.mark.asyncio
+async def test_message_download_with_url(downloader_cog, mock_channel, mock_guild, bot, mock_user):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+    
+    mock_message = MagicMock()
+    mock_message.content = "Check this out https://example.com/video"
+    
+    with patch('cogs.downloader.Downloader.download_standalone', return_value={
+        "status": "success",
+        "type": "file",
+        "content": "/tmp/test.mp4"
+    }):
+        with patch.object(downloader_cog, 'download', new_callable=AsyncMock) as mock_download:
+            await downloader_cog.message_download.callback(downloader_cog, ctx, message=mock_message)
+            mock_download.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_message_download_no_url(downloader_cog, mock_channel, mock_guild, bot):
+    ctx = MockApplicationContext(bot=bot, channel=mock_channel, guild=mock_guild)
+    
+    mock_message = MagicMock()
+    mock_message.content = "No URL here"
+    
+    with patch.object(ctx, 'respond', new_callable=AsyncMock) as mock_respond:
+        await downloader_cog.message_download.callback(downloader_cog, ctx, message=mock_message)
+        mock_respond.assert_called_once()
+        call_args = mock_respond.call_args
+        assert "Brak" in str(call_args)
\ No newline at end of file
-- 
cgit v1.2.3