1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

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,
},
]
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 rest_framework import status
from circuits.models import Circuit, CircuitTermination, CircuitType, Provider
from dcim.choices import *
from dcim.constants 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.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)
@ -1567,6 +1569,10 @@ class VirtualChassisTest(APIViewTestCases.APIViewTestCase):
},
]
cls.bulk_update_data = {
'domain': 'newdomain',
}
class PowerPanelTest(APIViewTestCases.APIViewTestCase):
model = PowerPanel
@ -1574,39 +1580,48 @@ class PowerPanelTest(APIViewTestCases.APIViewTestCase):
@classmethod
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 = (
RackGroup.objects.create(name='Rack Group 1', slug='rack-group-1', site=site),
RackGroup.objects.create(name='Rack Group 2', slug='rack-group-2', site=site),
RackGroup.objects.create(name='Rack Group 3', slug='rack-group-3', 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=sites[0]),
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 = (
PowerPanel(site=site, rack_group=rack_groups[0], name='Power Panel 1'),
PowerPanel(site=site, 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[0], name='Power Panel 1'),
PowerPanel(site=sites[0], rack_group=rack_groups[1], name='Power Panel 2'),
PowerPanel(site=sites[0], rack_group=rack_groups[2], name='Power Panel 3'),
)
PowerPanel.objects.bulk_create(power_panels)
cls.create_data = [
{
'name': 'Power Panel 4',
'site': site.pk,
'site': sites[0].pk,
'rack_group': rack_groups[0].pk,
},
{
'name': 'Power Panel 5',
'site': site.pk,
'site': sites[0].pk,
'rack_group': rack_groups[1].pk,
},
{
'name': 'Power Panel 6',
'site': site.pk,
'site': sites[0].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):
model = PowerFeed

View File

@ -36,6 +36,9 @@ class SecretRoleTest(APIViewTestCases.APIViewTestCase):
'slug': 'secret-role-6',
},
]
bulk_update_data = {
'description': 'New description',
}
@classmethod
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):
# Unlock the plaintext prior to evaluation of the instance
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):

View File

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