Add a very basic ownership validation for the owner cog

This commit is contained in:
Marc Di Luzio 2024-08-13 22:34:02 +01:00
parent f2bfa6d33c
commit f25dc8db35
5 changed files with 37 additions and 264 deletions

34
py/owner_cog_test.py Normal file
View 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")