From 9806f9c751f85b6ad6e9dea58d9e3dde36c7f2bc Mon Sep 17 00:00:00 2001 From: Barry Warsaw Date: Mon, 13 Apr 2015 10:46:37 -0400 Subject: Handle save/restore of the subscription workflow. To handle the case of restoring a saved workflow, allow the subscriber to be None in the constructor, but assert that it is not None after the first workflow step. Add setters/getters for handling save/restore. In the base workflow interface, rename key to token. --- src/mailman/model/tests/test_usermanager.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/mailman/model/tests/test_usermanager.py') diff --git a/src/mailman/model/tests/test_usermanager.py b/src/mailman/model/tests/test_usermanager.py index 31f1a7275..e441ed713 100644 --- a/src/mailman/model/tests/test_usermanager.py +++ b/src/mailman/model/tests/test_usermanager.py @@ -80,3 +80,8 @@ class TestUserManager(unittest.TestCase): user = self._usermanager.create_user('anne@example.com', 'Anne Person') other_user = self._usermanager.make_user('anne@example.com') self.assertIs(user, other_user) + + def test_get_user_by_id(self): + original = self._usermanager.make_user('anne@example.com') + copy = self._usermanager.get_user_by_id(original.user_id) + self.assertEqual(original, copy) -- cgit v1.2.3-70-g09d2