diff options
| author | salmonik <raisersoft@proton.me> | 2026-04-19 15:44:33 +0200 |
|---|---|---|
| committer | salmonik <raisersoft@proton.me> | 2026-04-19 15:44:33 +0200 |
| commit | b343381fea3eab26e334a1b3bba202d715f9ea2e (patch) | |
| tree | 792210018d1e11ac3b160edfe29cf878eb4e3f56 /tests/cogs/test_voice.py | |
| parent | chore: add pycord dependency, update main (diff) | |
| download | senator-sperma-b343381fea3eab26e334a1b3bba202d715f9ea2e.tar.gz senator-sperma-b343381fea3eab26e334a1b3bba202d715f9ea2e.zip | |
test: improve test assertions and edge cases
Diffstat (limited to '')
| -rw-r--r-- | tests/cogs/test_voice.py | 32 |
1 files changed, 16 insertions, 16 deletions
diff --git a/tests/cogs/test_voice.py b/tests/cogs/test_voice.py index f5784f6..ade29cb 100644 --- a/tests/cogs/test_voice.py +++ b/tests/cogs/test_voice.py | |||
| @@ -7,14 +7,14 @@ from tests.conftest import MockApplicationContext | |||
| 7 | @pytest.mark.asyncio | 7 | @pytest.mark.asyncio |
| 8 | async def test_get_voice_manager(): | 8 | async def test_get_voice_manager(): |
| 9 | from cogs.voice import get_voice_manager, VoiceManager | 9 | from cogs.voice import get_voice_manager, VoiceManager |
| 10 | 10 | ||
| 11 | mock_bot = MagicMock() | 11 | mock_bot = MagicMock() |
| 12 | mock_bot.user = MagicMock() | 12 | mock_bot.user = MagicMock() |
| 13 | mock_bot.user.id = 12345 | 13 | mock_bot.user.id = 12345 |
| 14 | 14 | ||
| 15 | manager1 = get_voice_manager(mock_bot) | 15 | manager1 = get_voice_manager(mock_bot) |
| 16 | manager2 = get_voice_manager(mock_bot) | 16 | manager2 = get_voice_manager(mock_bot) |
| 17 | 17 | ||
| 18 | assert isinstance(manager1, VoiceManager) | 18 | assert isinstance(manager1, VoiceManager) |
| 19 | assert manager1 is manager2 | 19 | assert manager1 is manager2 |
| 20 | 20 | ||
| @@ -22,41 +22,41 @@ async def test_get_voice_manager(): | |||
| 22 | @pytest.mark.asyncio | 22 | @pytest.mark.asyncio |
| 23 | async def test_voice_manager_join_channel_not_found(): | 23 | async def test_voice_manager_join_channel_not_found(): |
| 24 | from cogs.voice import VoiceManager | 24 | from cogs.voice import VoiceManager |
| 25 | 25 | ||
| 26 | mock_bot = MagicMock() | 26 | mock_bot = MagicMock() |
| 27 | mock_bot.get_channel.return_value = None | 27 | mock_bot.get_channel.return_value = None |
| 28 | 28 | ||
| 29 | manager = VoiceManager(mock_bot) | 29 | manager = VoiceManager(mock_bot) |
| 30 | result = await manager.join_by_channel_id(999) | 30 | result = await manager.join_by_channel_id(999) |
| 31 | 31 | ||
| 32 | assert result is None | 32 | assert result is None |
| 33 | 33 | ||
| 34 | 34 | ||
| 35 | @pytest.mark.asyncio | 35 | @pytest.mark.asyncio |
| 36 | async def test_voice_manager_join_invalid_channel(): | 36 | async def test_voice_manager_join_invalid_channel(): |
| 37 | from cogs.voice import VoiceManager | 37 | from cogs.voice import VoiceManager |
| 38 | 38 | ||
| 39 | mock_bot = MagicMock() | 39 | mock_bot = MagicMock() |
| 40 | mock_channel = MagicMock() | 40 | mock_channel = MagicMock() |
| 41 | mock_bot.get_channel.return_value = mock_channel | 41 | mock_bot.get_channel.return_value = mock_channel |
| 42 | 42 | ||
| 43 | manager = VoiceManager(mock_bot) | 43 | manager = VoiceManager(mock_bot) |
| 44 | result = await manager.join_by_channel_id(999) | 44 | result = await manager.join_by_channel_id(999) |
| 45 | 45 | ||
| 46 | assert result is None | 46 | assert result is None |
| 47 | 47 | ||
| 48 | 48 | ||
| 49 | @pytest.mark.asyncio | 49 | @pytest.mark.asyncio |
| 50 | async def test_voice_manager_get_voice_status_not_connected(): | 50 | async def test_voice_manager_get_voice_status_not_connected(): |
| 51 | from cogs.voice import VoiceManager | 51 | from cogs.voice import VoiceManager |
| 52 | 52 | ||
| 53 | mock_bot = MagicMock() | 53 | mock_bot = MagicMock() |
| 54 | mock_guild = MagicMock() | 54 | mock_guild = MagicMock() |
| 55 | 55 | ||
| 56 | with patch('cogs.voice.VoiceManager.get_voice_client', return_value=None): | 56 | with patch('cogs.voice.VoiceManager.get_voice_client', return_value=None): |
| 57 | manager = VoiceManager(mock_bot) | 57 | manager = VoiceManager(mock_bot) |
| 58 | voice_client, is_connected, is_playing = await manager.get_voice_status(mock_guild) | 58 | voice_client, is_connected, is_playing = await manager.get_voice_status(mock_guild) |
| 59 | 59 | ||
| 60 | assert voice_client is None | 60 | assert voice_client is None |
| 61 | assert is_connected == False | 61 | assert is_connected == False |
| 62 | assert is_playing == False | 62 | assert is_playing == False |
| @@ -65,15 +65,15 @@ async def test_voice_manager_get_voice_status_not_connected(): | |||
| 65 | @pytest.mark.asyncio | 65 | @pytest.mark.asyncio |
| 66 | async def test_voice_manager_stop(): | 66 | async def test_voice_manager_stop(): |
| 67 | from cogs.voice import VoiceManager | 67 | from cogs.voice import VoiceManager |
| 68 | 68 | ||
| 69 | mock_bot = MagicMock() | 69 | mock_bot = MagicMock() |
| 70 | mock_guild = MagicMock() | 70 | mock_guild = MagicMock() |
| 71 | mock_voice_client = MagicMock() | 71 | mock_voice_client = MagicMock() |
| 72 | mock_voice_client.disconnect = AsyncMock() | 72 | mock_voice_client.disconnect = AsyncMock() |
| 73 | 73 | ||
| 74 | with patch('cogs.voice.VoiceManager.get_voice_client', return_value=mock_voice_client): | 74 | with patch('cogs.voice.VoiceManager.get_voice_client', return_value=mock_voice_client): |
| 75 | manager = VoiceManager(mock_bot) | 75 | manager = VoiceManager(mock_bot) |
| 76 | await manager.stop(mock_guild) | 76 | await manager.stop(mock_guild) |
| 77 | 77 | ||
| 78 | mock_voice_client.stop.assert_called_once() | 78 | mock_voice_client.stop.assert_called_once() |
| 79 | mock_voice_client.disconnect.assert_called_once() \ No newline at end of file | 79 | mock_voice_client.disconnect.assert_called_once() |
