nonebot2/tests/test_permission.py

from typing import Tuple

import pytest
from nonebug import App

from utils import make_fake_event


@pytest.mark.asyncio
async def test_permission(app: App):
    from nonebot.permission import Permission
    from nonebot.exception import SkippedException

    async def falsy():
        return False

    async def truthy():
        return True

    async def skipped() -> bool:
        raise SkippedException

    def _is_eq(a: Permission, b: Permission) -> bool:
        return {d.call for d in a.checkers} == {d.call for d in b.checkers}

    assert _is_eq(Permission(truthy) | None, Permission(truthy))
    assert _is_eq(Permission(truthy) | falsy, Permission(truthy, falsy))
    assert _is_eq(Permission(truthy) | Permission(falsy), Permission(truthy, falsy))

    assert _is_eq(None | Permission(truthy), Permission(truthy))
    assert _is_eq(truthy | Permission(falsy), Permission(truthy, falsy))

    event = make_fake_event()()

    async with app.test_api() as ctx:
        bot = ctx.create_bot()
        assert await Permission(falsy)(bot, event) == False
        assert await Permission(truthy)(bot, event) == True
        assert await Permission(skipped)(bot, event) == False
        assert await Permission(truthy, falsy)(bot, event) == True
        assert await Permission(truthy, skipped)(bot, event) == True


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "type,expected",
    [
        ("message", True),
        ("notice", False),
    ],
)
async def test_message(
    app: App,
    type: str,
    expected: bool,
):
    from nonebot.permission import MESSAGE, Message

    dependent = list(MESSAGE.checkers)[0]
    checker = dependent.call

    assert isinstance(checker, Message)

    event = make_fake_event(_type=type)()
    assert await dependent(event=event) == expected


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "type,expected",
    [
        ("message", False),
        ("notice", True),
    ],
)
async def test_notice(
    app: App,
    type: str,
    expected: bool,
):
    from nonebot.permission import NOTICE, Notice

    dependent = list(NOTICE.checkers)[0]
    checker = dependent.call

    assert isinstance(checker, Notice)

    event = make_fake_event(_type=type)()
    assert await dependent(event=event) == expected


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "type,expected",
    [
        ("message", False),
        ("request", True),
    ],
)
async def test_request(
    app: App,
    type: str,
    expected: bool,
):
    from nonebot.permission import REQUEST, Request

    dependent = list(REQUEST.checkers)[0]
    checker = dependent.call

    assert isinstance(checker, Request)

    event = make_fake_event(_type=type)()
    assert await dependent(event=event) == expected


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "type,expected",
    [
        ("message", False),
        ("meta_event", True),
    ],
)
async def test_metaevent(
    app: App,
    type: str,
    expected: bool,
):
    from nonebot.permission import METAEVENT, MetaEvent

    dependent = list(METAEVENT.checkers)[0]
    checker = dependent.call

    assert isinstance(checker, MetaEvent)

    event = make_fake_event(_type=type)()
    assert await dependent(event=event) == expected


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "type,user_id,expected",
    [
        ("message", "test", True),
        ("message", "foo", False),
        ("message", "faketest", True),
        ("notice", "test", True),
    ],
)
async def test_superuser(
    app: App,
    type: str,
    user_id: str,
    expected: bool,
):
    from nonebot.permission import SUPERUSER, SuperUser

    dependent = list(SUPERUSER.checkers)[0]
    checker = dependent.call

    assert isinstance(checker, SuperUser)

    event = make_fake_event(_type=type, _user_id=user_id)()

    async with app.test_api() as ctx:
        bot = ctx.create_bot()
        assert await dependent(bot=bot, event=event) == expected


@pytest.mark.asyncio
@pytest.mark.parametrize(
    "session_ids,session_id,expected",
    [
        (("user", "foo"), "user", True),
        (("user", "foo"), "bar", False),
    ],
)
async def test_user(
    app: App, session_ids: Tuple[str, ...], session_id: str, expected: bool
):
    from nonebot.permission import USER, User

    dependent = list(USER(*session_ids).checkers)[0]
    checker = dependent.call

    assert isinstance(checker, User)

    event = make_fake_event(_session_id=session_id)()

    async with app.test_api() as ctx:
        bot = ctx.create_bot()
        assert await dependent(bot=bot, event=event) == expected