Add test for ensure_voice
This commit is contained in:
parent
e036835bf4
commit
d8dfc7119e
|
@ -25,6 +25,7 @@ PyNaCl==1.4.0
|
|||
pyparsing==2.4.7
|
||||
pytest==6.2.5
|
||||
pytest-cov==2.12.1
|
||||
pytest-mock==3.6.1
|
||||
python-dotenv==0.19.0
|
||||
six==1.16.0
|
||||
toml==0.10.2
|
||||
|
|
|
@ -32,7 +32,6 @@ class Music(commands.Cog):
|
|||
if ctx.author.voice:
|
||||
await ctx.author.voice.channel.connect()
|
||||
else:
|
||||
await ctx.send("You are not connected to a voice channel.")
|
||||
raise commands.CommandError("Author not connected to a voice channel.")
|
||||
elif ctx.voice_client.is_playing():
|
||||
ctx.voice_client.stop()
|
||||
|
|
|
@ -1,6 +1,19 @@
|
|||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
from pytest_mock import MockerFixture
|
||||
|
||||
# Add source to PATH such that imports within src modules are properly resolved.
|
||||
SRC_PATH = str(Path(__file__).parent / '..' / 'src')
|
||||
sys.path.insert(0, SRC_PATH)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def bot(mocker: MockerFixture):
|
||||
yield mocker.patch('discord.ext.commands.Bot', autospec=True)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def ctx(mocker: MockerFixture):
|
||||
yield mocker.patch('discord.ext.commands.Context', autospec=True)
|
||||
|
|
|
@ -1,3 +1,38 @@
|
|||
def test_bot():
|
||||
# placeholder test
|
||||
assert 1 + 1 == 2
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
import pytest
|
||||
from discord.ext import commands
|
||||
|
||||
from bot import Music
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_bot_ensure_voice(bot, ctx):
|
||||
mbot = Music(bot)
|
||||
|
||||
# 1. Inside a voice channel
|
||||
# 1.1 Does not call stop if no sound is playing
|
||||
ctx.voice_client.is_playing.return_value = False
|
||||
await mbot.ensure_voice(ctx)
|
||||
assert ctx.voice_client.stop.call_count == 0
|
||||
ctx.reset_mock(return_value=True)
|
||||
|
||||
# 1.2 Does call stop if sound is playing
|
||||
ctx.voice_client.is_playing.return_value = True
|
||||
await mbot.ensure_voice(ctx)
|
||||
assert ctx.voice_client.stop.call_count == 1
|
||||
ctx.reset_mock(return_value=True)
|
||||
|
||||
# 2. Not inside a voice channel
|
||||
# 2.1 Connects to voice channel of author if possible
|
||||
ctx.voice_client = None
|
||||
ctx.author.voice = AsyncMock()
|
||||
await mbot.ensure_voice(ctx)
|
||||
assert ctx.author.voice.channel.connect.call_count == 1
|
||||
ctx.reset_mock(return_value=True)
|
||||
|
||||
# 2.2 Error if author not inside a channel
|
||||
ctx.voice_client = None
|
||||
ctx.author.voice = None
|
||||
with pytest.raises(commands.CommandError):
|
||||
await mbot.ensure_voice(ctx)
|
||||
|
|
Loading…
Reference in New Issue