NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-25 08:46:10 -06:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Update serializer and tests

Browse Source
This commit is contained in:
Daniel Sheppard 2022-11-09 11:52:59 -06:00
parent 0a8a6d3023
commit 82b5e0e95c
2 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/dcim/api/serializers.py
View File

@ -682,8 +682,8 @@ class VirtualDeviceContextSerializer(NetBoxModelSerializer):
class Meta: class Meta:
model = VirtualDeviceContext model = VirtualDeviceContext
fields = [ fields = [
'id', 'url', 'display', 'name', 'device', 'tenant', 'primary_ip', 'primary_ip4', 'primary_ip6', 'comments', 'id', 'url', 'display', 'name', 'device', 'identifier', 'tenant', 'primary_ip', 'primary_ip4',
'tags', 'custom_fields', 'created', 'last_updated', 'primary_ip6', 'status', 'comments', 'tags', 'custom_fields', 'created', 'last_updated',
] ]

22
netbox/dcim/tests/test_api.py
View File

@ -2176,7 +2176,7 @@ class PowerFeedTest(APIViewTestCases.APIViewTestCase):
class VirtualDeviceContextTest(APIViewTestCases.APIViewTestCase): class VirtualDeviceContextTest(APIViewTestCases.APIViewTestCase):
model = VirtualDeviceContext model = VirtualDeviceContext
brief_fields = ['display', 'id', 'name', 'url', 'identifier', 'device'] brief_fields = ['device', 'display', 'id', 'identifier', 'name', 'url']
bulk_update_data = { bulk_update_data = {
'status': 'planned', 'status': 'planned',
} }
@ -2191,26 +2191,38 @@ class VirtualDeviceContextTest(APIViewTestCases.APIViewTestCase):
devices = ( devices = (
Device(name='Device 1', device_type=devicetype, device_role=devicerole, site=site), Device(name='Device 1', device_type=devicetype, device_role=devicerole, site=site),
Device(name='Device 2', device_type=devicetype, device_role=devicerole, site=site), Device(name='Device 2', device_type=devicetype, device_role=devicerole, site=site),
Device(name='Device 3', device_type=devicetype, device_role=devicerole, site=site),
) )
Device.objects.bulk_create(devices) 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 = [ cls.create_data = [
{ {
'device': devices[0].pk, 'device': devices[0].pk,
'status': 'active', 'status': 'active',
'name': 'VDC 1', 'name': 'VDC 1',
'identifier': '1', 'identifier': 1,
}, },
{ {
'device': devices[0].pk, 'device': devices[0].pk,
'status': 'active', 'status': 'active',
'name': 'VDC 2', 'name': 'VDC 2',
'identifier': '2', 'identifier': 2,
}, },
{ {
'device': devices[1].pk, 'device': devices[1].pk,
'status': 'active', 'status': 'active',
'name': 'VDC 1', 'name': 'VDC 3',
'identifier': '1', 'identifier': 3,
}, },
] ]