entities/test/unittest_wfobjs.py
changeset 8562 0d2fb4604265
parent 8559 8db29e17dd33
child 8577 3f082de68802
--- a/entities/test/unittest_wfobjs.py	Thu Sep 20 14:50:06 2012 +0200
+++ b/entities/test/unittest_wfobjs.py	Fri Sep 14 14:15:39 2012 +0200
@@ -20,7 +20,6 @@
 
 from cubicweb import ValidationError
 from cubicweb.devtools.testlib import CubicWebTC
-from cubicweb.server.session import security_enabled
 
 
 def add_wf(self, etype, name=None, default=False):
@@ -155,7 +154,7 @@
         wf = add_wf(self, 'CWUser')
         s = wf.add_state(u'foo', initial=True)
         self.commit()
-        with security_enabled(self.session, write=False):
+        with self.session.security_enabled(write=False):
             with self.assertRaises(ValidationError) as cm:
                 self.session.execute('SET X in_state S WHERE X eid %(x)s, S eid %(s)s',
                                      {'x': self.user().eid, 's': s.eid})