|
@@ -17,14 +17,15 @@ from virtualization.models import Cluster, ClusterType
|
|
|
|
|
|
|
|
class RegionTestCase(TestCase):
|
|
class RegionTestCase(TestCase):
|
|
|
queryset = Region.objects.all()
|
|
queryset = Region.objects.all()
|
|
|
|
|
+ filterset = RegionFilterSet
|
|
|
|
|
|
|
|
@classmethod
|
|
@classmethod
|
|
|
def setUpTestData(cls):
|
|
def setUpTestData(cls):
|
|
|
|
|
|
|
|
regions = (
|
|
regions = (
|
|
|
- Region(name='Region 1', slug='region-1'),
|
|
|
|
|
- Region(name='Region 2', slug='region-2'),
|
|
|
|
|
- Region(name='Region 3', slug='region-3'),
|
|
|
|
|
|
|
+ Region(name='Region 1', slug='region-1', description='A'),
|
|
|
|
|
+ Region(name='Region 2', slug='region-2', description='B'),
|
|
|
|
|
+ Region(name='Region 3', slug='region-3', description='C'),
|
|
|
)
|
|
)
|
|
|
for region in regions:
|
|
for region in regions:
|
|
|
region.save()
|
|
region.save()
|
|
@@ -43,22 +44,26 @@ class RegionTestCase(TestCase):
|
|
|
def test_id(self):
|
|
def test_id(self):
|
|
|
id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
params = {'id': [str(id) for id in id_list]}
|
|
params = {'id': [str(id) for id in id_list]}
|
|
|
- self.assertEqual(RegionFilterSet(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
def test_name(self):
|
|
def test_name(self):
|
|
|
params = {'name': ['Region 1', 'Region 2']}
|
|
params = {'name': ['Region 1', 'Region 2']}
|
|
|
- self.assertEqual(RegionFilterSet(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
def test_slug(self):
|
|
def test_slug(self):
|
|
|
params = {'slug': ['region-1', 'region-2']}
|
|
params = {'slug': ['region-1', 'region-2']}
|
|
|
- self.assertEqual(RegionFilterSet(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+
|
|
|
|
|
+ def test_description(self):
|
|
|
|
|
+ params = {'description': ['A', 'B']}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
def test_parent(self):
|
|
def test_parent(self):
|
|
|
parent_regions = Region.objects.filter(parent__isnull=True)[:2]
|
|
parent_regions = Region.objects.filter(parent__isnull=True)[:2]
|
|
|
params = {'parent_id': [parent_regions[0].pk, parent_regions[1].pk]}
|
|
params = {'parent_id': [parent_regions[0].pk, parent_regions[1].pk]}
|
|
|
- self.assertEqual(RegionFilterSet(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
params = {'parent': [parent_regions[0].slug, parent_regions[1].slug]}
|
|
params = {'parent': [parent_regions[0].slug, parent_regions[1].slug]}
|
|
|
- self.assertEqual(RegionFilterSet(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
|
|
|
|
|
|
|
class SiteTestCase(TestCase):
|
|
class SiteTestCase(TestCase):
|
|
@@ -81,7 +86,8 @@ class SiteTestCase(TestCase):
|
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
|
)
|
|
)
|
|
|
- TenantGroup.objects.bulk_create(tenant_groups)
|
|
|
|
|
|
|
+ for tenantgroup in tenant_groups:
|
|
|
|
|
+ tenantgroup.save()
|
|
|
|
|
|
|
|
tenants = (
|
|
tenants = (
|
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
@@ -138,11 +144,6 @@ class SiteTestCase(TestCase):
|
|
|
params = {'contact_email': ['contact1@example.com', 'contact2@example.com']}
|
|
params = {'contact_email': ['contact1@example.com', 'contact2@example.com']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
- def test_id__in(self):
|
|
|
|
|
- id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
|
|
- params = {'id__in': ','.join([str(id) for id in id_list])}
|
|
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
-
|
|
|
|
|
def test_status(self):
|
|
def test_status(self):
|
|
|
params = {'status': [SiteStatusChoices.STATUS_ACTIVE, SiteStatusChoices.STATUS_PLANNED]}
|
|
params = {'status': [SiteStatusChoices.STATUS_ACTIVE, SiteStatusChoices.STATUS_PLANNED]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
@@ -191,12 +192,21 @@ class RackGroupTestCase(TestCase):
|
|
|
)
|
|
)
|
|
|
Site.objects.bulk_create(sites)
|
|
Site.objects.bulk_create(sites)
|
|
|
|
|
|
|
|
|
|
+ parent_rack_groups = (
|
|
|
|
|
+ RackGroup(name='Parent Rack Group 1', slug='parent-rack-group-1', site=sites[0]),
|
|
|
|
|
+ RackGroup(name='Parent Rack Group 2', slug='parent-rack-group-2', site=sites[1]),
|
|
|
|
|
+ RackGroup(name='Parent Rack Group 3', slug='parent-rack-group-3', site=sites[2]),
|
|
|
|
|
+ )
|
|
|
|
|
+ for rackgroup in parent_rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
+
|
|
|
rack_groups = (
|
|
rack_groups = (
|
|
|
- RackGroup(name='Rack Group 1', slug='rack-group-1', site=sites[0]),
|
|
|
|
|
- RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
|
|
|
- RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
|
|
|
|
|
+ RackGroup(name='Rack Group 1', slug='rack-group-1', site=sites[0], parent=parent_rack_groups[0], description='A'),
|
|
|
|
|
+ RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1], parent=parent_rack_groups[1], description='B'),
|
|
|
|
|
+ RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2], parent=parent_rack_groups[2], description='C'),
|
|
|
)
|
|
)
|
|
|
- RackGroup.objects.bulk_create(rack_groups)
|
|
|
|
|
|
|
+ for rackgroup in rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
|
|
|
def test_id(self):
|
|
def test_id(self):
|
|
|
id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
@@ -211,18 +221,29 @@ class RackGroupTestCase(TestCase):
|
|
|
params = {'slug': ['rack-group-1', 'rack-group-2']}
|
|
params = {'slug': ['rack-group-1', 'rack-group-2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_description(self):
|
|
|
|
|
+ params = {'description': ['A', 'B']}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+
|
|
|
def test_region(self):
|
|
def test_region(self):
|
|
|
regions = Region.objects.all()[:2]
|
|
regions = Region.objects.all()[:2]
|
|
|
params = {'region_id': [regions[0].pk, regions[1].pk]}
|
|
params = {'region_id': [regions[0].pk, regions[1].pk]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
params = {'region': [regions[0].slug, regions[1].slug]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
|
|
def test_site(self):
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
params = {'site': [sites[0].slug, sites[1].slug]}
|
|
params = {'site': [sites[0].slug, sites[1].slug]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
+
|
|
|
|
|
+ def test_parent(self):
|
|
|
|
|
+ parent_groups = RackGroup.objects.filter(name__startswith='Parent')[:2]
|
|
|
|
|
+ params = {'parent_id': [parent_groups[0].pk, parent_groups[1].pk]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+ params = {'parent': [parent_groups[0].slug, parent_groups[1].slug]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
|
|
@@ -285,7 +306,8 @@ class RackTestCase(TestCase):
|
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
|
)
|
|
)
|
|
|
- RackGroup.objects.bulk_create(rack_groups)
|
|
|
|
|
|
|
+ for rackgroup in rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
|
|
|
rack_roles = (
|
|
rack_roles = (
|
|
|
RackRole(name='Rack Role 1', slug='rack-role-1'),
|
|
RackRole(name='Rack Role 1', slug='rack-role-1'),
|
|
@@ -299,7 +321,8 @@ class RackTestCase(TestCase):
|
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
|
)
|
|
)
|
|
|
- TenantGroup.objects.bulk_create(tenant_groups)
|
|
|
|
|
|
|
+ for tenantgroup in tenant_groups:
|
|
|
|
|
+ tenantgroup.save()
|
|
|
|
|
|
|
|
tenants = (
|
|
tenants = (
|
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
@@ -365,11 +388,6 @@ class RackTestCase(TestCase):
|
|
|
params = {'outer_unit': RackDimensionUnitChoices.UNIT_MILLIMETER}
|
|
params = {'outer_unit': RackDimensionUnitChoices.UNIT_MILLIMETER}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
- def test_id__in(self):
|
|
|
|
|
- id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
|
|
- params = {'id__in': ','.join([str(id) for id in id_list])}
|
|
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
-
|
|
|
|
|
def test_region(self):
|
|
def test_region(self):
|
|
|
regions = Region.objects.all()[:2]
|
|
regions = Region.objects.all()[:2]
|
|
|
params = {'region_id': [regions[0].pk, regions[1].pk]}
|
|
params = {'region_id': [regions[0].pk, regions[1].pk]}
|
|
@@ -442,7 +460,8 @@ class RackReservationTestCase(TestCase):
|
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
|
)
|
|
)
|
|
|
- RackGroup.objects.bulk_create(rack_groups)
|
|
|
|
|
|
|
+ for rackgroup in rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
|
|
|
racks = (
|
|
racks = (
|
|
|
Rack(name='Rack 1', site=sites[0], group=rack_groups[0]),
|
|
Rack(name='Rack 1', site=sites[0], group=rack_groups[0]),
|
|
@@ -463,7 +482,8 @@ class RackReservationTestCase(TestCase):
|
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
|
)
|
|
)
|
|
|
- TenantGroup.objects.bulk_create(tenant_groups)
|
|
|
|
|
|
|
+ for tenantgroup in tenant_groups:
|
|
|
|
|
+ tenantgroup.save()
|
|
|
|
|
|
|
|
tenants = (
|
|
tenants = (
|
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
@@ -479,11 +499,6 @@ class RackReservationTestCase(TestCase):
|
|
|
)
|
|
)
|
|
|
RackReservation.objects.bulk_create(reservations)
|
|
RackReservation.objects.bulk_create(reservations)
|
|
|
|
|
|
|
|
- def test_id__in(self):
|
|
|
|
|
- id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
|
|
- params = {'id__in': ','.join([str(id) for id in id_list])}
|
|
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
-
|
|
|
|
|
def test_site(self):
|
|
def test_site(self):
|
|
|
sites = Site.objects.all()[:2]
|
|
sites = Site.objects.all()[:2]
|
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
params = {'site_id': [sites[0].pk, sites[1].pk]}
|
|
@@ -529,9 +544,9 @@ class ManufacturerTestCase(TestCase):
|
|
|
def setUpTestData(cls):
|
|
def setUpTestData(cls):
|
|
|
|
|
|
|
|
manufacturers = (
|
|
manufacturers = (
|
|
|
- Manufacturer(name='Manufacturer 1', slug='manufacturer-1'),
|
|
|
|
|
- Manufacturer(name='Manufacturer 2', slug='manufacturer-2'),
|
|
|
|
|
- Manufacturer(name='Manufacturer 3', slug='manufacturer-3'),
|
|
|
|
|
|
|
+ Manufacturer(name='Manufacturer 1', slug='manufacturer-1', description='A'),
|
|
|
|
|
+ Manufacturer(name='Manufacturer 2', slug='manufacturer-2', description='B'),
|
|
|
|
|
+ Manufacturer(name='Manufacturer 3', slug='manufacturer-3', description='C'),
|
|
|
)
|
|
)
|
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
|
|
|
|
|
@@ -548,6 +563,10 @@ class ManufacturerTestCase(TestCase):
|
|
|
params = {'slug': ['manufacturer-1', 'manufacturer-2']}
|
|
params = {'slug': ['manufacturer-1', 'manufacturer-2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_description(self):
|
|
|
|
|
+ params = {'description': ['A', 'B']}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+
|
|
|
|
|
|
|
|
class DeviceTypeTestCase(TestCase):
|
|
class DeviceTypeTestCase(TestCase):
|
|
|
queryset = DeviceType.objects.all()
|
|
queryset = DeviceType.objects.all()
|
|
@@ -631,11 +650,6 @@ class DeviceTypeTestCase(TestCase):
|
|
|
params = {'subdevice_role': SubdeviceRoleChoices.ROLE_PARENT}
|
|
params = {'subdevice_role': SubdeviceRoleChoices.ROLE_PARENT}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
|
|
- def test_id__in(self):
|
|
|
|
|
- id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
|
|
- params = {'id__in': ','.join([str(id) for id in id_list])}
|
|
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
-
|
|
|
|
|
def test_manufacturer(self):
|
|
def test_manufacturer(self):
|
|
|
manufacturers = Manufacturer.objects.all()[:2]
|
|
manufacturers = Manufacturer.objects.all()[:2]
|
|
|
params = {'manufacturer_id': [manufacturers[0].pk, manufacturers[1].pk]}
|
|
params = {'manufacturer_id': [manufacturers[0].pk, manufacturers[1].pk]}
|
|
@@ -1080,9 +1094,9 @@ class PlatformTestCase(TestCase):
|
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
|
|
|
|
|
|
platforms = (
|
|
platforms = (
|
|
|
- Platform(name='Platform 1', slug='platform-1', manufacturer=manufacturers[0], napalm_driver='driver-1'),
|
|
|
|
|
- Platform(name='Platform 2', slug='platform-2', manufacturer=manufacturers[1], napalm_driver='driver-2'),
|
|
|
|
|
- Platform(name='Platform 3', slug='platform-3', manufacturer=manufacturers[2], napalm_driver='driver-3'),
|
|
|
|
|
|
|
+ Platform(name='Platform 1', slug='platform-1', manufacturer=manufacturers[0], napalm_driver='driver-1', description='A'),
|
|
|
|
|
+ Platform(name='Platform 2', slug='platform-2', manufacturer=manufacturers[1], napalm_driver='driver-2', description='B'),
|
|
|
|
|
+ Platform(name='Platform 3', slug='platform-3', manufacturer=manufacturers[2], napalm_driver='driver-3', description='C'),
|
|
|
)
|
|
)
|
|
|
Platform.objects.bulk_create(platforms)
|
|
Platform.objects.bulk_create(platforms)
|
|
|
|
|
|
|
@@ -1099,6 +1113,10 @@ class PlatformTestCase(TestCase):
|
|
|
params = {'slug': ['platform-1', 'platform-2']}
|
|
params = {'slug': ['platform-1', 'platform-2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_description(self):
|
|
|
|
|
+ params = {'description': ['A', 'B']}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+
|
|
|
def test_napalm_driver(self):
|
|
def test_napalm_driver(self):
|
|
|
params = {'napalm_driver': ['driver-1', 'driver-2']}
|
|
params = {'napalm_driver': ['driver-1', 'driver-2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
@@ -1166,7 +1184,8 @@ class DeviceTestCase(TestCase):
|
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
|
)
|
|
)
|
|
|
- RackGroup.objects.bulk_create(rack_groups)
|
|
|
|
|
|
|
+ for rackgroup in rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
|
|
|
racks = (
|
|
racks = (
|
|
|
Rack(name='Rack 1', site=sites[0], group=rack_groups[0]),
|
|
Rack(name='Rack 1', site=sites[0], group=rack_groups[0]),
|
|
@@ -1188,7 +1207,8 @@ class DeviceTestCase(TestCase):
|
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
TenantGroup(name='Tenant group 2', slug='tenant-group-2'),
|
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
TenantGroup(name='Tenant group 3', slug='tenant-group-3'),
|
|
|
)
|
|
)
|
|
|
- TenantGroup.objects.bulk_create(tenant_groups)
|
|
|
|
|
|
|
+ for tenantgroup in tenant_groups:
|
|
|
|
|
+ tenantgroup.save()
|
|
|
|
|
|
|
|
tenants = (
|
|
tenants = (
|
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]),
|
|
@@ -1242,8 +1262,8 @@ class DeviceTestCase(TestCase):
|
|
|
|
|
|
|
|
# Assign primary IPs for filtering
|
|
# Assign primary IPs for filtering
|
|
|
ipaddresses = (
|
|
ipaddresses = (
|
|
|
- IPAddress(family=4, address='192.0.2.1/24', interface=interfaces[0]),
|
|
|
|
|
- IPAddress(family=4, address='192.0.2.2/24', interface=interfaces[1]),
|
|
|
|
|
|
|
+ IPAddress(address='192.0.2.1/24', interface=interfaces[0]),
|
|
|
|
|
+ IPAddress(address='192.0.2.2/24', interface=interfaces[1]),
|
|
|
)
|
|
)
|
|
|
IPAddress.objects.bulk_create(ipaddresses)
|
|
IPAddress.objects.bulk_create(ipaddresses)
|
|
|
Device.objects.filter(pk=devices[0].pk).update(primary_ip4=ipaddresses[0])
|
|
Device.objects.filter(pk=devices[0].pk).update(primary_ip4=ipaddresses[0])
|
|
@@ -1283,11 +1303,6 @@ class DeviceTestCase(TestCase):
|
|
|
params = {'vc_priority': [1, 2]}
|
|
params = {'vc_priority': [1, 2]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
- def test_id__in(self):
|
|
|
|
|
- id_list = self.queryset.values_list('id', flat=True)[:2]
|
|
|
|
|
- params = {'id__in': ','.join([str(id) for id in id_list])}
|
|
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
-
|
|
|
|
|
def test_manufacturer(self):
|
|
def test_manufacturer(self):
|
|
|
manufacturers = Manufacturer.objects.all()[:2]
|
|
manufacturers = Manufacturer.objects.all()[:2]
|
|
|
params = {'manufacturer_id': [manufacturers[0].pk, manufacturers[1].pk]}
|
|
params = {'manufacturer_id': [manufacturers[0].pk, manufacturers[1].pk]}
|
|
@@ -2584,7 +2599,8 @@ class PowerPanelTestCase(TestCase):
|
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
RackGroup(name='Rack Group 2', slug='rack-group-2', site=sites[1]),
|
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
RackGroup(name='Rack Group 3', slug='rack-group-3', site=sites[2]),
|
|
|
)
|
|
)
|
|
|
- RackGroup.objects.bulk_create(rack_groups)
|
|
|
|
|
|
|
+ for rackgroup in rack_groups:
|
|
|
|
|
+ rackgroup.save()
|
|
|
|
|
|
|
|
power_panels = (
|
|
power_panels = (
|
|
|
PowerPanel(name='Power Panel 1', site=sites[0], rack_group=rack_groups[0]),
|
|
PowerPanel(name='Power Panel 1', site=sites[0], rack_group=rack_groups[0]),
|