Просмотр исходного кода

Fix RestrictedQuerySet evaluation in tests

Jeremy Stretch 5 лет назад
Родитель
Сommit
10e6b6ca66
2 измененных файлов с 3 добавлено и 3 удалено
  1. 1 1
      netbox/ipam/tests/test_api.py
  2. 2 2
      netbox/ipam/tests/test_models.py

+ 1 - 1
netbox/ipam/tests/test_api.py

@@ -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')

+ 2 - 2
netbox/ipam/tests/test_models.py

@@ -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})