server-client-python/test/test_permissionsrule.py at 621eb857d37faac86061595dabdbef9802c33e6c · jorwoods/server-client-python · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import tableauserverclient as TSC
from tableauserverclient.models.reference_item import ResourceReference


def test_and() -> None:
    grantee = ResourceReference("a", "user")
    rule1 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Deny,
            TSC.Permission.Capability.ViewComments: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )
    rule2 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )

    composite = rule1 & rule2

    assert composite.capabilities.get(TSC.Permission.Capability.ExportData) == TSC.Permission.Mode.Allow
    assert composite.capabilities.get(TSC.Permission.Capability.Delete) == TSC.Permission.Mode.Deny
    assert composite.capabilities.get(TSC.Permission.Capability.ViewComments) == None
    assert composite.capabilities.get(TSC.Permission.Capability.ExportXml) == TSC.Permission.Mode.Deny


def test_or() -> None:
    grantee = ResourceReference("a", "user")
    rule1 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Deny,
            TSC.Permission.Capability.ViewComments: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )
    rule2 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )

    composite = rule1 | rule2

    assert composite.capabilities.get(TSC.Permission.Capability.ExportData) == TSC.Permission.Mode.Allow
    assert composite.capabilities.get(TSC.Permission.Capability.Delete) == TSC.Permission.Mode.Allow
    assert composite.capabilities.get(TSC.Permission.Capability.ViewComments) == TSC.Permission.Mode.Allow
    assert composite.capabilities.get(TSC.Permission.Capability.ExportXml) == TSC.Permission.Mode.Deny


def test_eq_false() -> None:
    grantee = ResourceReference("a", "user")
    rule1 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Deny,
            TSC.Permission.Capability.ViewComments: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )
    rule2 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )

    assert rule1 != rule2


def test_eq_true() -> None:
    grantee = ResourceReference("a", "user")
    rule1 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Deny,
            TSC.Permission.Capability.ViewComments: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )
    rule2 = TSC.PermissionsRule(
        grantee,
        {
            TSC.Permission.Capability.ExportData: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.Delete: TSC.Permission.Mode.Deny,
            TSC.Permission.Capability.ViewComments: TSC.Permission.Mode.Allow,
            TSC.Permission.Capability.ExportXml: TSC.Permission.Mode.Deny,
        },
    )
    assert rule1 == rule2