tests factorised

This commit is contained in:
Qwann 2017-07-25 14:02:57 +02:00
parent a4da5d94c4
commit 3246552eba
6 changed files with 334 additions and 104 deletions

View file

@ -89,3 +89,7 @@ class ActivityTemplateSerializer(serializers.ModelSerializer):
for tag_data in tags_data]
activity_template.tags = tags
return activity_template
class ActivitySerializer(serializers.ModelSerializer):
pass

View file

@ -4,8 +4,8 @@ from django.db.models import Q
from rest_framework.viewsets import ModelViewSet
from api.event.serializers import EventSerializer, PlaceSerializer,\
ActivityTagSerializer, ActivityTemplateSerializer
from event.models import Event, Place, ActivityTag, ActivityTemplate
ActivityTagSerializer, ActivityTemplateSerializer, ActivitySerializer
from event.models import Event, Place, ActivityTag, ActivityTemplate, Activity
User = get_user_model()
@ -36,6 +36,10 @@ class EventSpecificViewSet(ModelViewSet):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)
def perform_update(self, serializer):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)
# ViewSets
class EventViewSet(ModelViewSet):
@ -73,3 +77,25 @@ class ActivityTemplateViewSet(ModelViewSet):
def perform_create(self, serializer):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)
def perform_update(self, serializer):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)
class ActivityViewSet(ModelViewSet):
"""
This viewset automatically provides `list`, `create`, `retrieve`,
`update` and `destroy` actions.
"""
queryset = Activity.objects.all()
serializer_class = ActivitySerializer
def perform_create(self, serializer):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)
def perform_update(self, serializer):
event_pk = self.kwargs.get('event_pk')
serializer.save(event_pk=event_pk)