This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/policies/tests/test_engine.py

59 lines
2.1 KiB
Python
Raw Normal View History

"""policy engine tests"""
from django.core.cache import cache
from django.test import TestCase
from passbook.core.models import Policy, User
from passbook.policies.dummy.models import DummyPolicy
from passbook.policies.engine import PolicyEngine
class PolicyTestEngine(TestCase):
"""PolicyEngine tests"""
def setUp(self):
cache.clear()
2019-12-31 11:51:16 +00:00
self.user = User.objects.create_user(username="policyuser")
self.policy_false = DummyPolicy.objects.create(
2019-12-31 11:51:16 +00:00
result=False, wait_min=0, wait_max=1
)
self.policy_true = DummyPolicy.objects.create(
2019-12-31 11:51:16 +00:00
result=True, wait_min=0, wait_max=1
)
self.policy_negate = DummyPolicy.objects.create(
2019-12-31 11:51:16 +00:00
negate=True, result=True, wait_min=0, wait_max=1
)
self.policy_raises = Policy.objects.create(name="raises")
def test_engine_empty(self):
"""Ensure empty policy list passes"""
engine = PolicyEngine([], self.user)
self.assertEqual(engine.build().passing, True)
def test_engine(self):
"""Ensure all policies passes (Mix of false and true -> false)"""
2019-12-31 11:51:16 +00:00
engine = PolicyEngine(
DummyPolicy.objects.filter(negate__exact=False), self.user
2019-12-31 11:51:16 +00:00
)
self.assertEqual(engine.build().passing, False)
def test_engine_negate(self):
"""Test negate flag"""
engine = PolicyEngine(DummyPolicy.objects.filter(negate__exact=True), self.user)
self.assertEqual(engine.build().passing, False)
def test_engine_policy_error(self):
"""Test negate flag"""
2019-12-31 11:51:16 +00:00
engine = PolicyEngine(Policy.objects.filter(name="raises"), self.user)
self.assertEqual(engine.build().passing, False)
def test_engine_cache(self):
"""Ensure empty policy list passes"""
2019-12-31 11:51:16 +00:00
engine = PolicyEngine(
DummyPolicy.objects.filter(negate__exact=False), self.user
2019-12-31 11:51:16 +00:00
)
self.assertEqual(len(cache.keys("policy_*")), 0)
self.assertEqual(engine.build().passing, False)
2019-12-31 11:51:16 +00:00
self.assertEqual(len(cache.keys("policy_*")), 2)
self.assertEqual(engine.build().passing, False)
2019-12-31 11:51:16 +00:00
self.assertEqual(len(cache.keys("policy_*")), 2)