|
|
@@ -3,13 +3,7 @@ from django.test import TestCase
|
|
|
|
|
|
from dcim.choices import *
|
|
|
from dcim.filters import *
|
|
|
-from dcim.models import (
|
|
|
- Cable, ConsolePort, ConsolePortTemplate, ConsoleServerPort, ConsoleServerPortTemplate, Device, DeviceBay,
|
|
|
- DeviceBayTemplate, DeviceRole, DeviceType, FrontPort, FrontPortTemplate, Interface, InterfaceTemplate,
|
|
|
- InventoryItem, Manufacturer, Platform, PowerFeed, PowerPanel, PowerPort, PowerPortTemplate, PowerOutlet,
|
|
|
- PowerOutletTemplate, Rack, Location, RackReservation, RackRole, RearPort, RearPortTemplate, Region, Site,
|
|
|
- VirtualChassis,
|
|
|
-)
|
|
|
+from dcim.models import *
|
|
|
from ipam.models import IPAddress
|
|
|
from tenancy.models import Tenant, TenantGroup
|
|
|
from virtualization.models import Cluster, ClusterType
|
|
|
@@ -80,6 +74,14 @@ class SiteTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
tenant_groups = (
|
|
|
TenantGroup(name='Tenant group 1', slug='tenant-group-1'),
|
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
|
@@ -96,9 +98,9 @@ class SiteTestCase(TestCase):
|
|
|
Tenant.objects.bulk_create(tenants)
|
|
|
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0], tenant=tenants[0], status=SiteStatusChoices.STATUS_ACTIVE, facility='Facility 1', asn=65001, latitude=10, longitude=10, contact_name='Contact 1', contact_phone='123-555-0001', contact_email='contact1@example.com'),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1], tenant=tenants[1], status=SiteStatusChoices.STATUS_PLANNED, facility='Facility 2', asn=65002, latitude=20, longitude=20, contact_name='Contact 2', contact_phone='123-555-0002', contact_email='contact2@example.com'),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2], tenant=tenants[2], status=SiteStatusChoices.STATUS_RETIRED, facility='Facility 3', asn=65003, latitude=30, longitude=30, contact_name='Contact 3', contact_phone='123-555-0003', contact_email='contact3@example.com'),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0], tenant=tenants[0], status=SiteStatusChoices.STATUS_ACTIVE, facility='Facility 1', asn=65001, latitude=10, longitude=10, contact_name='Contact 1', contact_phone='123-555-0001', contact_email='contact1@example.com'),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1], tenant=tenants[1], status=SiteStatusChoices.STATUS_PLANNED, facility='Facility 2', asn=65002, latitude=20, longitude=20, contact_name='Contact 2', contact_phone='123-555-0002', contact_email='contact2@example.com'),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2], tenant=tenants[2], status=SiteStatusChoices.STATUS_RETIRED, facility='Facility 3', asn=65003, latitude=30, longitude=30, contact_name='Contact 3', contact_phone='123-555-0003', contact_email='contact3@example.com'),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -153,6 +155,13 @@ class SiteTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'group_id': [groups[0].pk, groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'group': [groups[0].slug, groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_tenant(self):
|
|
|
tenants = Tenant.objects.all()[:2]
|
|
|
params = {'tenant_id': [tenants[0].pk, tenants[1].pk]}
|
|
|
@@ -183,10 +192,18 @@ class LocationTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -229,6 +246,13 @@ class LocationTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -290,10 +314,18 @@ class RackTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -388,6 +420,13 @@ class RackTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1161,10 +1200,18 @@ class DeviceTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -1324,6 +1371,13 @@ class DeviceTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1463,10 +1517,19 @@ class ConsolePortTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -1524,6 +1587,13 @@ class ConsolePortTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1559,10 +1629,19 @@ class ConsoleServerPortTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -1620,6 +1699,13 @@ class ConsoleServerPortTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1655,10 +1741,19 @@ class PowerPortTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -1724,6 +1819,13 @@ class PowerPortTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1759,10 +1861,19 @@ class PowerOutletTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -1825,6 +1936,13 @@ class PowerOutletTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -1860,10 +1978,19 @@ class InterfaceTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -1966,6 +2093,13 @@ class InterfaceTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2015,10 +2149,19 @@ class FrontPortTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -2082,6 +2225,13 @@ class FrontPortTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2117,10 +2267,19 @@ class RearPortTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -2178,6 +2337,13 @@ class RearPortTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2213,10 +2379,19 @@ class DeviceBayTestCase(TestCase):
|
|
|
)
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
+
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = Site.objects.bulk_create((
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
@@ -2256,6 +2431,13 @@ class DeviceBayTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2296,10 +2478,18 @@ class InventoryItemTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -2356,6 +2546,13 @@ class InventoryItemTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2409,10 +2606,18 @@ class VirtualChassisTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -2463,6 +2668,13 @@ class VirtualChassisTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2610,10 +2822,18 @@ class PowerPanelTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -2647,6 +2867,13 @@ class PowerPanelTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
@@ -2675,10 +2902,18 @@ class PowerFeedTestCase(TestCase):
|
|
|
for region in regions:
|
|
|
region.save()
|
|
|
|
|
|
+ groups = (
|
|
|
+ SiteGroup(name='Site Group 1', slug='site-group-1'),
|
|
|
+ SiteGroup(name='Site Group 2', slug='site-group-2'),
|
|
|
+ SiteGroup(name='Site Group 3', slug='site-group-3'),
|
|
|
+ )
|
|
|
+ for group in groups:
|
|
|
+ group.save()
|
|
|
+
|
|
|
sites = (
|
|
|
- Site(name='Site 1', slug='site-1', region=regions[0]),
|
|
|
- Site(name='Site 2', slug='site-2', region=regions[1]),
|
|
|
- Site(name='Site 3', slug='site-3', region=regions[2]),
|
|
|
+ Site(name='Site 1', slug='site-1', region=regions[0], group=groups[0]),
|
|
|
+ Site(name='Site 2', slug='site-2', region=regions[1], group=groups[1]),
|
|
|
+ Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
@@ -2759,6 +2994,13 @@ class PowerFeedTestCase(TestCase):
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_site_group(self):
|
|
|
+ site_groups = SiteGroup.objects.all()[:2]
|
|
|
+ params = {'site_group_id': [site_groups[0].pk, site_groups[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'site_group': [site_groups[0].slug, site_groups[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|