59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
from django.contrib.admin.options import ModelAdmin, TabularInline
|
|
from django.utils.deprecation import RemovedInDjango30Warning
|
|
|
|
from .models import Band, Song
|
|
from .test_checks import CheckTestCase
|
|
|
|
|
|
class HasAddPermissionObjTests(CheckTestCase):
|
|
def test_model_admin_inherited_valid(self):
|
|
class BandAdmin(ModelAdmin):
|
|
pass
|
|
|
|
self.assertIsValid(BandAdmin, Band)
|
|
|
|
def test_model_admin_valid(self):
|
|
class BandAdmin(ModelAdmin):
|
|
def has_add_permission(self, request):
|
|
return super().has_add_permission(request)
|
|
|
|
self.assertIsValid(BandAdmin, Band)
|
|
|
|
def test_inline_admin_inherited_valid(self):
|
|
class SongInlineAdmin(TabularInline):
|
|
model = Song
|
|
|
|
class BandAdmin(ModelAdmin):
|
|
inlines = [SongInlineAdmin]
|
|
|
|
self.assertIsValid(BandAdmin, Band)
|
|
|
|
def test_inline_admin_valid(self):
|
|
class SongInlineAdmin(TabularInline):
|
|
model = Song
|
|
|
|
def has_add_permission(self, request, obj):
|
|
return super().has_add_permission(request, obj)
|
|
|
|
class BandAdmin(ModelAdmin):
|
|
inlines = [SongInlineAdmin]
|
|
|
|
self.assertIsValid(BandAdmin, Band)
|
|
|
|
def test_inline_admin_warning(self):
|
|
class SongInlineAdmin(TabularInline):
|
|
model = Song
|
|
|
|
def has_add_permission(self, request):
|
|
return super().has_add_permission(request)
|
|
|
|
class BandAdmin(ModelAdmin):
|
|
inlines = [SongInlineAdmin]
|
|
|
|
msg = (
|
|
"Update SongInlineAdmin.has_add_permission() to accept a "
|
|
"positional `obj` argument."
|
|
)
|
|
with self.assertWarnsMessage(RemovedInDjango30Warning, msg):
|
|
self.assertIsValid(BandAdmin, Band)
|