diff options
Diffstat (limited to 'src/mailman/model/tests/test_workflow.py')
| -rw-r--r-- | src/mailman/model/tests/test_workflow.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/mailman/model/tests/test_workflow.py b/src/mailman/model/tests/test_workflow.py index 6081e5b57..ccf618c2b 100644 --- a/src/mailman/model/tests/test_workflow.py +++ b/src/mailman/model/tests/test_workflow.py @@ -108,3 +108,23 @@ class TestWorkflow(unittest.TestCase): self._manager.save(name, token) workflow = self._manager.restore('ewe', 'fly') self.assertIsNone(workflow) + + def test_restore_removes_record(self): + name = 'ant' + token = 'bee' + self.assertEqual(self._manager.count(), 0) + self._manager.save(name, token) + self.assertEqual(self._manager.count(), 1) + self._manager.restore(name, token) + self.assertEqual(self._manager.count(), 0) + + def test_save_after_restore(self): + name = 'ant' + token = 'bee' + self.assertEqual(self._manager.count(), 0) + self._manager.save(name, token) + self.assertEqual(self._manager.count(), 1) + self._manager.restore(name, token) + self.assertEqual(self._manager.count(), 0) + self._manager.save(name, token) + self.assertEqual(self._manager.count(), 1) |