policies/hibp: update for flows, add unittests

passbook/policies/hibp/tests.py

@@ -0,0 +1,29 @@
+"""HIBP Policy tests"""
+from django.test import TestCase
+from guardian.shortcuts import get_anonymous_user
+from oauth2_provider.generators import generate_client_secret
+
+from passbook.policies.hibp.models import HaveIBeenPwendPolicy
+from passbook.policies.types import PolicyRequest, PolicyResult
+
+
+class TestHIBPPolicy(TestCase):
+    """Test HIBP Policy"""
+
+    def test_false(self):
+        """Failing password case"""
+        policy = HaveIBeenPwendPolicy.objects.create(name="test_false",)
+        request = PolicyRequest(get_anonymous_user())
+        request.context["password"] = "password"
+        result: PolicyResult = policy.passes(request)
+        self.assertFalse(result.passing)
+        self.assertTrue(result.messages[0].startswith("Password exists on "))
+
+    def test_true(self):
+        """Positive password case"""
+        policy = HaveIBeenPwendPolicy.objects.create(name="test_true",)
+        request = PolicyRequest(get_anonymous_user())
+        request.context["password"] = generate_client_secret()
+        result: PolicyResult = policy.passes(request)
+        self.assertTrue(result.passing)
+        self.assertEqual(result.messages, tuple())