Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

16946 return empty queryset if filterset is not valid #17015

Merged
merged 7 commits into from
Aug 27, 2024
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 9 additions & 2 deletions netbox/netbox/graphql/filter_mixins.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import django_filters
import strawberry
import strawberry_django
from django.core.exceptions import FieldDoesNotExist
from django.core.exceptions import FieldDoesNotExist, ValidationError
from strawberry import auto
from ipam.fields import ASNField
from netbox.graphql.scalars import BigInt
Expand Down Expand Up @@ -201,4 +201,11 @@ def wrapper(cls):
class BaseFilterMixin:

def filter_by_filterset(self, queryset, key):
return self.filterset(data={key: getattr(self, key)}, queryset=queryset).qs
filterset = self.filterset(data={key: getattr(self, key)}, queryset=queryset)
if not filterset.is_valid():
# filterset.errors is errorDict - return first error as exception
k, v = next(iter(filterset.errors.items()))
jeremystretch marked this conversation as resolved.
Show resolved Hide resolved
return filterset.qs.none()
# We could raise validation error but strawberry logs it all to the
# console i.e. raise ValidationError(f"{k}: {v[0]}")
return filterset.qs
45 changes: 44 additions & 1 deletion netbox/netbox/tests/test_graphql.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,13 @@
import json

from django.test import override_settings
from django.urls import reverse

from utilities.testing import disable_warnings, TestCase
from core.models import ObjectType
from rest_framework import status
from users.models import ObjectPermission, Token
from users.models import User
from utilities.testing import disable_warnings, APITestCase, TestCase


class GraphQLTestCase(TestCase):
Expand Down Expand Up @@ -34,3 +40,40 @@ def test_graphiql_interface(self):
response = self.client.get(url, **header)
with disable_warnings('django.request'):
self.assertHttpStatus(response, 302) # Redirect to login page


class GraphQLAPITestCase(APITestCase):

@override_settings(LOGIN_REQUIRED=True)
@override_settings(EXEMPT_VIEW_PERMISSIONS=['*', 'auth.user'])
def test_graphql_filter_objects(self):
from dcim.models import Site, Location
jeremystretch marked this conversation as resolved.
Show resolved Hide resolved

site = Site.objects.create(name='Site A', slug='site-a')
location = Location.objects.create(site=site, name='Location A1', slug='location-a1')

url = reverse('graphql')
field_name = 'location_list'
query = '{location_list(filters: {site_id: "' + str(site.id) + '"}) {id site {id}}}'

# 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(Location))

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']['location_list']), 0)

query = '{location_list(filters: {site_id: "' + str(site.id + 1234) + '"}) {id site {id}}}'
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.assertEqual(len(data['data']['location_list']), 0)