260 lines
9.5 KiB
Python
260 lines
9.5 KiB
Python
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, Network, Host
|
|
|
|
|
|
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)
|
|
|
|
def test_empty_fields_on_modelformset(self):
|
|
"No fields passed to modelformset_factory should result in no fields on returned forms except for the id. See #14119."
|
|
UserFormSet = modelformset_factory(User, fields=())
|
|
formset = UserFormSet()
|
|
for form in formset.forms:
|
|
self.assertTrue('id' in form.fields)
|
|
self.assertEqual(len(form.fields), 1)
|
|
|
|
def test_save_as_new_with_new_inlines(self):
|
|
"""
|
|
Existing and new inlines are saved with save_as_new.
|
|
|
|
Regression for #14938.
|
|
|
|
"""
|
|
efnet = Network.objects.create(name="EFNet")
|
|
host1 = Host.objects.create(hostname="irc.he.net", network=efnet)
|
|
|
|
HostFormSet = inlineformset_factory(Network, Host)
|
|
|
|
# Add a new host, modify previous host, and save-as-new
|
|
data = {
|
|
'host_set-TOTAL_FORMS': u'2',
|
|
'host_set-INITIAL_FORMS': u'1',
|
|
'host_set-MAX_NUM_FORMS': u'0',
|
|
'host_set-0-id': unicode(host1.id),
|
|
'host_set-0-hostname': u'tranquility.hub.dal.net',
|
|
'host_set-1-hostname': u'matrix.de.eu.dal.net'
|
|
}
|
|
|
|
# To save a formset as new, it needs a new hub instance
|
|
dalnet = Network.objects.create(name="DALnet")
|
|
formset = HostFormSet(data, instance=dalnet, save_as_new=True)
|
|
|
|
self.assertTrue(formset.is_valid())
|
|
formset.save()
|
|
self.assertQuerysetEqual(
|
|
dalnet.host_set.order_by("hostname"),
|
|
["<Host: matrix.de.eu.dal.net>", "<Host: tranquility.hub.dal.net>"]
|
|
)
|
|
|
|
|
|
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)
|