|
|
@@ -1,4 +1,4 @@
|
|
|
-import netaddr
|
|
|
+from netaddr import IPNetwork, IPSet
|
|
|
from django.core.exceptions import ValidationError
|
|
|
from django.test import TestCase, override_settings
|
|
|
|
|
|
@@ -10,27 +10,27 @@ class TestAggregate(TestCase):
|
|
|
|
|
|
def test_get_utilization(self):
|
|
|
rir = RIR.objects.create(name='RIR 1', slug='rir-1')
|
|
|
- aggregate = Aggregate(prefix=netaddr.IPNetwork('10.0.0.0/8'), rir=rir)
|
|
|
+ aggregate = Aggregate(prefix=IPNetwork('10.0.0.0/8'), rir=rir)
|
|
|
aggregate.save()
|
|
|
|
|
|
# 25% utilization
|
|
|
Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/12')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.16.0.0/12')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.32.0.0/12')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.48.0.0/12')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/12')),
|
|
|
+ Prefix(prefix=IPNetwork('10.16.0.0/12')),
|
|
|
+ Prefix(prefix=IPNetwork('10.32.0.0/12')),
|
|
|
+ Prefix(prefix=IPNetwork('10.48.0.0/12')),
|
|
|
))
|
|
|
self.assertEqual(aggregate.get_utilization(), 25)
|
|
|
|
|
|
# 50% utilization
|
|
|
Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.64.0.0/10')),
|
|
|
+ Prefix(prefix=IPNetwork('10.64.0.0/10')),
|
|
|
))
|
|
|
self.assertEqual(aggregate.get_utilization(), 50)
|
|
|
|
|
|
# 100% utilization
|
|
|
Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.128.0.0/9')),
|
|
|
+ Prefix(prefix=IPNetwork('10.128.0.0/9')),
|
|
|
))
|
|
|
self.assertEqual(aggregate.get_utilization(), 100)
|
|
|
|
|
|
@@ -39,9 +39,9 @@ class TestPrefix(TestCase):
|
|
|
|
|
|
def test_get_duplicates(self):
|
|
|
prefixes = Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('192.0.2.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('192.0.2.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('192.0.2.0/24')),
|
|
|
))
|
|
|
duplicate_prefix_pks = [p.pk for p in prefixes[0].get_duplicates()]
|
|
|
|
|
|
@@ -54,11 +54,11 @@ class TestPrefix(TestCase):
|
|
|
VRF(name='VRF 3'),
|
|
|
))
|
|
|
prefixes = Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/16'), status=PrefixStatusChoices.STATUS_CONTAINER),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/24'), vrf=None),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.1.0/24'), vrf=vrfs[0]),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.2.0/24'), vrf=vrfs[1]),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.3.0/24'), vrf=vrfs[2]),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/16'), status=PrefixStatusChoices.STATUS_CONTAINER),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/24'), vrf=None),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.1.0/24'), vrf=vrfs[0]),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.2.0/24'), vrf=vrfs[1]),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.3.0/24'), vrf=vrfs[2]),
|
|
|
))
|
|
|
child_prefix_pks = {p.pk for p in prefixes[0].get_child_prefixes()}
|
|
|
|
|
|
@@ -79,13 +79,13 @@ class TestPrefix(TestCase):
|
|
|
VRF(name='VRF 3'),
|
|
|
))
|
|
|
parent_prefix = Prefix.objects.create(
|
|
|
- prefix=netaddr.IPNetwork('10.0.0.0/16'), status=PrefixStatusChoices.STATUS_CONTAINER
|
|
|
+ prefix=IPNetwork('10.0.0.0/16'), status=PrefixStatusChoices.STATUS_CONTAINER
|
|
|
)
|
|
|
ips = IPAddress.objects.bulk_create((
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.1/24'), vrf=None),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.1.1/24'), vrf=vrfs[0]),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.2.1/24'), vrf=vrfs[1]),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.3.1/24'), vrf=vrfs[2]),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.1/24'), vrf=None),
|
|
|
+ IPAddress(address=IPNetwork('10.0.1.1/24'), vrf=vrfs[0]),
|
|
|
+ IPAddress(address=IPNetwork('10.0.2.1/24'), vrf=vrfs[1]),
|
|
|
+ IPAddress(address=IPNetwork('10.0.3.1/24'), vrf=vrfs[2]),
|
|
|
))
|
|
|
child_ip_pks = {p.pk for p in parent_prefix.get_child_ips()}
|
|
|
|
|
|
@@ -102,16 +102,16 @@ class TestPrefix(TestCase):
|
|
|
def test_get_available_prefixes(self):
|
|
|
|
|
|
prefixes = Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/16')), # Parent prefix
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/20')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.32.0/20')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.128.0/18')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/16')), # Parent prefix
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/20')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.32.0/20')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.128.0/18')),
|
|
|
))
|
|
|
- missing_prefixes = netaddr.IPSet([
|
|
|
- netaddr.IPNetwork('10.0.16.0/20'),
|
|
|
- netaddr.IPNetwork('10.0.48.0/20'),
|
|
|
- netaddr.IPNetwork('10.0.64.0/18'),
|
|
|
- netaddr.IPNetwork('10.0.192.0/18'),
|
|
|
+ missing_prefixes = IPSet([
|
|
|
+ IPNetwork('10.0.16.0/20'),
|
|
|
+ IPNetwork('10.0.48.0/20'),
|
|
|
+ IPNetwork('10.0.64.0/18'),
|
|
|
+ IPNetwork('10.0.192.0/18'),
|
|
|
])
|
|
|
available_prefixes = prefixes[0].get_available_prefixes()
|
|
|
|
|
|
@@ -119,17 +119,17 @@ class TestPrefix(TestCase):
|
|
|
|
|
|
def test_get_available_ips(self):
|
|
|
|
|
|
- parent_prefix = Prefix.objects.create(prefix=netaddr.IPNetwork('10.0.0.0/28'))
|
|
|
+ parent_prefix = Prefix.objects.create(prefix=IPNetwork('10.0.0.0/28'))
|
|
|
IPAddress.objects.bulk_create((
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.1/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.3/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.5/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.7/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.9/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.11/26')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.13/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.1/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.3/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.5/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.7/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.9/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.11/26')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.13/26')),
|
|
|
))
|
|
|
- missing_ips = netaddr.IPSet([
|
|
|
+ missing_ips = IPSet([
|
|
|
'10.0.0.2/32',
|
|
|
'10.0.0.4/32',
|
|
|
'10.0.0.6/32',
|
|
|
@@ -145,39 +145,39 @@ class TestPrefix(TestCase):
|
|
|
def test_get_first_available_prefix(self):
|
|
|
|
|
|
prefixes = Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/16')), # Parent prefix
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/24')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.1.0/24')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.2.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/16')), # Parent prefix
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.1.0/24')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.2.0/24')),
|
|
|
))
|
|
|
- self.assertEqual(prefixes[0].get_first_available_prefix(), netaddr.IPNetwork('10.0.3.0/24'))
|
|
|
+ self.assertEqual(prefixes[0].get_first_available_prefix(), IPNetwork('10.0.3.0/24'))
|
|
|
|
|
|
- Prefix.objects.create(prefix=netaddr.IPNetwork('10.0.3.0/24'))
|
|
|
- self.assertEqual(prefixes[0].get_first_available_prefix(), netaddr.IPNetwork('10.0.4.0/22'))
|
|
|
+ Prefix.objects.create(prefix=IPNetwork('10.0.3.0/24'))
|
|
|
+ self.assertEqual(prefixes[0].get_first_available_prefix(), IPNetwork('10.0.4.0/22'))
|
|
|
|
|
|
def test_get_first_available_ip(self):
|
|
|
|
|
|
- parent_prefix = Prefix.objects.create(prefix=netaddr.IPNetwork('10.0.0.0/24'))
|
|
|
+ parent_prefix = Prefix.objects.create(prefix=IPNetwork('10.0.0.0/24'))
|
|
|
IPAddress.objects.bulk_create((
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.1/24')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.2/24')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('10.0.0.3/24')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.1/24')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.2/24')),
|
|
|
+ IPAddress(address=IPNetwork('10.0.0.3/24')),
|
|
|
))
|
|
|
self.assertEqual(parent_prefix.get_first_available_ip(), '10.0.0.4/24')
|
|
|
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('10.0.0.4/24'))
|
|
|
+ IPAddress.objects.create(address=IPNetwork('10.0.0.4/24'))
|
|
|
self.assertEqual(parent_prefix.get_first_available_ip(), '10.0.0.5/24')
|
|
|
|
|
|
def test_get_utilization(self):
|
|
|
|
|
|
# Container Prefix
|
|
|
prefix = Prefix.objects.create(
|
|
|
- prefix=netaddr.IPNetwork('10.0.0.0/24'),
|
|
|
+ prefix=IPNetwork('10.0.0.0/24'),
|
|
|
status=PrefixStatusChoices.STATUS_CONTAINER
|
|
|
)
|
|
|
Prefix.objects.bulk_create((
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.0/26')),
|
|
|
- Prefix(prefix=netaddr.IPNetwork('10.0.0.128/26')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.0/26')),
|
|
|
+ Prefix(prefix=IPNetwork('10.0.0.128/26')),
|
|
|
))
|
|
|
self.assertEqual(prefix.get_utilization(), 50)
|
|
|
|
|
|
@@ -186,7 +186,7 @@ class TestPrefix(TestCase):
|
|
|
prefix.save()
|
|
|
IPAddress.objects.bulk_create(
|
|
|
# Create 32 IPAddresses within the Prefix
|
|
|
- [IPAddress(address=netaddr.IPNetwork('10.0.0.{}/24'.format(i))) for i in range(1, 33)]
|
|
|
+ [IPAddress(address=IPNetwork('10.0.0.{}/24'.format(i))) for i in range(1, 33)]
|
|
|
)
|
|
|
self.assertEqual(prefix.get_utilization(), 12) # ~= 12%
|
|
|
|
|
|
@@ -196,36 +196,234 @@ class TestPrefix(TestCase):
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=False)
|
|
|
def test_duplicate_global(self):
|
|
|
- Prefix.objects.create(prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
- duplicate_prefix = Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
+ Prefix.objects.create(prefix=IPNetwork('192.0.2.0/24'))
|
|
|
+ duplicate_prefix = Prefix(prefix=IPNetwork('192.0.2.0/24'))
|
|
|
self.assertIsNone(duplicate_prefix.clean())
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=True)
|
|
|
def test_duplicate_global_unique(self):
|
|
|
- Prefix.objects.create(prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
- duplicate_prefix = Prefix(prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
+ Prefix.objects.create(prefix=IPNetwork('192.0.2.0/24'))
|
|
|
+ duplicate_prefix = Prefix(prefix=IPNetwork('192.0.2.0/24'))
|
|
|
self.assertRaises(ValidationError, duplicate_prefix.clean)
|
|
|
|
|
|
def test_duplicate_vrf(self):
|
|
|
vrf = VRF.objects.create(name='Test', rd='1:1', enforce_unique=False)
|
|
|
- Prefix.objects.create(vrf=vrf, prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
- duplicate_prefix = Prefix(vrf=vrf, prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
+ Prefix.objects.create(vrf=vrf, prefix=IPNetwork('192.0.2.0/24'))
|
|
|
+ duplicate_prefix = Prefix(vrf=vrf, prefix=IPNetwork('192.0.2.0/24'))
|
|
|
self.assertIsNone(duplicate_prefix.clean())
|
|
|
|
|
|
def test_duplicate_vrf_unique(self):
|
|
|
vrf = VRF.objects.create(name='Test', rd='1:1', enforce_unique=True)
|
|
|
- Prefix.objects.create(vrf=vrf, prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
- duplicate_prefix = Prefix(vrf=vrf, prefix=netaddr.IPNetwork('192.0.2.0/24'))
|
|
|
+ Prefix.objects.create(vrf=vrf, prefix=IPNetwork('192.0.2.0/24'))
|
|
|
+ duplicate_prefix = Prefix(vrf=vrf, prefix=IPNetwork('192.0.2.0/24'))
|
|
|
self.assertRaises(ValidationError, duplicate_prefix.clean)
|
|
|
|
|
|
|
|
|
+class TestPrefixHierarchy(TestCase):
|
|
|
+ """
|
|
|
+ Test the automatic updating of depth and child count in response to changes made within
|
|
|
+ the prefix hierarchy.
|
|
|
+ """
|
|
|
+ @classmethod
|
|
|
+ def setUpTestData(cls):
|
|
|
+
|
|
|
+ prefixes = (
|
|
|
+
|
|
|
+ # IPv4
|
|
|
+ Prefix(prefix='10.0.0.0/8', _depth=0, _children=2),
|
|
|
+ Prefix(prefix='10.0.0.0/16', _depth=1, _children=1),
|
|
|
+ Prefix(prefix='10.0.0.0/24', _depth=2, _children=0),
|
|
|
+
|
|
|
+ # IPv6
|
|
|
+ Prefix(prefix='2001:db8::/32', _depth=0, _children=2),
|
|
|
+ Prefix(prefix='2001:db8::/40', _depth=1, _children=1),
|
|
|
+ Prefix(prefix='2001:db8::/48', _depth=2, _children=0),
|
|
|
+
|
|
|
+ )
|
|
|
+ Prefix.objects.bulk_create(prefixes)
|
|
|
+
|
|
|
+ def test_create_prefix4(self):
|
|
|
+ # Create 10.0.0.0/12
|
|
|
+ Prefix(prefix='10.0.0.0/12').save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=4)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 3)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/12'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 2)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[2]._children, 1)
|
|
|
+ self.assertEqual(prefixes[3].prefix, IPNetwork('10.0.0.0/24'))
|
|
|
+ self.assertEqual(prefixes[3]._depth, 3)
|
|
|
+ self.assertEqual(prefixes[3]._children, 0)
|
|
|
+
|
|
|
+ def test_create_prefix6(self):
|
|
|
+ # Create 2001:db8::/36
|
|
|
+ Prefix(prefix='2001:db8::/36').save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=6)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 3)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/36'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 2)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[2]._children, 1)
|
|
|
+ self.assertEqual(prefixes[3].prefix, IPNetwork('2001:db8::/48'))
|
|
|
+ self.assertEqual(prefixes[3]._depth, 3)
|
|
|
+ self.assertEqual(prefixes[3]._children, 0)
|
|
|
+
|
|
|
+ def test_update_prefix4(self):
|
|
|
+ # Change 10.0.0.0/24 to 10.0.0.0/12
|
|
|
+ p = Prefix.objects.get(prefix='10.0.0.0/24')
|
|
|
+ p.prefix = '10.0.0.0/12'
|
|
|
+ p.save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=4)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 2)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/12'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 1)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[2]._children, 0)
|
|
|
+
|
|
|
+ def test_update_prefix6(self):
|
|
|
+ # Change 2001:db8::/48 to 2001:db8::/36
|
|
|
+ p = Prefix.objects.get(prefix='2001:db8::/48')
|
|
|
+ p.prefix = '2001:db8::/36'
|
|
|
+ p.save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=6)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 2)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/36'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 1)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[2]._children, 0)
|
|
|
+
|
|
|
+ def test_update_prefix_vrf4(self):
|
|
|
+ vrf = VRF(name='VRF A')
|
|
|
+ vrf.save()
|
|
|
+
|
|
|
+ # Move 10.0.0.0/16 to a VRF
|
|
|
+ p = Prefix.objects.get(prefix='10.0.0.0/16')
|
|
|
+ p.vrf = vrf
|
|
|
+ p.save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(vrf__isnull=True, prefix__family=4)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 1)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/24'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 0)
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(vrf=vrf)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/16'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 0)
|
|
|
+
|
|
|
+ def test_update_prefix_vrf6(self):
|
|
|
+ vrf = VRF(name='VRF A')
|
|
|
+ vrf.save()
|
|
|
+
|
|
|
+ # Move 2001:db8::/40 to a VRF
|
|
|
+ p = Prefix.objects.get(prefix='2001:db8::/40')
|
|
|
+ p.vrf = vrf
|
|
|
+ p.save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(vrf__isnull=True, prefix__family=6)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 1)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/48'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 0)
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(vrf=vrf)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/40'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 0)
|
|
|
+
|
|
|
+ def test_delete_prefix4(self):
|
|
|
+ # Delete 10.0.0.0/16
|
|
|
+ Prefix.objects.filter(prefix='10.0.0.0/16').delete()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=4)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 1)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/24'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 0)
|
|
|
+
|
|
|
+ def test_delete_prefix6(self):
|
|
|
+ # Delete 2001:db8::/40
|
|
|
+ Prefix.objects.filter(prefix='2001:db8::/40').delete()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=6)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 1)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/48'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 0)
|
|
|
+
|
|
|
+ def test_duplicate_prefix4(self):
|
|
|
+ # Duplicate 10.0.0.0/16
|
|
|
+ Prefix(prefix='10.0.0.0/16').save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=4)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 3)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/16'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 1)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[2]._children, 1)
|
|
|
+ self.assertEqual(prefixes[3].prefix, IPNetwork('10.0.0.0/24'))
|
|
|
+ self.assertEqual(prefixes[3]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[3]._children, 0)
|
|
|
+
|
|
|
+ def test_duplicate_prefix6(self):
|
|
|
+ # Duplicate 2001:db8::/40
|
|
|
+ Prefix(prefix='2001:db8::/40').save()
|
|
|
+
|
|
|
+ prefixes = Prefix.objects.filter(prefix__family=6)
|
|
|
+ self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
|
|
|
+ self.assertEqual(prefixes[0]._depth, 0)
|
|
|
+ self.assertEqual(prefixes[0]._children, 3)
|
|
|
+ self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/40'))
|
|
|
+ self.assertEqual(prefixes[1]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[1]._children, 1)
|
|
|
+ self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
|
|
|
+ self.assertEqual(prefixes[2]._depth, 1)
|
|
|
+ self.assertEqual(prefixes[2]._children, 1)
|
|
|
+ self.assertEqual(prefixes[3].prefix, IPNetwork('2001:db8::/48'))
|
|
|
+ self.assertEqual(prefixes[3]._depth, 2)
|
|
|
+ self.assertEqual(prefixes[3]._children, 0)
|
|
|
+
|
|
|
+
|
|
|
class TestIPAddress(TestCase):
|
|
|
|
|
|
def test_get_duplicates(self):
|
|
|
ips = IPAddress.objects.bulk_create((
|
|
|
- IPAddress(address=netaddr.IPNetwork('192.0.2.1/24')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('192.0.2.1/24')),
|
|
|
- IPAddress(address=netaddr.IPNetwork('192.0.2.1/24')),
|
|
|
+ IPAddress(address=IPNetwork('192.0.2.1/24')),
|
|
|
+ IPAddress(address=IPNetwork('192.0.2.1/24')),
|
|
|
+ IPAddress(address=IPNetwork('192.0.2.1/24')),
|
|
|
))
|
|
|
duplicate_ip_pks = [p.pk for p in ips[0].get_duplicates()]
|
|
|
|
|
|
@@ -237,44 +435,44 @@ class TestIPAddress(TestCase):
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=False)
|
|
|
def test_duplicate_global(self):
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
- duplicate_ip = IPAddress(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'))
|
|
|
+ duplicate_ip = IPAddress(address=IPNetwork('192.0.2.1/24'))
|
|
|
self.assertIsNone(duplicate_ip.clean())
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=True)
|
|
|
def test_duplicate_global_unique(self):
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
- duplicate_ip = IPAddress(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'))
|
|
|
+ duplicate_ip = IPAddress(address=IPNetwork('192.0.2.1/24'))
|
|
|
self.assertRaises(ValidationError, duplicate_ip.clean)
|
|
|
|
|
|
def test_duplicate_vrf(self):
|
|
|
vrf = VRF.objects.create(name='Test', rd='1:1', enforce_unique=False)
|
|
|
- IPAddress.objects.create(vrf=vrf, address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
- duplicate_ip = IPAddress(vrf=vrf, address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
+ IPAddress.objects.create(vrf=vrf, address=IPNetwork('192.0.2.1/24'))
|
|
|
+ duplicate_ip = IPAddress(vrf=vrf, address=IPNetwork('192.0.2.1/24'))
|
|
|
self.assertIsNone(duplicate_ip.clean())
|
|
|
|
|
|
def test_duplicate_vrf_unique(self):
|
|
|
vrf = VRF.objects.create(name='Test', rd='1:1', enforce_unique=True)
|
|
|
- IPAddress.objects.create(vrf=vrf, address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
- duplicate_ip = IPAddress(vrf=vrf, address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
+ IPAddress.objects.create(vrf=vrf, address=IPNetwork('192.0.2.1/24'))
|
|
|
+ duplicate_ip = IPAddress(vrf=vrf, address=IPNetwork('192.0.2.1/24'))
|
|
|
self.assertRaises(ValidationError, duplicate_ip.clean)
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=True)
|
|
|
def test_duplicate_nonunique_nonrole_role(self):
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
- duplicate_ip = IPAddress(address=netaddr.IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'))
|
|
|
+ duplicate_ip = IPAddress(address=IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
self.assertRaises(ValidationError, duplicate_ip.clean)
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=True)
|
|
|
def test_duplicate_nonunique_role_nonrole(self):
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
- duplicate_ip = IPAddress(address=netaddr.IPNetwork('192.0.2.1/24'))
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
+ duplicate_ip = IPAddress(address=IPNetwork('192.0.2.1/24'))
|
|
|
self.assertRaises(ValidationError, duplicate_ip.clean)
|
|
|
|
|
|
@override_settings(ENFORCE_GLOBAL_UNIQUE=True)
|
|
|
def test_duplicate_nonunique_role(self):
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
- IPAddress.objects.create(address=netaddr.IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
+ IPAddress.objects.create(address=IPNetwork('192.0.2.1/24'), role=IPAddressRoleChoices.ROLE_VIP)
|
|
|
|
|
|
|
|
|
class TestVLANGroup(TestCase):
|