mirror of https://github.com/jumpserver/jumpserver
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.test import TestCase
|
|
|
|
from assets.models import Node, SystemUser
|
|
from .asset_permission import FlatPermission
|
|
from ..models import ActionFlag
|
|
|
|
|
|
class TestFlatPermissionEqual(TestCase):
|
|
def setUp(self):
|
|
node1 = Node(value="parent", key="1:1")
|
|
node2 = Node(value="child", key="1:1:1")
|
|
|
|
system_user1 = SystemUser(username="name1", name="name1", priority=20)
|
|
system_user2 = SystemUser(username="name2", name="name2", priority=10)
|
|
|
|
action1 = ActionFlag.ALL
|
|
action2 = ActionFlag.CONNECT
|
|
action3 = ActionFlag.UPDOWNLOAD
|
|
|
|
perm1 = FlatPermission(node1, system_user1, action1)
|
|
perm2 = FlatPermission(node2, system_user1, action1)
|
|
perm3 = FlatPermission(node2, system_user2, action1)
|
|
|
|
self.groups = (
|
|
(perm1, perm2, True),
|
|
(perm1, perm3, True),
|
|
)
|
|
|
|
def test_equal(self):
|
|
for k, k2, wanted in self.groups:
|
|
if (k == k2) != wanted:
|
|
print("Not equal {} {}", k, k2)
|
|
|
|
|