PageRenderTime 1141ms CodeModel.GetById 795ms app.highlight 165ms RepoModel.GetById 177ms app.codeStats 0ms

/tests/regressiontests/generic_inline_admin/tests.py

https://code.google.com/p/mango-py/
Python | 216 lines | 182 code | 12 blank | 22 comment | 0 complexity | 76d14895471d2ea0ec7c307e7eb065df MD5 | raw file
  1# coding: utf-8
  2
  3from django.conf import settings
  4from django.contrib.contenttypes.generic import generic_inlineformset_factory
  5from django.test import TestCase
  6
  7# local test models
  8from models import Episode, EpisodeExtra, EpisodeMaxNum, EpisodeExclude, \
  9                   Media, EpisodePermanent, MediaPermanentInline, Category
 10
 11
 12class GenericAdminViewTest(TestCase):
 13    fixtures = ['users.xml']
 14
 15    def setUp(self):
 16        # set TEMPLATE_DEBUG to True to ensure {% include %} will raise
 17        # exceptions since that is how inlines are rendered and #9498 will
 18        # bubble up if it is an issue.
 19        self.original_template_debug = settings.TEMPLATE_DEBUG
 20        settings.TEMPLATE_DEBUG = True
 21        self.client.login(username='super', password='secret')
 22
 23        # Can't load content via a fixture (since the GenericForeignKey
 24        # relies on content type IDs, which will vary depending on what
 25        # other tests have been run), thus we do it here.
 26        e = Episode.objects.create(name='This Week in Django')
 27        self.episode_pk = e.pk
 28        m = Media(content_object=e, url='http://example.com/podcast.mp3')
 29        m.save()
 30        self.mp3_media_pk = m.pk
 31
 32        m = Media(content_object=e, url='http://example.com/logo.png')
 33        m.save()
 34        self.png_media_pk = m.pk
 35
 36    def tearDown(self):
 37        self.client.logout()
 38        settings.TEMPLATE_DEBUG = self.original_template_debug
 39
 40    def testBasicAddGet(self):
 41        """
 42        A smoke test to ensure GET on the add_view works.
 43        """
 44        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/add/')
 45        self.assertEqual(response.status_code, 200)
 46
 47    def testBasicEditGet(self):
 48        """
 49        A smoke test to ensure GET on the change_view works.
 50        """
 51        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/%d/' % self.episode_pk)
 52        self.assertEqual(response.status_code, 200)
 53
 54    def testBasicAddPost(self):
 55        """
 56        A smoke test to ensure POST on add_view works.
 57        """
 58        post_data = {
 59            "name": u"This Week in Django",
 60            # inline data
 61            "generic_inline_admin-media-content_type-object_id-TOTAL_FORMS": u"1",
 62            "generic_inline_admin-media-content_type-object_id-INITIAL_FORMS": u"0",
 63            "generic_inline_admin-media-content_type-object_id-MAX_NUM_FORMS": u"0",
 64        }
 65        response = self.client.post('/generic_inline_admin/admin/generic_inline_admin/episode/add/', post_data)
 66        self.assertEqual(response.status_code, 302) # redirect somewhere
 67
 68    def testBasicEditPost(self):
 69        """
 70        A smoke test to ensure POST on edit_view works.
 71        """
 72        post_data = {
 73            "name": u"This Week in Django",
 74            # inline data
 75            "generic_inline_admin-media-content_type-object_id-TOTAL_FORMS": u"3",
 76            "generic_inline_admin-media-content_type-object_id-INITIAL_FORMS": u"2",
 77            "generic_inline_admin-media-content_type-object_id-MAX_NUM_FORMS": u"0",
 78            "generic_inline_admin-media-content_type-object_id-0-id": u"%d" % self.mp3_media_pk,
 79            "generic_inline_admin-media-content_type-object_id-0-url": u"http://example.com/podcast.mp3",
 80            "generic_inline_admin-media-content_type-object_id-1-id": u"%d" % self.png_media_pk,
 81            "generic_inline_admin-media-content_type-object_id-1-url": u"http://example.com/logo.png",
 82            "generic_inline_admin-media-content_type-object_id-2-id": u"",
 83            "generic_inline_admin-media-content_type-object_id-2-url": u"",
 84        }
 85        url = '/generic_inline_admin/admin/generic_inline_admin/episode/%d/' % self.episode_pk
 86        response = self.client.post(url, post_data)
 87        self.assertEqual(response.status_code, 302) # redirect somewhere
 88
 89    def testGenericInlineFormset(self):
 90        EpisodeMediaFormSet = generic_inlineformset_factory(Media, can_delete=False, extra=3)
 91        e = Episode.objects.get(name='This Week in Django')
 92
 93        # Works with no queryset
 94        formset = EpisodeMediaFormSet(instance=e)
 95        self.assertEqual(len(formset.forms), 5)
 96        self.assertEqual(formset.forms[0].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-0-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-0-url" type="text" name="generic_inline_admin-media-content_type-object_id-0-url" value="http://example.com/podcast.mp3" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-0-id" value="%s" id="id_generic_inline_admin-media-content_type-object_id-0-id" /></p>' % self.mp3_media_pk)
 97        self.assertEqual(formset.forms[1].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-1-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-1-url" type="text" name="generic_inline_admin-media-content_type-object_id-1-url" value="http://example.com/logo.png" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-1-id" value="%s" id="id_generic_inline_admin-media-content_type-object_id-1-id" /></p>' % self.png_media_pk)
 98        self.assertEqual(formset.forms[2].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-2-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-2-url" type="text" name="generic_inline_admin-media-content_type-object_id-2-url" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-2-id" id="id_generic_inline_admin-media-content_type-object_id-2-id" /></p>')
 99
100        # A queryset can be used to alter display ordering
101        formset = EpisodeMediaFormSet(instance=e, queryset=Media.objects.order_by('url'))
102        self.assertEqual(len(formset.forms), 5)
103        self.assertEqual(formset.forms[0].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-0-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-0-url" type="text" name="generic_inline_admin-media-content_type-object_id-0-url" value="http://example.com/logo.png" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-0-id" value="%s" id="id_generic_inline_admin-media-content_type-object_id-0-id" /></p>' % self.png_media_pk)
104        self.assertEqual(formset.forms[1].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-1-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-1-url" type="text" name="generic_inline_admin-media-content_type-object_id-1-url" value="http://example.com/podcast.mp3" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-1-id" value="%s" id="id_generic_inline_admin-media-content_type-object_id-1-id" /></p>' % self.mp3_media_pk)
105        self.assertEqual(formset.forms[2].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-2-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-2-url" type="text" name="generic_inline_admin-media-content_type-object_id-2-url" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-2-id" id="id_generic_inline_admin-media-content_type-object_id-2-id" /></p>')
106
107
108        # Works with a queryset that omits items
109        formset = EpisodeMediaFormSet(instance=e, queryset=Media.objects.filter(url__endswith=".png"))
110        self.assertEqual(len(formset.forms), 4)
111        self.assertEqual(formset.forms[0].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-0-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-0-url" type="text" name="generic_inline_admin-media-content_type-object_id-0-url" value="http://example.com/logo.png" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-0-id" value="%s" id="id_generic_inline_admin-media-content_type-object_id-0-id" /></p>' % self.png_media_pk)
112        self.assertEqual(formset.forms[1].as_p(), '<p><label for="id_generic_inline_admin-media-content_type-object_id-1-url">Url:</label> <input id="id_generic_inline_admin-media-content_type-object_id-1-url" type="text" name="generic_inline_admin-media-content_type-object_id-1-url" maxlength="200" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-1-id" id="id_generic_inline_admin-media-content_type-object_id-1-id" /></p>')
113
114    def testGenericInlineFormsetFactory(self):
115        # Regression test for #10522.
116        inline_formset = generic_inlineformset_factory(Media,
117            exclude=('url',))
118
119        # Regression test for #12340.
120        e = Episode.objects.get(name='This Week in Django')
121        formset = inline_formset(instance=e)
122        self.assertTrue(formset.get_queryset().ordered)
123
124class GenericInlineAdminParametersTest(TestCase):
125    fixtures = ['users.xml']
126
127    def setUp(self):
128        self.client.login(username='super', password='secret')
129
130    def tearDown(self):
131        self.client.logout()
132
133    def _create_object(self, model):
134        """
135        Create a model with an attached Media object via GFK. We can't
136        load content via a fixture (since the GenericForeignKey relies on
137        content type IDs, which will vary depending on what other tests
138        have been run), thus we do it here.
139        """
140        e = model.objects.create(name='This Week in Django')
141        Media.objects.create(content_object=e, url='http://example.com/podcast.mp3')
142        return e
143
144    def testNoParam(self):
145        """
146        With one initial form, extra (default) at 3, there should be 4 forms.
147        """
148        e = self._create_object(Episode)
149        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/%s/' % e.pk)
150        formset = response.context['inline_admin_formsets'][0].formset
151        self.assertEqual(formset.total_form_count(), 4)
152        self.assertEqual(formset.initial_form_count(), 1)
153
154    def testExtraParam(self):
155        """
156        With extra=0, there should be one form.
157        """
158        e = self._create_object(EpisodeExtra)
159        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodeextra/%s/' % e.pk)
160        formset = response.context['inline_admin_formsets'][0].formset
161        self.assertEqual(formset.total_form_count(), 1)
162        self.assertEqual(formset.initial_form_count(), 1)
163
164    def testMaxNumParam(self):
165        """
166        With extra=5 and max_num=2, there should be only 2 forms.
167        """
168        e = self._create_object(EpisodeMaxNum)
169        inline_form_data = '<input type="hidden" name="generic_inline_admin-media-content_type-object_id-TOTAL_FORMS" value="2" id="id_generic_inline_admin-media-content_type-object_id-TOTAL_FORMS" /><input type="hidden" name="generic_inline_admin-media-content_type-object_id-INITIAL_FORMS" value="1" id="id_generic_inline_admin-media-content_type-object_id-INITIAL_FORMS" />'
170        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodemaxnum/%s/' % e.pk)
171        formset = response.context['inline_admin_formsets'][0].formset
172        self.assertEqual(formset.total_form_count(), 2)
173        self.assertEqual(formset.initial_form_count(), 1)
174
175    def testExcludeParam(self):
176        """
177        Generic inline formsets should respect include.
178        """
179        e = self._create_object(EpisodeExclude)
180        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodeexclude/%s/' % e.pk)
181        formset = response.context['inline_admin_formsets'][0].formset
182        self.assertFalse('url' in formset.forms[0], 'The formset has excluded "url" field.')
183
184class GenericInlineAdminWithUniqueTogetherTest(TestCase):
185    fixtures = ['users.xml']
186
187    def setUp(self):
188        self.client.login(username='super', password='secret')
189
190    def tearDown(self):
191        self.client.logout()
192
193    def testAdd(self):
194        category_id = Category.objects.create(name='male').pk
195        post_data = {
196            "name": u"John Doe",
197            # inline data
198            "generic_inline_admin-phonenumber-content_type-object_id-TOTAL_FORMS": u"1",
199            "generic_inline_admin-phonenumber-content_type-object_id-INITIAL_FORMS": u"0",
200            "generic_inline_admin-phonenumber-content_type-object_id-MAX_NUM_FORMS": u"0",
201            "generic_inline_admin-phonenumber-content_type-object_id-0-id": "",
202            "generic_inline_admin-phonenumber-content_type-object_id-0-phone_number": "555-555-5555",
203            "generic_inline_admin-phonenumber-content_type-object_id-0-category": "%s" % category_id,
204        }
205        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/contact/add/')
206        self.assertEqual(response.status_code, 200)
207        response = self.client.post('/generic_inline_admin/admin/generic_inline_admin/contact/add/', post_data)
208        self.assertEqual(response.status_code, 302) # redirect somewhere
209
210class NoInlineDeletionTest(TestCase):
211    def test_no_deletion(self):
212        fake_site = object()
213        inline = MediaPermanentInline(EpisodePermanent, fake_site)
214        fake_request = object()
215        formset = inline.get_formset(fake_request)
216        self.assertFalse(formset.can_delete)