"""
:Copyright: 2006-2021 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

from unittest.mock import patch

import pytest

with patch('flask.current_app'):
    from byceps.util.framework.permission_registry import PermissionRegistry
from byceps.util.authorization import create_permission_enum


ItemPermission = create_permission_enum('item', ['view', 'create', 'update'])


@pytest.mark.parametrize(
    'permission_id, expected',
    [
        ('item.create'   , ItemPermission.create),  # enum member exists
        ('item.delete'   , None                 ),  # enum exists, but member does not
        ('article.create', None                 ),  # enum does not exist
    ],
)
def test_get_enum_member(permission_id, expected):
    registry = create_registry_with_registered_enum()

    assert registry.get_enum_member(permission_id) == expected


@pytest.mark.parametrize(
    'permission_ids, expected',
    [
        ({'item.create', 'item.create'   }, {ItemPermission.create                       }),  # duplicates are ignored
        ({'item.create', 'item.update'   }, {ItemPermission.create, ItemPermission.update}),  # multiple different enums are returned
        ({'item.create', 'article.create'}, {ItemPermission.create                       }),  # unknown IDs are ignored
        ({'article.create'               }, frozenset()                                   ),  # unknown IDs are ignored
    ],
)
def test_get_enum_members(permission_ids, expected):
    registry = create_registry_with_registered_enum()

    assert registry.get_enum_members(permission_ids) == expected


def create_registry_with_registered_enum():
    registry = PermissionRegistry()
    registry.register_enum(ItemPermission)
    return registry

"""
:Copyright: 2006-2021 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

from flask import Flask

import pytest

from byceps.util.authorization import create_permission_enum
from byceps.util.framework.permission_registry import PermissionRegistry


ItemPermission = create_permission_enum('item', ['view', 'create', 'update'])


@pytest.mark.parametrize(
    'permission_id, expected',
    [
        ('item.create'   , ItemPermission.create),  # enum member exists
        ('item.delete'   , None                 ),  # enum exists, but member does not
        ('article.create', None                 ),  # enum does not exist
    ],
)
def test_get_enum_member(permission_id, expected):
    registry = create_registry_with_registered_enum()

    with Flask('dummy').app_context():
        assert registry.get_enum_member(permission_id) == expected


@pytest.mark.parametrize(
    'permission_ids, expected',
    [
        ({'item.create', 'item.create'   }, {ItemPermission.create                       }),  # duplicates are ignored
        ({'item.create', 'item.update'   }, {ItemPermission.create, ItemPermission.update}),  # multiple different enums are returned
        ({'item.create', 'article.create'}, {ItemPermission.create                       }),  # unknown IDs are ignored
        ({'article.create'               }, frozenset()                                   ),  # unknown IDs are ignored
    ],
)
def test_get_enum_members(permission_ids, expected):
    registry = create_registry_with_registered_enum()

    assert registry.get_enum_members(permission_ids) == expected


def create_registry_with_registered_enum():
    registry = PermissionRegistry()
    registry.register_enum(ItemPermission)
    return registry

Use dummy Flask app instead of `unittest.mock`