cst1/kpsul
1
0
Fork 0
forked from DGNum/gestioCOF
Code Pull requests Activity

Avoid using get_by_natural_key

Browse source
This commit is contained in:
Martin Pépin 2019-12-24 17:13:27 +01:00
parent 64c792b11f
commit 1f945d1af3
No known key found for this signature in database
GPG key ID: E7520278B1774448
3 changed files with 15 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

16
kfet/open/tests.py
View file

@ -84,7 +84,9 @@ class OpenKfetTest(ChannelTestCase):
def test_export_team(self):
"""Export all values for a team member."""
user = User.objects.create_user("team", "", "team")
is_team = Permission.objects.get_by_natural_key("is_team", "kfet", "account")
is_team = Permission.objects.get(
codename="is_team", content_type__app_label="kfet"
)
user.user_permissions.add(is_team)
export = self.kfet_open.export(user)
self.assertSetEqual(set(["status", "admin_status", "force_close"]), set(export))
@ -115,11 +117,11 @@ class OpenKfetViewsTest(ChannelTestCase):
# get some permissions
perms = {
"kfet.is_team": Permission.objects.get_by_natural_key(
"is_team", "kfet", "account"
"kfet.is_team": Permission.objects.get(
codename="is_team", content_type__app_label="kfet"
),
"kfet.can_force_close": Permission.objects.get_by_natural_key(
"can_force_close", "kfet", "account"
"kfet.can_force_close": Permission.objects.get(
codename="can_force_close", content_type__app_label="kfet"
),
}
@ -204,7 +206,9 @@ class OpenKfetConsumerTest(ChannelTestCase):
"""Team user is added to kfet.open.team group."""
# setup team user and its client
t = User.objects.create_user("team", "", "team")
is_team = Permission.objects.get_by_natural_key("is_team", "kfet", "account")
is_team = Permission.objects.get(
codename="is_team", content_type__app_label="kfet"
)
t.user_permissions.add(is_team)
c = WSClient()
c.force_login(t)