/tests/regressiontests/admin_registration/tests.py
https://code.google.com/p/mango-py/ · Python · 61 lines · 47 code · 10 blank · 4 comment · 0 complexity · 6863daa3f26e403ad3d75c4edeee22ae MD5 · raw file
- from django.test import TestCase
- from django.core.exceptions import ImproperlyConfigured
- from django.contrib import admin
- from models import Person, Place, Location
- class NameAdmin(admin.ModelAdmin):
- list_display = ['name']
- save_on_top = True
- class TestRegistration(TestCase):
- def setUp(self):
- self.site = admin.AdminSite()
- def test_bare_registration(self):
- self.site.register(Person)
- self.assertTrue(
- isinstance(self.site._registry[Person], admin.options.ModelAdmin)
- )
- def test_registration_with_model_admin(self):
- self.site.register(Person, NameAdmin)
- self.assertTrue(
- isinstance(self.site._registry[Person], NameAdmin)
- )
- def test_prevent_double_registration(self):
- self.site.register(Person)
- self.assertRaises(admin.sites.AlreadyRegistered,
- self.site.register,
- Person)
- def test_registration_with_star_star_options(self):
- self.site.register(Person, search_fields=['name'])
- self.assertEqual(self.site._registry[Person].search_fields, ['name'])
- def test_star_star_overrides(self):
- self.site.register(Person, NameAdmin,
- search_fields=["name"], list_display=['__str__'])
- self.assertEqual(self.site._registry[Person].search_fields, ['name'])
- self.assertEqual(self.site._registry[Person].list_display,
- ['action_checkbox', '__str__'])
- self.assertTrue(self.site._registry[Person].save_on_top)
- def test_iterable_registration(self):
- self.site.register([Person, Place], search_fields=['name'])
- self.assertTrue(
- isinstance(self.site._registry[Person], admin.options.ModelAdmin)
- )
- self.assertEqual(self.site._registry[Person].search_fields, ['name'])
- self.assertTrue(
- isinstance(self.site._registry[Place], admin.options.ModelAdmin)
- )
- self.assertEqual(self.site._registry[Place].search_fields, ['name'])
- def test_abstract_model(self):
- """
- Exception is raised when trying to register an abstract model.
- Refs #12004.
- """
- self.assertRaises(ImproperlyConfigured, self.site.register, Location)