def post(self, request): """ Create new team --- request_serializer: TeamSerializer response_serializer: TeamSerializer """ serializer = TeamSerializer(request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data, status=status.HTTP_201_CREATED) else: return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def put(self, request, pk=None, slug=None): """ Update details of selected team --- request_serializer: TeamSerializer request_serializer: TeamSerializer """ team = Teams.get_team(pk, slug) serializer = TeamSerializer(player, data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data) else: return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def test_get_team_list(self): test_1 = Team.objects.create(team_name='qwerty') test_2 = Team.objects.create(team_name='dsds') url = reverse('team_list') response = self.client.get(url) data = TeamSerializer([test_1, test_2], many=True).data expected_data = [{ 'id': test_1.id, 'team_name': 'qwerty', }, { 'id': test_2.id, 'team_name': 'dsds', }] self.assertEqual(status.HTTP_200_OK, response.status_code) self.assertEqual(expected_data, data)