summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/mailman/database/tests/test_migrations.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/mailman/database/tests/test_migrations.py b/src/mailman/database/tests/test_migrations.py
index c264948ae..d983f9891 100644
--- a/src/mailman/database/tests/test_migrations.py
+++ b/src/mailman/database/tests/test_migrations.py
@@ -42,12 +42,13 @@ from zope.component import getUtility
from mailman.interfaces.database import IDatabaseFactory
from mailman.interfaces.domain import IDomainManager
from mailman.interfaces.archiver import ArchivePolicy
+from mailman.interfaces.bounce import BounceContext
from mailman.interfaces.listmanager import IListManager
from mailman.interfaces.mailinglist import IAcceptableAliasSet
from mailman.interfaces.nntp import NewsgroupModeration
from mailman.interfaces.subscriptions import ISubscriptionService
from mailman.model.bans import Ban
-from mailman.model.bounce import BounceContext, BounceEvent
+from mailman.model.bounce import BounceEvent
from mailman.testing.helpers import temporary_db
from mailman.testing.layers import ConfigLayer
@@ -474,5 +475,5 @@ class TestMigration20130406MigratedData(MigrationTestBase):
self.assertEqual(events[0].email, 'anne@example.com')
self.assertEqual(events[0].timestamp, datetime(2013, 4, 6, 21, 12))
self.assertEqual(events[0].message_id, '<abc@example.com>')
- self.assertEqual(events[0].context, BounceContext[1])
+ self.assertEqual(events[0].context, BounceContext.normal)
self.assertFalse(events[0].processed)