ipam/tests/test_models.py

41         prefixes = Prefix.objects.bulk_create((
46         duplicate_prefix_pks = [p.pk for p in prefixes[0].get_duplicates()]
48         self.assertSetEqual(set(duplicate_prefix_pks), {prefixes[1].pk, prefixes[2].pk})
56         prefixes = Prefix.objects.bulk_create((
63         child_prefix_pks = {p.pk for p in prefixes[0].get_child_prefixes()}
66 …self.assertSetEqual(child_prefix_pks, {prefixes[1].pk, prefixes[2].pk, prefixes[3].pk, prefixes[4]…
68         prefixes[0].vrf = vrfs[0]
69         prefixes[0].save()
70         child_prefix_pks = {p.pk for p in prefixes[0].get_child_prefixes()}
73         self.assertSetEqual(child_prefix_pks, {prefixes[2].pk})
121         prefixes = Prefix.objects.bulk_create((
133         available_prefixes = prefixes[0].get_available_prefixes()
164         prefixes = Prefix.objects.bulk_create((
170         self.assertEqual(prefixes[0].get_first_available_prefix(), IPNetwork('10.0.3.0/24'))
173         self.assertEqual(prefixes[0].get_first_available_prefix(), IPNetwork('10.0.4.0/22'))
189         prefixes = (
194         Prefix.objects.bulk_create(prefixes)
195         self.assertEqual(prefixes[0].get_utilization(), 50)  # 50% utilization
250         prefixes = (
263         Prefix.objects.bulk_create(prefixes)
269         prefixes = Prefix.objects.filter(prefix__family=4)
270         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
271         self.assertEqual(prefixes[0]._depth, 0)
272         self.assertEqual(prefixes[0]._children, 3)
273         self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/12'))
274         self.assertEqual(prefixes[1]._depth, 1)
275         self.assertEqual(prefixes[1]._children, 2)
276         self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
277         self.assertEqual(prefixes[2]._depth, 2)
278         self.assertEqual(prefixes[2]._children, 1)
279         self.assertEqual(prefixes[3].prefix, IPNetwork('10.0.0.0/24'))
280         self.assertEqual(prefixes[3]._depth, 3)
281         self.assertEqual(prefixes[3]._children, 0)
287         prefixes = Prefix.objects.filter(prefix__family=6)
288         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
289         self.assertEqual(prefixes[0]._depth, 0)
290         self.assertEqual(prefixes[0]._children, 3)
291         self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/36'))
292         self.assertEqual(prefixes[1]._depth, 1)
293         self.assertEqual(prefixes[1]._children, 2)
294         self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
295         self.assertEqual(prefixes[2]._depth, 2)
296         self.assertEqual(prefixes[2]._children, 1)
297         self.assertEqual(prefixes[3].prefix, IPNetwork('2001:db8::/48'))
298         self.assertEqual(prefixes[3]._depth, 3)
299         self.assertEqual(prefixes[3]._children, 0)
307         prefixes = Prefix.objects.filter(prefix__family=4)
308         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
309         self.assertEqual(prefixes[0]._depth, 0)
310         self.assertEqual(prefixes[0]._children, 2)
311         self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/12'))
312         self.assertEqual(prefixes[1]._depth, 1)
313         self.assertEqual(prefixes[1]._children, 1)
314         self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
315         self.assertEqual(prefixes[2]._depth, 2)
316         self.assertEqual(prefixes[2]._children, 0)
324         prefixes = Prefix.objects.filter(prefix__family=6)
325         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
326         self.assertEqual(prefixes[0]._depth, 0)
327         self.assertEqual(prefixes[0]._children, 2)
328         self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/36'))
329         self.assertEqual(prefixes[1]._depth, 1)
330         self.assertEqual(prefixes[1]._children, 1)
331         self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
332         self.assertEqual(prefixes[2]._depth, 2)
333         self.assertEqual(prefixes[2]._children, 0)
344         prefixes = Prefix.objects.filter(vrf__isnull=True, prefix__family=4)
345         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
346         self.assertEqual(prefixes[0]._depth, 0)
347         self.assertEqual(prefixes[0]._children, 1)
348         self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/24'))
349         self.assertEqual(prefixes[1]._depth, 1)
350         self.assertEqual(prefixes[1]._children, 0)
352         prefixes = Prefix.objects.filter(vrf=vrf)
353         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/16'))
354         self.assertEqual(prefixes[0]._depth, 0)
355         self.assertEqual(prefixes[0]._children, 0)
366         prefixes = Prefix.objects.filter(vrf__isnull=True, prefix__family=6)
367         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
368         self.assertEqual(prefixes[0]._depth, 0)
369         self.assertEqual(prefixes[0]._children, 1)
370         self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/48'))
371         self.assertEqual(prefixes[1]._depth, 1)
372         self.assertEqual(prefixes[1]._children, 0)
374         prefixes = Prefix.objects.filter(vrf=vrf)
375         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/40'))
376         self.assertEqual(prefixes[0]._depth, 0)
377         self.assertEqual(prefixes[0]._children, 0)
383         prefixes = Prefix.objects.filter(prefix__family=4)
384         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
385         self.assertEqual(prefixes[0]._depth, 0)
386         self.assertEqual(prefixes[0]._children, 1)
387         self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/24'))
388         self.assertEqual(prefixes[1]._depth, 1)
389         self.assertEqual(prefixes[1]._children, 0)
395         prefixes = Prefix.objects.filter(prefix__family=6)
396         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
397         self.assertEqual(prefixes[0]._depth, 0)
398         self.assertEqual(prefixes[0]._children, 1)
399         self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/48'))
400         self.assertEqual(prefixes[1]._depth, 1)
401         self.assertEqual(prefixes[1]._children, 0)
407         prefixes = Prefix.objects.filter(prefix__family=4)
408         self.assertEqual(prefixes[0].prefix, IPNetwork('10.0.0.0/8'))
409         self.assertEqual(prefixes[0]._depth, 0)
410         self.assertEqual(prefixes[0]._children, 3)
411         self.assertEqual(prefixes[1].prefix, IPNetwork('10.0.0.0/16'))
412         self.assertEqual(prefixes[1]._depth, 1)
413         self.assertEqual(prefixes[1]._children, 1)
414         self.assertEqual(prefixes[2].prefix, IPNetwork('10.0.0.0/16'))
415         self.assertEqual(prefixes[2]._depth, 1)
416         self.assertEqual(prefixes[2]._children, 1)
417         self.assertEqual(prefixes[3].prefix, IPNetwork('10.0.0.0/24'))
418         self.assertEqual(prefixes[3]._depth, 2)
419         self.assertEqual(prefixes[3]._children, 0)
425         prefixes = Prefix.objects.filter(prefix__family=6)
426         self.assertEqual(prefixes[0].prefix, IPNetwork('2001:db8::/32'))
427         self.assertEqual(prefixes[0]._depth, 0)
428         self.assertEqual(prefixes[0]._children, 3)
429         self.assertEqual(prefixes[1].prefix, IPNetwork('2001:db8::/40'))
430         self.assertEqual(prefixes[1]._depth, 1)
431         self.assertEqual(prefixes[1]._children, 1)
432         self.assertEqual(prefixes[2].prefix, IPNetwork('2001:db8::/40'))
433         self.assertEqual(prefixes[2]._depth, 1)
434         self.assertEqual(prefixes[2]._children, 1)
435         self.assertEqual(prefixes[3].prefix, IPNetwork('2001:db8::/48'))
436         self.assertEqual(prefixes[3]._depth, 2)
437         self.assertEqual(prefixes[3]._children, 0)