Feature/transaction state

.gitignore

@@ -20,3 +20,4 @@
 coverage
 node_modules
 npm-debug.log
+env/*

platalbank_core/tests/test_transactions.py

@@ -1,5 +1,7 @@
 from platalbank_core.models import Event, Transaction, Account
 
+from django.core.urlresolvers import reverse
+from rest_framework import status
 from rest_framework.test import APITestCase
 
 def reload(obj):
@@ -40,7 +42,22 @@ def test_not_duplicate_transfer(self):
             debited_account=self.accA,
             credited_account=self.accB,
             event=self.event
-        ).save()
+        )
         self.accA, self.accB = reload(self.accA), reload(self.accB)
         self.assertEqual(self.accA.balance, 3700)
         self.assertEqual(self.accB.balance, 1837)
+
+    def test_transaction_setState(self):
+        Transaction.objects.create(
+            amount=500,
+            label='Test',
+            debited_account=self.accA,
+            credited_account=self.accB,
+            event=self.event
+        )
+        self.assertEqual(Transaction.objects.get(id=1).state, 'P')
+        url = reverse('transaction-setState',args=[1])
+        data = {'state': 'C'}
+        response = self.client.post(url, data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(Transaction.objects.get(id=1).state, 'C')

platalbank_core/views/transaction.py

@@ -1,8 +1,39 @@
+from rest_framework.response import Response
 from rest_framework import viewsets
+from rest_framework import exceptions
+from rest_framework.decorators import detail_route
+from django.http import HttpResponse
 
 from platalbank_core.models import Transaction
 from platalbank_core.serializers import TransactionSerializer
 
 class TransactionViewSet(viewsets.ModelViewSet):
     queryset = Transaction.objects.all()
     serializer_class = TransactionSerializer
+
+    @detail_route(methods=['post'])
+    def setState(self, request, pk=None):
+
+        try :
+            transaction = Transaction.objects.get(id=pk)
+        except Transaction.DoesNotExist:
+            return HttpResponse(status=404)
+
+        state = request.data.get('state')
+
+        #Une fois les permissions implementees, ce type de changement d'etat demandera des permissions Khube ou User
+        if (state in [Transaction.AUTHORIZED , Transaction.REJECTED]):
+            transaction.state = state
+            transaction.save()
+
+            return Response({'status':'State modified'})
+
+        #Ce type de changement d'etat demandera des permissions seller
+        elif (state in [Transaction.ABORTED,Transaction.COMPLETED]):
+            transaction.state = state
+            transaction.save()
+
+            return Response({'status':'State modified'})
+
+        else :
+            raise exceptions.PermissionDenied