python/lib/django_1_4/tests/regressiontests/admin_registration/tests.py

   1 from __future__ import absolute_import
   2
   3 from django.contrib import admin
   4 from django.core.exceptions import ImproperlyConfigured
   5 from django.test import TestCase
   6
   7 from .models import Person, Place, Location
   8
   9
  10 class NameAdmin(admin.ModelAdmin):
  11     list_display = ['name']
  12     save_on_top = True
  13
  14 class TestRegistration(TestCase):
  15     def setUp(self):
  16         self.site = admin.AdminSite()
  17
  18     def test_bare_registration(self):
  19         self.site.register(Person)
  20         self.assertTrue(
  21             isinstance(self.site._registry[Person], admin.options.ModelAdmin)
  22         )
  23
  24     def test_registration_with_model_admin(self):
  25         self.site.register(Person, NameAdmin)
  26         self.assertTrue(
  27             isinstance(self.site._registry[Person], NameAdmin)
  28         )
  29
  30     def test_prevent_double_registration(self):
  31         self.site.register(Person)
  32         self.assertRaises(admin.sites.AlreadyRegistered,
  33                           self.site.register,
  34                           Person)
  35
  36     def test_registration_with_star_star_options(self):
  37         self.site.register(Person, search_fields=['name'])
  38         self.assertEqual(self.site._registry[Person].search_fields, ['name'])
  39
  40     def test_star_star_overrides(self):
  41         self.site.register(Person, NameAdmin,
  42                            search_fields=["name"], list_display=['__str__'])
  43         self.assertEqual(self.site._registry[Person].search_fields, ['name'])
  44         self.assertEqual(self.site._registry[Person].list_display,
  45                          ['__str__'])
  46         self.assertTrue(self.site._registry[Person].save_on_top)
  47
  48     def test_iterable_registration(self):
  49         self.site.register([Person, Place], search_fields=['name'])
  50         self.assertTrue(
  51             isinstance(self.site._registry[Person], admin.options.ModelAdmin)
  52         )
  53         self.assertEqual(self.site._registry[Person].search_fields, ['name'])
  54         self.assertTrue(
  55             isinstance(self.site._registry[Place], admin.options.ModelAdmin)
  56         )
  57         self.assertEqual(self.site._registry[Place].search_fields, ['name'])
  58
  59     def test_abstract_model(self):
  60         """
  61         Exception is raised when trying to register an abstract model.
  62         Refs #12004.
  63         """
  64         self.assertRaises(ImproperlyConfigured, self.site.register, Location)