Exemplo n.º 1
0
class PublicationAdminTests(TestCase):
    def setUp(self):
        self.admin = PublicationAdmin(Article, AdminSite())
        self.request = RequestFactory().get("/")

    def get_list_display(self, *args, **kwargs):
        return self.admin.get_list_display(self.request, *args, **kwargs)

    def test_default_list_display(self):
        self.assertEqual(self.get_list_display(), ("__str__", "author_name", "publication_start_date", "enabled"))

    def test_custom_list_display(self):
        self.admin.list_display = ("title", "content")
        self.assertEqual(
            self.get_list_display(), ("title", "content", "author_name", "publication_start_date", "enabled")
        )

    def test_extend_list_display_is_disabled(self):
        self.admin.list_display = ("title", "content")
        self.admin.extend_list_display = False
        self.assertEqual(self.get_list_display(), ("title", "content"))

    def test_extend_list_display_is_disabled_by_argument(self):
        self.admin.list_display = ("title", "content")
        self.assertEqual(self.get_list_display(extend=False), ("title", "content"))
Exemplo n.º 2
0
 def setUp(self):
     self.admin = PublicationAdmin(Article, AdminSite())
     self.request = RequestFactory().get("/")