1 files changed, 81 insertions, 0 deletions

diff --git a/src/mailman_pgp/workflows/tests/test_mod_approval.py b/src/mailman_pgp/workflows/tests/test_mod_approval.py
index 8b6b4d1..49e4204 100644
--- a/src/mailman_pgp/workflows/tests/test_mod_approval.py
+++ b/src/mailman_pgp/workflows/tests/test_mod_approval.py
@@ -14,3 +14,84 @@
 #
 # You should have received a copy of the GNU General Public License along with
 # this program.  If not, see <http://www.gnu.org/licenses/>.
+
+""""""
+from unittest import TestCase
+
+from mailman.app.lifecycle import create_list
+from mailman.interfaces.subscriptions import TokenOwner
+from mailman.interfaces.usermanager import IUserManager
+from mailman.interfaces.workflows import IWorkflow
+from mailman.testing.helpers import get_queue_messages
+from zope.component import getUtility
+from zope.interface import implementer
+
+from mailman_pgp.database import mm_transaction, transaction
+from mailman_pgp.model.address import PGPAddress
+from mailman_pgp.model.list import PGPMailingList
+from mailman_pgp.pgp.wrapper import PGPWrapper
+from mailman_pgp.testing.layers import PGPConfigLayer
+from mailman_pgp.testing.pgp import load_key
+from mailman_pgp.workflows.key_change import KeyChangeBase
+from mailman_pgp.workflows.mod_approval import ModeratorApprovalMixin
+
+
+@implementer(IWorkflow)
+class PGPTestWorkflow(KeyChangeBase, ModeratorApprovalMixin):
+    name = 'test-workflow'
+    description = ''
+    initial_state = 'mod_approval'
+
+
+class TestModeratorApprovalMixin(TestCase):
+    layer = PGPConfigLayer
+
+    def setUp(self):
+        with mm_transaction():
+            self.mlist = create_list('test@example.com',
+                                     style_name='pgp-default')
+        with transaction():
+            self.pgp_list = PGPMailingList.for_list(self.mlist)
+            self.pgp_list.key = load_key('ecc_p256.priv.asc')
+
+        self.sender_key = load_key('rsa_1024.priv.asc')
+        self.sender_new_key = load_key('ecc_p256.priv.asc')
+        self.sender = getUtility(IUserManager).create_address(
+                'anne@example.org')
+
+    def test_get_approval(self):
+        with transaction() as t:
+            pgp_address = PGPAddress(self.sender)
+            pgp_address.key = self.sender_key.pubkey
+            pgp_address.key_confirmed = True
+            t.add(pgp_address)
+
+        workflow = PGPTestWorkflow(self.mlist, pgp_address,
+                                   self.sender_new_key.pubkey)
+        list(workflow)
+        items = get_queue_messages('virgin', expected_count=1)
+        message = items[0].msg
+
+        self.assertEqual(message['Subject'],
+                         'New key change request from {}'.format(
+                                 pgp_address.email))
+        wrapped = PGPWrapper(message)
+        self.assertTrue(wrapped.has_keys())
+        keys = list(wrapped.keys())
+        self.assertEqual(len(keys), 1)
+        key = keys.pop()
+        self.assertEqual(key.fingerprint, self.sender_new_key.fingerprint)
+
+    def test_receive_approval(self):
+        with transaction() as t:
+            pgp_address = PGPAddress(self.sender)
+            pgp_address.key = self.sender_key.pubkey
+            pgp_address.key_confirmed = True
+            t.add(pgp_address)
+
+        workflow = PGPTestWorkflow(self.mlist, pgp_address,
+                                   self.sender_new_key.pubkey)
+        list(workflow)
+        get_queue_messages('virgin', expected_count=1)
+        list(workflow)
+        self.assertEqual(workflow.token_owner, TokenOwner.no_one)