Assign owner to start task

tests/examples/shipment/tests.py

@@ -3,6 +3,7 @@
 
 from django.test import TestCase
 from viewflow.graphviz import diagram
+from viewflow.models import Task
 from viewflow.test import FlowTest
 
 from examples.shipment.flows import ShipmentFlow
@@ -20,6 +21,9 @@ def test_diagram_creation_succeed(self):
 
 
 class ShipmentFlowTests(TestCase):
+    def task(self, process, flow_task):
+        return Task.objects.get(process=process, flow_task=flow_task)
+
     def setUp(self):
         self.default_carrier = Carrier.objects.get(name=Carrier.DEFAULT)
         self.special_carrier = Carrier.objects.exclude(name=Carrier.DEFAULT)[0]
@@ -29,7 +33,8 @@ def test_normal_post_succeed(self):
             # Clerk start process
             flow.Task(ShipmentFlow.start).User('shipment/clerk') \
                 .Execute({'goods_tag': 'TST123'}) \
-                .Assert(lambda p: p.created is not None)
+                .Assert(lambda p: p.created is not None) \
+                .Assert(lambda p: self.task(p, ShipmentFlow.start).owner.username == 'shipment/clerk')
 
             # Clerk decides that is special shipment post
             flow.Task(ShipmentFlow.shipment_type).User('shipment/clerk') \

viewflow/flow/start.py

@@ -75,6 +75,9 @@ def get_management_form_cls(self):
         else:
             return self.flow_cls.management_form_cls
 
+    def assign(self, user):
+        self.task.assign(user=user)
+
     def prepare(self, data=None):
         super(StartViewActivation, self).prepare()
 
@@ -130,6 +133,7 @@ def dispatch(self, request, activation, **kwargs):
         if not self.activation.flow_task.has_perm(request.user):
             raise PermissionDenied
 
+        self.activation.assign(user=request.user)
         self.activation.prepare(request.POST or None)
         return super(StartViewMixin, self).dispatch(request, **kwargs)
 
@@ -195,6 +199,7 @@ def dispatch(self, request, *args, **kwargs):
         if not self.flow_task.has_perm(request.user):
             raise PermissionDenied
 
+        self.assign(user=request.user)
         self.prepare(request.POST or None)
         return super(StartView, self).dispatch(request, *args, **kwargs)
 

viewflow/models.py

@@ -100,7 +100,7 @@ def _in_db(self):
         """
         return self.pk
 
-    @transition(field=status, source=STATUS.NEW, target=STATUS.ASSIGNED, conditions=[_in_db])
+    @transition(field=status, source=STATUS.NEW, target=STATUS.ASSIGNED)
     def assign(self, user=None, external_task_id=None):
         """
         Tasks that perform some activity should be associated with