67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
"""admin tests"""
|
|
from importlib import import_module
|
|
from typing import Callable
|
|
|
|
from django.forms import ModelForm
|
|
from django.shortcuts import reverse
|
|
from django.test import Client, TestCase
|
|
from django.urls.exceptions import NoReverseMatch
|
|
|
|
from passbook.admin.urls import urlpatterns
|
|
from passbook.core.models import Group, User
|
|
from passbook.lib.utils.reflection import get_apps
|
|
|
|
|
|
class TestAdmin(TestCase):
|
|
"""Generic admin tests"""
|
|
|
|
def setUp(self):
|
|
self.user = User.objects.create_user(username="test")
|
|
self.user.pb_groups.add(Group.objects.filter(is_superuser=True).first())
|
|
self.user.save()
|
|
self.client = Client()
|
|
self.client.force_login(self.user)
|
|
|
|
|
|
def generic_view_tester(view_name: str) -> Callable:
|
|
"""This is used instead of subTest for better visibility"""
|
|
|
|
def tester(self: TestAdmin):
|
|
try:
|
|
full_url = reverse(f"passbook_admin:{view_name}")
|
|
response = self.client.get(full_url)
|
|
self.assertTrue(response.status_code < 500)
|
|
except NoReverseMatch:
|
|
pass
|
|
|
|
return tester
|
|
|
|
|
|
for url in urlpatterns:
|
|
method_name = url.name.replace("-", "_")
|
|
setattr(TestAdmin, f"test_view_{method_name}", generic_view_tester(url.name))
|
|
|
|
|
|
def generic_form_tester(form: ModelForm) -> Callable:
|
|
"""Test a form"""
|
|
|
|
def tester(self: TestAdmin):
|
|
form_inst = form()
|
|
self.assertFalse(form_inst.is_valid())
|
|
|
|
return tester
|
|
|
|
|
|
# Load the forms module from every app, so we have all forms loaded
|
|
for app in get_apps():
|
|
module = app.__module__.replace(".apps", ".forms")
|
|
try:
|
|
import_module(module)
|
|
except ImportError:
|
|
pass
|
|
|
|
for form_class in ModelForm.__subclasses__():
|
|
setattr(
|
|
TestAdmin, f"test_form_{form_class.__name__}", generic_form_tester(form_class)
|
|
)
|