1. from django.contrib import admin

  2. from django.contrib.admin.decorators import register

  3. from django.contrib.admin.sites import site

  4. from django.core.exceptions import ImproperlyConfigured

  5. from django.test import SimpleTestCase

  6. 

  7. from .models import Location, Person, Place, Traveler

  8. 

  9. 

  10. class NameAdmin(admin.ModelAdmin):

  11.     list_display = ["name"]

  12.     save_on_top = True

  13. 

  14. 

  15. class CustomSite(admin.AdminSite):

  16.     pass

  17. 

  18. 

  19. class TestRegistration(SimpleTestCase):

  20.     def setUp(self):

  21.         self.site = admin.AdminSite()

  22. 

  23.     def test_bare_registration(self):

  24.         self.site.register(Person)

  25.         self.assertIsInstance(self.site._registry[Person], admin.ModelAdmin)

  26.         self.site.unregister(Person)

  27.         self.assertEqual(self.site._registry, {})

  28. 

  29.     def test_registration_with_model_admin(self):

  30.         self.site.register(Person, NameAdmin)

  31.         self.assertIsInstance(self.site._registry[Person], NameAdmin)

  32.         self.site.unregister(Person)

  33.         self.assertEqual(self.site._registry, {})

  34. 

  35.     def test_prevent_double_registration(self):

  36.         self.site.register(Person)

  37.         msg = "The model Person is already registered in app 'admin_registration'."

  38.         with self.assertRaisesMessage(admin.sites.AlreadyRegistered, msg):

  39.             self.site.register(Person)

  40. 

  41.     def test_prevent_double_registration_for_custom_admin(self):

  42.         class PersonAdmin(admin.ModelAdmin):

  43.             pass

  44. 

  45.         self.site.register(Person, PersonAdmin)

  46.         msg = (

  47.             "The model Person is already registered with "

  48.             "'admin_registration.PersonAdmin'."

  49.         )

  50.         with self.assertRaisesMessage(admin.sites.AlreadyRegistered, msg):

  51.             self.site.register(Person, PersonAdmin)

  52. 

  53.     def test_unregister_unregistered_model(self):

  54.         msg = "The model Person is not registered"

  55.         with self.assertRaisesMessage(admin.sites.NotRegistered, msg):

  56.             self.site.unregister(Person)

  57. 

  58.     def test_registration_with_star_star_options(self):

  59.         self.site.register(Person, search_fields=["name"])

  60.         self.assertEqual(self.site._registry[Person].search_fields, ["name"])

  61. 

  62.     def test_star_star_overrides(self):

  63.         self.site.register(

  64.             Person, NameAdmin, search_fields=["name"], list_display=["__str__"]

  65.         )

  66.         self.assertEqual(self.site._registry[Person].search_fields, ["name"])

  67.         self.assertEqual(self.site._registry[Person].list_display, ["__str__"])

  68.         self.assertTrue(self.site._registry[Person].save_on_top)

  69. 

  70.     def test_iterable_registration(self):

  71.         self.site.register([Person, Place], search_fields=["name"])

  72.         self.assertIsInstance(self.site._registry[Person], admin.ModelAdmin)

  73.         self.assertEqual(self.site._registry[Person].search_fields, ["name"])

  74.         self.assertIsInstance(self.site._registry[Place], admin.ModelAdmin)

  75.         self.assertEqual(self.site._registry[Place].search_fields, ["name"])

  76.         self.site.unregister([Person, Place])

  77.         self.assertEqual(self.site._registry, {})

  78. 

  79.     def test_abstract_model(self):

  80.         """

  81.         Exception is raised when trying to register an abstract model.

  82.         Refs #12004.

  83.         """

  84.         msg = "The model Location is abstract, so it cannot be registered with admin."

  85.         with self.assertRaisesMessage(ImproperlyConfigured, msg):

  86.             self.site.register(Location)

  87. 

  88.     def test_is_registered_model(self):

  89.         "Checks for registered models should return true."

  90.         self.site.register(Person)

  91.         self.assertTrue(self.site.is_registered(Person))

  92. 

  93.     def test_is_registered_not_registered_model(self):

  94.         "Checks for unregistered models should return false."

  95.         self.assertFalse(self.site.is_registered(Person))

  96. 

  97. 

  98. class TestRegistrationDecorator(SimpleTestCase):

  99.     """

  100.     Tests the register decorator in admin.decorators

  101. 

  102.     For clarity:

  103. 

  104.         @register(Person)

  105.         class AuthorAdmin(ModelAdmin):

  106.             pass

  107. 

  108.     is functionally equal to (the way it is written in these tests):

  109. 

  110.         AuthorAdmin = register(Person)(AuthorAdmin)

  111.     """

  112. 

  113.     def setUp(self):

  114.         self.default_site = site

  115.         self.custom_site = CustomSite()

  116. 

  117.     def test_basic_registration(self):

  118.         register(Person)(NameAdmin)

  119.         self.assertIsInstance(self.default_site._registry[Person], admin.ModelAdmin)

  120.         self.default_site.unregister(Person)

  121. 

  122.     def test_custom_site_registration(self):

  123.         register(Person, site=self.custom_site)(NameAdmin)

  124.         self.assertIsInstance(self.custom_site._registry[Person], admin.ModelAdmin)

  125. 

  126.     def test_multiple_registration(self):

  127.         register(Traveler, Place)(NameAdmin)

  128.         self.assertIsInstance(self.default_site._registry[Traveler], admin.ModelAdmin)

  129.         self.default_site.unregister(Traveler)

  130.         self.assertIsInstance(self.default_site._registry[Place], admin.ModelAdmin)

  131.         self.default_site.unregister(Place)

  132. 

  133.     def test_wrapped_class_not_a_model_admin(self):

  134.         with self.assertRaisesMessage(

  135.             ValueError, "Wrapped class must subclass ModelAdmin."

  136.         ):

  137.             register(Person)(CustomSite)

  138. 

  139.     def test_custom_site_not_an_admin_site(self):

  140.         with self.assertRaisesMessage(ValueError, "site must subclass AdminSite"):

  141.             register(Person, site=Traveler)(NameAdmin)

  142. 

  143.     def test_empty_models_list_registration_fails(self):

  144.         with self.assertRaisesMessage(

  145.             ValueError, "At least one model must be passed to register."

  146.         ):

  147.             register()(NameAdmin)
About Sign Up Leave Feedback Contact