From 7dc2f9314897eca9904a386e9e9a508b755992cb Mon Sep 17 00:00:00 2001 From: Diederik van der Boor Date: Fri, 3 Feb 2017 22:04:27 +0100 Subject: [PATCH] Added plain tests for admin importing --- polymorphic/tests/__init__.py | 1 + polymorphic/tests/test_admin.py | 31 +++++++++++++++++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 polymorphic/tests/test_admin.py diff --git a/polymorphic/tests/__init__.py b/polymorphic/tests/__init__.py index 89e3a94..892e570 100644 --- a/polymorphic/tests/__init__.py +++ b/polymorphic/tests/__init__.py @@ -411,6 +411,7 @@ class DateModel(PolymorphicModel): # Import tests +from .test_admin import * from .test_orm import * from .test_multidb import * from .test_regression import * diff --git a/polymorphic/tests/test_admin.py b/polymorphic/tests/test_admin.py new file mode 100644 index 0000000..a5d72ca --- /dev/null +++ b/polymorphic/tests/test_admin.py @@ -0,0 +1,31 @@ +from django.contrib.admin import AdminSite +from django.test import TestCase + +from polymorphic.admin import PolymorphicParentModelAdmin, PolymorphicChildModelAdmin, PolymorphicChildModelFilter +from polymorphic.tests 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)