mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 03:08:55 -05:00
afa08713e0
* Move all fixtures to pytest plugin folder * Add core dunder all * Update other dunder all's * Black reformat
45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from redbot.pytest.admin import *
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_serverlock_check(admin, coroutine):
|
|
await admin.conf.serverlocked.set(True)
|
|
guild = MagicMock()
|
|
guild.leave = coroutine
|
|
|
|
# noinspection PyProtectedMember
|
|
ret = await admin._serverlock_check(guild)
|
|
|
|
assert ret is True
|
|
|
|
|
|
def test_announcer_initial_state(announcer):
|
|
assert announcer.active is None
|
|
|
|
|
|
def test_announcer_start(announcer):
|
|
announcer.announcer = object
|
|
announcer.start()
|
|
|
|
assert announcer.ctx.bot.loop.create_task.called
|
|
assert announcer.active is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_announcer_ignore(announcer, empty_guild, empty_channel):
|
|
await announcer.config.guild(empty_guild).announce_channel.set(empty_channel.id)
|
|
|
|
guild = MagicMock()
|
|
guild.id = empty_guild.id
|
|
|
|
guild.get_channel.return_value = empty_channel
|
|
|
|
ret = await announcer._get_announce_channel(guild)
|
|
|
|
assert guild.get_channel.called
|
|
assert ret == empty_channel
|