mirror of
https://github.com/netbox-community/netbox.git
synced 2025-08-13 02:58:17 -06:00
16078 add tests for graphql filtering
This commit is contained in:
parent
3fc82207c8
commit
27a2bc21ad
@ -648,6 +648,9 @@ class IPAddressTest(APIViewTestCases.APIViewTestCase):
|
|||||||
bulk_update_data = {
|
bulk_update_data = {
|
||||||
'description': 'New description',
|
'description': 'New description',
|
||||||
}
|
}
|
||||||
|
graphql_filter = {
|
||||||
|
'address': '192.168.0.1/24',
|
||||||
|
}
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def setUpTestData(cls):
|
def setUpTestData(cls):
|
||||||
|
@ -443,8 +443,8 @@ class APIViewTestCases:
|
|||||||
def _build_query(self, name, **filters):
|
def _build_query(self, name, **filters):
|
||||||
type_class = get_graphql_type_for_model(self.model)
|
type_class = get_graphql_type_for_model(self.model)
|
||||||
if filters:
|
if filters:
|
||||||
filter_string = ', '.join(f'{k}:{v}' for k, v in filters.items())
|
filter_string = ', '.join(f'{k}: "{v}"' for k, v in filters.items())
|
||||||
filter_string = f'({filter_string})'
|
filter_string = f'(filters: {{{filter_string}}})'
|
||||||
else:
|
else:
|
||||||
filter_string = ''
|
filter_string = ''
|
||||||
|
|
||||||
@ -550,6 +550,31 @@ class APIViewTestCases:
|
|||||||
self.assertNotIn('errors', data)
|
self.assertNotIn('errors', data)
|
||||||
self.assertGreater(len(data['data'][field_name]), 0)
|
self.assertGreater(len(data['data'][field_name]), 0)
|
||||||
|
|
||||||
|
@override_settings(LOGIN_REQUIRED=True)
|
||||||
|
@override_settings(EXEMPT_VIEW_PERMISSIONS=['*', 'auth.user'])
|
||||||
|
def test_graphql_filter_objects(self):
|
||||||
|
if not hasattr(self, 'graphql_filter'):
|
||||||
|
return
|
||||||
|
|
||||||
|
url = reverse('graphql')
|
||||||
|
field_name = f'{self._get_graphql_base_name()}_list'
|
||||||
|
query = self._build_query(field_name, **self.graphql_filter)
|
||||||
|
|
||||||
|
# Add object-level permission
|
||||||
|
obj_perm = ObjectPermission(
|
||||||
|
name='Test permission',
|
||||||
|
actions=['view']
|
||||||
|
)
|
||||||
|
obj_perm.save()
|
||||||
|
obj_perm.users.add(self.user)
|
||||||
|
obj_perm.object_types.add(ObjectType.objects.get_for_model(self.model))
|
||||||
|
|
||||||
|
response = self.client.post(url, data={'query': query}, format="json", **self.header)
|
||||||
|
self.assertHttpStatus(response, status.HTTP_200_OK)
|
||||||
|
data = json.loads(response.content)
|
||||||
|
self.assertNotIn('errors', data)
|
||||||
|
self.assertGreater(len(data['data'][field_name]), 0)
|
||||||
|
|
||||||
class APIViewTestCase(
|
class APIViewTestCase(
|
||||||
GetObjectViewTestCase,
|
GetObjectViewTestCase,
|
||||||
ListObjectsViewTestCase,
|
ListObjectsViewTestCase,
|
||||||
|
Loading…
Reference in New Issue
Block a user