class AdminCategoryTest(TestCase):

    def setUp(self):
        self.admin = CategoryAdmin(Category, AdminSite())
        self.user = user.make()
        self.category = category.make(
            author=self.user, _quantity=1
        )

    def test_default_fields(self):
        self.assertEqual(
            list(self.admin.get_form(request).base_fields),
            ['title', 'slug', 'author', 'description']
        )

        self.assertEqual(
            list(self.admin.get_fields(request)),
            ['title', 'slug', 'author', 'description']
        )

        self.assertEqual(
            list(self.admin.get_fields(request, self.category[0])),
            ['title', 'slug', 'author', 'description']
        )

    def tearDown(self):
        pass
 def setUp(self):
     self.admin = CategoryAdmin(Category, AdminSite())
     self.user = user.make()
     self.category = category.make(
         author=self.user, _quantity=1
     )