mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Update filter tests to use unrestricted querysets

Browse Source
This commit is contained in:
Jeremy Stretch
2020-07-07 14:26:14 -04:00
parent 88033c0801
commit 25e1319864
7 changed files with 199 additions and 199 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
netbox/secrets/tests/test_filters.py
View File

@@ -6,7 +6,7 @@ from secrets.models import Secret, SecretRole
class SecretRoleTestCase(TestCase):
queryset = SecretRole.objects.all()
queryset = SecretRole.objects.unrestricted()
filterset = SecretRoleFilterSet
@classmethod
@@ -33,7 +33,7 @@ class SecretRoleTestCase(TestCase):
class SecretTestCase(TestCase):
queryset = Secret.objects.all()
queryset = Secret.objects.unrestricted()
filterset = SecretFilterSet
@classmethod
@@ -76,14 +76,14 @@ class SecretTestCase(TestCase):
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_role(self):
roles = SecretRole.objects.all()[:2]
roles = SecretRole.objects.unrestricted()[:2]
params = {'role_id': [roles[0].pk, roles[1].pk]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
params = {'role': [roles[0].slug, roles[1].slug]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_device(self):
devices = Device.objects.all()[:2]
devices = Device.objects.unrestricted()[:2]
params = {'device_id': [devices[0].pk, devices[1].pk]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
params = {'device': [devices[0].name, devices[1].name]}