From ad7c5f7c4c530a3587949c8eaf9ad064419a6e33 Mon Sep 17 00:00:00 2001 From: Barry Warsaw Date: Sat, 6 Feb 2016 17:15:41 -0500 Subject: Add a set_preferred() helper. This refactors the setting of a user's preferred address to the first in their list of linked addresses. --- src/mailman/testing/helpers.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'src/mailman/testing/helpers.py') diff --git a/src/mailman/testing/helpers.py b/src/mailman/testing/helpers.py index 2e3809f03..631424661 100644 --- a/src/mailman/testing/helpers.py +++ b/src/mailman/testing/helpers.py @@ -31,6 +31,7 @@ __all__ = [ 'make_digest_messages', 'make_testable_runner', 'reset_the_world', + 'set_preferred', 'specialized_message_from_string', 'subscribe', 'temporary_db', @@ -554,3 +555,13 @@ message triggering a digest volume=1, digest_number=1) runner = make_testable_runner(DigestRunner, 'digest') runner.run() + + + +def set_preferred(user): + # Avoid circular imports. + from mailman.utilities.datetime import now + preferred = list(user.addresses)[0] + preferred.verified_on = now() + user.preferred_address = preferred + return preferred -- cgit v1.2.3-70-g09d2