Ticket #18681: django-bug-get_fieldsets.shar

# This is a shell archive.  Save it in a file, remove anything before
# this line, and then unpack it by entering "sh file".  Note, it may
# create directories; files and directories will be owned by you and
# have default permissions.
#
# This archive contains:
#
#       bookmarks
#       bookmarks/__init__.py
#       bookmarks/admin.py
#       bookmarks/models.py
#       bookmarks/tests.py
#
echo c - bookmarks
mkdir -p bookmarks > /dev/null 2>&1
echo x - bookmarks/__init__.py
sed 's/^X//' >bookmarks/__init__.py << '20a1c7151f244b5c6bc388b74c2b6612'
20a1c7151f244b5c6bc388b74c2b6612
echo x - bookmarks/admin.py
sed 's/^X//' >bookmarks/admin.py << '947cf57c8598ad9386f34efacea96cd9'
X# vim: ts=4 sw=4 et tw=78
Xfrom django.contrib import admin
Xfrom bookmarks.models import Bookmark
X
Xclass BookmarkAdmin(admin.ModelAdmin) :
X    def get_fieldsets(self, request, obj=None) :
X        if obj is None :
X            return [ (None, { 'fields': ('url',) }) ]
X        else :
X            return [ (None, { 'fields': ('url', 'added_by')}) ]
X
Xclass BookmarkAdminHack(BookmarkAdmin) :
X       def get_form(self, request, obj=None, **kwargs) :
X               self.fieldsets = self.get_fieldsets(request, obj)
X               return super(BookmarkAdminHack, self).get_form(request, obj, **kwargs)
947cf57c8598ad9386f34efacea96cd9
echo x - bookmarks/models.py
sed 's/^X//' >bookmarks/models.py << '43ade1173a2a323a2b7de45cc78547dd'
Xfrom django.db import models
Xfrom django.contrib.auth.models import User
X
Xclass Bookmark(models.Model) :
X       url = models.URLField(max_length=64)
X       added_by = models.ForeignKey(User)
X
43ade1173a2a323a2b7de45cc78547dd
echo x - bookmarks/tests.py
sed 's/^X//' >bookmarks/tests.py << '7e892ef10c1a1bfd433c456bb538fb06'
X# vim: ts=4 sw=4 et tw=78
Xfrom django.test import TestCase
Xfrom django.http import HttpRequest
Xfrom bookmarks.admin import BookmarkAdmin, BookmarkAdminHack
Xfrom bookmarks.models import Bookmark
Xfrom django.contrib import admin
Xfrom django.contrib.auth.models import User
X
Xclass SimpleTest(TestCase):
X    def do_admin_get_fields_test(self, adm) :
X        request = HttpRequest()
X        request.user = User(username='admin')
X        request.POST = { 'url': u'http://example.org' }
X        request.FILES = {}
X        request.method = 'POST'
X        # This is the same logic used as in ModelAdmin.add_view()
X        ModelForm = adm.get_form(request)
X        form = ModelForm(request.POST, request.FILES)
X        # This is form.is_valid() but split out to show errors more clearly
X        self.assertEqual(form.is_bound, True)
X        form.full_clean()
X        self.assertEqual(form.errors, {})
X
X    def test_admin_get_fields(self) :
X        try :
X            admin.site.unregister(Bookmark)
X        except :
X            pass
X        admin.site.register(Bookmark, BookmarkAdmin)
X        adm = BookmarkAdmin(Bookmark, admin.site)
X        self.do_admin_get_fields_test(adm)
X
X    def test_admin_get_fields_hack(self) :
X        try :
X            admin.site.unregister(Bookmark)
X        except :
X            pass
X        admin.site.register(Bookmark, BookmarkAdminHack)
X        adm = BookmarkAdminHack(Bookmark, admin.site)
X        self.do_admin_get_fields_test(adm)
7e892ef10c1a1bfd433c456bb538fb06
exit