tests factorised
This commit is contained in:
parent
a4da5d94c4
commit
3246552eba
6 changed files with 334 additions and 104 deletions
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue