32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from django.contrib.admin import AdminSite
|
|
from django.test import TestCase
|
|
|
|
from polymorphic.admin import PolymorphicParentModelAdmin, PolymorphicChildModelAdmin, PolymorphicChildModelFilter
|
|
from polymorphic.tests.models import Model2A, Model2B, Model2C, Model2D
|
|
|
|
|
|
class MultipleDatabasesTests(TestCase):
|
|
|
|
def test_admin_registration(self):
|
|
"""
|
|
Test how the registration works
|
|
"""
|
|
class Model2Admin(PolymorphicParentModelAdmin):
|
|
base_model = Model2A
|
|
list_filter = (PolymorphicChildModelFilter,)
|
|
child_models = (Model2B, Model2C, Model2D)
|
|
|
|
class Model2ChildAdmin(PolymorphicChildModelAdmin):
|
|
base_model = Model2A
|
|
base_fieldsets = (
|
|
("Base fields", {
|
|
'fields': ('field1',)
|
|
}),
|
|
)
|
|
|
|
admin_site = AdminSite()
|
|
admin_site.register(Model2A, Model2Admin)
|
|
admin_site.register(Model2B, Model2ChildAdmin)
|
|
admin_site.register(Model2C, Model2ChildAdmin)
|
|
admin_site.register(Model2D, Model2ChildAdmin)
|