Coverage for vpn/api/views.py: 98%

1from rest_framework.routers import APIRootView 

2 

3from netbox.api.viewsets import NetBoxModelViewSet 

4from vpn import filtersets 

5from vpn.models import * 

6 

7from . import serializers 

8 

9__all__ = ( 

10 'IKEPolicyViewSet', 

11 'IKEProposalViewSet', 

12 'IPSecPolicyViewSet', 

13 'IPSecProfileViewSet', 

14 'IPSecProposalViewSet', 

15 'L2VPNTerminationViewSet', 

16 'L2VPNViewSet', 

17 'TunnelGroupViewSet', 

18 'TunnelTerminationViewSet', 

19 'TunnelViewSet', 

20 'VPNRootView', 

21) 

22 

23 

24class VPNRootView(APIRootView): 

25 """ 

26 VPN API root view 

27 """ 

28 def get_view_name(self): 

29 return 'VPN' 

30 

31 

32# 

33# Viewsets 

34# 

35 

36class TunnelGroupViewSet(NetBoxModelViewSet): 

37 queryset = TunnelGroup.objects.all() 

38 serializer_class = serializers.TunnelGroupSerializer 

39 filterset_class = filtersets.TunnelGroupFilterSet 

40 

41 

42class TunnelViewSet(NetBoxModelViewSet): 

43 queryset = Tunnel.objects.all() 

44 serializer_class = serializers.TunnelSerializer 

45 filterset_class = filtersets.TunnelFilterSet 

46 

47 

48class TunnelTerminationViewSet(NetBoxModelViewSet): 

49 queryset = TunnelTermination.objects.all() 

50 serializer_class = serializers.TunnelTerminationSerializer 

51 filterset_class = filtersets.TunnelTerminationFilterSet 

52 

53 

54class IKEProposalViewSet(NetBoxModelViewSet): 

55 queryset = IKEProposal.objects.all() 

56 serializer_class = serializers.IKEProposalSerializer 

57 filterset_class = filtersets.IKEProposalFilterSet 

58 

59 

60class IKEPolicyViewSet(NetBoxModelViewSet): 

61 queryset = IKEPolicy.objects.all() 

62 serializer_class = serializers.IKEPolicySerializer 

63 filterset_class = filtersets.IKEPolicyFilterSet 

64 

65 

66class IPSecProposalViewSet(NetBoxModelViewSet): 

67 queryset = IPSecProposal.objects.all() 

68 serializer_class = serializers.IPSecProposalSerializer 

69 filterset_class = filtersets.IPSecProposalFilterSet 

70 

71 

72class IPSecPolicyViewSet(NetBoxModelViewSet): 

73 queryset = IPSecPolicy.objects.all() 

74 serializer_class = serializers.IPSecPolicySerializer 

75 filterset_class = filtersets.IPSecPolicyFilterSet 

76 

77 

78class IPSecProfileViewSet(NetBoxModelViewSet): 

79 queryset = IPSecProfile.objects.all() 

80 serializer_class = serializers.IPSecProfileSerializer 

81 filterset_class = filtersets.IPSecProfileFilterSet 

82 

83 

84class L2VPNViewSet(NetBoxModelViewSet): 

85 queryset = L2VPN.objects.all() 

86 serializer_class = serializers.L2VPNSerializer 

87 filterset_class = filtersets.L2VPNFilterSet 

88 

89 

90class L2VPNTerminationViewSet(NetBoxModelViewSet): 

91 queryset = L2VPNTermination.objects.all() 

92 serializer_class = serializers.L2VPNTerminationSerializer 

93 filterset_class = filtersets.L2VPNTerminationFilterSet