Add a very basic ownership validation for the owner cog
This commit is contained in:
parent
f2bfa6d33c
commit
f25dc8db35
5 changed files with 37 additions and 264 deletions
34
py/owner_cog_test.py
Normal file
34
py/owner_cog_test.py
Normal file
|
@ -0,0 +1,34 @@
|
|||
import discord
|
||||
import discord.ext.commands as commands
|
||||
import pytest
|
||||
import pytest_asyncio
|
||||
import discord.ext.test as dpytest
|
||||
|
||||
from owner_cog import OwnerCog
|
||||
|
||||
# Primarily borrowing from https://dpytest.readthedocs.io/en/latest/tutorials/using_pytest.html
|
||||
# TODO: Test more somehow, though it seems like dpytest is pretty incomplete
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def bot():
|
||||
# Setup
|
||||
intents = discord.Intents.default()
|
||||
intents.members = True
|
||||
intents.message_content = True
|
||||
b = commands.Bot(command_prefix="$",
|
||||
intents=intents)
|
||||
await b._async_setup_hook()
|
||||
await b.add_cog(OwnerCog(b))
|
||||
dpytest.configure(b)
|
||||
yield b
|
||||
await dpytest.empty_queue()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_must_be_owner(bot):
|
||||
with pytest.raises(commands.errors.NotOwner):
|
||||
await dpytest.message("$sync")
|
||||
|
||||
with pytest.raises(commands.errors.NotOwner):
|
||||
await dpytest.message("$close")
|
Loading…
Add table
Add a link
Reference in a new issue