Added coverage testing
This commit is contained in:
@@ -167,16 +167,20 @@ from rest_framework import generics
|
||||
|
||||
class CollectionExportView(AuthorizedResourceMixin, generics.RetrieveAPIView):
|
||||
serializer_class = CollectionSerializer
|
||||
lookup_url_kwarg = 'collection'
|
||||
|
||||
def get_queryset(self):
|
||||
if self.request.user.is_superuser:
|
||||
return Collection.objects.all()
|
||||
return Collection.objects.filter(administrators=self.request.user)
|
||||
|
||||
class WorkExportView(AuthorizedResourceMixin, generics.RetrieveAPIView):
|
||||
serializer_class = WorkSerializer
|
||||
|
||||
def get_queryset(self):
|
||||
return Work.objects.filter(collection__administrators=self.request.user, collection=self.kwargs['collection'])
|
||||
works = Work.objects.filter(collection=self.kwargs['collection'])
|
||||
if self.request.user.is_superuser:
|
||||
return works
|
||||
return works.filter(collection__administrators=self.request.user)
|
||||
|
||||
class WorkImportView(AuthorizedResourceMixin, generics.CreateAPIView):
|
||||
serializer_class = WorkSerializer
|
||||
|
||||
Reference in New Issue
Block a user