~homeworkprod/byceps

ref: 1181a35dd06d48c779208a6d2a4b3672dd38d812 byceps/tests/blueprints/authorization/test_registry.py -rw-r--r-- 1.7 KiB
1181a35d — Jochen Kupperschmidt Ported tests from nose2 to pytest 4 years ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""
:Copyright: 2006-2017 Jochen Kupperschmidt
:License: Modified BSD, see LICENSE for details.
"""

from unittest.mock import patch

import pytest

with patch('flask.current_app'):
    from byceps.blueprints.authorization.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