~homeworkprod/byceps

ref: ee44f550bd6c9f49e72424cd5826669dd1f57c1c byceps/tests/unit/util/test_permission_registry.py -rw-r--r-- 1.7 KiB
ee44f550 — Jochen Kupperschmidt Avoid same German translation for different meanings of "cancel" 1 year, 6 months 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
45
46
47
48
49
"""
: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, 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