bike/workstand
1
0
Fork 0
mirror of https://github.com/fspc/workstand.git synced 2025-10-31 08:25:35 -04:00
Code Issues Projects Releases Wiki Activity

Claim endpoint.

Browse Source
This commit is contained in:
Drew Larson 2017-01-07 19:03:14 -06:00
parent 7ce619911f
commit 8a13c0356c
2 changed files with 92 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
bikeshop_project/bike/tests.py
View File

@ -5,6 +5,8 @@ from model_mommy import timezone
from rest_framework.test import APIClient
from model_mommy import mommy
from rest_framework import status
from registration.models import Member
from .models import Bike, BikeState
@ -150,3 +152,77 @@ class TestGet(TestCase):
self.assertEqual(result.status_code, status.HTTP_200_OK)
self.assertEqual(result.data['state'], BikeState.AVAILABLE)
def test_claim_cannot_transition_wrong_state(self):
member = mommy.make(Member)
data = {
"colour": "black",
"make": "Miyata",
"serial_number": "12345676",
"source": Bike.COS_BIKE_DIVERSION_PILOT,
"donated_by": "Greg",
"donated_at": "2017-01-01",
"size": Bike.SMALL,
"price": Decimal('68.00'),
"state": BikeState.ASSESSED,
"stolen": False,
"cpic_searched_at": timezone.now(),
}
bike = Bike.objects.create(**data)
client = APIClient()
client.force_authenticate(user=self.user, token='blah')
result = client.put(f'/api/v1/bikes/{bike.id}/claim/', data={'member': member.id}, format='json')
self.assertEqual(result.status_code, status.HTTP_400_BAD_REQUEST)
def test_claim_cannot_transition_claimed(self):
member = mommy.make(Member)
data = {
"colour": "black",
"make": "Miyata",
"serial_number": "12345676",
"source": Bike.COS_BIKE_DIVERSION_PILOT,
"donated_by": "Greg",
"donated_at": "2017-01-01",
"size": Bike.SMALL,
"price": Decimal('68.00'),
"state": BikeState.CLAIMED,
"stolen": False,
"cpic_searched_at": timezone.now(),
"claimed_by": member,
"last_worked_on": timezone.now()
}
bike = Bike.objects.create(**data)
client = APIClient()
client.force_authenticate(user=self.user, token='blah')
result = client.put('/api/v1/bikes/{bike_id}/claim/'.format(bike_id=bike.id), data={'member': member.id}, format='json')
self.assertEqual(result.status_code, status.HTTP_400_BAD_REQUEST)
def test_claim_can_transition(self):
member = mommy.make(Member)
data = {
"colour": "black",
"make": "Miyata",
"serial_number": "12345676",
"source": Bike.COS_BIKE_DIVERSION_PILOT,
"donated_by": "Greg",
"donated_at": "2017-01-01",
"size": Bike.SMALL,
"price": Decimal('68.00'),
"state": BikeState.AVAILABLE,
"stolen": False,
"cpic_searched_at": timezone.now(),
}
bike = Bike.objects.create(**data)
self.assertEqual(bike.state, BikeState.AVAILABLE)
client = APIClient()
client.force_authenticate(user=self.user, token='blah')
result = client.put(f'/api/v1/bikes/{bike.id}/claim/', data={'member': member.id}, format='json')
self.assertEqual(result.status_code, status.HTTP_200_OK)
self.assertEqual(result.data['state'], BikeState.CLAIMED)

16
bikeshop_project/bike/views.py
View File

@ -13,6 +13,8 @@ from bike.serializers import BikeSerializer
from rest_framework import status
from registration.models import Member
@method_decorator(login_required, name='dispatch')
class BikesView(TemplateView):
@ -52,3 +54,17 @@ class BikeViewSet(viewsets.ModelViewSet):
serializer = BikeSerializer(bike, context={'request': request})
return Response(serializer.data, status=status.HTTP_200_OK)
@detail_route(methods=['put'])
def claim(self, request, pk):
bike = get_object_or_404(Bike, pk=pk)
member = get_object_or_404(Member, id=request.data.get('member'))
state = BikeState.CLAIMED
if not can_proceed(bike.claim):
raise ValidationError(detail=f'Transition from {bike.state} to {state}')
bike.claim(member)
bike.save()
serializer = BikeSerializer(bike, context={'request': request})
return Response(serializer.data, status=status.HTTP_200_OK)