from django.test import override_settings from django.urls import reverse from django.utils.translation import gettext as _ from rest_framework import status from dcim.choices import * from dcim.constants import * from dcim.models import * from extras.models import ConfigTemplate from ipam.models import ASN, RIR, VLAN, VRF from netbox.api.serializers import GenericObjectSerializer from tenancy.models import Tenant from users.models import User from utilities.testing import APITestCase, APIViewTestCases, create_test_device from virtualization.models import Cluster, ClusterType from wireless.choices import WirelessChannelChoices from wireless.models import WirelessLAN class AppTest(APITestCase): def test_root(self): url = reverse('dcim-api:api-root') response = self.client.get('{}?format=api'.format(url), **self.header) self.assertEqual(response.status_code, 200) class Mixins: class ComponentTraceMixin(APITestCase): peer_termination_type = None def test_trace(self): """ Test tracing a device component's attached cable. """ obj = self.model.objects.first() peer_device = Device.objects.create( site=Site.objects.first(), device_type=DeviceType.objects.first(), role=DeviceRole.objects.first(), name='Peer Device' ) if self.peer_termination_type is None: raise NotImplementedError(_("Test case must set peer_termination_type")) peer_obj = self.peer_termination_type.objects.create( device=peer_device, name='Peer Termination' ) cable = Cable(a_terminations=[obj], b_terminations=[peer_obj], label='Cable 1') cable.save() self.add_permissions(f'dcim.view_{self.model._meta.model_name}') url = reverse(f'dcim-api:{self.model._meta.model_name}-trace', kwargs={'pk': obj.pk}) response = self.client.get(url, **self.header) self.assertHttpStatus(response, status.HTTP_200_OK) self.assertEqual(len(response.data), 1) segment1 = response.data[0] self.assertEqual(segment1[0][0]['name'], obj.name) self.assertEqual(segment1[1]['label'], cable.label) self.assertEqual(segment1[2][0]['name'], peer_obj.name) class RegionTest(APIViewTestCases.APIViewTestCase): model = Region brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'site_count', 'slug', 'url'] create_data = [ { 'name': 'Region 4', 'slug': 'region-4', }, { 'name': 'Region 5', 'slug': 'region-5', }, { 'name': 'Region 6', 'slug': 'region-6', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): Region.objects.create(name='Region 1', slug='region-1') Region.objects.create(name='Region 2', slug='region-2') Region.objects.create(name='Region 3', slug='region-3') class SiteGroupTest(APIViewTestCases.APIViewTestCase): model = SiteGroup brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'site_count', 'slug', 'url'] create_data = [ { 'name': 'Site Group 4', 'slug': 'site-group-4', }, { 'name': 'Site Group 5', 'slug': 'site-group-5', }, { 'name': 'Site Group 6', 'slug': 'site-group-6', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): SiteGroup.objects.create(name='Site Group 1', slug='site-group-1') SiteGroup.objects.create(name='Site Group 2', slug='site-group-2') SiteGroup.objects.create(name='Site Group 3', slug='site-group-3') class SiteTest(APIViewTestCases.APIViewTestCase): model = Site brief_fields = ['description', 'display', 'id', 'name', 'slug', 'url'] bulk_update_data = { 'status': 'planned', } @classmethod def setUpTestData(cls): regions = ( Region.objects.create(name='Region 1', slug='region-1'), Region.objects.create(name='Region 2', slug='region-2'), ) groups = ( SiteGroup.objects.create(name='Site Group 1', slug='site-group-1'), SiteGroup.objects.create(name='Site Group 2', slug='site-group-2'), ) sites = ( Site(region=regions[0], group=groups[0], name='Site 1', slug='site-1'), Site(region=regions[0], group=groups[0], name='Site 2', slug='site-2'), Site(region=regions[0], group=groups[0], name='Site 3', slug='site-3'), ) Site.objects.bulk_create(sites) rir = RIR.objects.create(name='RFC 6996', is_private=True) tenant = Tenant.objects.create(name='Tenant 1', slug='tenant-1') asns = [ ASN(asn=65000 + i, rir=rir) for i in range(8) ] ASN.objects.bulk_create(asns) cls.create_data = [ { 'name': 'Site 4', 'slug': 'site-4', 'region': regions[1].pk, 'group': groups[1].pk, 'status': SiteStatusChoices.STATUS_ACTIVE, 'asns': [asns[0].pk, asns[1].pk], 'tenant': tenant.pk, }, { 'name': 'Site 5', 'slug': 'site-5', 'region': regions[1].pk, 'group': groups[1].pk, 'status': SiteStatusChoices.STATUS_ACTIVE, 'asns': [asns[2].pk, asns[3].pk], }, { 'name': 'Site 6', 'slug': 'site-6', 'region': regions[1].pk, 'group': groups[1].pk, 'status': SiteStatusChoices.STATUS_ACTIVE, 'asns': [asns[4].pk, asns[5].pk], }, ] class LocationTest(APIViewTestCases.APIViewTestCase): model = Location brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'rack_count', 'slug', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_site',) @classmethod def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) parent_locations = ( Location.objects.create(site=sites[0], name='Parent Location 1', slug='parent-location-1', status=LocationStatusChoices.STATUS_ACTIVE), Location.objects.create(site=sites[1], name='Parent Location 2', slug='parent-location-2', status=LocationStatusChoices.STATUS_ACTIVE), ) Location.objects.create(site=sites[0], name='Location 1', slug='location-1', parent=parent_locations[0], status=LocationStatusChoices.STATUS_ACTIVE) Location.objects.create(site=sites[0], name='Location 2', slug='location-2', parent=parent_locations[0], status=LocationStatusChoices.STATUS_ACTIVE) Location.objects.create(site=sites[0], name='Location 3', slug='location-3', parent=parent_locations[0], status=LocationStatusChoices.STATUS_ACTIVE) cls.create_data = [ { 'name': 'Test Location 4', 'slug': 'test-location-4', 'site': sites[1].pk, 'parent': parent_locations[1].pk, 'status': LocationStatusChoices.STATUS_PLANNED, }, { 'name': 'Test Location 5', 'slug': 'test-location-5', 'site': sites[1].pk, 'parent': parent_locations[1].pk, 'status': LocationStatusChoices.STATUS_PLANNED, }, { 'name': 'Test Location 6', 'slug': 'test-location-6', 'site': sites[1].pk, # Omit parent to test uniqueness constraint 'status': LocationStatusChoices.STATUS_PLANNED, }, ] class RackRoleTest(APIViewTestCases.APIViewTestCase): model = RackRole brief_fields = ['description', 'display', 'id', 'name', 'rack_count', 'slug', 'url'] create_data = [ { 'name': 'Rack Role 4', 'slug': 'rack-role-4', 'color': 'ffff00', }, { 'name': 'Rack Role 5', 'slug': 'rack-role-5', 'color': 'ffff00', }, { 'name': 'Rack Role 6', 'slug': 'rack-role-6', 'color': 'ffff00', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): rack_roles = ( RackRole(name='Rack Role 1', slug='rack-role-1', color='ff0000'), RackRole(name='Rack Role 2', slug='rack-role-2', color='00ff00'), RackRole(name='Rack Role 3', slug='rack-role-3', color='0000ff'), ) RackRole.objects.bulk_create(rack_roles) class RackTypeTest(APIViewTestCases.APIViewTestCase): model = RackType brief_fields = ['description', 'display', 'id', 'manufacturer', 'model', 'slug', 'url'] bulk_update_data = { 'description': 'new description', } user_permissions = ('dcim.view_manufacturer',) @classmethod def setUpTestData(cls): manufacturers = ( Manufacturer(name='Manufacturer 1', slug='manufacturer-1'), Manufacturer(name='Manufacturer 2', slug='manufacturer-2'), ) Manufacturer.objects.bulk_create(manufacturers) rack_types = ( RackType(manufacturer=manufacturers[0], model='Rack Type 1', slug='rack-type-1', form_factor=RackFormFactorChoices.TYPE_CABINET,), RackType(manufacturer=manufacturers[0], model='Rack Type 2', slug='rack-type-2', form_factor=RackFormFactorChoices.TYPE_CABINET,), RackType(manufacturer=manufacturers[0], model='Rack Type 3', slug='rack-type-3', form_factor=RackFormFactorChoices.TYPE_CABINET,), ) RackType.objects.bulk_create(rack_types) cls.create_data = [ { 'manufacturer': manufacturers[1].pk, 'model': 'Rack Type 4', 'slug': 'rack-type-4', 'form_factor': RackFormFactorChoices.TYPE_CABINET, }, { 'manufacturer': manufacturers[1].pk, 'model': 'Rack Type 5', 'slug': 'rack-type-5', 'form_factor': RackFormFactorChoices.TYPE_CABINET, }, { 'manufacturer': manufacturers[1].pk, 'model': 'Rack Type 6', 'slug': 'rack-type-6', 'form_factor': RackFormFactorChoices.TYPE_CABINET, }, ] class RackTest(APIViewTestCases.APIViewTestCase): model = Rack brief_fields = ['description', 'device_count', 'display', 'id', 'name', 'url'] bulk_update_data = { 'status': 'planned', } user_permissions = ('dcim.view_site', ) @classmethod def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) locations = ( Location.objects.create(site=sites[0], name='Location 1', slug='location-1'), Location.objects.create(site=sites[1], name='Location 2', slug='location-2'), ) rack_roles = ( RackRole(name='Rack Role 1', slug='rack-role-1', color='ff0000'), RackRole(name='Rack Role 2', slug='rack-role-2', color='00ff00'), ) RackRole.objects.bulk_create(rack_roles) racks = ( Rack(site=sites[0], location=locations[0], role=rack_roles[0], name='Rack 1'), Rack(site=sites[0], location=locations[0], role=rack_roles[0], name='Rack 2'), Rack(site=sites[0], location=locations[0], role=rack_roles[0], name='Rack 3'), ) Rack.objects.bulk_create(racks) cls.create_data = [ { 'name': 'Test Rack 4', 'site': sites[1].pk, 'location': locations[1].pk, 'role': rack_roles[1].pk, }, { 'name': 'Test Rack 5', 'site': sites[1].pk, 'location': locations[1].pk, 'role': rack_roles[1].pk, }, { 'name': 'Test Rack 6', 'site': sites[1].pk, 'location': locations[1].pk, 'role': rack_roles[1].pk, }, ] def test_get_rack_elevation(self): """ GET a single rack elevation. """ rack = Rack.objects.first() self.add_permissions('dcim.view_rack') url = reverse('dcim-api:rack-elevation', kwargs={'pk': rack.pk}) # Retrieve all units response = self.client.get(url, **self.header) self.assertEqual(response.data['count'], 84) # Search for specific units response = self.client.get(f'{url}?q=3', **self.header) self.assertEqual(response.data['count'], 26) response = self.client.get(f'{url}?q=U3', **self.header) self.assertEqual(response.data['count'], 22) response = self.client.get(f'{url}?q=U10', **self.header) self.assertEqual(response.data['count'], 2) def test_get_rack_elevation_svg(self): """ GET a single rack elevation in SVG format. """ rack = Rack.objects.first() self.add_permissions('dcim.view_rack') url = '{}?render=svg'.format(reverse('dcim-api:rack-elevation', kwargs={'pk': rack.pk})) response = self.client.get(url, **self.header) self.assertHttpStatus(response, status.HTTP_200_OK) self.assertEqual(response.get('Content-Type'), 'image/svg+xml') class RackReservationTest(APIViewTestCases.APIViewTestCase): model = RackReservation brief_fields = ['description', 'display', 'id', 'units', 'url', 'user'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_rack', 'users.view_user') @classmethod def setUpTestData(cls): user = User.objects.create(username='user1', is_active=True) site = Site.objects.create(name='Test Site 1', slug='test-site-1') racks = ( Rack(site=site, name='Rack 1'), Rack(site=site, name='Rack 2'), ) Rack.objects.bulk_create(racks) rack_reservations = ( RackReservation(rack=racks[0], units=[1, 2, 3], user=user, description='Reservation #1'), RackReservation(rack=racks[0], units=[4, 5, 6], user=user, description='Reservation #2'), RackReservation(rack=racks[0], units=[7, 8, 9], user=user, description='Reservation #3'), ) RackReservation.objects.bulk_create(rack_reservations) cls.create_data = [ { 'rack': racks[1].pk, 'units': [10, 11, 12], 'user': user.pk, 'description': 'Reservation #4', }, { 'rack': racks[1].pk, 'units': [13, 14, 15], 'user': user.pk, 'description': 'Reservation #5', }, { 'rack': racks[1].pk, 'units': [16, 17, 18], 'user': user.pk, 'description': 'Reservation #6', }, ] class ManufacturerTest(APIViewTestCases.APIViewTestCase): model = Manufacturer brief_fields = ['description', 'devicetype_count', 'display', 'id', 'name', 'slug', 'url'] create_data = [ { 'name': 'Manufacturer 4', 'slug': 'manufacturer-4', }, { 'name': 'Manufacturer 5', 'slug': 'manufacturer-5', }, { 'name': 'Manufacturer 6', 'slug': 'manufacturer-6', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturers = ( Manufacturer(name='Manufacturer 1', slug='manufacturer-1'), Manufacturer(name='Manufacturer 2', slug='manufacturer-2'), Manufacturer(name='Manufacturer 3', slug='manufacturer-3'), ) Manufacturer.objects.bulk_create(manufacturers) class DeviceTypeTest(APIViewTestCases.APIViewTestCase): model = DeviceType brief_fields = ['description', 'device_count', 'display', 'id', 'manufacturer', 'model', 'slug', 'url'] bulk_update_data = { 'part_number': 'ABC123', } user_permissions = ('dcim.view_manufacturer', ) @classmethod def setUpTestData(cls): manufacturers = ( Manufacturer(name='Manufacturer 1', slug='manufacturer-1'), Manufacturer(name='Manufacturer 2', slug='manufacturer-2'), ) Manufacturer.objects.bulk_create(manufacturers) device_types = ( DeviceType(manufacturer=manufacturers[0], model='Device Type 1', slug='device-type-1'), DeviceType(manufacturer=manufacturers[0], model='Device Type 2', slug='device-type-2'), DeviceType(manufacturer=manufacturers[0], model='Device Type 3', slug='device-type-3'), ) DeviceType.objects.bulk_create(device_types) cls.create_data = [ { 'manufacturer': manufacturers[1].pk, 'model': 'Device Type 4', 'slug': 'device-type-4', 'u_height': 0, }, { 'manufacturer': manufacturers[1].pk, 'model': 'Device Type 5', 'slug': 'device-type-5', 'u_height': 0.5, }, { 'manufacturer': manufacturers[1].pk, 'model': 'Device Type 6', 'slug': 'device-type-6', 'u_height': 1, }, ] class ModuleTypeTest(APIViewTestCases.APIViewTestCase): model = ModuleType brief_fields = ['description', 'display', 'id', 'manufacturer', 'model', 'url'] bulk_update_data = { 'part_number': 'ABC123', } user_permissions = ('dcim.view_manufacturer', ) @classmethod def setUpTestData(cls): manufacturers = ( Manufacturer(name='Manufacturer 1', slug='manufacturer-1'), Manufacturer(name='Manufacturer 2', slug='manufacturer-2'), ) Manufacturer.objects.bulk_create(manufacturers) module_types = ( ModuleType(manufacturer=manufacturers[0], model='Module Type 1'), ModuleType(manufacturer=manufacturers[0], model='Module Type 2'), ModuleType(manufacturer=manufacturers[0], model='Module Type 3'), ) ModuleType.objects.bulk_create(module_types) cls.create_data = [ { 'manufacturer': manufacturers[1].pk, 'model': 'Module Type 4', }, { 'manufacturer': manufacturers[1].pk, 'model': 'Module Type 5', }, { 'manufacturer': manufacturers[1].pk, 'model': 'Module Type 6', }, ] class ConsolePortTemplateTest(APIViewTestCases.APIViewTestCase): model = ConsolePortTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) console_port_templates = ( ConsolePortTemplate(device_type=devicetype, name='Console Port Template 1'), ConsolePortTemplate(device_type=devicetype, name='Console Port Template 2'), ConsolePortTemplate(device_type=devicetype, name='Console Port Template 3'), ) ConsolePortTemplate.objects.bulk_create(console_port_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Console Port Template 4', }, { 'device_type': devicetype.pk, 'name': 'Console Port Template 5', }, { 'module_type': moduletype.pk, 'name': 'Console Port Template 6', }, { 'module_type': moduletype.pk, 'name': 'Console Port Template 7', }, ] class ConsoleServerPortTemplateTest(APIViewTestCases.APIViewTestCase): model = ConsoleServerPortTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) console_server_port_templates = ( ConsoleServerPortTemplate(device_type=devicetype, name='Console Server Port Template 1'), ConsoleServerPortTemplate(device_type=devicetype, name='Console Server Port Template 2'), ConsoleServerPortTemplate(device_type=devicetype, name='Console Server Port Template 3'), ) ConsoleServerPortTemplate.objects.bulk_create(console_server_port_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Console Server Port Template 4', }, { 'device_type': devicetype.pk, 'name': 'Console Server Port Template 5', }, { 'module_type': moduletype.pk, 'name': 'Console Server Port Template 6', }, { 'module_type': moduletype.pk, 'name': 'Console Server Port Template 7', }, ] class PowerPortTemplateTest(APIViewTestCases.APIViewTestCase): model = PowerPortTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) power_port_templates = ( PowerPortTemplate(device_type=devicetype, name='Power Port Template 1'), PowerPortTemplate(device_type=devicetype, name='Power Port Template 2'), PowerPortTemplate(device_type=devicetype, name='Power Port Template 3'), ) PowerPortTemplate.objects.bulk_create(power_port_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Power Port Template 4', }, { 'device_type': devicetype.pk, 'name': 'Power Port Template 5', }, { 'module_type': moduletype.pk, 'name': 'Power Port Template 6', }, { 'module_type': moduletype.pk, 'name': 'Power Port Template 7', }, ] class PowerOutletTemplateTest(APIViewTestCases.APIViewTestCase): model = PowerOutletTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_devicetype', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) power_port_templates = ( PowerPortTemplate(device_type=devicetype, name='Power Port Template 1'), PowerPortTemplate(device_type=devicetype, name='Power Port Template 2'), ) PowerPortTemplate.objects.bulk_create(power_port_templates) power_outlet_templates = ( PowerOutletTemplate(device_type=devicetype, name='Power Outlet Template 1'), PowerOutletTemplate(device_type=devicetype, name='Power Outlet Template 2'), PowerOutletTemplate(device_type=devicetype, name='Power Outlet Template 3'), ) PowerOutletTemplate.objects.bulk_create(power_outlet_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Power Outlet Template 4', 'power_port': power_port_templates[0].pk, }, { 'device_type': devicetype.pk, 'name': 'Power Outlet Template 5', 'power_port': power_port_templates[1].pk, }, { 'device_type': devicetype.pk, 'name': 'Power Outlet Template 6', 'power_port': None, }, { 'module_type': moduletype.pk, 'name': 'Power Outlet Template 7', }, { 'module_type': moduletype.pk, 'name': 'Power Outlet Template 8', }, ] class InterfaceTemplateTest(APIViewTestCases.APIViewTestCase): model = InterfaceTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) interface_templates = ( InterfaceTemplate(device_type=devicetype, name='Interface Template 1', type='1000base-t'), InterfaceTemplate(device_type=devicetype, name='Interface Template 2', type='1000base-t'), InterfaceTemplate(device_type=devicetype, name='Interface Template 3', type='1000base-t'), ) InterfaceTemplate.objects.bulk_create(interface_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Interface Template 4', 'type': '1000base-t', }, { 'device_type': devicetype.pk, 'name': 'Interface Template 5', 'type': '1000base-t', }, { 'module_type': moduletype.pk, 'name': 'Interface Template 6', 'type': '1000base-t', }, { 'module_type': moduletype.pk, 'name': 'Interface Template 7', 'type': '1000base-t', }, ] class FrontPortTemplateTest(APIViewTestCases.APIViewTestCase): model = FrontPortTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_rearporttemplate', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) rear_port_templates = ( RearPortTemplate(device_type=devicetype, name='Rear Port Template 1', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(device_type=devicetype, name='Rear Port Template 2', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(device_type=devicetype, name='Rear Port Template 3', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(device_type=devicetype, name='Rear Port Template 4', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(module_type=moduletype, name='Rear Port Template 5', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(module_type=moduletype, name='Rear Port Template 6', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(module_type=moduletype, name='Rear Port Template 7', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(module_type=moduletype, name='Rear Port Template 8', type=PortTypeChoices.TYPE_8P8C), ) RearPortTemplate.objects.bulk_create(rear_port_templates) front_port_templates = ( FrontPortTemplate( device_type=devicetype, name='Front Port Template 1', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_port_templates[0] ), FrontPortTemplate( device_type=devicetype, name='Front Port Template 2', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_port_templates[1] ), FrontPortTemplate( module_type=moduletype, name='Front Port Template 5', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_port_templates[4] ), FrontPortTemplate( module_type=moduletype, name='Front Port Template 6', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_port_templates[5] ), ) FrontPortTemplate.objects.bulk_create(front_port_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Front Port Template 3', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_port_templates[2].pk, 'rear_port_position': 1, }, { 'device_type': devicetype.pk, 'name': 'Front Port Template 4', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_port_templates[3].pk, 'rear_port_position': 1, }, { 'module_type': moduletype.pk, 'name': 'Front Port Template 7', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_port_templates[6].pk, 'rear_port_position': 1, }, { 'module_type': moduletype.pk, 'name': 'Front Port Template 8', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_port_templates[7].pk, 'rear_port_position': 1, }, ] class RearPortTemplateTest(APIViewTestCases.APIViewTestCase): model = RearPortTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) moduletype = ModuleType.objects.create( manufacturer=manufacturer, model='Module Type 1' ) rear_port_templates = ( RearPortTemplate(device_type=devicetype, name='Rear Port Template 1', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(device_type=devicetype, name='Rear Port Template 2', type=PortTypeChoices.TYPE_8P8C), RearPortTemplate(device_type=devicetype, name='Rear Port Template 3', type=PortTypeChoices.TYPE_8P8C), ) RearPortTemplate.objects.bulk_create(rear_port_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Rear Port Template 4', 'type': PortTypeChoices.TYPE_8P8C, }, { 'device_type': devicetype.pk, 'name': 'Rear Port Template 5', 'type': PortTypeChoices.TYPE_8P8C, }, { 'module_type': moduletype.pk, 'name': 'Rear Port Template 6', 'type': PortTypeChoices.TYPE_8P8C, }, { 'module_type': moduletype.pk, 'name': 'Rear Port Template 7', 'type': PortTypeChoices.TYPE_8P8C, }, ] class ModuleBayTemplateTest(APIViewTestCases.APIViewTestCase): model = ModuleBayTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_devicetype', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1', subdevice_role=SubdeviceRoleChoices.ROLE_PARENT ) module_bay_templates = ( ModuleBayTemplate(device_type=devicetype, name='Module Bay Template 1'), ModuleBayTemplate(device_type=devicetype, name='Module Bay Template 2'), ModuleBayTemplate(device_type=devicetype, name='Module Bay Template 3'), ) ModuleBayTemplate.objects.bulk_create(module_bay_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Module Bay Template 4', }, { 'device_type': devicetype.pk, 'name': 'Module Bay Template 5', }, { 'device_type': devicetype.pk, 'name': 'Module Bay Template 6', }, ] class DeviceBayTemplateTest(APIViewTestCases.APIViewTestCase): model = DeviceBayTemplate brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_devicetype', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1', subdevice_role=SubdeviceRoleChoices.ROLE_PARENT ) device_bay_templates = ( DeviceBayTemplate(device_type=devicetype, name='Device Bay Template 1'), DeviceBayTemplate(device_type=devicetype, name='Device Bay Template 2'), DeviceBayTemplate(device_type=devicetype, name='Device Bay Template 3'), ) DeviceBayTemplate.objects.bulk_create(device_bay_templates) cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Device Bay Template 4', }, { 'device_type': devicetype.pk, 'name': 'Device Bay Template 5', }, { 'device_type': devicetype.pk, 'name': 'Device Bay Template 6', }, ] class InventoryItemTemplateTest(APIViewTestCases.APIViewTestCase): model = InventoryItemTemplate brief_fields = ['_depth', 'description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_devicetype', 'dcim.view_manufacturer',) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1' ) role = InventoryItemRole.objects.create(name='Inventory Item Role 1', slug='inventory-item-role-1') inventory_item_templates = ( InventoryItemTemplate(device_type=devicetype, name='Inventory Item Template 1', manufacturer=manufacturer, role=role), InventoryItemTemplate(device_type=devicetype, name='Inventory Item Template 2', manufacturer=manufacturer, role=role), InventoryItemTemplate(device_type=devicetype, name='Inventory Item Template 3', manufacturer=manufacturer, role=role), InventoryItemTemplate(device_type=devicetype, name='Inventory Item Template 4', manufacturer=manufacturer, role=role), ) for item in inventory_item_templates: item.save() cls.create_data = [ { 'device_type': devicetype.pk, 'name': 'Inventory Item Template 5', 'manufacturer': manufacturer.pk, 'role': role.pk, 'parent': inventory_item_templates[3].pk, }, { 'device_type': devicetype.pk, 'name': 'Inventory Item Template 6', 'manufacturer': manufacturer.pk, 'role': role.pk, 'parent': inventory_item_templates[3].pk, }, { 'device_type': devicetype.pk, 'name': 'Inventory Item Template 7', 'manufacturer': manufacturer.pk, 'role': role.pk, 'parent': inventory_item_templates[3].pk, }, ] class DeviceRoleTest(APIViewTestCases.APIViewTestCase): model = DeviceRole brief_fields = ['description', 'device_count', 'display', 'id', 'name', 'slug', 'url', 'virtualmachine_count'] create_data = [ { 'name': 'Device Role 4', 'slug': 'device-role-4', 'color': 'ffff00', }, { 'name': 'Device Role 5', 'slug': 'device-role-5', 'color': 'ffff00', }, { 'name': 'Device Role 6', 'slug': 'device-role-6', 'color': 'ffff00', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): roles = ( DeviceRole(name='Device Role 1', slug='device-role-1', color='ff0000'), DeviceRole(name='Device Role 2', slug='device-role-2', color='00ff00'), DeviceRole(name='Device Role 3', slug='device-role-3', color='0000ff'), ) DeviceRole.objects.bulk_create(roles) class PlatformTest(APIViewTestCases.APIViewTestCase): model = Platform brief_fields = ['description', 'device_count', 'display', 'id', 'name', 'slug', 'url', 'virtualmachine_count'] create_data = [ { 'name': 'Platform 4', 'slug': 'platform-4', }, { 'name': 'Platform 5', 'slug': 'platform-5', }, { 'name': 'Platform 6', 'slug': 'platform-6', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): platforms = ( Platform(name='Platform 1', slug='platform-1'), Platform(name='Platform 2', slug='platform-2'), Platform(name='Platform 3', slug='platform-3'), ) Platform.objects.bulk_create(platforms) class DeviceTest(APIViewTestCases.APIViewTestCase): model = Device brief_fields = ['description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'status': 'failed', } user_permissions = ( 'dcim.view_site', 'dcim.view_rack', 'dcim.view_location', 'dcim.view_devicerole', 'dcim.view_devicetype', 'extras.view_configtemplate', ) @classmethod def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) racks = ( Rack(name='Rack 1', site=sites[0]), Rack(name='Rack 2', site=sites[1]), ) Rack.objects.bulk_create(racks) manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') device_types = ( DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'), DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2', u_height=2), ) DeviceType.objects.bulk_create(device_types) roles = ( DeviceRole(name='Device Role 1', slug='device-role-1', color='ff0000'), DeviceRole(name='Device Role 2', slug='device-role-2', color='00ff00'), ) DeviceRole.objects.bulk_create(roles) cluster_type = ClusterType.objects.create(name='Cluster Type 1', slug='cluster-type-1') clusters = ( Cluster(name='Cluster 1', type=cluster_type), Cluster(name='Cluster 2', type=cluster_type), ) Cluster.objects.bulk_create(clusters) devices = ( Device( device_type=device_types[0], role=roles[0], name='Device 1', site=sites[0], rack=racks[0], cluster=clusters[0], local_context_data={'A': 1} ), Device( device_type=device_types[0], role=roles[0], name='Device 2', site=sites[0], rack=racks[0], cluster=clusters[0], local_context_data={'B': 2} ), Device( device_type=device_types[0], role=roles[0], name='Device 3', site=sites[0], rack=racks[0], cluster=clusters[0], local_context_data={'C': 3} ), ) Device.objects.bulk_create(devices) cls.create_data = [ { 'device_type': device_types[1].pk, 'role': roles[1].pk, 'name': 'Test Device 4', 'site': sites[1].pk, 'rack': racks[1].pk, 'cluster': clusters[1].pk, }, { 'device_type': device_types[1].pk, 'role': roles[1].pk, 'name': 'Test Device 5', 'site': sites[1].pk, 'rack': racks[1].pk, 'cluster': clusters[1].pk, }, { 'device_type': device_types[1].pk, 'role': roles[1].pk, 'name': 'Test Device 6', 'site': sites[1].pk, 'rack': racks[1].pk, 'cluster': clusters[1].pk, }, ] def test_config_context_included_by_default_in_list_view(self): """ Check that config context data is included by default in the devices list. """ self.add_permissions('dcim.view_device') url = reverse('dcim-api:device-list') + '?slug=device-with-context-data' response = self.client.get(url, **self.header) self.assertEqual(response.data['results'][0].get('config_context', {}).get('A'), 1) def test_config_context_excluded(self): """ Check that config context data can be excluded by passing ?exclude=config_context. """ self.add_permissions('dcim.view_device') url = reverse('dcim-api:device-list') + '?exclude=config_context' response = self.client.get(url, **self.header) self.assertFalse('config_context' in response.data['results'][0]) def test_unique_name_per_site_constraint(self): """ Check that creating a device with a duplicate name within a site fails. """ device = Device.objects.first() data = { 'device_type': device.device_type.pk, 'role': device.role.pk, 'site': device.site.pk, 'name': device.name, } self.add_permissions('dcim.add_device') url = reverse('dcim-api:device-list') response = self.client.post(url, data, format='json', **self.header) self.assertHttpStatus(response, status.HTTP_400_BAD_REQUEST) def test_rack_fit(self): """ Check that creating multiple devices with overlapping position fails. """ device = Device.objects.first() device_type = DeviceType.objects.all()[1] data = [ { 'device_type': device_type.pk, 'role': device.role.pk, 'site': device.site.pk, 'name': 'Test Device 7', 'rack': device.rack.pk, 'face': 'front', 'position': 1 }, { 'device_type': device_type.pk, 'role': device.role.pk, 'site': device.site.pk, 'name': 'Test Device 8', 'rack': device.rack.pk, 'face': 'front', 'position': 2 } ] self.add_permissions('dcim.add_device') url = reverse('dcim-api:device-list') response = self.client.post(url, data, format='json', **self.header) self.assertHttpStatus(response, status.HTTP_400_BAD_REQUEST) def test_render_config(self): configtemplate = ConfigTemplate.objects.create( name='Config Template 1', template_code='Config for device {{ device.name }}' ) device = Device.objects.first() device.config_template = configtemplate device.save() self.add_permissions('dcim.add_device') url = reverse('dcim-api:device-detail', kwargs={'pk': device.pk}) + 'render-config/' response = self.client.post(url, {}, format='json', **self.header) self.assertHttpStatus(response, status.HTTP_200_OK) self.assertEqual(response.data['content'], f'Config for device {device.name}') class ModuleTest(APIViewTestCases.APIViewTestCase): model = Module brief_fields = ['description', 'device', 'display', 'id', 'module_bay', 'module_type', 'url'] bulk_update_data = { 'serial': '1234ABCD', } user_permissions = ('dcim.view_modulebay', 'dcim.view_moduletype', 'dcim.view_device') @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Generic', slug='generic') device = create_test_device('Test Device 1') module_types = ( ModuleType(manufacturer=manufacturer, model='Module Type 1'), ModuleType(manufacturer=manufacturer, model='Module Type 2'), ModuleType(manufacturer=manufacturer, model='Module Type 3'), ) ModuleType.objects.bulk_create(module_types) module_bays = ( ModuleBay(device=device, name='Module Bay 1'), ModuleBay(device=device, name='Module Bay 2'), ModuleBay(device=device, name='Module Bay 3'), ModuleBay(device=device, name='Module Bay 4'), ModuleBay(device=device, name='Module Bay 5'), ModuleBay(device=device, name='Module Bay 6'), ) for module_bay in module_bays: module_bay.save() modules = ( Module(device=device, module_bay=module_bays[0], module_type=module_types[0]), Module(device=device, module_bay=module_bays[1], module_type=module_types[1]), Module(device=device, module_bay=module_bays[2], module_type=module_types[2]), ) Module.objects.bulk_create(modules) cls.create_data = [ { 'device': device.pk, 'module_bay': module_bays[3].pk, 'module_type': module_types[0].pk, 'status': ModuleStatusChoices.STATUS_ACTIVE, 'serial': 'ABC123', 'asset_tag': 'Foo1', }, { 'device': device.pk, 'module_bay': module_bays[4].pk, 'module_type': module_types[1].pk, 'status': ModuleStatusChoices.STATUS_ACTIVE, 'serial': 'DEF456', 'asset_tag': 'Foo2', }, { 'device': device.pk, 'module_bay': module_bays[5].pk, 'module_type': module_types[2].pk, 'status': ModuleStatusChoices.STATUS_ACTIVE, 'serial': 'GHI789', 'asset_tag': 'Foo3', }, ] class ConsolePortTest(Mixins.ComponentTraceMixin, APIViewTestCases.APIViewTestCase): model = ConsolePort brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = ConsoleServerPort user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) console_ports = ( ConsolePort(device=device, name='Console Port 1'), ConsolePort(device=device, name='Console Port 2'), ConsolePort(device=device, name='Console Port 3'), ) ConsolePort.objects.bulk_create(console_ports) cls.create_data = [ { 'device': device.pk, 'name': 'Console Port 4', 'speed': 9600, }, { 'device': device.pk, 'name': 'Console Port 5', 'speed': 115200, }, { 'device': device.pk, 'name': 'Console Port 6', 'speed': None, }, ] class ConsoleServerPortTest(Mixins.ComponentTraceMixin, APIViewTestCases.APIViewTestCase): model = ConsoleServerPort brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = ConsolePort user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) console_server_ports = ( ConsoleServerPort(device=device, name='Console Server Port 1'), ConsoleServerPort(device=device, name='Console Server Port 2'), ConsoleServerPort(device=device, name='Console Server Port 3'), ) ConsoleServerPort.objects.bulk_create(console_server_ports) cls.create_data = [ { 'device': device.pk, 'name': 'Console Server Port 4', 'speed': 9600, }, { 'device': device.pk, 'name': 'Console Server Port 5', 'speed': 115200, }, { 'device': device.pk, 'name': 'Console Server Port 6', 'speed': None, }, ] class PowerPortTest(Mixins.ComponentTraceMixin, APIViewTestCases.APIViewTestCase): model = PowerPort brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = PowerOutlet user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) power_ports = ( PowerPort(device=device, name='Power Port 1'), PowerPort(device=device, name='Power Port 2'), PowerPort(device=device, name='Power Port 3'), ) PowerPort.objects.bulk_create(power_ports) cls.create_data = [ { 'device': device.pk, 'name': 'Power Port 4', }, { 'device': device.pk, 'name': 'Power Port 5', }, { 'device': device.pk, 'name': 'Power Port 6', }, ] class PowerOutletTest(Mixins.ComponentTraceMixin, APIViewTestCases.APIViewTestCase): model = PowerOutlet brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = PowerPort user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) power_ports = ( PowerPort(device=device, name='Power Port 1'), PowerPort(device=device, name='Power Port 2'), ) PowerPort.objects.bulk_create(power_ports) power_outlets = ( PowerOutlet(device=device, name='Power Outlet 1'), PowerOutlet(device=device, name='Power Outlet 2'), PowerOutlet(device=device, name='Power Outlet 3'), ) PowerOutlet.objects.bulk_create(power_outlets) cls.create_data = [ { 'device': device.pk, 'name': 'Power Outlet 4', 'power_port': power_ports[0].pk, }, { 'device': device.pk, 'name': 'Power Outlet 5', 'power_port': power_ports[1].pk, }, { 'device': device.pk, 'name': 'Power Outlet 6', 'power_port': None, }, ] class InterfaceTest(Mixins.ComponentTraceMixin, APIViewTestCases.APIViewTestCase): model = Interface brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = Interface user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) interfaces = ( Interface(device=device, name='Interface 1', type='1000base-t'), Interface(device=device, name='Interface 2', type='1000base-t'), Interface(device=device, name='Interface 3', type='1000base-t'), ) Interface.objects.bulk_create(interfaces) vdcs = ( VirtualDeviceContext(name='VDC 1', identifier=1, device=device), VirtualDeviceContext(name='VDC 2', identifier=2, device=device) ) VirtualDeviceContext.objects.bulk_create(vdcs) vlans = ( VLAN(name='VLAN 1', vid=1), VLAN(name='VLAN 2', vid=2), VLAN(name='VLAN 3', vid=3), ) VLAN.objects.bulk_create(vlans) wireless_lans = ( WirelessLAN(ssid='WLAN1'), WirelessLAN(ssid='WLAN2'), ) WirelessLAN.objects.bulk_create(wireless_lans) vrfs = ( VRF(name='VRF 1'), VRF(name='VRF 2'), VRF(name='VRF 3'), ) VRF.objects.bulk_create(vrfs) cls.create_data = [ { 'device': device.pk, 'name': 'Interface 4', 'type': '1000base-t', 'mode': InterfaceModeChoices.MODE_TAGGED, 'speed': 1000000, 'duplex': 'full', 'vrf': vrfs[0].pk, 'poe_mode': InterfacePoEModeChoices.MODE_PD, 'poe_type': InterfacePoETypeChoices.TYPE_1_8023AF, 'tagged_vlans': [vlans[0].pk, vlans[1].pk], 'untagged_vlan': vlans[2].pk, }, { 'device': device.pk, 'name': 'Interface 5', 'type': '1000base-t', 'mode': InterfaceModeChoices.MODE_TAGGED, 'bridge': interfaces[0].pk, 'speed': 100000, 'duplex': 'half', 'vrf': vrfs[1].pk, 'tagged_vlans': [vlans[0].pk, vlans[1].pk], 'untagged_vlan': vlans[2].pk, }, { 'device': device.pk, 'vdcs': [vdcs[0].pk], 'name': 'Interface 6', 'type': 'virtual', 'mode': InterfaceModeChoices.MODE_TAGGED, 'parent': interfaces[1].pk, 'vrf': vrfs[2].pk, 'tagged_vlans': [vlans[0].pk, vlans[1].pk], 'untagged_vlan': vlans[2].pk, }, { 'device': device.pk, 'vdcs': [vdcs[1].pk], 'name': 'Interface 7', 'type': InterfaceTypeChoices.TYPE_80211A, 'tx_power': 10, 'wireless_lans': [wireless_lans[0].pk, wireless_lans[1].pk], 'rf_channel': WirelessChannelChoices.CHANNEL_5G_32, }, { 'device': device.pk, 'vdcs': [vdcs[1].pk], 'name': 'Interface 8', 'type': InterfaceTypeChoices.TYPE_80211A, 'tx_power': 10, 'wireless_lans': [wireless_lans[0].pk, wireless_lans[1].pk], 'rf_channel': "", }, ] def test_bulk_delete_child_interfaces(self): interface1 = Interface.objects.get(name='Interface 1') device = interface1.device self.add_permissions('dcim.delete_interface') # Create a child interface child = Interface.objects.create( device=device, name='Interface 1A', type=InterfaceTypeChoices.TYPE_VIRTUAL, parent=interface1 ) self.assertEqual(device.interfaces.count(), 4) # Attempt to delete only the parent interface url = self._get_detail_url(interface1) self.client.delete(url, **self.header) self.assertEqual(device.interfaces.count(), 4) # Parent was not deleted # Attempt to bulk delete parent & child together data = [ {"id": interface1.pk}, {"id": child.pk}, ] self.client.delete(self._get_list_url(), data, format='json', **self.header) self.assertEqual(device.interfaces.count(), 2) # Child & parent were both deleted class FrontPortTest(APIViewTestCases.APIViewTestCase): model = FrontPort brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = Interface user_permissions = ('dcim.view_device', 'dcim.view_rearport') @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) rear_ports = ( RearPort(device=device, name='Rear Port 1', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 2', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 3', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 4', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 5', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 6', type=PortTypeChoices.TYPE_8P8C), ) RearPort.objects.bulk_create(rear_ports) front_ports = ( FrontPort(device=device, name='Front Port 1', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_ports[0]), FrontPort(device=device, name='Front Port 2', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_ports[1]), FrontPort(device=device, name='Front Port 3', type=PortTypeChoices.TYPE_8P8C, rear_port=rear_ports[2]), ) FrontPort.objects.bulk_create(front_ports) cls.create_data = [ { 'device': device.pk, 'name': 'Front Port 4', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_ports[3].pk, 'rear_port_position': 1, }, { 'device': device.pk, 'name': 'Front Port 5', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_ports[4].pk, 'rear_port_position': 1, }, { 'device': device.pk, 'name': 'Front Port 6', 'type': PortTypeChoices.TYPE_8P8C, 'rear_port': rear_ports[5].pk, 'rear_port_position': 1, }, ] class RearPortTest(APIViewTestCases.APIViewTestCase): model = RearPort brief_fields = ['_occupied', 'cable', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } peer_termination_type = Interface user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) rear_ports = ( RearPort(device=device, name='Rear Port 1', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 2', type=PortTypeChoices.TYPE_8P8C), RearPort(device=device, name='Rear Port 3', type=PortTypeChoices.TYPE_8P8C), ) RearPort.objects.bulk_create(rear_ports) cls.create_data = [ { 'device': device.pk, 'name': 'Rear Port 4', 'type': PortTypeChoices.TYPE_8P8C, }, { 'device': device.pk, 'name': 'Rear Port 5', 'type': PortTypeChoices.TYPE_8P8C, }, { 'device': device.pk, 'name': 'Rear Port 6', 'type': PortTypeChoices.TYPE_8P8C, }, ] class ModuleBayTest(APIViewTestCases.APIViewTestCase): model = ModuleBay brief_fields = ['description', 'display', 'id', 'installed_module', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') device = Device.objects.create(device_type=device_type, role=role, name='Device 1', site=site) module_bays = ( ModuleBay(device=device, name='Device Bay 1'), ModuleBay(device=device, name='Device Bay 2'), ModuleBay(device=device, name='Device Bay 3'), ) for module_bay in module_bays: module_bay.save() cls.create_data = [ { 'device': device.pk, 'name': 'Device Bay 4', }, { 'device': device.pk, 'name': 'Device Bay 5', }, { 'device': device.pk, 'name': 'Device Bay 6', }, ] class DeviceBayTest(APIViewTestCases.APIViewTestCase): model = DeviceBay brief_fields = ['description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_device', ) @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device_types = ( DeviceType( manufacturer=manufacturer, model='Device Type 1', slug='device-type-1', subdevice_role=SubdeviceRoleChoices.ROLE_PARENT ), DeviceType( manufacturer=manufacturer, model='Device Type 2', slug='device-type-2', subdevice_role=SubdeviceRoleChoices.ROLE_CHILD ), ) DeviceType.objects.bulk_create(device_types) devices = ( Device(device_type=device_types[0], role=role, name='Device 1', site=site), Device(device_type=device_types[1], role=role, name='Device 2', site=site), Device(device_type=device_types[1], role=role, name='Device 3', site=site), Device(device_type=device_types[1], role=role, name='Device 4', site=site), ) Device.objects.bulk_create(devices) device_bays = ( DeviceBay(device=devices[0], name='Device Bay 1'), DeviceBay(device=devices[0], name='Device Bay 2'), DeviceBay(device=devices[0], name='Device Bay 3'), ) DeviceBay.objects.bulk_create(device_bays) cls.create_data = [ { 'device': devices[0].pk, 'name': 'Device Bay 4', 'installed_device': devices[1].pk, }, { 'device': devices[0].pk, 'name': 'Device Bay 5', 'installed_device': devices[2].pk, }, { 'device': devices[0].pk, 'name': 'Device Bay 6', 'installed_device': devices[3].pk, }, ] class InventoryItemTest(APIViewTestCases.APIViewTestCase): model = InventoryItem brief_fields = ['_depth', 'description', 'device', 'display', 'id', 'name', 'url'] bulk_update_data = { 'description': 'New description', } user_permissions = ('dcim.view_device', 'dcim.view_manufacturer') @classmethod def setUpTestData(cls): manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') site = Site.objects.create(name='Site 1', slug='site-1') role = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1', color='ff0000') device = Device.objects.create(device_type=devicetype, role=role, name='Device 1', site=site) roles = ( InventoryItemRole(name='Inventory Item Role 1', slug='inventory-item-role-1'), InventoryItemRole(name='Inventory Item Role 2', slug='inventory-item-role-2'), ) InventoryItemRole.objects.bulk_create(roles) interfaces = ( Interface(device=device, name='Interface 1'), Interface(device=device, name='Interface 2'), Interface(device=device, name='Interface 3'), ) Interface.objects.bulk_create(interfaces) InventoryItem.objects.create(device=device, name='Inventory Item 1', role=roles[0], manufacturer=manufacturer, component=interfaces[0]) InventoryItem.objects.create(device=device, name='Inventory Item 2', role=roles[0], manufacturer=manufacturer, component=interfaces[1]) InventoryItem.objects.create(device=device, name='Inventory Item 3', role=roles[0], manufacturer=manufacturer, component=interfaces[2]) cls.create_data = [ { 'device': device.pk, 'name': 'Inventory Item 4', 'role': roles[1].pk, 'manufacturer': manufacturer.pk, 'component_type': 'dcim.interface', 'component_id': interfaces[0].pk, }, { 'device': device.pk, 'name': 'Inventory Item 5', 'role': roles[1].pk, 'manufacturer': manufacturer.pk, 'component_type': 'dcim.interface', 'component_id': interfaces[1].pk, }, { 'device': device.pk, 'name': 'Inventory Item 6', 'role': roles[1].pk, 'manufacturer': manufacturer.pk, 'component_type': 'dcim.interface', 'component_id': interfaces[2].pk, }, ] class InventoryItemRoleTest(APIViewTestCases.APIViewTestCase): model = InventoryItemRole brief_fields = ['description', 'display', 'id', 'inventoryitem_count', 'name', 'slug', 'url'] create_data = [ { 'name': 'Inventory Item Role 4', 'slug': 'inventory-item-role-4', 'color': 'ffff00', }, { 'name': 'Inventory Item Role 5', 'slug': 'inventory-item-role-5', 'color': 'ffff00', }, { 'name': 'Inventory Item Role 6', 'slug': 'inventory-item-role-6', 'color': 'ffff00', }, ] bulk_update_data = { 'description': 'New description', } @classmethod def setUpTestData(cls): roles = ( InventoryItemRole(name='Inventory Item Role 1', slug='inventory-item-role-1', color='ff0000'), InventoryItemRole(name='Inventory Item Role 2', slug='inventory-item-role-2', color='00ff00'), InventoryItemRole(name='Inventory Item Role 3', slug='inventory-item-role-3', color='0000ff'), ) InventoryItemRole.objects.bulk_create(roles) class CableTest(APIViewTestCases.APIViewTestCase): model = Cable brief_fields = ['description', 'display', 'id', 'label', 'url'] bulk_update_data = { 'length': 100, 'length_unit': 'm', } # TODO: Allow updating cable terminations test_update_object = None def model_to_dict(self, *args, **kwargs): data = super().model_to_dict(*args, **kwargs) # Serialize termination objects if 'a_terminations' in data: data['a_terminations'] = GenericObjectSerializer(data['a_terminations'], many=True).data if 'b_terminations' in data: data['b_terminations'] = GenericObjectSerializer(data['b_terminations'], many=True).data return data @classmethod def setUpTestData(cls): site = Site.objects.create(name='Site 1', slug='site-1') manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1', color='ff0000') devices = ( Device(device_type=devicetype, role=role, name='Device 1', site=site), Device(device_type=devicetype, role=role, name='Device 2', site=site), ) Device.objects.bulk_create(devices) interfaces = [] for device in devices: for i in range(0, 10): interfaces.append(Interface(device=device, type=InterfaceTypeChoices.TYPE_1GE_FIXED, name=f'eth{i}')) Interface.objects.bulk_create(interfaces) cables = ( Cable(a_terminations=[interfaces[0]], b_terminations=[interfaces[10]], label='Cable 1'), Cable(a_terminations=[interfaces[1]], b_terminations=[interfaces[11]], label='Cable 2'), Cable(a_terminations=[interfaces[2]], b_terminations=[interfaces[12]], label='Cable 3'), ) for cable in cables: cable.save() cls.create_data = [ { 'a_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[4].pk, }], 'b_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[14].pk, }], 'label': 'Cable 4', }, { 'a_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[5].pk, }], 'b_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[15].pk, }], 'label': 'Cable 5', }, { 'a_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[6].pk, }], 'b_terminations': [{ 'object_type': 'dcim.interface', 'object_id': interfaces[16].pk, }], 'label': 'Cable 6', }, ] class ConnectedDeviceTest(APITestCase): @classmethod def setUpTestData(cls): site = Site.objects.create(name='Site 1', slug='site-1') manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1', color='ff0000') devices = ( Device(device_type=devicetype, role=role, name='TestDevice1', site=site), Device(device_type=devicetype, role=role, name='TestDevice2', site=site), ) Device.objects.bulk_create(devices) interfaces = ( Interface(device=devices[0], name='eth0'), Interface(device=devices[1], name='eth0'), Interface(device=devices[0], name='eth1'), # Not connected ) Interface.objects.bulk_create(interfaces) cable = Cable(a_terminations=[interfaces[0]], b_terminations=[interfaces[1]]) cable.save() @override_settings(EXEMPT_VIEW_PERMISSIONS=['*']) def test_get_connected_device(self): url = reverse('dcim-api:connected-device-list') url_params = f'?peer_device=TestDevice1&peer_interface=eth0' response = self.client.get(url + url_params, **self.header) self.assertHttpStatus(response, status.HTTP_200_OK) self.assertEqual(response.data['name'], 'TestDevice2') url_params = f'?peer_device=TestDevice1&peer_interface=eth1' response = self.client.get(url + url_params, **self.header) self.assertHttpStatus(response, status.HTTP_404_NOT_FOUND) class VirtualChassisTest(APIViewTestCases.APIViewTestCase): model = VirtualChassis brief_fields = ['description', 'display', 'id', 'master', 'member_count', 'name', 'url'] @classmethod def setUpTestData(cls): site = Site.objects.create(name='Test Site', slug='test-site') manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type', slug='device-type') role = DeviceRole.objects.create(name='Device Role', slug='device-role', color='ff0000') devices = ( Device(name='Device 1', device_type=devicetype, role=role, site=site), Device(name='Device 2', device_type=devicetype, role=role, site=site), Device(name='Device 3', device_type=devicetype, role=role, site=site), Device(name='Device 4', device_type=devicetype, role=role, site=site), Device(name='Device 5', device_type=devicetype, role=role, site=site), Device(name='Device 6', device_type=devicetype, role=role, site=site), Device(name='Device 7', device_type=devicetype, role=role, site=site), Device(name='Device 8', device_type=devicetype, role=role, site=site), Device(name='Device 9', device_type=devicetype, role=role, site=site), Device(name='Device 10', device_type=devicetype, role=role, site=site), Device(name='Device 11', device_type=devicetype, role=role, site=site), Device(name='Device 12', device_type=devicetype, role=role, site=site), ) Device.objects.bulk_create(devices) # Create 12 interfaces per device interfaces = [] for i, device in enumerate(devices): for j in range(0, 13): interfaces.append( # Interface name starts with parent device's position in VC; e.g. 1/1, 1/2, 1/3... Interface(device=device, name=f'{i % 3 + 1}/{j}', type=InterfaceTypeChoices.TYPE_1GE_FIXED) ) Interface.objects.bulk_create(interfaces) # Create three VirtualChassis with three members each virtual_chassis = ( VirtualChassis(name='Virtual Chassis 1', master=devices[0], domain='domain-1'), VirtualChassis(name='Virtual Chassis 2', master=devices[3], domain='domain-2'), VirtualChassis(name='Virtual Chassis 3', master=devices[6], domain='domain-3'), ) VirtualChassis.objects.bulk_create(virtual_chassis) Device.objects.filter(pk=devices[0].pk).update(virtual_chassis=virtual_chassis[0], vc_position=1) Device.objects.filter(pk=devices[1].pk).update(virtual_chassis=virtual_chassis[0], vc_position=2) Device.objects.filter(pk=devices[2].pk).update(virtual_chassis=virtual_chassis[0], vc_position=3) Device.objects.filter(pk=devices[3].pk).update(virtual_chassis=virtual_chassis[1], vc_position=1) Device.objects.filter(pk=devices[4].pk).update(virtual_chassis=virtual_chassis[1], vc_position=2) Device.objects.filter(pk=devices[5].pk).update(virtual_chassis=virtual_chassis[1], vc_position=3) Device.objects.filter(pk=devices[6].pk).update(virtual_chassis=virtual_chassis[2], vc_position=1) Device.objects.filter(pk=devices[7].pk).update(virtual_chassis=virtual_chassis[2], vc_position=2) Device.objects.filter(pk=devices[8].pk).update(virtual_chassis=virtual_chassis[2], vc_position=3) cls.update_data = { 'name': 'Virtual Chassis X', 'domain': 'domain-x', 'master': devices[1].pk, } cls.create_data = [ { 'name': 'Virtual Chassis 4', 'domain': 'domain-4', }, { 'name': 'Virtual Chassis 5', 'domain': 'domain-5', }, { 'name': 'Virtual Chassis 6', 'domain': 'domain-6', }, ] cls.bulk_update_data = { 'domain': 'newdomain', 'master': None } class PowerPanelTest(APIViewTestCases.APIViewTestCase): model = PowerPanel brief_fields = ['description', 'display', 'id', 'name', 'powerfeed_count', 'url'] user_permissions = ('dcim.view_site', ) @classmethod def setUpTestData(cls): sites = ( Site.objects.create(name='Site 1', slug='site-1'), Site.objects.create(name='Site 2', slug='site-2'), ) locations = ( Location.objects.create(name='Location 1', slug='location-1', site=sites[0]), Location.objects.create(name='Location 2', slug='location-2', site=sites[0]), Location.objects.create(name='Location 3', slug='location-3', site=sites[0]), Location.objects.create(name='Location 4', slug='location-3', site=sites[1]), ) power_panels = ( PowerPanel(site=sites[0], location=locations[0], name='Power Panel 1'), PowerPanel(site=sites[0], location=locations[1], name='Power Panel 2'), PowerPanel(site=sites[0], location=locations[2], name='Power Panel 3'), ) PowerPanel.objects.bulk_create(power_panels) cls.create_data = [ { 'name': 'Power Panel 4', 'site': sites[0].pk, 'location': locations[0].pk, }, { 'name': 'Power Panel 5', 'site': sites[0].pk, 'location': locations[1].pk, }, { 'name': 'Power Panel 6', 'site': sites[0].pk, 'location': locations[2].pk, }, ] cls.bulk_update_data = { 'site': sites[1].pk, 'location': locations[3].pk } class PowerFeedTest(APIViewTestCases.APIViewTestCase): model = PowerFeed brief_fields = ['_occupied', 'cable', 'description', 'display', 'id', 'name', 'url'] bulk_update_data = { 'status': 'planned', } user_permissions = ('dcim.view_powerpanel', ) @classmethod def setUpTestData(cls): site = Site.objects.create(name='Site 1', slug='site-1') location = Location.objects.create(site=site, name='Location 1', slug='location-1') rackrole = RackRole.objects.create(name='Rack Role 1', slug='rack-role-1', color='ff0000') racks = ( Rack(site=site, location=location, role=rackrole, name='Rack 1'), Rack(site=site, location=location, role=rackrole, name='Rack 2'), Rack(site=site, location=location, role=rackrole, name='Rack 3'), Rack(site=site, location=location, role=rackrole, name='Rack 4'), ) Rack.objects.bulk_create(racks) power_panels = ( PowerPanel(site=site, location=location, name='Power Panel 1'), PowerPanel(site=site, location=location, name='Power Panel 2'), ) PowerPanel.objects.bulk_create(power_panels) PRIMARY = PowerFeedTypeChoices.TYPE_PRIMARY REDUNDANT = PowerFeedTypeChoices.TYPE_REDUNDANT power_feeds = ( PowerFeed(power_panel=power_panels[0], rack=racks[0], name='Power Feed 1A', type=PRIMARY), PowerFeed(power_panel=power_panels[1], rack=racks[0], name='Power Feed 1B', type=REDUNDANT), PowerFeed(power_panel=power_panels[0], rack=racks[1], name='Power Feed 2A', type=PRIMARY), PowerFeed(power_panel=power_panels[1], rack=racks[1], name='Power Feed 2B', type=REDUNDANT), PowerFeed(power_panel=power_panels[0], rack=racks[2], name='Power Feed 3A', type=PRIMARY), PowerFeed(power_panel=power_panels[1], rack=racks[2], name='Power Feed 3B', type=REDUNDANT), ) PowerFeed.objects.bulk_create(power_feeds) cls.create_data = [ { 'name': 'Power Feed 4A', 'power_panel': power_panels[0].pk, 'rack': racks[3].pk, 'type': PRIMARY, }, { 'name': 'Power Feed 4B', 'power_panel': power_panels[1].pk, 'rack': racks[3].pk, 'type': REDUNDANT, }, ] class VirtualDeviceContextTest(APIViewTestCases.APIViewTestCase): model = VirtualDeviceContext brief_fields = ['description', 'device', 'display', 'id', 'identifier', 'name', 'url'] bulk_update_data = { 'status': 'planned', } @classmethod def setUpTestData(cls): site = Site.objects.create(name='Test Site', slug='test-site') manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type', slug='device-type') role = DeviceRole.objects.create(name='Device Role', slug='device-role', color='ff0000') devices = ( Device(name='Device 1', device_type=devicetype, role=role, site=site), Device(name='Device 2', device_type=devicetype, role=role, site=site), Device(name='Device 3', device_type=devicetype, role=role, site=site), ) Device.objects.bulk_create(devices) vdcs = ( VirtualDeviceContext(device=devices[1], name='VDC 1', identifier=1, status='active'), VirtualDeviceContext(device=devices[1], name='VDC 2', identifier=2, status='active'), VirtualDeviceContext(device=devices[2], name='VDC 1', identifier=1, status='active'), VirtualDeviceContext(device=devices[2], name='VDC 2', identifier=2, status='active'), VirtualDeviceContext(device=devices[2], name='VDC 3', identifier=3, status='active'), VirtualDeviceContext(device=devices[2], name='VDC 4', identifier=4, status='active'), VirtualDeviceContext(device=devices[2], name='VDC 5', identifier=5, status='active'), ) VirtualDeviceContext.objects.bulk_create(vdcs) cls.create_data = [ { 'device': devices[0].pk, 'status': 'active', 'name': 'VDC 1', 'identifier': 1, }, { 'device': devices[0].pk, 'status': 'active', 'name': 'VDC 2', 'identifier': 2, }, { 'device': devices[1].pk, 'status': 'active', 'name': 'VDC 3', # Omit identifier to test uniqueness constraint }, ]