1 files changed, 43 insertions, 1 deletions
diff --git a/src/mailman/rest/tests/test_validator.py b/src/mailman/rest/tests/test_validator.py
index 0d197032e..95855d170 100644
--- a/src/mailman/rest/tests/test_validator.py
+++ b/src/mailman/rest/tests/test_validator.py
@@ -22,9 +22,14 @@ import unittest
 from mailman.core.api import API30, API31
 from mailman.interfaces.action import Action
 from mailman.interfaces.usermanager import IUserManager
+from mailman.interfaces.workflows import (ISubscriptionWorkflow,
+                                          IUnsubscriptionWorkflow)
 from mailman.rest.validator import (
-    enum_validator, list_of_strings_validator, subscriber_validator)
+    enum_validator, list_of_strings_validator, policy_validator,
+    subscriber_validator)
 from mailman.testing.layers import RESTLayer
+from mailman.workflows.subscription import ConfirmSubscriptionPolicy
+from mailman.workflows.unsubscription import ConfirmUnsubscriptionPolicy
 from zope.component import getUtility
 
 
@@ -91,3 +96,40 @@ class TestValidators(unittest.TestCase):
 
     def test_enum_validator_blank(self):
         self.assertEqual(enum_validator(Action, allow_blank=True)(''), None)
+
+    def test_policy_validator_wrong_policy_class(self):
+        self.assertRaises(ValueError, policy_validator, None)
+
+    def test_policy_validator_sub_name(self):
+        self.assertEqual(policy_validator(ISubscriptionWorkflow)(
+                ConfirmSubscriptionPolicy.name), ConfirmSubscriptionPolicy)
+
+    def test_policy_validator_sub_class(self):
+        self.assertEqual(policy_validator(ISubscriptionWorkflow)(
+                ConfirmSubscriptionPolicy), ConfirmSubscriptionPolicy)
+
+    def test_policy_validator_sub_backward_compat(self):
+        self.assertEqual(policy_validator(ISubscriptionWorkflow)('confirm'),
+                         ConfirmSubscriptionPolicy)
+
+    def test_policy_validator_sub_wrong_policy(self):
+        validator = policy_validator(ISubscriptionWorkflow)
+        with self.assertRaises(ValueError):
+            validator('not a subscription policy')
+
+    def test_policy_validator_unsub_name(self):
+        self.assertEqual(policy_validator(IUnsubscriptionWorkflow)(
+                ConfirmUnsubscriptionPolicy.name), ConfirmUnsubscriptionPolicy)
+
+    def test_policy_validator_unsub_class(self):
+        self.assertEqual(policy_validator(IUnsubscriptionWorkflow)(
+                ConfirmUnsubscriptionPolicy), ConfirmUnsubscriptionPolicy)
+
+    def test_policy_validator_unsub_backward_compat(self):
+        self.assertEqual(policy_validator(IUnsubscriptionWorkflow)('confirm'),
+                         ConfirmUnsubscriptionPolicy)
+
+    def test_policy_validator_unsub_wrong_policy(self):
+        validator = policy_validator(IUnsubscriptionWorkflow)
+        with self.assertRaises(ValueError):
+            validator('not an unsubscription policy')