44d7e81a93
* add helper to mass-compile and re-use mappings Signed-off-by: Jens Langhammer <jens@goauthentik.io> * implement for scim Signed-off-by: Jens Langhammer <jens@goauthentik.io> * actually make it even simpler Signed-off-by: Jens Langhammer <jens@goauthentik.io> * migrate google Signed-off-by: Jens Langhammer <jens@goauthentik.io> * migrate microsoft too Signed-off-by: Jens Langhammer <jens@goauthentik.io> * remove sleeps in tests with freezegun Signed-off-by: Jens Langhammer <jens@goauthentik.io> * migrate ldap to propertymapping helper Signed-off-by: Jens Langhammer <jens@goauthentik.io> * move mapper to generic sync Signed-off-by: Jens Langhammer <jens@goauthentik.io> * fix Signed-off-by: Jens Langhammer <jens@goauthentik.io> * apparently that doesn't work Signed-off-by: Jens Langhammer <jens@goauthentik.io> * forgot a sleep Signed-off-by: Jens Langhammer <jens@goauthentik.io> * backport fixes from #9783 Signed-off-by: Jens Langhammer <jens@goauthentik.io> --------- Signed-off-by: Jens Langhammer <jens@goauthentik.io>
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
"""authentik core models tests"""
|
|
|
|
from collections.abc import Callable
|
|
from datetime import timedelta
|
|
|
|
from django.test import RequestFactory, TestCase
|
|
from django.utils.timezone import now
|
|
from freezegun import freeze_time
|
|
from guardian.shortcuts import get_anonymous_user
|
|
|
|
from authentik.core.models import Provider, Source, Token
|
|
from authentik.flows.models import Stage
|
|
from authentik.lib.utils.reflection import all_subclasses
|
|
|
|
|
|
class TestModels(TestCase):
|
|
"""Test Models"""
|
|
|
|
def test_token_expire(self):
|
|
"""Test token expiring"""
|
|
with freeze_time() as freeze:
|
|
token = Token.objects.create(expires=now(), user=get_anonymous_user())
|
|
freeze.tick(timedelta(seconds=1))
|
|
self.assertTrue(token.is_expired)
|
|
|
|
def test_token_expire_no_expire(self):
|
|
"""Test token expiring with "expiring" set"""
|
|
with freeze_time() as freeze:
|
|
token = Token.objects.create(expires=now(), user=get_anonymous_user(), expiring=False)
|
|
freeze.tick(timedelta(seconds=1))
|
|
self.assertFalse(token.is_expired)
|
|
|
|
|
|
def source_tester_factory(test_model: type[Stage]) -> Callable:
|
|
"""Test source"""
|
|
|
|
factory = RequestFactory()
|
|
request = factory.get("/")
|
|
|
|
def tester(self: TestModels):
|
|
model_class = None
|
|
if test_model._meta.abstract: # pragma: no cover
|
|
model_class = test_model.__bases__[0]()
|
|
else:
|
|
model_class = test_model()
|
|
model_class.slug = "test"
|
|
self.assertIsNotNone(model_class.component)
|
|
_ = model_class.ui_login_button(request)
|
|
_ = model_class.ui_user_settings()
|
|
|
|
return tester
|
|
|
|
|
|
def provider_tester_factory(test_model: type[Stage]) -> Callable:
|
|
"""Test provider"""
|
|
|
|
def tester(self: TestModels):
|
|
model_class = None
|
|
if test_model._meta.abstract: # pragma: no cover
|
|
return
|
|
model_class = test_model()
|
|
self.assertIsNotNone(model_class.component)
|
|
|
|
return tester
|
|
|
|
|
|
for model in all_subclasses(Source):
|
|
setattr(TestModels, f"test_source_{model.__name__}", source_tester_factory(model))
|
|
for model in all_subclasses(Provider):
|
|
setattr(TestModels, f"test_provider_{model.__name__}", provider_tester_factory(model))
|