import datetime

import re

from datetime import date

from decimal import Decimal

from django import forms

from django.db import models

from django.forms.models import (_get_foreign_key, inlineformset_factory,

    modelformset_factory, modelformset_factory)

from django.test import TestCase

from modeltests.model_formsets.models import (

    Author, BetterAuthor, Book, BookWithCustomPK, Editor,

    BookWithOptionalAltEditor, AlternateBook, AuthorMeeting, CustomPrimaryKey,

    Place, Owner, Location, OwnerProfile, Restaurant, Product, Price,

    MexicanRestaurant, ClassyMexicanRestaurant, Repository, Revision,

    Person, Membership, Team, Player, Poet, Poem, Post)

class DeletionTests(TestCase):

    def test_deletion(self):

        PoetFormSet = modelformset_factory(Poet, can_delete=True)

        poet = Poet.objects.create(name='test')

        data = {

            'form-TOTAL_FORMS': u'1',

            'form-INITIAL_FORMS': u'1',

            'form-MAX_NUM_FORMS': u'0',

            'form-0-id': str(poet.pk),

            'form-0-name': u'test',

            'form-0-DELETE': u'on',

        }

        formset = PoetFormSet(data, queryset=Poet.objects.all())

        formset.save()

        self.assertTrue(formset.is_valid())

        self.assertEqual(Poet.objects.count(), 0)

    def test_add_form_deletion_when_invalid(self):

        """

        Make sure that an add form that is filled out, but marked for deletion

        doesn't cause validation errors.

        """

        PoetFormSet = modelformset_factory(Poet, can_delete=True)

        data = {

            'form-TOTAL_FORMS': u'1',

            'form-INITIAL_FORMS': u'0',

            'form-MAX_NUM_FORMS': u'0',

            'form-0-id': u'',

            'form-0-name': u'x' * 1000,

        }

        formset = PoetFormSet(data, queryset=Poet.objects.all())

        # Make sure this form doesn't pass validation.

        self.assertEqual(formset.is_valid(), False)

        self.assertEqual(Poet.objects.count(), 0)

        # Then make sure that it *does* pass validation and delete the object,

        # even though the data isn't actually valid.

        data['form-0-DELETE'] = 'on'

        formset = PoetFormSet(data, queryset=Poet.objects.all())

        self.assertEqual(formset.is_valid(), True)

        formset.save()

        self.assertEqual(Poet.objects.count(), 0)

    def test_change_form_deletion_when_invalid(self):

        """

        Make sure that an add form that is filled out, but marked for deletion

        doesn't cause validation errors.

        """

        PoetFormSet = modelformset_factory(Poet, can_delete=True)

        poet = Poet.objects.create(name='test')

        data = {

            'form-TOTAL_FORMS': u'1',

            'form-INITIAL_FORMS': u'1',

            'form-MAX_NUM_FORMS': u'0',

            'form-0-id': u'1',

            'form-0-name': u'x' * 1000,

        }

        formset = PoetFormSet(data, queryset=Poet.objects.all())

        # Make sure this form doesn't pass validation.

        self.assertEqual(formset.is_valid(), False)

        self.assertEqual(Poet.objects.count(), 1)

        # Then make sure that it *does* pass validation and delete the object,

        # even though the data isn't actually valid.

        data['form-0-DELETE'] = 'on'

        formset = PoetFormSet(data, queryset=Poet.objects.all())

        self.assertEqual(formset.is_valid(), True)

        formset.save()

        self.assertEqual(Poet.objects.count(), 0)

class ModelFormsetTest(TestCase):

    def test_simple_save(self):

        qs = Author.objects.all()

        AuthorFormSet = modelformset_factory(Author, extra=3)

        formset = AuthorFormSet(queryset=qs)

        self.assertEqual(len(formset.forms), 3)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_form-0-name">Name:</label> <input id="id_form-0-name" type="text" name="form-0-name" maxlength="100" /><input type="hidden" name="form-0-id" id="id_form-0-id" /></p>')

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_form-1-name">Name:</label> <input id="id_form-1-name" type="text" name="form-1-name" maxlength="100" /><input type="hidden" name="form-1-id" id="id_form-1-id" /></p>')

        self.assertEqual(formset.forms[2].as_p(),

            '<p><label for="id_form-2-name">Name:</label> <input id="id_form-2-name" type="text" name="form-2-name" maxlength="100" /><input type="hidden" name="form-2-id" id="id_form-2-id" /></p>')

        data = {

            'form-TOTAL_FORMS': '3', # the number of forms rendered

            'form-INITIAL_FORMS': '0', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-name': 'Charles Baudelaire',

            'form-1-name': 'Arthur Rimbaud',

            'form-2-name': '',

        }

        formset = AuthorFormSet(data=data, queryset=qs)

        self.assertTrue(formset.is_valid())

        saved = formset.save()

        self.assertEqual(len(saved), 2)

        author1, author2 = saved

        self.assertEqual(author1, Author.objects.get(name='Charles Baudelaire'))

        self.assertEqual(author2, Author.objects.get(name='Arthur Rimbaud'))

        authors = list(Author.objects.order_by('name'))

        self.assertEqual(authors, [author2, author1])

        # Gah! We forgot Paul Verlaine. Let's create a formset to edit the

        # existing authors with an extra form to add him. We *could* pass in a

        # queryset to restrict the Author objects we edit, but in this case

        # we'll use it to display them in alphabetical order by name.

        qs = Author.objects.order_by('name')

        AuthorFormSet = modelformset_factory(Author, extra=1, can_delete=False)

        formset = AuthorFormSet(queryset=qs)

        self.assertEqual(len(formset.forms), 3)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_form-0-name">Name:</label> <input id="id_form-0-name" type="text" name="form-0-name" value="Arthur Rimbaud" maxlength="100" /><input type="hidden" name="form-0-id" value="%d" id="id_form-0-id" /></p>' % author2.id)

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_form-1-name">Name:</label> <input id="id_form-1-name" type="text" name="form-1-name" value="Charles Baudelaire" maxlength="100" /><input type="hidden" name="form-1-id" value="%d" id="id_form-1-id" /></p>' % author1.id)

        self.assertEqual(formset.forms[2].as_p(),

            '<p><label for="id_form-2-name">Name:</label> <input id="id_form-2-name" type="text" name="form-2-name" maxlength="100" /><input type="hidden" name="form-2-id" id="id_form-2-id" /></p>')

        data = {

            'form-TOTAL_FORMS': '3', # the number of forms rendered

            'form-INITIAL_FORMS': '2', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-id': str(author2.id),

            'form-0-name': 'Arthur Rimbaud',

            'form-1-id': str(author1.id),

            'form-1-name': 'Charles Baudelaire',

            'form-2-name': 'Paul Verlaine',

        }

        formset = AuthorFormSet(data=data, queryset=qs)

        self.assertTrue(formset.is_valid())

        # Only changed or new objects are returned from formset.save()

        saved = formset.save()

        self.assertEqual(len(saved), 1)

        author3 = saved[0]

        self.assertEqual(author3, Author.objects.get(name='Paul Verlaine'))

        authors = list(Author.objects.order_by('name'))

        self.assertEqual(authors, [author2, author1, author3])

        # This probably shouldn't happen, but it will. If an add form was

        # marked for deletion, make sure we don't save that form.

        qs = Author.objects.order_by('name')

        AuthorFormSet = modelformset_factory(Author, extra=1, can_delete=True)

        formset = AuthorFormSet(queryset=qs)

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

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_form-0-name">Name:</label> <input id="id_form-0-name" type="text" name="form-0-name" value="Arthur Rimbaud" maxlength="100" /></p>\n'

            '<p><label for="id_form-0-DELETE">Delete:</label> <input type="checkbox" name="form-0-DELETE" id="id_form-0-DELETE" /><input type="hidden" name="form-0-id" value="%d" id="id_form-0-id" /></p>' % author2.id)

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_form-1-name">Name:</label> <input id="id_form-1-name" type="text" name="form-1-name" value="Charles Baudelaire" maxlength="100" /></p>\n'

            '<p><label for="id_form-1-DELETE">Delete:</label> <input type="checkbox" name="form-1-DELETE" id="id_form-1-DELETE" /><input type="hidden" name="form-1-id" value="%d" id="id_form-1-id" /></p>' % author1.id)

        self.assertEqual(formset.forms[2].as_p(),

            '<p><label for="id_form-2-name">Name:</label> <input id="id_form-2-name" type="text" name="form-2-name" value="Paul Verlaine" maxlength="100" /></p>\n'

            '<p><label for="id_form-2-DELETE">Delete:</label> <input type="checkbox" name="form-2-DELETE" id="id_form-2-DELETE" /><input type="hidden" name="form-2-id" value="%d" id="id_form-2-id" /></p>' % author3.id)

        self.assertEqual(formset.forms[3].as_p(),

            '<p><label for="id_form-3-name">Name:</label> <input id="id_form-3-name" type="text" name="form-3-name" maxlength="100" /></p>\n'

            '<p><label for="id_form-3-DELETE">Delete:</label> <input type="checkbox" name="form-3-DELETE" id="id_form-3-DELETE" /><input type="hidden" name="form-3-id" id="id_form-3-id" /></p>')

        data = {

            'form-TOTAL_FORMS': '4', # the number of forms rendered

            'form-INITIAL_FORMS': '3', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-id': str(author2.id),

            'form-0-name': 'Arthur Rimbaud',

            'form-1-id': str(author1.id),

            'form-1-name': 'Charles Baudelaire',

            'form-2-id': str(author3.id),

            'form-2-name': 'Paul Verlaine',

            'form-3-name': 'Walt Whitman',

            'form-3-DELETE': 'on',

        }

        formset = AuthorFormSet(data=data, queryset=qs)

        self.assertTrue(formset.is_valid())

        # No objects were changed or saved so nothing will come back.

        self.assertEqual(formset.save(), [])

        authors = list(Author.objects.order_by('name'))

        self.assertEqual(authors, [author2, author1, author3])

        # Let's edit a record to ensure save only returns that one record.

        data = {

            'form-TOTAL_FORMS': '4', # the number of forms rendered

            'form-INITIAL_FORMS': '3', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-id': str(author2.id),

            'form-0-name': 'Walt Whitman',

            'form-1-id': str(author1.id),

            'form-1-name': 'Charles Baudelaire',

            'form-2-id': str(author3.id),

            'form-2-name': 'Paul Verlaine',

            'form-3-name': '',

            'form-3-DELETE': '',

        }

        formset = AuthorFormSet(data=data, queryset=qs)

        self.assertTrue(formset.is_valid())

        # One record has changed.

        saved = formset.save()

        self.assertEqual(len(saved), 1)

        self.assertEqual(saved[0], Author.objects.get(name='Walt Whitman'))

    def test_commit_false(self):

        # Test the behavior of commit=False and save_m2m

        author1 = Author.objects.create(name='Charles Baudelaire')

        author2 = Author.objects.create(name='Paul Verlaine')

        author3 = Author.objects.create(name='Walt Whitman')

        meeting = AuthorMeeting.objects.create(created=date.today())

        meeting.authors = Author.objects.all()

        # create an Author instance to add to the meeting.

        author4 = Author.objects.create(name=u'John Steinbeck')

        AuthorMeetingFormSet = modelformset_factory(AuthorMeeting, extra=1, can_delete=True)

        data = {

            'form-TOTAL_FORMS': '2', # the number of forms rendered

            'form-INITIAL_FORMS': '1', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-id': '1',

            'form-0-name': '2nd Tuesday of the Week Meeting',

            'form-0-authors': [2, 1, 3, 4],

            'form-1-name': '',

            'form-1-authors': '',

            'form-1-DELETE': '',

        }

        formset = AuthorMeetingFormSet(data=data, queryset=AuthorMeeting.objects.all())

        self.assertTrue(formset.is_valid())

        instances = formset.save(commit=False)

        for instance in instances:

            instance.created = date.today()

            instance.save()

        formset.save_m2m()

        self.assertQuerysetEqual(instances[0].authors.all(), [

            '<Author: Charles Baudelaire>',

            '<Author: John Steinbeck>',

            '<Author: Paul Verlaine>',

            '<Author: Walt Whitman>',

        ])

    def test_max_num(self):

        # Test the behavior of max_num with model formsets. It should allow

        # all existing related objects/inlines for a given object to be

        # displayed, but not allow the creation of new inlines beyond max_num.

        author1 = Author.objects.create(name='Charles Baudelaire')

        author2 = Author.objects.create(name='Paul Verlaine')

        author3 = Author.objects.create(name='Walt Whitman')

        qs = Author.objects.order_by('name')

        AuthorFormSet = modelformset_factory(Author, max_num=None, extra=3)

        formset = AuthorFormSet(queryset=qs)

        self.assertEqual(len(formset.forms), 6)

        self.assertEqual(len(formset.extra_forms), 3)

        AuthorFormSet = modelformset_factory(Author, max_num=4, extra=3)

        formset = AuthorFormSet(queryset=qs)

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

        self.assertEqual(len(formset.extra_forms), 1)

        AuthorFormSet = modelformset_factory(Author, max_num=0, extra=3)

        formset = AuthorFormSet(queryset=qs)

        self.assertEqual(len(formset.forms), 3)

        self.assertEqual(len(formset.extra_forms), 0)

        AuthorFormSet = modelformset_factory(Author, max_num=None)

        formset = AuthorFormSet(queryset=qs)

        self.assertQuerysetEqual(formset.get_queryset(), [

            '<Author: Charles Baudelaire>',

            '<Author: Paul Verlaine>',

            '<Author: Walt Whitman>',

        ])

        AuthorFormSet = modelformset_factory(Author, max_num=0)

        formset = AuthorFormSet(queryset=qs)

        self.assertQuerysetEqual(formset.get_queryset(), [

            '<Author: Charles Baudelaire>',

            '<Author: Paul Verlaine>',

            '<Author: Walt Whitman>',

        ])

        AuthorFormSet = modelformset_factory(Author, max_num=4)

        formset = AuthorFormSet(queryset=qs)

        self.assertQuerysetEqual(formset.get_queryset(), [

            '<Author: Charles Baudelaire>',

            '<Author: Paul Verlaine>',

            '<Author: Walt Whitman>',

        ])

    def test_custom_save_method(self):

        class PoetForm(forms.ModelForm):

            def save(self, commit=True):

                # change the name to "Vladimir Mayakovsky" just to be a jerk.

                author = super(PoetForm, self).save(commit=False)

                author.name = u"Vladimir Mayakovsky"

                if commit:

                    author.save()

                return author

        PoetFormSet = modelformset_factory(Poet, form=PoetForm)

        data = {

            'form-TOTAL_FORMS': '3', # the number of forms rendered

            'form-INITIAL_FORMS': '0', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-name': 'Walt Whitman',

            'form-1-name': 'Charles Baudelaire',

            'form-2-name': '',

        }

        qs = Poet.objects.all()

        formset = PoetFormSet(data=data, queryset=qs)

        self.assertTrue(formset.is_valid())

        poets = formset.save()

        self.assertEqual(len(poets), 2)

        poet1, poet2 = poets

        self.assertEqual(poet1.name, 'Vladimir Mayakovsky')

        self.assertEqual(poet2.name, 'Vladimir Mayakovsky')

    def test_model_inheritance(self):

        BetterAuthorFormSet = modelformset_factory(BetterAuthor)

        formset = BetterAuthorFormSet()

        self.assertEqual(len(formset.forms), 1)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_form-0-name">Name:</label> <input id="id_form-0-name" type="text" name="form-0-name" maxlength="100" /></p>\n'

            '<p><label for="id_form-0-write_speed">Write speed:</label> <input type="text" name="form-0-write_speed" id="id_form-0-write_speed" /><input type="hidden" name="form-0-author_ptr" id="id_form-0-author_ptr" /></p>')

        data = {

            'form-TOTAL_FORMS': '1', # the number of forms rendered

            'form-INITIAL_FORMS': '0', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-author_ptr': '',

            'form-0-name': 'Ernest Hemingway',

            'form-0-write_speed': '10',

        }

        formset = BetterAuthorFormSet(data)

        self.assertTrue(formset.is_valid())

        saved = formset.save()

        self.assertEqual(len(saved), 1)

        author1, = saved

        self.assertEqual(author1, BetterAuthor.objects.get(name='Ernest Hemingway'))

        hemingway_id = BetterAuthor.objects.get(name="Ernest Hemingway").pk

        formset = BetterAuthorFormSet()

        self.assertEqual(len(formset.forms), 2)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_form-0-name">Name:</label> <input id="id_form-0-name" type="text" name="form-0-name" value="Ernest Hemingway" maxlength="100" /></p>\n'

            '<p><label for="id_form-0-write_speed">Write speed:</label> <input type="text" name="form-0-write_speed" value="10" id="id_form-0-write_speed" /><input type="hidden" name="form-0-author_ptr" value="%d" id="id_form-0-author_ptr" /></p>' % hemingway_id)

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_form-1-name">Name:</label> <input id="id_form-1-name" type="text" name="form-1-name" maxlength="100" /></p>\n'

            '<p><label for="id_form-1-write_speed">Write speed:</label> <input type="text" name="form-1-write_speed" id="id_form-1-write_speed" /><input type="hidden" name="form-1-author_ptr" id="id_form-1-author_ptr" /></p>')

        data = {

            'form-TOTAL_FORMS': '2', # the number of forms rendered

            'form-INITIAL_FORMS': '1', # the number of forms with initial data

            'form-MAX_NUM_FORMS': '', # the max number of forms

            'form-0-author_ptr': hemingway_id,

            'form-0-name': 'Ernest Hemingway',

            'form-0-write_speed': '10',

            'form-1-author_ptr': '',

            'form-1-name': '',

            'form-1-write_speed': '',

        }

        formset = BetterAuthorFormSet(data)

        self.assertTrue(formset.is_valid())

        self.assertEqual(formset.save(), [])

    def test_inline_formsets(self):

        # We can also create a formset that is tied to a parent model. This is

        # how the admin system's edit inline functionality works.

        AuthorBooksFormSet = inlineformset_factory(Author, Book, can_delete=False, extra=3)

        author = Author.objects.create(name='Charles Baudelaire')

        formset = AuthorBooksFormSet(instance=author)

        self.assertEqual(len(formset.forms), 3)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_book_set-0-title">Title:</label> <input id="id_book_set-0-title" type="text" name="book_set-0-title" maxlength="100" /><input type="hidden" name="book_set-0-author" value="%d" id="id_book_set-0-author" /><input type="hidden" name="book_set-0-id" id="id_book_set-0-id" /></p>'  % author.id)

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_book_set-1-title">Title:</label> <input id="id_book_set-1-title" type="text" name="book_set-1-title" maxlength="100" /><input type="hidden" name="book_set-1-author" value="%d" id="id_book_set-1-author" /><input type="hidden" name="book_set-1-id" id="id_book_set-1-id" /></p>' % author.id)

        self.assertEqual(formset.forms[2].as_p(),

            '<p><label for="id_book_set-2-title">Title:</label> <input id="id_book_set-2-title" type="text" name="book_set-2-title" maxlength="100" /><input type="hidden" name="book_set-2-author" value="%d" id="id_book_set-2-author" /><input type="hidden" name="book_set-2-id" id="id_book_set-2-id" /></p>' % author.id)

        data = {

            'book_set-TOTAL_FORMS': '3', # the number of forms rendered

            'book_set-INITIAL_FORMS': '0', # the number of forms with initial data

            'book_set-MAX_NUM_FORMS': '', # the max number of forms

            'book_set-0-title': 'Les Fleurs du Mal',

            'book_set-1-title': '',

            'book_set-2-title': '',

        }

        formset = AuthorBooksFormSet(data, instance=author)

        self.assertTrue(formset.is_valid())

        saved = formset.save()

        self.assertEqual(len(saved), 1)

        book1, = saved

        self.assertEqual(book1, Book.objects.get(title='Les Fleurs du Mal'))

        self.assertQuerysetEqual(author.book_set.all(), ['<Book: Les Fleurs du Mal>'])

        # Now that we've added a book to Charles Baudelaire, let's try adding

        # another one. This time though, an edit form will be available for

        # every existing book.

        AuthorBooksFormSet = inlineformset_factory(Author, Book, can_delete=False, extra=2)

        author = Author.objects.get(name='Charles Baudelaire')

        formset = AuthorBooksFormSet(instance=author)

        self.assertEqual(len(formset.forms), 3)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_book_set-0-title">Title:</label> <input id="id_book_set-0-title" type="text" name="book_set-0-title" value="Les Fleurs du Mal" maxlength="100" /><input type="hidden" name="book_set-0-author" value="%d" id="id_book_set-0-author" /><input type="hidden" name="book_set-0-id" value="%d" id="id_book_set-0-id" /></p>' % (author.id, book1.id))

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_book_set-1-title">Title:</label> <input id="id_book_set-1-title" type="text" name="book_set-1-title" maxlength="100" /><input type="hidden" name="book_set-1-author" value="%d" id="id_book_set-1-author" /><input type="hidden" name="book_set-1-id" id="id_book_set-1-id" /></p>' % author.id)

        self.assertEqual(formset.forms[2].as_p(),

            '<p><label for="id_book_set-2-title">Title:</label> <input id="id_book_set-2-title" type="text" name="book_set-2-title" maxlength="100" /><input type="hidden" name="book_set-2-author" value="%d" id="id_book_set-2-author" /><input type="hidden" name="book_set-2-id" id="id_book_set-2-id" /></p>' % author.id)

        data = {

            'book_set-TOTAL_FORMS': '3', # the number of forms rendered

            'book_set-INITIAL_FORMS': '1', # the number of forms with initial data

            'book_set-MAX_NUM_FORMS': '', # the max number of forms

            'book_set-0-id': '1',

            'book_set-0-title': 'Les Fleurs du Mal',

            'book_set-1-title': 'Les Paradis Artificiels',

            'book_set-2-title': '',

        }

        formset = AuthorBooksFormSet(data, instance=author)

        self.assertTrue(formset.is_valid())

        saved = formset.save()

        self.assertEqual(len(saved), 1)

        book2, = saved

        self.assertEqual(book2, Book.objects.get(title='Les Paradis Artificiels'))

        # As you can see, 'Les Paradis Artificiels' is now a book belonging to

        # Charles Baudelaire.

        self.assertQuerysetEqual(author.book_set.order_by('title'), [

            '<Book: Les Fleurs du Mal>',

            '<Book: Les Paradis Artificiels>',

        ])

    def test_inline_formsets_save_as_new(self):

        # The save_as_new parameter lets you re-associate the data to a new

        # instance.  This is used in the admin for save_as functionality.

        AuthorBooksFormSet = inlineformset_factory(Author, Book, can_delete=False, extra=2)

        author = Author.objects.create(name='Charles Baudelaire')

        data = {

            'book_set-TOTAL_FORMS': '3', # the number of forms rendered

            'book_set-INITIAL_FORMS': '2', # the number of forms with initial data

            'book_set-MAX_NUM_FORMS': '', # the max number of forms

            'book_set-0-id': '1',

            'book_set-0-title': 'Les Fleurs du Mal',

            'book_set-1-id': '2',

            'book_set-1-title': 'Les Paradis Artificiels',

            'book_set-2-title': '',

        }

        formset = AuthorBooksFormSet(data, instance=Author(), save_as_new=True)

        self.assertTrue(formset.is_valid())

        new_author = Author.objects.create(name='Charles Baudelaire')

        formset = AuthorBooksFormSet(data, instance=new_author, save_as_new=True)

        saved = formset.save()

        self.assertEqual(len(saved), 2)

        book1, book2 = saved

        self.assertEqual(book1.title, 'Les Fleurs du Mal')

        self.assertEqual(book2.title, 'Les Paradis Artificiels')

        # Test using a custom prefix on an inline formset.

        formset = AuthorBooksFormSet(prefix="test")

        self.assertEqual(len(formset.forms), 2)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_test-0-title">Title:</label> <input id="id_test-0-title" type="text" name="test-0-title" maxlength="100" /><input type="hidden" name="test-0-author" id="id_test-0-author" /><input type="hidden" name="test-0-id" id="id_test-0-id" /></p>')

        self.assertEqual(formset.forms[1].as_p(),

            '<p><label for="id_test-1-title">Title:</label> <input id="id_test-1-title" type="text" name="test-1-title" maxlength="100" /><input type="hidden" name="test-1-author" id="id_test-1-author" /><input type="hidden" name="test-1-id" id="id_test-1-id" /></p>')

    def test_inline_formsets_with_custom_pk(self):

        # Test inline formsets where the inline-edited object has a custom

        # primary key that is not the fk to the parent object.

        AuthorBooksFormSet2 = inlineformset_factory(Author, BookWithCustomPK, can_delete=False, extra=1)

        author = Author.objects.create(pk=1, name='Charles Baudelaire')

        formset = AuthorBooksFormSet2(instance=author)

        self.assertEqual(len(formset.forms), 1)

        self.assertEqual(formset.forms[0].as_p(),

            '<p><label for="id_bookwithcustompk_set-0-my_pk">My pk:</label> <input type="text" name="bookwithcustompk_set-0-my_pk" id="id_bookwithcustompk_set-0-my_pk" /></p>\n'

            '<p><label for="id_bookwithcustompk_set-0-title">Title:</label> <input id="id_bookwithcustompk_set-0-title" type="text" name="bookwithcustompk_set-0-title" maxlength="100" /><input type="hidden" name="bookwithcustompk_set-0-author" value="1" id="id_bookwithcustompk_set-0-author" /></p>')