17170 test fixes

This commit is contained in:
Arthur 2025-03-13 11:09:11 -07:00
parent e331be39f9
commit 82233c1e3e
2 changed files with 8 additions and 5 deletions

View File

@ -3,7 +3,7 @@ from drf_spectacular.types import OpenApiTypes
from drf_spectacular.utils import extend_schema_field
from rest_framework import serializers
from netbox.api.fields import ChoiceField, ContentTypeField
from netbox.api.fields import ChoiceField, ContentTypeField, SerializedPKRelatedField
from netbox.api.serializers import NestedGroupModelSerializer, NetBoxModelSerializer
from tenancy.choices import ContactPriorityChoices
from tenancy.models import ContactAssignment, Contact, ContactGroup, ContactRole
@ -43,7 +43,12 @@ class ContactRoleSerializer(NetBoxModelSerializer):
class ContactSerializer(NetBoxModelSerializer):
groups = ContactGroupSerializer(many=True, nested=True, required=False, allow_null=True, default=None)
groups = SerializedPKRelatedField(
queryset=ContactGroup.objects.all(),
serializer=ContactGroupSerializer,
required=False,
many=True
)
class Meta:
model = Contact

View File

@ -170,7 +170,7 @@ class ContactTest(APIViewTestCases.APIViewTestCase):
model = Contact
brief_fields = ['description', 'display', 'id', 'name', 'url']
bulk_update_data = {
'groups': None,
'groups': [],
'comments': 'New comments',
}
@ -195,11 +195,9 @@ class ContactTest(APIViewTestCases.APIViewTestCase):
cls.create_data = [
{
'name': 'Contact 4',
'groups': [contact_groups[1].pk],
},
{
'name': 'Contact 5',
'groups': [contact_groups[1].pk],
},
{
'name': 'Contact 6',