def setUp(self):
self.james_joyce = Author.objects.create(name='James Joyce')
self.charles_dickens = Author.objects.create(name='Charles Dickens')
self.paper = NewsPaper.objects.create(title='the daily record')
self.article = Article.objects.create(
paper=self.paper,
title='Test article',
authors=[self.james_joyce],
)
ArticleFormset = childformset_factory(NewsPaper, Article, exclude=['categories', 'tags'], extra=3)
self.formset = ArticleFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 10,
'form-0-id': self.article.id,
'form-0-title': self.article.title,
'form-0-authors': [self.james_joyce.id, self.charles_dickens.id],
}, instance=self.paper)
ArticleTagsFormset = childformset_factory(NewsPaper, Article, exclude=['categories', 'authors'], extra=3)
self.tags_formset = ArticleTagsFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 10,
'form-0-id': self.article.id,
'form-0-title': self.article.title,
'form-0-tags': 'tag1, tagtwo',
}, instance=self.paper)
def setUp(self):
self.james_joyce = Author.objects.create(name='James Joyce')
self.charles_dickens = Author.objects.create(name='Charles Dickens')
self.paper = NewsPaper.objects.create(title='the daily record')
self.article = Article.objects.create(
paper=self.paper,
title='Test article',
authors=[self.james_joyce],
)
ArticleFormset = childformset_factory(NewsPaper, Article, exclude=['categories', 'tags'], extra=3)
self.formset = ArticleFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 10,
'form-0-id': self.article.id,
'form-0-title': self.article.title,
'form-0-authors': [self.james_joyce.id, self.charles_dickens.id],
}, instance=self.paper)
ArticleTagsFormset = childformset_factory(NewsPaper, Article, exclude=['categories', 'authors'], extra=3)
self.tags_formset = ArticleTagsFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 10,
'form-0-id': self.article.id,
'form-0-title': self.article.title,
'form-0-tags': 'tag1, tagtwo',
}, instance=self.paper)
def test_child_updates_without_ids(self):
john = BandMember(name='John Lennon')
beatles = Band(name='The Beatles', members=[
john
])
beatles.save()
john_id = john.id
paul = BandMember(name='Paul McCartney')
beatles.members.add(paul)
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 2,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': john_id,
'form-1-name': 'Paul McCartney', # NB no way to know programmatically that this form corresponds to the 'paul' object
'form-1-id': '',
}, instance=beatles)
self.assertTrue(band_members_formset.is_valid())
band_members_formset.save(commit=False)
self.assertEqual(2, beatles.members.count())
def test_child_updates_without_ids(self):
john = BandMember(name='John Lennon')
beatles = Band(name='The Beatles', members=[
john
])
beatles.save()
john_id = john.id
paul = BandMember(name='Paul McCartney')
beatles.members.add(paul)
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 2,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': john_id,
'form-1-name': 'Paul McCartney', # NB no way to know programmatically that this form corresponds to the 'paul' object
'form-1-id': '',
}, instance=beatles)
self.assertTrue(band_members_formset.is_valid())
band_members_formset.save(commit=False)
self.assertEqual(2, beatles.members.count())
def test_saving_formset_preserves_order(self):
AlbumsFormset = childformset_factory(Band,
Album,
extra=3,
can_order=True)
beatles = Band(name='The Beatles')
albums_formset = AlbumsFormset(
{
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 0,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'With The Beatles',
'form-0-id': '',
'form-0-ORDER': '2',
'form-1-name': 'Please Please Me',
'form-1-id': '',
'form-1-ORDER': '1',
},
instance=beatles)
self.assertTrue(albums_formset.is_valid())
albums_formset.save(commit=False)
album_names = [album.name for album in beatles.albums.all()]
self.assertEqual(['Please Please Me', 'With The Beatles'], album_names)
def test_child_updates_without_ids(self):
first_song = Song(name='I Saw Her Standing There')
album = Album(name='Please Please Me', songs=[first_song])
beatles = Band(name='The Beatles', albums=[album])
beatles.save()
first_song_id = first_song.id
second_song = Song(name='Misery')
album.songs.add(second_song)
AlbumsFormset = childformset_factory(Band, Album, form=ClusterForm, extra=3)
albums_formset = AlbumsFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'Please Please Me',
'form-0-id': album.id,
'form-0-songs-TOTAL_FORMS': 2,
'form-0-songs-INITIAL_FORMS': 2,
'form-0-songs-MAX_NUM_FORMS': 1000,
'form-0-songs-0-name': 'I Saw Her Standing There',
'form-0-songs-0-id': first_song_id,
'form-0-songs-1-name': 'Misery',
'form-0-songs-1-id': '',
}, instance=beatles)
self.assertTrue(albums_formset.is_valid())
albums_formset.save(commit=False)
self.assertEqual(2, beatles.albums.first().songs.count())
def test_save_commit_false(self):
first_song = Song(name='I Saw Her Standing There')
second_song = Song(name='Mystery')
album = Album(name='Please Please Me', songs=[first_song, second_song])
beatles = Band(name='The Beatles', albums=[album])
beatles.save()
first_song_id, second_song_id = first_song.id, second_song.id
AlbumsFormset = childformset_factory(Band, Album, form=ClusterForm, extra=3)
albums_formset = AlbumsFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'Please Please Me',
'form-0-id': album.id,
'form-0-songs-TOTAL_FORMS': 4,
'form-0-songs-INITIAL_FORMS': 2,
'form-0-songs-MAX_NUM_FORMS': 1000,
'form-0-songs-0-name': 'I Saw Her Standing There',
'form-0-songs-0-DELETE': 'form-0-songs-0-DELETE',
'form-0-songs-0-id': first_song_id,
'form-0-songs-1-name': 'Misery', # changing data of an existing record
'form-0-songs-1-id': second_song_id,
'form-0-songs-2-name': '',
'form-0-songs-2-id': '',
'form-0-songs-3-name': 'Chains', # adding a record
'form-0-songs-3-id': '',
}, instance=beatles)
self.assertTrue(albums_formset.is_valid())
updated_albums = albums_formset.save(commit=False)
# updated_members should only include the items that have been changed and not deleted
self.assertEqual(1, len(updated_albums))
self.assertEqual('Please Please Me', updated_albums[0].name)
self.assertEqual(2, updated_albums[0].songs.count())
self.assertEqual('Misery', updated_albums[0].songs.first().name)
self.assertEqual(second_song_id, updated_albums[0].songs.first().id)
self.assertEqual('Chains', updated_albums[0].songs.all()[1].name)
self.assertEqual(None, updated_albums[0].songs.all()[1].id)
# Changes should not be committed to the db yet
self.assertTrue(Song.objects.filter(name='I Saw Her Standing There', id=first_song_id).exists())
self.assertEqual('Mystery', Song.objects.get(id=second_song_id).name)
self.assertFalse(Song.objects.filter(name='Chains').exists())
beatles.albums.first().songs.commit()
# this should create/update/delete database entries
self.assertEqual('Misery', Song.objects.get(id=second_song_id).name)
self.assertTrue(Song.objects.filter(name='Chains').exists())
self.assertFalse(Song.objects.filter(name='I Saw Her Standing There').exists())
def test_save_commit_false(self):
john = BandMember(name='John Lennon')
paul = BandMember(name='Paul McCartney')
ringo = BandMember(name='Richard Starkey')
beatles = Band(name='The Beatles', members=[john, paul, ringo])
beatles.save()
john_id, paul_id, ringo_id = john.id, paul.id, ringo.id
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset(
{
'form-TOTAL_FORMS': 5,
'form-INITIAL_FORMS': 3,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-DELETE': 'form-0-DELETE',
'form-0-id': john_id,
'form-1-name': 'Paul McCartney',
'form-1-id': paul_id,
'form-2-name':
'Ringo Starr', # changing data of an existing record
'form-2-id': ringo_id,
'form-3-name': '',
'form-3-id': '',
'form-4-name': 'George Harrison', # adding a record
'form-4-id': '',
},
instance=beatles)
self.assertTrue(band_members_formset.is_valid())
updated_members = band_members_formset.save(commit=False)
# updated_members should only include the items that have been changed and not deleted
self.assertEqual(2, len(updated_members))
self.assertEqual('Ringo Starr', updated_members[0].name)
self.assertEqual(ringo_id, updated_members[0].id)
self.assertEqual('George Harrison', updated_members[1].name)
self.assertEqual(None, updated_members[1].id)
# Changes should not be committed to the db yet
self.assertTrue(
BandMember.objects.filter(name='John Lennon', id=john_id).exists())
self.assertEqual('Richard Starkey',
BandMember.objects.get(id=ringo_id).name)
self.assertFalse(
BandMember.objects.filter(name='George Harrison').exists())
beatles.members.commit()
# this should create/update/delete database entries
self.assertEqual('Ringo Starr',
BandMember.objects.get(id=ringo_id).name)
self.assertTrue(
BandMember.objects.filter(name='George Harrison').exists())
self.assertFalse(
BandMember.objects.filter(name='John Lennon').exists())
def test_can_create_formset(self):
beatles = Band(name='The Beatles', members=[
BandMember(name='John Lennon'),
BandMember(name='Paul McCartney'),
])
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset(instance=beatles)
self.assertEqual(5, len(band_members_formset.forms))
self.assertEqual('John Lennon', band_members_formset.forms[0].instance.name)
def test_can_create_formset(self):
beatles = Band(name='The Beatles', members=[
BandMember(name='John Lennon'),
BandMember(name='Paul McCartney'),
])
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset(instance=beatles)
self.assertEqual(5, len(band_members_formset.forms))
self.assertEqual('John Lennon', band_members_formset.forms[0].instance.name)
def test_save_commit_false(self):
john = BandMember(name='John Lennon')
paul = BandMember(name='Paul McCartney')
ringo = BandMember(name='Richard Starkey')
beatles = Band(name='The Beatles', members=[
john, paul, ringo
])
beatles.save()
john_id, paul_id, ringo_id = john.id, paul.id, ringo.id
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 5,
'form-INITIAL_FORMS': 3,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-DELETE': 'form-0-DELETE',
'form-0-id': john_id,
'form-1-name': 'Paul McCartney',
'form-1-id': paul_id,
'form-2-name': 'Ringo Starr', # changing data of an existing record
'form-2-id': ringo_id,
'form-3-name': '',
'form-3-id': '',
'form-4-name': 'George Harrison', # adding a record
'form-4-id': '',
}, instance=beatles)
self.assertTrue(band_members_formset.is_valid())
updated_members = band_members_formset.save(commit=False)
# updated_members should only include the items that have been changed and not deleted
self.assertEqual(2, len(updated_members))
self.assertEqual('Ringo Starr', updated_members[0].name)
self.assertEqual(ringo_id, updated_members[0].id)
self.assertEqual('George Harrison', updated_members[1].name)
self.assertEqual(None, updated_members[1].id)
# Changes should not be committed to the db yet
self.assertTrue(BandMember.objects.filter(name='John Lennon', id=john_id).exists())
self.assertEqual('Richard Starkey', BandMember.objects.get(id=ringo_id).name)
self.assertFalse(BandMember.objects.filter(name='George Harrison').exists())
beatles.members.commit()
# this should create/update/delete database entries
self.assertEqual('Ringo Starr', BandMember.objects.get(id=ringo_id).name)
self.assertTrue(BandMember.objects.filter(name='George Harrison').exists())
self.assertFalse(BandMember.objects.filter(name='John Lennon').exists())
def test_min_num_ignored_in_validation_when_validate_max_false(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
})
self.assertTrue(band_members_formset.is_valid())
def test_min_num_ignored_in_validation_when_validate_max_false(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
})
self.assertTrue(band_members_formset.is_valid())
def test_min_num_fail_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2, validate_min=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
})
self.assertFalse(band_members_formset.is_valid())
self.assertEqual(band_members_formset.non_form_errors()[0], "Please submit 2 or more forms.")
def test_min_num_fail_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2, validate_min=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
})
self.assertFalse(band_members_formset.is_valid())
self.assertEqual(band_members_formset.non_form_errors()[0], "Please submit 2 or more forms.")
def test_min_num_pass_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2, validate_min=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
'form-1-name': 'Paul McCartney',
'form-1-id': '',
})
self.assertTrue(band_members_formset.is_valid())
def test_min_num_pass_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, min_num=2, validate_min=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
'form-1-name': 'Paul McCartney',
'form-1-id': '',
})
self.assertTrue(band_members_formset.is_valid())
def test_can_create_formset(self):
beatles = Band(name='The Beatles', albums=[
Album(name='Please Please Me', songs=[
Song(name='I Saw Her Standing There'),
Song(name='Misery')
])
])
AlbumsFormset = childformset_factory(Band, Album, form=ClusterForm, extra=3)
albums_formset = AlbumsFormset(instance=beatles)
self.assertEqual(4, len(albums_formset.forms))
self.assertEqual('Please Please Me', albums_formset.forms[0].instance.name)
self.assertEqual(5, len(albums_formset.forms[0].formsets['songs'].forms))
self.assertEqual(
'I Saw Her Standing There',
albums_formset.forms[0].formsets['songs'].forms[0].instance.name
)
def test_max_num_fail_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, max_num=2, validate_max=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 3,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
'form-1-name': 'Paul McCartney',
'form-1-id': '',
'form-2-name': 'Ringo Starr',
'form-2-id': '',
})
self.assertFalse(band_members_formset.is_valid())
self.assertEqual(band_members_formset.non_form_errors()[0], "Please submit 2 or fewer forms.")
def test_max_num_fail_validation(self):
BandMembersFormset = childformset_factory(Band, BandMember, max_num=2, validate_max=True)
band_members_formset = BandMembersFormset({
'form-TOTAL_FORMS': 3,
'form-INITIAL_FORMS': 1,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'John Lennon',
'form-0-id': '',
'form-1-name': 'Paul McCartney',
'form-1-id': '',
'form-2-name': 'Ringo Starr',
'form-2-id': '',
})
self.assertFalse(band_members_formset.is_valid())
self.assertEqual(band_members_formset.non_form_errors()[0], "Please submit 2 or fewer forms.")
def test_saving_formset_preserves_order(self):
AlbumsFormset = childformset_factory(Band, Album, extra=3, can_order=True)
beatles = Band(name='The Beatles')
albums_formset = AlbumsFormset({
'form-TOTAL_FORMS': 2,
'form-INITIAL_FORMS': 0,
'form-MAX_NUM_FORMS': 1000,
'form-0-name': 'With The Beatles',
'form-0-id': '',
'form-0-ORDER': '2',
'form-1-name': 'Please Please Me',
'form-1-id': '',
'form-1-ORDER': '1',
}, instance=beatles)
self.assertTrue(albums_formset.is_valid())
albums_formset.save(commit=False)
album_names = [album.name for album in beatles.albums.all()]
self.assertEqual(['Please Please Me', 'With The Beatles'], album_names)
def test_empty_formset(self):
AlbumsFormset = childformset_factory(Band, Album, form=ClusterForm, extra=3)
albums_formset = AlbumsFormset()
self.assertEqual(3, len(albums_formset.forms))
self.assertEqual(3, len(albums_formset.forms[0].formsets['songs'].forms))
def test_empty_formset(self):
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset()
self.assertEqual(3, len(band_members_formset.forms))
def test_empty_formset(self):
BandMembersFormset = childformset_factory(Band, BandMember, extra=3)
band_members_formset = BandMembersFormset()
self.assertEqual(3, len(band_members_formset.forms))
