def test_fieldsets(self):
model_admin = ModelAdmin(User, self.site)
self.assertEqual(
model_admin.get_fieldsets(request),
[(None, {'fields': [
'password',
'last_login',
'is_superuser',
'groups',
'user_permissions',
'is_removed',
'email',
'is_staff',
'is_active',
'date_joined',
'name'
]})]
)
self.assertEqual(
model_admin.get_fieldsets(request, self.user),
[(None, {'fields': [
'password',
'last_login',
'is_superuser',
'groups',
'user_permissions',
'is_removed',
'email',
'is_staff',
'is_active',
'date_joined',
'name'
]})]
)
def test_default_fieldsets(self):
# fieldsets_add and fieldsets_change should return a special data structure that
# is used in the templates. They should generate the "right thing" whether we
# have specified a custom form, the fields argument, or nothing at all.
#
# Here's the default case. There are no custom form_add/form_change methods,
# no fields argument, and no fieldsets argument.
ma = ModelAdmin(Band, self.site)
self.assertEqual(ma.get_fieldsets(request), [(None, {'fields': ['name', 'bio', 'sign_date']})])
self.assertEqual(ma.get_fieldsets(request, self.band), [(None, {'fields': ['name', 'bio', 'sign_date']})])
def test_default_fieldsets(self):
# fieldsets_add and fieldsets_change should return a special data structure that
# is used in the templates. They should generate the "right thing" whether we
# have specified a custom form, the fields argument, or nothing at all.
#
# Here's the default case. There are no custom form_add/form_change methods,
# no fields argument, and no fieldsets argument.
ma = ModelAdmin(Band, self.site)
self.assertEqual(ma.get_fieldsets(request), [(None, {'fields': ['name', 'bio', 'sign_date']})])
self.assertEqual(ma.get_fieldsets(request, self.band), [(None, {'fields': ['name', 'bio', 'sign_date']})])
def test_fieldsets(self):
user = UserFactory()
ma = ModelAdmin(CustomUser, self.site)
expected = [(None, {
'fields': [
'password', 'last_login', 'is_superuser', 'groups',
'user_permissions', 'email', 'is_active', 'is_admin',
'is_staff', 'is_company_admin', 'avatar', 'name',
'activation_key', 'company'
]
})]
self.assertEqual(expected, ma.get_fieldsets(None, user))
def test_fieldsets(self):
user = UserFactory()
ma = ModelAdmin(CustomUser, self.site)
expected = [
(None, {
'fields': [
'password', 'last_login', 'is_superuser', 'groups',
'user_permissions', 'email', 'is_active', 'is_admin',
'is_staff', 'is_company_admin', 'avatar', 'name',
'activation_key', 'company']
})
]
self.assertEqual(expected, ma.get_fieldsets(None, user))
