142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
from django.contrib import admin
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.utils.html import escape
|
|
|
|
from polymorphic.admin import (
|
|
PolymorphicChildModelAdmin,
|
|
PolymorphicChildModelFilter,
|
|
PolymorphicInlineSupportMixin,
|
|
PolymorphicParentModelAdmin,
|
|
StackedPolymorphicInline,
|
|
)
|
|
from polymorphic.tests.admintestcase import AdminTestCase
|
|
from polymorphic.tests.models import (
|
|
InlineModelA,
|
|
InlineModelB,
|
|
InlineParent,
|
|
Model2A,
|
|
Model2B,
|
|
Model2C,
|
|
Model2D,
|
|
)
|
|
|
|
|
|
class PolymorphicAdminTests(AdminTestCase):
|
|
def test_admin_registration(self):
|
|
"""
|
|
Test how the registration works
|
|
"""
|
|
|
|
@self.register(Model2A)
|
|
class Model2Admin(PolymorphicParentModelAdmin):
|
|
base_model = Model2A
|
|
list_filter = (PolymorphicChildModelFilter,)
|
|
child_models = (Model2B, Model2C, Model2D)
|
|
|
|
@self.register(Model2B)
|
|
@self.register(Model2C)
|
|
@self.register(Model2D)
|
|
class Model2ChildAdmin(PolymorphicChildModelAdmin):
|
|
base_model = Model2A
|
|
base_fieldsets = (("Base fields", {"fields": ("field1",)}),)
|
|
|
|
# -- add page
|
|
ct_id = ContentType.objects.get_for_model(Model2D).pk
|
|
self.admin_get_add(Model2A) # shows type page
|
|
self.admin_get_add(Model2A, qs="?ct_id={}".format(ct_id)) # shows type page
|
|
|
|
self.admin_get_add(Model2A) # shows type page
|
|
self.admin_post_add(
|
|
Model2A,
|
|
{"field1": "A", "field2": "B", "field3": "C", "field4": "D"},
|
|
qs="?ct_id={}".format(ct_id),
|
|
)
|
|
|
|
d_obj = Model2A.objects.all()[0]
|
|
self.assertEqual(d_obj.__class__, Model2D)
|
|
self.assertEqual(d_obj.field1, "A")
|
|
self.assertEqual(d_obj.field2, "B")
|
|
|
|
# -- list page
|
|
self.admin_get_changelist(Model2A) # asserts 200
|
|
|
|
# -- edit
|
|
response = self.admin_get_change(Model2A, d_obj.pk)
|
|
self.assertContains(response, "field4")
|
|
self.admin_post_change(
|
|
Model2A,
|
|
d_obj.pk,
|
|
{"field1": "A2", "field2": "B2", "field3": "C2", "field4": "D2"},
|
|
)
|
|
|
|
d_obj.refresh_from_db()
|
|
self.assertEqual(d_obj.field1, "A2")
|
|
self.assertEqual(d_obj.field2, "B2")
|
|
self.assertEqual(d_obj.field3, "C2")
|
|
self.assertEqual(d_obj.field4, "D2")
|
|
|
|
# -- history
|
|
self.admin_get_history(Model2A, d_obj.pk)
|
|
|
|
# -- delete
|
|
self.admin_get_delete(Model2A, d_obj.pk)
|
|
self.admin_post_delete(Model2A, d_obj.pk)
|
|
self.assertRaises(Model2A.DoesNotExist, lambda: d_obj.refresh_from_db())
|
|
|
|
def test_admin_inlines(self):
|
|
"""
|
|
Test the registration of inline models.
|
|
"""
|
|
|
|
class InlineModelAChild(StackedPolymorphicInline.Child):
|
|
model = InlineModelA
|
|
|
|
class InlineModelBChild(StackedPolymorphicInline.Child):
|
|
model = InlineModelB
|
|
|
|
class Inline(StackedPolymorphicInline):
|
|
model = InlineModelA
|
|
child_inlines = (InlineModelAChild, InlineModelBChild)
|
|
|
|
@self.register(InlineParent)
|
|
class InlineParentAdmin(PolymorphicInlineSupportMixin, admin.ModelAdmin):
|
|
inlines = (Inline,)
|
|
|
|
parent = InlineParent.objects.create(title="FOO")
|
|
self.assertEqual(parent.inline_children.count(), 0)
|
|
|
|
# -- get edit page
|
|
response = self.admin_get_change(InlineParent, parent.pk)
|
|
|
|
# Make sure the fieldset has the right data exposed in data-inline-formset
|
|
self.assertContains(response, "childTypes")
|
|
self.assertContains(response, escape('"type": "inlinemodela"'))
|
|
self.assertContains(response, escape('"type": "inlinemodelb"'))
|
|
|
|
# -- post edit page
|
|
self.admin_post_change(
|
|
InlineParent,
|
|
parent.pk,
|
|
{
|
|
"title": "FOO2",
|
|
"inline_children-INITIAL_FORMS": 0,
|
|
"inline_children-TOTAL_FORMS": 1,
|
|
"inline_children-MIN_NUM_FORMS": 0,
|
|
"inline_children-MAX_NUM_FORMS": 1000,
|
|
"inline_children-0-parent": parent.pk,
|
|
"inline_children-0-polymorphic_ctype": ContentType.objects.get_for_model(
|
|
InlineModelB
|
|
).pk,
|
|
"inline_children-0-field1": "A2",
|
|
"inline_children-0-field2": "B2",
|
|
},
|
|
)
|
|
|
|
parent.refresh_from_db()
|
|
self.assertEqual(parent.title, "FOO2")
|
|
self.assertEqual(parent.inline_children.count(), 1)
|
|
child = parent.inline_children.all()[0]
|
|
self.assertEqual(child.__class__, InlineModelB)
|
|
self.assertEqual(child.field1, "A2")
|
|
self.assertEqual(child.field2, "B2")
|