mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Add tests for wireless

Browse Source
This commit is contained in:
jeremystretch
2021-10-20 13:34:39 -04:00
parent 4a7159389e
commit 6a4becfb46
13 changed files with 529 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/wireless/graphql/schema.py
View File

@@ -5,11 +5,11 @@ from .types import *
class WirelessQuery(graphene.ObjectType):
wirelesslan = ObjectField(WirelessLANType)
wirelesslan_list = ObjectListField(WirelessLANType)
wireless_lan = ObjectField(WirelessLANType)
wireless_lan_list = ObjectListField(WirelessLANType)
wirelesslangroup = ObjectField(WirelessLANGroupType)
wirelesslangroup_list = ObjectListField(WirelessLANGroupType)
wireless_lan_group = ObjectField(WirelessLANGroupType)
wireless_lan_group_list = ObjectListField(WirelessLANGroupType)
wirelesslink = ObjectField(WirelessLinkType)
wirelesslink_list = ObjectListField(WirelessLinkType)
wireless_link = ObjectField(WirelessLinkType)
wireless_link_list = ObjectListField(WirelessLinkType)

18
netbox/wireless/graphql/types.py
View File

@@ -1,5 +1,5 @@
from wireless import filtersets, models
from netbox.graphql.types import ObjectType
from netbox.graphql.types import ObjectType, PrimaryObjectType
__all__ = (
'WirelessLANType',
@@ -16,17 +16,29 @@ class WirelessLANGroupType(ObjectType):
filterset_class = filtersets.WirelessLANGroupFilterSet
class WirelessLANType(ObjectType):
class WirelessLANType(PrimaryObjectType):
class Meta:
model = models.WirelessLAN
fields = '__all__'
filterset_class = filtersets.WirelessLANFilterSet
def resolve_auth_type(self, info):
return self.auth_type or None
class WirelessLinkType(ObjectType):
def resolve_auth_cipher(self, info):
return self.auth_cipher or None
class WirelessLinkType(PrimaryObjectType):
class Meta:
model = models.WirelessLink
fields = '__all__'
filterset_class = filtersets.WirelessLinkFilterSet
def resolve_auth_type(self, info):
return self.auth_type or None
def resolve_auth_cipher(self, info):
return self.auth_cipher or None