Ticket #12241: preserve_query_string_with_tests.diff

django/contrib/admin/options.py

@@ -634 +634 @@
             "admin/change_form.html"
         ], context, context_instance=context_instance)
 
+    def handle_continue_redirect(self, request, url_continue, msg):
+        self.message_user(request, msg + ' ' + _("You may edit it again below."))
+        if request.POST.has_key("_popup"):
+            url_continue += ("?_popup=1" + request.META.get('QUERY_STRING'))
+        else:
+            url_continue += ("?" + request.META.get('QUERY_STRING'))
+        return url_continue
+
     def response_add(self, request, obj, post_url_continue='../%s/'):
         """
         Determines the HttpResponse for the add_view stage.
@@ -645 +653 @@
         # Here, we distinguish between different save types by checking for
         # the presence of keys in request.POST.
         if request.POST.has_key("_continue"):
-            self.message_user(request, msg + ' ' + _("You may edit it again below."))
-            if request.POST.has_key("_popup"):
-                post_url_continue += "?_popup=1"
-            return HttpResponseRedirect(post_url_continue % pk_value)
+            url = self.handle_continue_redirect(request, post_url_continue, msg)
+            return HttpResponseRedirect(url % pk_value)
 
         if request.POST.has_key("_popup"):
             return HttpResponse('<script type="text/javascript">opener.dismissAddAnotherPopup(window, "%s", "%s");</script>' % \
@@ -678 +684 @@
 
         msg = _('The %(name)s "%(obj)s" was changed successfully.') % {'name': force_unicode(opts.verbose_name), 'obj': force_unicode(obj)}
         if request.POST.has_key("_continue"):
-            self.message_user(request, msg + ' ' + _("You may edit it again below."))
-            if request.REQUEST.has_key('_popup'):
-                return HttpResponseRedirect(request.path + "?_popup=1")
-            else:
-                return HttpResponseRedirect(request.path)
+            url = self.handle_continue_redirect(request, request.path, msg)
+            return HttpResponseRedirect(url)
         elif request.POST.has_key("_saveasnew"):
             msg = _('The %(name)s "%(obj)s" was added successfully. You may edit it again below.') % {'name': force_unicode(opts.verbose_name), 'obj': obj}
             self.message_user(request, msg)

tests/regressiontests/admin_views/tests.py

@@ -1513 +1513 @@
             else:
                 self.assertEqual(response.status_code, 404)
 
+class AdminResponseChangeTest(TestCase):
+    fixtures = ['admin-views-users.xml']
+
+    def setUp(self):
+        self.client.login(username='super', password='secret')
+        self.data = {'title':"I Could Go Anywhere", 'content':"Versatile", 'date':datetime.datetime.now()}
+        article = Article.objects.create(**self.data)
+        self.pk = article.id
+
+    def test_response_change(self):
+        """Regression test for #12241: ModelAdmin.response_change and
+        ModelAdmin.response_add should preserve the request
+        query string after the redirection if the "save and continue" button
+        is used within the admin."""
+        self.data['_continue'] = True
+        self.data['date_0'] = u"2008-03-18"
+        self.data['date_1'] = u"11:54:58"
+        response = self.client.post('/test_admin/admin/admin_views/article/%d/?test=1' % self.pk, self.data)
+        self.assertEqual(response.status_code, 302)
+        self.assertTrue("?test=1" in response['Location'])
+        response = self.client.post('/test_admin/admin/admin_views/article/add/?test=1', self.data)
+        self.assertEqual(response.status_code, 302)
+        self.assertTrue("?test=1" in response['Location'])
+
+
 class AdminInlineFileUploadTest(TestCase):
     fixtures = ['admin-views-users.xml', 'admin-views-actions.xml']
     urlbit = 'admin'