Testing utils

Author: barseghyanartur

examples/simple/search_indexes/urls.py

@@ -19,6 +19,8 @@
     BookOrderingByScoreCompoundSearchBackendDocumentViewSet,
     BookOrderingByScoreDocumentViewSet,
     BookPermissionsDocumentViewSet,
+    BookNoPermissionsDocumentViewSet,
+    BookNoRecordsDocumentViewSet,
     BookSimpleQueryStringBoostSearchFilterBackendDocumentViewSet,
     BookSimpleQueryStringSearchFilterBackendDocumentViewSet,
     BookSourceSearchBackendDocumentViewSet,
@@ -108,6 +110,18 @@
     basename='bookdocument_permissions'
 )
 
+router.register(
+    r'books-no-permissions',
+    BookNoPermissionsDocumentViewSet,
+    basename='bookdocument_no_permissions'
+)
+
+router.register(
+    r'books-no-records',
+    BookNoRecordsDocumentViewSet,
+    basename='bookdocument_no_records'
+)
+
 router.register(
     r'books-custom',
     BookCustomDocumentViewSet,

examples/simple/search_indexes/viewsets/__init__.py

@@ -1,32 +1,11 @@
-from .address import AddressDocumentViewSet, FrontAddressDocumentViewSet
-from .author import AuthorDocumentViewSet
-from .book import (
-    BookCompoundFuzzySearchBackendDocumentViewSet,
-    BookCompoundSearchBackendDocumentViewSet,
-    BookCompoundSearchBoostSearchBackendDocumentViewSet,
-    BookCustomDocumentViewSet,
-    BookDefaultFilterLookupDocumentViewSet,
-    BookDocumentViewSet,
-    BookFrontendDocumentViewSet,
-    BookFunctionalSuggesterDocumentViewSet,
-    BookIgnoreIndexErrorsDocumentViewSet,
-    BookMoreLikeThisDocumentViewSet,
-    BookMoreLikeThisNoOptionsDocumentViewSet,
-    BookMultiMatchOptionsPhasePrefixSearchFilterBackendDocumentViewSet,
-    BookMultiMatchSearchFilterBackendDocumentViewSet,
-    BookOrderingByScoreCompoundSearchBackendDocumentViewSet,
-    BookOrderingByScoreDocumentViewSet,
-    BookPermissionsDocumentViewSet,
-    BookSimpleQueryStringBoostSearchFilterBackendDocumentViewSet,
-    BookSimpleQueryStringSearchFilterBackendDocumentViewSet,
-    BookSourceSearchBackendDocumentViewSet,
-    QueryFriendlyPaginationBookDocumentViewSet,
-)
-from .city import CityDocumentViewSet, CityCompoundSearchBackendDocumentViewSet
-from .journal import JournalDocumentViewSet
-from .location import LocationDocumentViewSet
-from .publisher import PublisherDocumentViewSet
-from .tag import TagDocumentViewSet
+from .address import *
+from .author import *
+from .book import *
+from .city import *
+from .journal import *
+from .location import *
+from .publisher import *
+from .tag import *
 
 __all__ = (
     'AddressDocumentViewSet',
@@ -47,6 +26,8 @@
     'BookOrderingByScoreCompoundSearchBackendDocumentViewSet',
     'BookOrderingByScoreDocumentViewSet',
     'BookPermissionsDocumentViewSet',
+    'BookNoPermissionsDocumentViewSet',
+    'BookNoRecordsDocumentViewSet',
     'BookSimpleQueryStringBoostSearchFilterBackendDocumentViewSet',
     'BookSimpleQueryStringSearchFilterBackendDocumentViewSet',
     'BookSourceSearchBackendDocumentViewSet',

examples/simple/search_indexes/viewsets/book/permissions.py

@@ -31,6 +31,7 @@
     DocumentViewSet,
 )
 from django_elasticsearch_dsl_drf.pagination import PageNumberPagination
+from django_elasticsearch_dsl_drf.utils import EmptySearch
 
 from elasticsearch_dsl import DateHistogramFacet, RangeFacet
 
@@ -42,6 +43,8 @@
 
 __all__ = (
     'BookPermissionsDocumentViewSet',
+    'BookNoPermissionsDocumentViewSet',
+    'BookNoRecordsDocumentViewSet',
 )
 
 
@@ -60,6 +63,21 @@ def has_object_permission(self, request, view, obj):
         return False
 
 
+class CustomNoPermission(BasePermission):
+
+    def has_permission(self, request, view):
+        """
+        Return `True` if permission is granted, `False` otherwise.
+        """
+        return False
+
+    def has_object_permission(self, request, view, obj):
+        """
+        Return `True` if permission is granted, `False` otherwise.
+        """
+        return False
+
+
 class CustomPageNumberPagination(PageNumberPagination):
     """This is needed in order to make page size customisation possible."""
 
@@ -331,3 +349,14 @@ class BookPermissionsDocumentViewSet(DocumentViewSet):
         'tag_suggest': 'tags.suggest',
         'summary_suggest': 'summary',
     }
+
+
+class BookNoPermissionsDocumentViewSet(BookPermissionsDocumentViewSet):
+    permission_classes = (CustomNoPermission,)
+
+
+class BookNoRecordsDocumentViewSet(BookPermissionsDocumentViewSet):
+    def get_queryset(self):
+        queryset = EmptySearch()
+        queryset.model = self.document.Django.model
+        return queryset

src/django_elasticsearch_dsl_drf/tests/data_mixins.py

@@ -6,16 +6,11 @@
 
 import uuid
 
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 from books import constants
 import factories
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.data_mixins'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_faceted_search.py

@@ -7,8 +7,7 @@
 import unittest
 
 from django.core.management import call_command
-
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 import pytest
 
@@ -19,11 +18,6 @@
 
 from .base import BaseRestFrameworkTestCase
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_faceted_search'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_filtering_common.py

@@ -7,8 +7,7 @@
 import unittest
 
 from django.core.management import call_command
-
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 import pytest
 
@@ -32,11 +31,6 @@
 )
 from .data_mixins import AddressesMixin, BooksMixin
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_filtering_common'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_filtering_geo_spatial.py

@@ -7,11 +7,10 @@
 import unittest
 
 from django.core.management import call_command
+from django.urls import reverse
 
 from elasticsearch.connection.base import TransportError
 
-from nine.versions import DJANGO_GTE_1_10
-
 import pytest
 
 from rest_framework import status
@@ -21,11 +20,6 @@
 from ..constants import SEPARATOR_LOOKUP_COMPLEX_VALUE
 from .base import BaseRestFrameworkTestCase
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_filtering_geo_spatial'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_filtering_global_aggregations.py

@@ -7,8 +7,7 @@
 import unittest
 
 from django.core.management import call_command
-
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 import pytest
 
@@ -20,11 +19,6 @@
 from .base import BaseRestFrameworkTestCase
 from .data_mixins import AddressesMixin, BooksMixin
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_filtering_post'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_filtering_post_filter.py

@@ -7,8 +7,7 @@
 import unittest
 
 from django.core.management import call_command
-
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 import pytest
 
@@ -26,11 +25,6 @@
 )
 from .data_mixins import AddressesMixin, BooksMixin
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_filtering_post'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'

src/django_elasticsearch_dsl_drf/tests/test_functional_suggesters.py

@@ -8,8 +8,7 @@
 import unittest
 
 from django.core.management import call_command
-
-from nine.versions import DJANGO_GTE_1_10
+from django.urls import reverse
 
 import pytest
 
@@ -20,11 +19,6 @@
 from .base import BaseRestFrameworkTestCase
 from .data_mixins import AddressesMixin
 
-if DJANGO_GTE_1_10:
-    from django.urls import reverse
-else:
-    from django.core.urlresolvers import reverse
-
 __title__ = 'django_elasticsearch_dsl_drf.tests.test_suggesters'
 __author__ = 'Artur Barseghyan <[email protected]>'
 __copyright__ = '2017-2020 Artur Barseghyan'