--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/regressiontests/model_formsets_regress/tests.py Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,218 @@
+from django import forms
+from django.forms.models import modelform_factory, inlineformset_factory, modelformset_factory
+from django.test import TestCase
+
+from models import User, UserSite, Restaurant, Manager
+
+
+class InlineFormsetTests(TestCase):
+ def test_formset_over_to_field(self):
+ "A formset over a ForeignKey with a to_field can be saved. Regression for #10243"
+ Form = modelform_factory(User)
+ FormSet = inlineformset_factory(User, UserSite)
+
+ # Instantiate the Form and FormSet to prove
+ # you can create a form with no data
+ form = Form()
+ form_set = FormSet(instance=User())
+
+ # Now create a new User and UserSite instance
+ data = {
+ 'serial': u'1',
+ 'username': u'apollo13',
+ 'usersite_set-TOTAL_FORMS': u'1',
+ 'usersite_set-INITIAL_FORMS': u'0',
+ 'usersite_set-MAX_NUM_FORMS': u'0',
+ 'usersite_set-0-data': u'10',
+ 'usersite_set-0-user': u'apollo13'
+ }
+ user = User()
+ form = Form(data)
+ if form.is_valid():
+ user = form.save()
+ else:
+ self.fail('Errors found on form:%s' % form_set)
+
+ form_set = FormSet(data, instance=user)
+ if form_set.is_valid():
+ form_set.save()
+ usersite = UserSite.objects.all().values()
+ self.assertEqual(usersite[0]['data'], 10)
+ self.assertEqual(usersite[0]['user_id'], u'apollo13')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ # Now update the UserSite instance
+ data = {
+ 'usersite_set-TOTAL_FORMS': u'1',
+ 'usersite_set-INITIAL_FORMS': u'1',
+ 'usersite_set-MAX_NUM_FORMS': u'0',
+ 'usersite_set-0-id': unicode(usersite[0]['id']),
+ 'usersite_set-0-data': u'11',
+ 'usersite_set-0-user': u'apollo13'
+ }
+ form_set = FormSet(data, instance=user)
+ if form_set.is_valid():
+ form_set.save()
+ usersite = UserSite.objects.all().values()
+ self.assertEqual(usersite[0]['data'], 11)
+ self.assertEqual(usersite[0]['user_id'], u'apollo13')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ # Now add a new UserSite instance
+ data = {
+ 'usersite_set-TOTAL_FORMS': u'2',
+ 'usersite_set-INITIAL_FORMS': u'1',
+ 'usersite_set-MAX_NUM_FORMS': u'0',
+ 'usersite_set-0-id': unicode(usersite[0]['id']),
+ 'usersite_set-0-data': u'11',
+ 'usersite_set-0-user': u'apollo13',
+ 'usersite_set-1-data': u'42',
+ 'usersite_set-1-user': u'apollo13'
+ }
+ form_set = FormSet(data, instance=user)
+ if form_set.is_valid():
+ form_set.save()
+ usersite = UserSite.objects.all().values().order_by('data')
+ self.assertEqual(usersite[0]['data'], 11)
+ self.assertEqual(usersite[0]['user_id'], u'apollo13')
+ self.assertEqual(usersite[1]['data'], 42)
+ self.assertEqual(usersite[1]['user_id'], u'apollo13')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ def test_formset_over_inherited_model(self):
+ "A formset over a ForeignKey with a to_field can be saved. Regression for #11120"
+ Form = modelform_factory(Restaurant)
+ FormSet = inlineformset_factory(Restaurant, Manager)
+
+ # Instantiate the Form and FormSet to prove
+ # you can create a form with no data
+ form = Form()
+ form_set = FormSet(instance=Restaurant())
+
+ # Now create a new Restaurant and Manager instance
+ data = {
+ 'name': u"Guido's House of Pasta",
+ 'manager_set-TOTAL_FORMS': u'1',
+ 'manager_set-INITIAL_FORMS': u'0',
+ 'manager_set-MAX_NUM_FORMS': u'0',
+ 'manager_set-0-name': u'Guido Van Rossum'
+ }
+ restaurant = User()
+ form = Form(data)
+ if form.is_valid():
+ restaurant = form.save()
+ else:
+ self.fail('Errors found on form:%s' % form_set)
+
+ form_set = FormSet(data, instance=restaurant)
+ if form_set.is_valid():
+ form_set.save()
+ manager = Manager.objects.all().values()
+ self.assertEqual(manager[0]['name'], 'Guido Van Rossum')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ # Now update the Manager instance
+ data = {
+ 'manager_set-TOTAL_FORMS': u'1',
+ 'manager_set-INITIAL_FORMS': u'1',
+ 'manager_set-MAX_NUM_FORMS': u'0',
+ 'manager_set-0-id': unicode(manager[0]['id']),
+ 'manager_set-0-name': u'Terry Gilliam'
+ }
+ form_set = FormSet(data, instance=restaurant)
+ if form_set.is_valid():
+ form_set.save()
+ manager = Manager.objects.all().values()
+ self.assertEqual(manager[0]['name'], 'Terry Gilliam')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ # Now add a new Manager instance
+ data = {
+ 'manager_set-TOTAL_FORMS': u'2',
+ 'manager_set-INITIAL_FORMS': u'1',
+ 'manager_set-MAX_NUM_FORMS': u'0',
+ 'manager_set-0-id': unicode(manager[0]['id']),
+ 'manager_set-0-name': u'Terry Gilliam',
+ 'manager_set-1-name': u'John Cleese'
+ }
+ form_set = FormSet(data, instance=restaurant)
+ if form_set.is_valid():
+ form_set.save()
+ manager = Manager.objects.all().values().order_by('name')
+ self.assertEqual(manager[0]['name'], 'John Cleese')
+ self.assertEqual(manager[1]['name'], 'Terry Gilliam')
+ else:
+ self.fail('Errors found on formset:%s' % form_set.errors)
+
+ def test_formset_with_none_instance(self):
+ "A formset with instance=None can be created. Regression for #11872"
+ Form = modelform_factory(User)
+ FormSet = inlineformset_factory(User, UserSite)
+
+ # Instantiate the Form and FormSet to prove
+ # you can create a formset with an instance of None
+ form = Form(instance=None)
+ formset = FormSet(instance=None)
+
+
+class CustomWidget(forms.CharField):
+ pass
+
+
+class UserSiteForm(forms.ModelForm):
+ class Meta:
+ model = UserSite
+ widgets = {'data': CustomWidget}
+
+
+class Callback(object):
+
+ def __init__(self):
+ self.log = []
+
+ def __call__(self, db_field, **kwargs):
+ self.log.append((db_field, kwargs))
+ return db_field.formfield(**kwargs)
+
+
+class FormfieldCallbackTests(TestCase):
+ """
+ Regression for #13095: Using base forms with widgets
+ defined in Meta should not raise errors.
+ """
+
+ def test_inlineformset_factory_default(self):
+ Formset = inlineformset_factory(User, UserSite, form=UserSiteForm)
+ form = Formset({}).forms[0]
+ self.assertTrue(isinstance(form['data'].field.widget, CustomWidget))
+
+ def test_modelformset_factory_default(self):
+ Formset = modelformset_factory(UserSite, form=UserSiteForm)
+ form = Formset({}).forms[0]
+ self.assertTrue(isinstance(form['data'].field.widget, CustomWidget))
+
+ def assertCallbackCalled(self, callback):
+ id_field, user_field, data_field = UserSite._meta.fields
+ expected_log = [
+ (id_field, {}),
+ (user_field, {}),
+ (data_field, {'widget': CustomWidget}),
+ ]
+ self.assertEqual(callback.log, expected_log)
+
+ def test_inlineformset_custom_callback(self):
+ callback = Callback()
+ inlineformset_factory(User, UserSite, form=UserSiteForm,
+ formfield_callback=callback)
+ self.assertCallbackCalled(callback)
+
+ def test_modelformset_custom_callback(self):
+ callback = Callback()
+ modelformset_factory(UserSite, form=UserSiteForm,
+ formfield_callback=callback)
+ self.assertCallbackCalled(callback)