# HG changeset patch
# User Julien Jehannet <julien@smaf.org>
# Date 1295470169 -3600
# Node ID 2d40f3c48f319e2060b0f3abd8107a081a0d79d6
# Parent  f691757792f9642a4e3b17820b7c3eb81d17f66f
[tests] use migration api to create workflow

Prefer high-level migration api thru admin session
instead of raw rql in helper method

diff -r f691757792f9 -r 2d40f3c48f31 entities/test/unittest_wfobjs.py
--- a/entities/test/unittest_wfobjs.py	Fri Jan 21 12:20:59 2011 +0100
+++ b/entities/test/unittest_wfobjs.py	Wed Jan 19 21:49:29 2011 +0100
@@ -18,20 +18,16 @@
 
 from __future__ import with_statement
 
+from cubicweb import ValidationError
 from cubicweb.devtools.testlib import CubicWebTC
-from cubicweb import ValidationError
 from cubicweb.server.session import security_enabled
 
+
 def add_wf(self, etype, name=None, default=False):
     if name is None:
         name = etype
-    wf = self.execute('INSERT Workflow X: X name %(n)s', {'n': unicode(name)}).get_entity(0, 0)
-    self.execute('SET WF workflow_of ET WHERE WF eid %(wf)s, ET name %(et)s',
-                 {'wf': wf.eid, 'et': etype})
-    if default:
-        self.execute('SET ET default_workflow WF WHERE WF eid %(wf)s, ET name %(et)s',
-                     {'wf': wf.eid, 'et': etype})
-    return wf
+    return self.shell().add_workflow(name, etype, default=default,
+                                     ensure_workflowable=False)
 
 def parse_hist(wfhist):
     return [(ti.previous_state.name, ti.new_state.name,