django-admin2/djadmin2/tests/test_core.py

from django.contrib.auth.models import Group, User
from django.contrib.sites.models import Site
from django.core.exceptions import ImproperlyConfigured
from django.test import TestCase

from djadmin2.site import djadmin2_site
from .models import SmallThing
from ..core import Admin2
from ..types import ModelAdmin2

APP_LABEL, APP_VERBOSE_NAME = 'app_one_label', 'App One Verbose Name'


class Admin2Test(TestCase):
    def setUp(self):
        self.admin2 = Admin2()

    def test_register(self):
        self.admin2.register(SmallThing)
        self.assertTrue(isinstance(self.admin2.registry[SmallThing], ModelAdmin2))

    def test_register_error(self):
        self.admin2.register(SmallThing)
        self.assertRaises(ImproperlyConfigured, self.admin2.register, SmallThing)

    def test_deregister(self):
        self.admin2.register(SmallThing)
        self.admin2.deregister(SmallThing)
        self.assertTrue(SmallThing not in self.admin2.registry)

    def test_deregister_error(self):
        self.assertRaises(ImproperlyConfigured, self.admin2.deregister, SmallThing)

    def test_register_app_verbose_name(self):
        self.admin2.register_app_verbose_name(APP_LABEL, APP_VERBOSE_NAME)
        self.assertEquals(
            self.admin2.app_verbose_names[APP_LABEL],
            APP_VERBOSE_NAME
        )

    def test_register_app_verbose_name_error(self):
        self.admin2.register_app_verbose_name(APP_LABEL, APP_VERBOSE_NAME)
        self.assertRaises(
            ImproperlyConfigured,
            self.admin2.register_app_verbose_name,
            APP_LABEL,
            APP_VERBOSE_NAME
        )

    def test_deregister_app_verbose_name(self):
        self.admin2.register_app_verbose_name(APP_LABEL, APP_VERBOSE_NAME)
        self.admin2.deregister_app_verbose_name(APP_LABEL)
        self.assertTrue(APP_LABEL not in self.admin2.app_verbose_names)

    def test_deregister_app_verbose_name_error(self):
        self.assertRaises(
            ImproperlyConfigured,
            self.admin2.deregister_app_verbose_name,
            APP_LABEL
        )

    def test_get_urls(self):
        self.admin2.register(SmallThing)
        self.assertEquals(8, len(self.admin2.get_urls()))

    def test_default_entries(self):
        expected_default_models = (User, Group, Site)
        for model in expected_default_models:
            self.assertTrue(isinstance(djadmin2_site.registry[model], ModelAdmin2))