Coverage for wireless/api/serializers_/nested.py: 100%

1from drf_spectacular.utils import extend_schema_serializer 

2from rest_framework import serializers 

3 

4from netbox.api.serializers import WritableNestedSerializer 

5from wireless import models 

6 

7__all__ = ( 

8 'NestedWirelessLANGroupSerializer', 

9 'NestedWirelessLinkSerializer', 

10) 

11 

12 

13@extend_schema_serializer( 

14 exclude_fields=('wirelesslan_count',), 

15) 

16class NestedWirelessLANGroupSerializer(WritableNestedSerializer): 

17 wirelesslan_count = serializers.IntegerField(read_only=True) 

18 _depth = serializers.IntegerField(source='level', read_only=True) 

19 

20 class Meta: 

21 model = models.WirelessLANGroup 

22 fields = ['id', 'url', 'display_url', 'display', 'name', 'slug', 'wirelesslan_count', '_depth'] 

23 

24 

25class NestedWirelessLinkSerializer(WritableNestedSerializer): 

26 

27 class Meta: 

28 model = models.WirelessLink 

29 fields = ['id', 'url', 'display_url', 'display', 'ssid']