34 lines
934 B
Python
34 lines
934 B
Python
"""base model tests"""
|
|
from typing import Callable
|
|
|
|
from django.test import TestCase
|
|
from rest_framework.serializers import BaseSerializer
|
|
|
|
from authentik.flows.models import Stage
|
|
from authentik.lib.models import SerializerModel
|
|
from authentik.lib.utils.reflection import all_subclasses
|
|
|
|
|
|
class TestModels(TestCase):
|
|
"""Generic model properties tests"""
|
|
|
|
|
|
def model_tester_factory(test_model: type[Stage]) -> Callable:
|
|
"""Test a form"""
|
|
|
|
def tester(self: TestModels):
|
|
try:
|
|
model_class = None
|
|
if test_model._meta.abstract: # pragma: no cover
|
|
return
|
|
model_class = test_model()
|
|
self.assertTrue(issubclass(model_class.serializer, BaseSerializer))
|
|
except NotImplementedError:
|
|
pass
|
|
|
|
return tester
|
|
|
|
|
|
for model in all_subclasses(SerializerModel):
|
|
setattr(TestModels, f"test_model_{model.__name__}", model_tester_factory(model))
|