testERP5Type.py: add compatibility for new workflow.

product/ERP5Type/tests/testERP5Type.py

@@ -2500,7 +2500,7 @@ class TestERP5Type(PropertySheetTestCase, LogInterceptor):
 
       from Products.DCWorkflow.Transitions import TRIGGER_WORKFLOW_METHOD
       if wf.getPortalType() == 'Workflow':
-        wf._getOb('transition_dummy_workflow_method').setTriggerType(TRIGGER_WORKFLOW_METHOD)
+        wf._getOb('transition_dummy_workflow_method')._edit(trigger_type=TRIGGER_WORKFLOW_METHOD)
       else:
         wf.transitions.dummy_workflow_method.setProperties(
           title='', new_state_id='', trigger_type=TRIGGER_WORKFLOW_METHOD)
@@ -2509,7 +2509,6 @@ class TestERP5Type(PropertySheetTestCase, LogInterceptor):
       self.assertTrue(hasattr(doc, 'dummyWorkflowMethod'))
 
       wf.deleteTransitions(['dummy_workflow_method'])
-
       self.commit()
       self.assertFalse(hasattr(doc, 'dummyWorkflowMethod'))
 
@@ -2518,9 +2517,11 @@ class TestERP5Type(PropertySheetTestCase, LogInterceptor):
       self.portal.portal_workflow.addWorkflowToType(self.portal.portal_types._getOb('Person'), ('delivery_causality_workflow',))
       self.commit()
       self.assertTrue(hasattr(doc, 'getCausalityState'))
-      wf = self.portal.portal_workflow.delivery_causality_workflow
-      wf.setStateVariable('dummy_state')
-
+      wf = self.portal.portal_workflow._getOb('delivery_causality_workflow')
+      if wf.getPortalType() == 'Workflow':
+        wf._edit(state_variable='dummy_state')
+      else:
+        wf.setStateVariable('dummy_state')
       self.commit()
       self.assertTrue(hasattr(doc, 'getDummyState'))