plugin

1 files changed, 4 insertions, 3 deletions

diff --git a/src/mailman/runners/tests/test_leave.py b/src/mailman/runners/tests/test_leave.py
index 7069e6710..c20637c3f 100644
--- a/src/mailman/runners/tests/test_leave.py
+++ b/src/mailman/runners/tests/test_leave.py
@@ -23,13 +23,14 @@ from email.iterators import body_line_iterator
 from mailman.app.lifecycle import create_list
 from mailman.config import config
 from mailman.database.transaction import transaction
-from mailman.interfaces.mailinglist import SubscriptionPolicy
 from mailman.interfaces.usermanager import IUserManager
 from mailman.runners.command import CommandRunner
 from mailman.testing.helpers import (
     get_queue_messages, make_testable_runner, set_preferred,
     specialized_message_from_string as mfs)
 from mailman.testing.layers import ConfigLayer
+from mailman.workflows.unsubscription import (ConfirmUnsubscriptionPolicy,
+                                              OpenUnsubscriptionPolicy)
 from zope.component import getUtility
 
 
@@ -65,7 +66,7 @@ class TestLeave(unittest.TestCase):
 
     def test_leave(self):
         with transaction():
-            self._mlist.unsubscription_policy = SubscriptionPolicy.confirm
+            self._mlist.unsubscription_policy = ConfirmUnsubscriptionPolicy
             anne = getUtility(IUserManager).create_user('anne@example.org')
             set_preferred(anne)
             self._mlist.subscribe(anne.preferred_address)
@@ -100,7 +101,7 @@ leave
         # sent to the -leave address and it contains the 'leave' command, we
         # should only process one command per email.
         with transaction():
-            self._mlist.unsubscription_policy = SubscriptionPolicy.open
+            self._mlist.unsubscription_policy = OpenUnsubscriptionPolicy
             anne = getUtility(IUserManager).create_user('anne@example.org')
             set_preferred(anne)
             self._mlist.subscribe(anne.preferred_address)