From 2f69eddb572dc52726628a93faac34b81850f1b0 Mon Sep 17 00:00:00 2001 From: Arthur Date: Mon, 23 Oct 2023 13:13:37 -0700 Subject: [PATCH] 8356 review changes --- netbox/virtualization/api/nested_serializers.py | 2 +- netbox/virtualization/api/views.py | 2 +- netbox/virtualization/tests/test_api.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/netbox/virtualization/api/nested_serializers.py b/netbox/virtualization/api/nested_serializers.py index 8812f4405..927ce5f49 100644 --- a/netbox/virtualization/api/nested_serializers.py +++ b/netbox/virtualization/api/nested_serializers.py @@ -81,4 +81,4 @@ class NestedVirtualDiskSerializer(WritableNestedSerializer): class Meta: model = VirtualDisk - fields = ['id', 'url', 'virtual_machine', 'name', 'size'] + fields = ['id', 'url', 'display', 'virtual_machine', 'name'] diff --git a/netbox/virtualization/api/views.py b/netbox/virtualization/api/views.py index 8e8005fcd..df106774c 100644 --- a/netbox/virtualization/api/views.py +++ b/netbox/virtualization/api/views.py @@ -5,7 +5,7 @@ from extras.api.mixins import ConfigContextQuerySetMixin from netbox.api.viewsets import NetBoxModelViewSet from utilities.utils import count_related from virtualization import filtersets -from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualDisk, VirtualMachine, VMInterface +from virtualization.models import * from . import serializers diff --git a/netbox/virtualization/tests/test_api.py b/netbox/virtualization/tests/test_api.py index 3349025cd..bd5bdb16d 100644 --- a/netbox/virtualization/tests/test_api.py +++ b/netbox/virtualization/tests/test_api.py @@ -6,7 +6,7 @@ from dcim.models import Site from ipam.models import VLAN, VRF from utilities.testing import APITestCase, APIViewTestCases, create_test_device from virtualization.choices import * -from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualDisk, VirtualMachine, VMInterface +from virtualization.models import * class AppTest(APITestCase):