Coverage for wireless/api/views.py: 95%

1from rest_framework.routers import APIRootView 

2 

3from netbox.api.viewsets import NetBoxModelViewSet 

4from wireless import filtersets 

5from wireless.models import * 

6 

7from . import serializers 

8 

9 

10class WirelessRootView(APIRootView): 

11 """ 

12 Wireless API root view 

13 """ 

14 def get_view_name(self): 

15 return 'Wireless' 

16 

17 

18class WirelessLANGroupViewSet(NetBoxModelViewSet): 

19 queryset = WirelessLANGroup.objects.add_related_count( 

20 WirelessLANGroup.objects.all(), 

21 WirelessLAN, 

22 'group', 

23 'wirelesslan_count', 

24 cumulative=True 

25 ) 

26 serializer_class = serializers.WirelessLANGroupSerializer 

27 filterset_class = filtersets.WirelessLANGroupFilterSet 

28 

29 

30class WirelessLANViewSet(NetBoxModelViewSet): 

31 queryset = WirelessLAN.objects.all() 

32 serializer_class = serializers.WirelessLANSerializer 

33 filterset_class = filtersets.WirelessLANFilterSet 

34 

35 

36class WirelessLinkViewSet(NetBoxModelViewSet): 

37 queryset = WirelessLink.objects.all() 

38 serializer_class = serializers.WirelessLinkSerializer 

39 filterset_class = filtersets.WirelessLinkFilterSet