# coding: utf-8

from django.conf import settings

from django.contrib.contenttypes.generic import generic_inlineformset_factory

from django.test import TestCase

# local test models

from models import Episode, EpisodeExtra, EpisodeMaxNum, EpisodeExclude, \

                   Media, EpisodePermanent, MediaPermanentInline

class GenericAdminViewTest(TestCase):

    fixtures = ['users.xml']

    def setUp(self):

        # set TEMPLATE_DEBUG to True to ensure {% include %} will raise

        # exceptions since that is how inlines are rendered and #9498 will

        # bubble up if it is an issue.

        self.original_template_debug = settings.TEMPLATE_DEBUG

        settings.TEMPLATE_DEBUG = True

        self.client.login(username='super', password='secret')

        # Can't load content via a fixture (since the GenericForeignKey

        # relies on content type IDs, which will vary depending on what

        # other tests have been run), thus we do it here.

        e = Episode.objects.create(name='This Week in Django')

        self.episode_pk = e.pk

        m = Media(content_object=e, url='http://example.com/podcast.mp3')

        m.save()

        self.mp3_media_pk = m.pk

        m = Media(content_object=e, url='http://example.com/logo.png')

        m.save()

        self.png_media_pk = m.pk

    def tearDown(self):

        self.client.logout()

        settings.TEMPLATE_DEBUG = self.original_template_debug

    def testBasicAddGet(self):

        """

        A smoke test to ensure GET on the add_view works.

        """

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/add/')

        self.assertEqual(response.status_code, 200)

    def testBasicEditGet(self):

        """

        A smoke test to ensure GET on the change_view works.

        """

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/%d/' % self.episode_pk)

        self.assertEqual(response.status_code, 200)

    def testBasicAddPost(self):

        """

        A smoke test to ensure POST on add_view works.

        """

        post_data = {

            "name": u"This Week in Django",

            # inline data

            "generic_inline_admin-media-content_type-object_id-TOTAL_FORMS": u"1",

            "generic_inline_admin-media-content_type-object_id-INITIAL_FORMS": u"0",

            "generic_inline_admin-media-content_type-object_id-MAX_NUM_FORMS": u"0",

        }

        response = self.client.post('/generic_inline_admin/admin/generic_inline_admin/episode/add/', post_data)

        self.assertEqual(response.status_code, 302) # redirect somewhere

    def testBasicEditPost(self):

        """

        A smoke test to ensure POST on edit_view works.

        """

        post_data = {

            "name": u"This Week in Django",

            # inline data

            "generic_inline_admin-media-content_type-object_id-TOTAL_FORMS": u"3",

            "generic_inline_admin-media-content_type-object_id-INITIAL_FORMS": u"2",

            "generic_inline_admin-media-content_type-object_id-MAX_NUM_FORMS": u"0",

            "generic_inline_admin-media-content_type-object_id-0-id": u"%d" % self.mp3_media_pk,

            "generic_inline_admin-media-content_type-object_id-0-url": u"http://example.com/podcast.mp3",

            "generic_inline_admin-media-content_type-object_id-1-id": u"%d" % self.png_media_pk,

            "generic_inline_admin-media-content_type-object_id-1-url": u"http://example.com/logo.png",

            "generic_inline_admin-media-content_type-object_id-2-id": u"",

            "generic_inline_admin-media-content_type-object_id-2-url": u"",

        }

        url = '/generic_inline_admin/admin/generic_inline_admin/episode/%d/' % self.episode_pk

        response = self.client.post(url, post_data)

        self.assertEqual(response.status_code, 302) # redirect somewhere

    def testGenericInlineFormset(self):

        EpisodeMediaFormSet = generic_inlineformset_factory(Media, can_delete=False, extra=3)

        e = Episode.objects.get(name='This Week in Django')

        # Works with no queryset

        formset = EpisodeMediaFormSet(instance=e)

        self.assertEquals(len(formset.forms), 5)

        self.assertEquals(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)

        self.assertEquals(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)

        self.assertEquals(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>')

        # A queryset can be used to alter display ordering

        formset = EpisodeMediaFormSet(instance=e, queryset=Media.objects.order_by('url'))

        self.assertEquals(len(formset.forms), 5)

        self.assertEquals(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)

        self.assertEquals(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)

        self.assertEquals(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>')

        # Works with a queryset that omits items

        formset = EpisodeMediaFormSet(instance=e, queryset=Media.objects.filter(url__endswith=".png"))

        self.assertEquals(len(formset.forms), 4)

        self.assertEquals(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)

        self.assertEquals(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>')

    def testGenericInlineFormsetFactory(self):

        # Regression test for #10522.

        inline_formset = generic_inlineformset_factory(Media,

            exclude=('url',))

        # Regression test for #12340.

        e = Episode.objects.get(name='This Week in Django')

        formset = inline_formset(instance=e)

        self.assertTrue(formset.get_queryset().ordered)

class GenericInlineAdminParametersTest(TestCase):

    fixtures = ['users.xml']

    def setUp(self):

        self.client.login(username='super', password='secret')

    def tearDown(self):

        self.client.logout()

    def _create_object(self, model):

        """

        Create a model with an attached Media object via GFK. We can't

        load content via a fixture (since the GenericForeignKey relies on

        content type IDs, which will vary depending on what other tests

        have been run), thus we do it here.

        """

        e = model.objects.create(name='This Week in Django')

        Media.objects.create(content_object=e, url='http://example.com/podcast.mp3')

        return e

    def testNoParam(self):

        """

        With one initial form, extra (default) at 3, there should be 4 forms.

        """

        e = self._create_object(Episode)

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episode/%s/' % e.pk)

        formset = response.context['inline_admin_formsets'][0].formset

        self.assertEqual(formset.total_form_count(), 4)

        self.assertEqual(formset.initial_form_count(), 1)

    def testExtraParam(self):

        """

        With extra=0, there should be one form.

        """

        e = self._create_object(EpisodeExtra)

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodeextra/%s/' % e.pk)

        formset = response.context['inline_admin_formsets'][0].formset

        self.assertEqual(formset.total_form_count(), 1)

        self.assertEqual(formset.initial_form_count(), 1)

    def testMaxNumParam(self):

        """

        With extra=5 and max_num=2, there should be only 2 forms.

        """

        e = self._create_object(EpisodeMaxNum)

        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" />'

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodemaxnum/%s/' % e.pk)

        formset = response.context['inline_admin_formsets'][0].formset

        self.assertEqual(formset.total_form_count(), 2)

        self.assertEqual(formset.initial_form_count(), 1)

    def testExcludeParam(self):

        """

        Generic inline formsets should respect include.

        """

        e = self._create_object(EpisodeExclude)

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/episodeexclude/%s/' % e.pk)

        formset = response.context['inline_admin_formsets'][0].formset

        self.assertFalse('url' in formset.forms[0], 'The formset has excluded "url" field.')

class GenericInlineAdminWithUniqueTogetherTest(TestCase):

    fixtures = ['users.xml']

    def setUp(self):

        self.client.login(username='super', password='secret')

    def tearDown(self):

        self.client.logout()

    def testAdd(self):

        post_data = {

            "name": u"John Doe",

            # inline data

            "generic_inline_admin-phonenumber-content_type-object_id-TOTAL_FORMS": u"1",

            "generic_inline_admin-phonenumber-content_type-object_id-INITIAL_FORMS": u"0",

            "generic_inline_admin-phonenumber-content_type-object_id-MAX_NUM_FORMS": u"0",

            "generic_inline_admin-phonenumber-content_type-object_id-0-id": "",

            "generic_inline_admin-phonenumber-content_type-object_id-0-phone_number": "555-555-5555",

        }

        response = self.client.get('/generic_inline_admin/admin/generic_inline_admin/contact/add/')

        self.assertEqual(response.status_code, 200)

        response = self.client.post('/generic_inline_admin/admin/generic_inline_admin/contact/add/', post_data)

        self.assertEqual(response.status_code, 302) # redirect somewhere

class NoInlineDeletionTest(TestCase):

    def test_no_deletion(self):

        fake_site = object()

        inline = MediaPermanentInline(EpisodePermanent, fake_site)

        fake_request = object()

        formset = inline.get_formset(fake_request)

        self.assertFalse(formset.can_delete)