Add missing bulk update test data

This commit is contained in:
Jeremy Stretch 2020-11-02 16:56:04 -05:00
parent 9fae11a42f
commit 2bb79a2773
5 changed files with 44 additions and 11 deletions

View File

@ -174,3 +174,7 @@ class CircuitTerminationTest(APIViewTestCases.APIViewTestCase):
'port_speed': 200000, 'port_speed': 200000,
}, },
] ]
cls.bulk_update_data = {
'port_speed': 123456
}

View File

@ -2,7 +2,6 @@ from django.contrib.auth.models import User
from django.urls import reverse from django.urls import reverse
from rest_framework import status from rest_framework import status
from circuits.models import Circuit, CircuitTermination, CircuitType, Provider
from dcim.choices import * from dcim.choices import *
from dcim.constants import * from dcim.constants import *
from dcim.models import ( from dcim.models import (
@ -1539,10 +1538,13 @@ class VirtualChassisTest(APIViewTestCases.APIViewTestCase):
VirtualChassis(name='Virtual Chassis 3', master=devices[6], domain='domain-3'), VirtualChassis(name='Virtual Chassis 3', master=devices[6], domain='domain-3'),
) )
VirtualChassis.objects.bulk_create(virtual_chassis) 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[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[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[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[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[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) Device.objects.filter(pk=devices[8].pk).update(virtual_chassis=virtual_chassis[2], vc_position=3)
@ -1567,6 +1569,10 @@ class VirtualChassisTest(APIViewTestCases.APIViewTestCase):
}, },
] ]
cls.bulk_update_data = {
'domain': 'newdomain',
}
class PowerPanelTest(APIViewTestCases.APIViewTestCase): class PowerPanelTest(APIViewTestCases.APIViewTestCase):
model = PowerPanel model = PowerPanel
@ -1574,39 +1580,48 @@ class PowerPanelTest(APIViewTestCases.APIViewTestCase):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
site = Site.objects.create(name='Site 1', slug='site-1') sites = (
Site.objects.create(name='Site 1', slug='site-1'),
Site.objects.create(name='Site 2', slug='site-2'),
)
rack_groups = ( rack_groups = (
RackGroup.objects.create(name='Rack Group 1', slug='rack-group-1', site=site), RackGroup.objects.create(name='Rack Group 1', slug='rack-group-1', site=sites[0]),
RackGroup.objects.create(name='Rack Group 2', slug='rack-group-2', site=site), RackGroup.objects.create(name='Rack Group 2', slug='rack-group-2', site=sites[0]),
RackGroup.objects.create(name='Rack Group 3', slug='rack-group-3', site=site), RackGroup.objects.create(name='Rack Group 3', slug='rack-group-3', site=sites[0]),
RackGroup.objects.create(name='Rack Group 4', slug='rack-group-3', site=sites[1]),
) )
power_panels = ( power_panels = (
PowerPanel(site=site, rack_group=rack_groups[0], name='Power Panel 1'), PowerPanel(site=sites[0], rack_group=rack_groups[0], name='Power Panel 1'),
PowerPanel(site=site, rack_group=rack_groups[1], name='Power Panel 2'), PowerPanel(site=sites[0], rack_group=rack_groups[1], name='Power Panel 2'),
PowerPanel(site=site, rack_group=rack_groups[2], name='Power Panel 3'), PowerPanel(site=sites[0], rack_group=rack_groups[2], name='Power Panel 3'),
) )
PowerPanel.objects.bulk_create(power_panels) PowerPanel.objects.bulk_create(power_panels)
cls.create_data = [ cls.create_data = [
{ {
'name': 'Power Panel 4', 'name': 'Power Panel 4',
'site': site.pk, 'site': sites[0].pk,
'rack_group': rack_groups[0].pk, 'rack_group': rack_groups[0].pk,
}, },
{ {
'name': 'Power Panel 5', 'name': 'Power Panel 5',
'site': site.pk, 'site': sites[0].pk,
'rack_group': rack_groups[1].pk, 'rack_group': rack_groups[1].pk,
}, },
{ {
'name': 'Power Panel 6', 'name': 'Power Panel 6',
'site': site.pk, 'site': sites[0].pk,
'rack_group': rack_groups[2].pk, 'rack_group': rack_groups[2].pk,
}, },
] ]
cls.bulk_update_data = {
'site': sites[1].pk,
'rack_group': rack_groups[3].pk
}
class PowerFeedTest(APIViewTestCases.APIViewTestCase): class PowerFeedTest(APIViewTestCases.APIViewTestCase):
model = PowerFeed model = PowerFeed

View File

@ -36,6 +36,9 @@ class SecretRoleTest(APIViewTestCases.APIViewTestCase):
'slug': 'secret-role-6', 'slug': 'secret-role-6',
}, },
] ]
bulk_update_data = {
'description': 'New description',
}
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
@ -112,6 +115,10 @@ class SecretTest(APIViewTestCases.APIViewTestCase):
}, },
] ]
self.bulk_update_data = {
'role': secret_roles[1].pk,
}
def prepare_instance(self, instance): def prepare_instance(self, instance):
# Unlock the plaintext prior to evaluation of the instance # Unlock the plaintext prior to evaluation of the instance
instance.decrypt(self.master_key) instance.decrypt(self.master_key)

View File

@ -132,6 +132,10 @@ class ObjectPermissionTest(APIViewTestCases.APIViewTestCase):
}, },
] ]
cls.bulk_update_data = {
'description': 'New description',
}
class UserConfigTest(APITestCase): class UserConfigTest(APITestCase):

View File

@ -84,6 +84,9 @@ class ClusterGroupTest(APIViewTestCases.APIViewTestCase):
class ClusterTest(APIViewTestCases.APIViewTestCase): class ClusterTest(APIViewTestCases.APIViewTestCase):
model = Cluster model = Cluster
brief_fields = ['id', 'name', 'url', 'virtualmachine_count'] brief_fields = ['id', 'name', 'url', 'virtualmachine_count']
bulk_update_data = {
'comments': 'New comment',
}
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):