30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
"""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())
|