|
|
@@ -73,7 +73,7 @@ class APIViewTestCases:
|
|
|
|
|
|
class GetObjectViewTestCase(APITestCase):
|
|
|
|
|
|
- @override_settings(EXEMPT_VIEW_PERMISSIONS=['*'])
|
|
|
+ @override_settings(EXEMPT_VIEW_PERMISSIONS=['*'], LOGIN_REQUIRED=False)
|
|
|
def test_get_object_anonymous(self):
|
|
|
"""
|
|
|
GET a single object as an unauthenticated user.
|
|
|
@@ -135,7 +135,7 @@ class APIViewTestCases:
|
|
|
class ListObjectsViewTestCase(APITestCase):
|
|
|
brief_fields = []
|
|
|
|
|
|
- @override_settings(EXEMPT_VIEW_PERMISSIONS=['*'])
|
|
|
+ @override_settings(EXEMPT_VIEW_PERMISSIONS=['*'], LOGIN_REQUIRED=False)
|
|
|
def test_list_objects_anonymous(self):
|
|
|
"""
|
|
|
GET a list of objects as an unauthenticated user.
|
|
|
@@ -440,13 +440,12 @@ class APIViewTestCases:
|
|
|
base_name = self.model._meta.verbose_name.lower().replace(' ', '_')
|
|
|
return getattr(self, 'graphql_base_name', base_name)
|
|
|
|
|
|
- def _build_query(self, name, **filters):
|
|
|
+ def _build_query_with_filter(self, name, filter_string):
|
|
|
+ """
|
|
|
+ Called by either _build_query or _build_filtered_query - construct the actual
|
|
|
+ query given a name and filter string
|
|
|
+ """
|
|
|
type_class = get_graphql_type_for_model(self.model)
|
|
|
- if filters:
|
|
|
- filter_string = ', '.join(f'{k}:{v}' for k, v in filters.items())
|
|
|
- filter_string = f'({filter_string})'
|
|
|
- else:
|
|
|
- filter_string = ''
|
|
|
|
|
|
# Compile list of fields to include
|
|
|
fields_string = ''
|
|
|
@@ -492,6 +491,30 @@ class APIViewTestCases:
|
|
|
|
|
|
return query
|
|
|
|
|
|
+ def _build_filtered_query(self, name, **filters):
|
|
|
+ """
|
|
|
+ Create a filtered query: i.e. ip_address_list(filters: {address: "1.1.1.1/24"}){.
|
|
|
+ """
|
|
|
+ if filters:
|
|
|
+ filter_string = ', '.join(f'{k}: "{v}"' for k, v in filters.items())
|
|
|
+ filter_string = f'(filters: {{{filter_string}}})'
|
|
|
+ else:
|
|
|
+ filter_string = ''
|
|
|
+
|
|
|
+ return self._build_query_with_filter(name, filter_string)
|
|
|
+
|
|
|
+ def _build_query(self, name, **filters):
|
|
|
+ """
|
|
|
+ Create a normal query - unfiltered or with a string query: i.e. site(name: "aaa"){.
|
|
|
+ """
|
|
|
+ if filters:
|
|
|
+ filter_string = ', '.join(f'{k}:{v}' for k, v in filters.items())
|
|
|
+ filter_string = f'({filter_string})'
|
|
|
+ else:
|
|
|
+ filter_string = ''
|
|
|
+
|
|
|
+ return self._build_query_with_filter(name, filter_string)
|
|
|
+
|
|
|
@override_settings(LOGIN_REQUIRED=True)
|
|
|
@override_settings(EXEMPT_VIEW_PERMISSIONS=['*', 'auth.user'])
|
|
|
def test_graphql_get_object(self):
|
|
|
@@ -550,6 +573,31 @@ class APIViewTestCases:
|
|
|
self.assertNotIn('errors', data)
|
|
|
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_filtered_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(
|
|
|
GetObjectViewTestCase,
|
|
|
ListObjectsViewTestCase,
|