test/unittest_predicates.py
changeset 9668 501f14e095d1
parent 9653 ac723e09704f
child 10476 62251bfdfd79
--- a/test/unittest_predicates.py	Mon Apr 07 12:03:06 2014 +0200
+++ b/test/unittest_predicates.py	Mon Apr 07 14:28:47 2014 +0200
@@ -75,12 +75,13 @@
         super(WorkflowSelectorTC, self).tearDown()
 
     def setup_database(self):
-        wf = self.shell().add_workflow("wf_test", 'StateFull', default=True)
-        created   = wf.add_state('created', initial=True)
-        validated = wf.add_state('validated')
-        abandoned = wf.add_state('abandoned')
-        wf.add_transition('validate', created, validated, ('managers',))
-        wf.add_transition('forsake', (created, validated,), abandoned, ('managers',))
+        with self.admin_access.shell() as shell:
+            wf = shell.add_workflow("wf_test", 'StateFull', default=True)
+            created   = wf.add_state('created', initial=True)
+            validated = wf.add_state('validated')
+            abandoned = wf.add_state('abandoned')
+            wf.add_transition('validate', created, validated, ('managers',))
+            wf.add_transition('forsake', (created, validated,), abandoned, ('managers',))
 
     @contextmanager
     def statefull_stuff(self):