summaryrefslogtreecommitdiff
path: root/src/mailman/model/tests/test_workflow.py
diff options
context:
space:
mode:
authorBarry Warsaw2015-04-13 18:05:26 -0400
committerBarry Warsaw2015-04-13 18:05:26 -0400
commitf7a4e76d24898ec942a0f3a9a932916d9d0662bc (patch)
tree7953a2772cfdddc3c6464b9e02b2dab8cd29f400 /src/mailman/model/tests/test_workflow.py
parent3126d190d9c8a9b37da952cba42ea6e3b838a2c3 (diff)
downloadmailman-f7a4e76d24898ec942a0f3a9a932916d9d0662bc.tar.gz
mailman-f7a4e76d24898ec942a0f3a9a932916d9d0662bc.tar.zst
mailman-f7a4e76d24898ec942a0f3a9a932916d9d0662bc.zip
Diffstat (limited to 'src/mailman/model/tests/test_workflow.py')
-rw-r--r--src/mailman/model/tests/test_workflow.py20
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)