plugin

1 files changed, 12 insertions, 12 deletions

diff --git a/src/mailman/model/tests/test_workflow.py b/src/mailman/model/tests/test_workflow.py
index 9cc5446b7..900efbb25 100644
--- a/src/mailman/model/tests/test_workflow.py
+++ b/src/mailman/model/tests/test_workflow.py
@@ -19,7 +19,7 @@
 
 import unittest
 
-from mailman.interfaces.workflow import IWorkflowStateManager
+from mailman.interfaces.workflows import IWorkflowStateManager
 from mailman.testing.layers import ConfigLayer
 from zope.component import getUtility
 
@@ -33,12 +33,12 @@ class TestWorkflow(unittest.TestCase):
     def test_save_restore_workflow(self):
         # Save and restore a workflow.
         token = 'bee'
-        step = 'cat'
+        steps = 'cat'
         data = 'dog'
-        self._manager.save(token, step, data)
+        self._manager.save(token, steps, data)
         state = self._manager.restore(token)
         self.assertEqual(state.token, token)
-        self.assertEqual(state.step, step)
+        self.assertEqual(state.steps, steps)
         self.assertEqual(state.data, data)
 
     def test_save_restore_workflow_without_step(self):
@@ -48,17 +48,17 @@ class TestWorkflow(unittest.TestCase):
         self._manager.save(token, data=data)
         state = self._manager.restore(token)
         self.assertEqual(state.token, token)
-        self.assertIsNone(state.step)
+        self.assertIsNone(state.steps)
         self.assertEqual(state.data, data)
 
     def test_save_restore_workflow_without_data(self):
         # Save and restore a workflow that contains no data.
         token = 'bee'
-        step = 'cat'
-        self._manager.save(token, step)
+        steps = 'cat'
+        self._manager.save(token, steps)
         state = self._manager.restore(token)
         self.assertEqual(state.token, token)
-        self.assertEqual(state.step, step)
+        self.assertEqual(state.steps, steps)
         self.assertIsNone(state.data)
 
     def test_save_restore_workflow_without_step_or_data(self):
@@ -67,7 +67,7 @@ class TestWorkflow(unittest.TestCase):
         self._manager.save(token)
         state = self._manager.restore(token)
         self.assertEqual(state.token, token)
-        self.assertIsNone(state.step)
+        self.assertIsNone(state.steps)
         self.assertIsNone(state.data)
 
     def test_restore_workflow_with_no_matching_token(self):
@@ -106,13 +106,13 @@ class TestWorkflow(unittest.TestCase):
         self._manager.discard('token2')
         self.assertEqual(self._manager.count, 3)
         state = self._manager.restore('token1')
-        self.assertEqual(state.step, 'one')
+        self.assertEqual(state.steps, 'one')
         state = self._manager.restore('token2')
         self.assertIsNone(state)
         state = self._manager.restore('token3')
-        self.assertEqual(state.step, 'three')
+        self.assertEqual(state.steps, 'three')
         state = self._manager.restore('token4')
-        self.assertEqual(state.step, 'four')
+        self.assertEqual(state.steps, 'four')
 
     def test_discard_missing_workflow(self):
         self._manager.discard('bogus-token')