Add ListStyle tests.

1 files changed, 4 insertions, 3 deletions

diff --git a/src/mailman_pgp/rules/tests/test_signature.py b/src/mailman_pgp/rules/tests/test_signature.py
index 86141a9..084f599 100644
--- a/src/mailman_pgp/rules/tests/test_signature.py
+++ b/src/mailman_pgp/rules/tests/test_signature.py
@@ -18,9 +18,9 @@ from unittest import TestCase
 
 from mailman.app.lifecycle import create_list
 from mailman.config import config
-from mailman.database.transaction import transaction as mailman_transaction
 from mailman.testing.helpers import specialized_message_from_string as mfs
 
+from mailman_pgp.database import mm_transaction
 from mailman_pgp.rules.signature import Signature
 from mailman_pgp.testing.layers import PGPConfigLayer
 
@@ -29,7 +29,7 @@ class TestSignature(TestCase):
     layer = PGPConfigLayer
 
     def setUp(self):
-        with mailman_transaction():
+        with mm_transaction():
             self.mlist = create_list('nobody@example.com',
                                      style_name='pgp-default')
 
@@ -37,7 +37,8 @@ class TestSignature(TestCase):
         self.assertIn(Signature.name, config.rules.keys())
 
     def test_no_pgp_list(self):
-        ordinary_list = create_list('ordinary@example.com')
+        with mm_transaction():
+            ordinary_list = create_list('ordinary@example.com')
         msg = mfs("""\
 From: anne@example.com
 To: ordinary@example.com