test: before all tests

api_man/views.py

@@ -1,6 +1,5 @@
 from rest_framework import viewsets
 from rest_framework.exceptions import PermissionDenied
-from rest_framework.pagination import LimitOffsetPagination
 from rest_framework.permissions import IsAuthenticated, DjangoModelPermissions
 from rest_framework.decorators import action
 from rest_framework.response import Response
@@ -9,6 +8,7 @@ from django_filters import rest_framework as dj_filters
 from django.views.decorators.cache import cache_page
 from django.utils.decorators import method_decorator
 
+from flower.viewsets import LimitedModelViewSet
 from . import serializers
 
 
@@ -86,8 +86,7 @@ class BasicInfoFilterMixin:
         return qs
 
 
-class BaseViewSet(BasicInfoFilterMixin, viewsets.ModelViewSet):
-    pagination_class = LimitOffsetPagination
+class BaseViewSet(BasicInfoFilterMixin, LimitedModelViewSet):
 
     def get_queryset(self):
         qs = super().get_queryset()
@@ -106,10 +105,9 @@ class BaseViewSet(BasicInfoFilterMixin, viewsets.ModelViewSet):
 
 
 @method_decorator(cache_page(20), name='list')
-class QuickInputViewSet(BasicInfoFilterMixin, viewsets.ModelViewSet):
+class QuickInputViewSet(BasicInfoFilterMixin, LimitedModelViewSet):
     queryset = serializers.basic_models.QuickInput.objects.all()
     serializer_class = serializers.QuickInputSerializer
-    pagination_class = LimitOffsetPagination
 
     def filter_queryset(self, queryset):
         qs = super().filter_queryset(queryset)