Fix RestrictedQuerySet evaluation in tests

This commit is contained in:
Jeremy Stretch 2020-06-29 11:27:23 -04:00
parent 5732466e56
commit 10e6b6ca66
2 changed files with 3 additions and 3 deletions

View File

@ -416,7 +416,7 @@ class VLANTest(APIViewTestCases.APIViewTestCase):
"""
Attempt and fail to delete a VLAN with a Prefix assigned to it.
"""
vlan = VLAN.objects.first()
vlan = VLAN.objects.unrestricted().first()
Prefix.objects.create(prefix=IPNetwork('192.0.2.0/24'), vlan=vlan)
self.add_permissions('ipam.delete_vlan')

View File

@ -43,7 +43,7 @@ class TestPrefix(TestCase):
Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24')),
Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24')),
))
duplicate_prefix_pks = [p.pk for p in prefixes[0].get_duplicates()]
duplicate_prefix_pks = [p.pk for p in prefixes[0].get_duplicates().unrestricted()]
self.assertSetEqual(set(duplicate_prefix_pks), {prefixes[1].pk, prefixes[2].pk})
@ -227,7 +227,7 @@ class TestIPAddress(TestCase):
IPAddress(address=netaddr.IPNetwork('192.0.2.1/24')),
IPAddress(address=netaddr.IPNetwork('192.0.2.1/24')),
))
duplicate_ip_pks = [p.pk for p in ips[0].get_duplicates()]
duplicate_ip_pks = [p.pk for p in ips[0].get_duplicates().unrestricted()]
self.assertSetEqual(set(duplicate_ip_pks), {ips[1].pk, ips[2].pk})