Ticket #15274: tests_for_get_object.patch

tests/regressiontests/generic_views/views.py

@@ -114 +114 @@
     success_url = '/list/authors/'
 
 
+class OneAuthorUpdate(generic.UpdateView):
+
+    def get_object(self):
+        return Author.objects.get(pk=1)
+
+
+class OneAuthorWithRedirectUpdate(generic.UpdateView):
+    success_url = '/list/authors/'
+
+    def get_object(self):
+        return Author.objects.get(pk=1)
+
+
 class SpecializedAuthorUpdate(generic.UpdateView):
     model = Author
     form_class = AuthorForm

tests/regressiontests/generic_views/edit.py

@@ -201 +201 @@
         except ImproperlyConfigured:
             pass
 
+    def test_update_get_object(self):
+        a = Author.objects.create(
+            name='Randall Munroe',
+            slug='randall-munroe',
+        )
+        res = self.client.get('/edit/author/update/')
+        self.assertEqual(res.status_code, 200)
+        self.assertTrue(isinstance(res.context['form'], forms.ModelForm))
+        self.assertEqual(res.context['object'], Author.objects.get(pk=a.pk))
+        self.assertEqual(res.context['author'], Author.objects.get(pk=a.pk))
+        self.assertTemplateUsed(res, 'generic_views/author_form.html')
+
+        # Modification with both POST and PUT (browser compatible)
+        res = self.client.post('/edit/author/update/' % a.pk,
+                        {'name': 'Randall Munroe (xkcd)', 'slug': 'randall-munroe'})
+        self.assertEqual(res.status_code, 302)
+        self.assertRedirects(res, 'http://testserver/list/authors/')
+        self.assertQuerysetEqual(Author.objects.all(), ['<Author: Randall Munroe (xkcd)>'])
+
+    def test_update_get_object_with_redirect(self):
+        a = Author.objects.create(
+            name='Randall Munroe',
+            slug='randall-munroe',
+        )
+        res = self.client.get('/edit/author/update/redirect')
+        self.assertEqual(res.status_code, 200)
+        self.assertTrue(isinstance(res.context['form'], forms.ModelForm))
+        self.assertEqual(res.context['object'], Author.objects.get(pk=a.pk))
+        self.assertEqual(res.context['author'], Author.objects.get(pk=a.pk))
+        self.assertTemplateUsed(res, 'generic_views/author_form.html')
+
+        # Modification with both POST and PUT (browser compatible)
+        res = self.client.post('/edit/author/update/redirect' % a.pk,
+                        {'name': 'Randall Munroe (xkcd)', 'slug': 'randall-munroe'})
+        self.assertEqual(res.status_code, 302)
+        self.assertRedirects(res, 'http://testserver/list/authors/')
+        self.assertQuerysetEqual(Author.objects.all(), ['<Author: Randall Munroe (xkcd)>'])
+
 class DeleteViewTests(TestCase):
     urls = 'regressiontests.generic_views.urls'
 

tests/regressiontests/generic_views/urls.py

@@ -72 +72 @@
         views.NaiveAuthorUpdate.as_view(success_url='/edit/author/%(id)d/update/')),
     (r'^edit/author/(?P<pk>\d+)/update/$',
         views.AuthorUpdate.as_view()),
+    (r'^edit/author/update/$',
+        views.OneAuthorUpdate.as_view()),
+    (r'^edit/author/update/redirect$',
+        views.OneAuthorWithRedirectUpdate.as_view()),
     (r'^edit/author/(?P<pk>\d+)/update/special/$',
         views.SpecializedAuthorUpdate.as_view()),
     (r'^edit/author/(?P<pk>\d+)/delete/naive/$',