Projet

Général

Profil

0002-tests_rbac-factorize-get_-_model-calls-57500.patch

Benjamin Dauvergne, 01 février 2022 00:48

Télécharger (2,33 ko)

Voir les différences:

Subject: [PATCH 2/6] tests_rbac: factorize get_*_model calls (#57500)

 tests_rbac/test_rbac.py | 20 ++++++--------------
 1 file changed, 6 insertions(+), 14 deletions(-)
tests_rbac/test_rbac.py
24 24

  
25 25
from django_rbac import backends, models, utils
26 26

  
27
OU = OrganizationalUnit = utils.get_ou_model()
28
Permission = utils.get_permission_model()
29
RoleParenting = utils.get_role_parenting_model()
30
Role = utils.get_role_model()
31
User = get_user_model()
32

  
27 33

  
28 34
def test_role_parenting(db):
29
    OrganizationalUnit = utils.get_ou_model()
30
    Role = utils.get_role_model()
31
    RoleParenting = utils.get_role_parenting_model()
32 35

  
33 36
    ou = OrganizationalUnit.objects.create(name='ou')
34 37
    roles = []
......
83 86

  
84 87
@pytest.mark.slow
85 88
def test_massive_role_parenting(db):
86

  
87
    User = get_user_model()
88
    Role = utils.get_role_model()
89
    RoleParenting = utils.get_role_parenting_model()
90
    Permission = utils.get_permission_model()
91 89
    user = User.objects.create(username='user')
92 90
    roles = []
93 91
    # Try a depth=10 tree of roles
......
124 122

  
125 123

  
126 124
def test_rbac_backend(db):
127
    Permission = utils.get_permission_model()
128
    User = get_user_model()
129
    OU = utils.get_ou_model()
130 125
    ou1 = OU.objects.create(name='ou1', slug='ou1')
131 126
    ou2 = OU.objects.create(name='ou2', slug='ou2')
132 127
    user1 = User.objects.create(username='john.doe')
133
    Role = utils.get_role_model()
134 128
    ct_ct = ContentType.objects.get_for_model(ContentType)
135 129
    role_ct = ContentType.objects.get_for_model(Role)
136 130
    change_op = models.Operation.objects.get(slug='change')
......
217 211

  
218 212

  
219 213
def test_all_members(db):
220
    User = get_user_model()
221 214
    u1 = User.objects.create(username='john.doe')
222 215
    u2 = User.objects.create(username='donald.knuth')
223 216
    u3 = User.objects.create(username='alan.turing')
224
    Role = utils.get_role_model()
225 217
    r1 = Role.objects.create(name='r1')
226 218
    r1.members.add(u1)
227 219
    r1.members.add(u3)
228
-