PageRenderTime 33ms CodeModel.GetById 8ms RepoModel.GetById 0ms app.codeStats 0ms

/tests/regressiontests/forms/tests/models.py

https://code.google.com/p/mango-py/
Python | 161 lines | 148 code | 10 blank | 3 comment | 2 complexity | 4661a4199e39432daa89eeac0e682b0a MD5 | raw file
Possible License(s): BSD-3-Clause
  1. # -*- coding: utf-8 -*-
  2. import datetime
  3. from django.core.files.uploadedfile import SimpleUploadedFile
  4. from django.forms import Form, ModelForm, FileField, ModelChoiceField
  5. from django.test import TestCase
  6. from regressiontests.forms.models import ChoiceModel, ChoiceOptionModel, ChoiceFieldModel, FileModel, Group, BoundaryModel, Defaults
  7. class ChoiceFieldForm(ModelForm):
  8. class Meta:
  9. model = ChoiceFieldModel
  10. class FileForm(Form):
  11. file1 = FileField()
  12. class TestTicket12510(TestCase):
  13. ''' It is not necessary to generate choices for ModelChoiceField (regression test for #12510). '''
  14. def setUp(self):
  15. self.groups = [Group.objects.create(name=name) for name in 'abc']
  16. def test_choices_not_fetched_when_not_rendering(self):
  17. def test():
  18. field = ModelChoiceField(Group.objects.order_by('-name'))
  19. self.assertEqual('a', field.clean(self.groups[0].pk).name)
  20. # only one query is required to pull the model from DB
  21. self.assertNumQueries(1, test)
  22. class ModelFormCallableModelDefault(TestCase):
  23. def test_no_empty_option(self):
  24. "If a model's ForeignKey has blank=False and a default, no empty option is created (Refs #10792)."
  25. option = ChoiceOptionModel.objects.create(name='default')
  26. choices = list(ChoiceFieldForm().fields['choice'].choices)
  27. self.assertEqual(len(choices), 1)
  28. self.assertEqual(choices[0], (option.pk, unicode(option)))
  29. def test_callable_initial_value(self):
  30. "The initial value for a callable default returning a queryset is the pk (refs #13769)"
  31. obj1 = ChoiceOptionModel.objects.create(id=1, name='default')
  32. obj2 = ChoiceOptionModel.objects.create(id=2, name='option 2')
  33. obj3 = ChoiceOptionModel.objects.create(id=3, name='option 3')
  34. self.assertEqual(ChoiceFieldForm().as_p(), """<p><label for="id_choice">Choice:</label> <select name="choice" id="id_choice">
  35. <option value="1" selected="selected">ChoiceOption 1</option>
  36. <option value="2">ChoiceOption 2</option>
  37. <option value="3">ChoiceOption 3</option>
  38. </select><input type="hidden" name="initial-choice" value="1" id="initial-id_choice" /></p>
  39. <p><label for="id_choice_int">Choice int:</label> <select name="choice_int" id="id_choice_int">
  40. <option value="1" selected="selected">ChoiceOption 1</option>
  41. <option value="2">ChoiceOption 2</option>
  42. <option value="3">ChoiceOption 3</option>
  43. </select><input type="hidden" name="initial-choice_int" value="1" id="initial-id_choice_int" /></p>
  44. <p><label for="id_multi_choice">Multi choice:</label> <select multiple="multiple" name="multi_choice" id="id_multi_choice">
  45. <option value="1" selected="selected">ChoiceOption 1</option>
  46. <option value="2">ChoiceOption 2</option>
  47. <option value="3">ChoiceOption 3</option>
  48. </select><input type="hidden" name="initial-multi_choice" value="1" id="initial-id_multi_choice_0" /> <span class="helptext"> Hold down "Control", or "Command" on a Mac, to select more than one.</span></p>
  49. <p><label for="id_multi_choice_int">Multi choice int:</label> <select multiple="multiple" name="multi_choice_int" id="id_multi_choice_int">
  50. <option value="1" selected="selected">ChoiceOption 1</option>
  51. <option value="2">ChoiceOption 2</option>
  52. <option value="3">ChoiceOption 3</option>
  53. </select><input type="hidden" name="initial-multi_choice_int" value="1" id="initial-id_multi_choice_int_0" /> <span class="helptext"> Hold down "Control", or "Command" on a Mac, to select more than one.</span></p>""")
  54. def test_initial_instance_value(self):
  55. "Initial instances for model fields may also be instances (refs #7287)"
  56. obj1 = ChoiceOptionModel.objects.create(id=1, name='default')
  57. obj2 = ChoiceOptionModel.objects.create(id=2, name='option 2')
  58. obj3 = ChoiceOptionModel.objects.create(id=3, name='option 3')
  59. self.assertEqual(ChoiceFieldForm(initial={
  60. 'choice': obj2,
  61. 'choice_int': obj2,
  62. 'multi_choice': [obj2,obj3],
  63. 'multi_choice_int': ChoiceOptionModel.objects.exclude(name="default"),
  64. }).as_p(), """<p><label for="id_choice">Choice:</label> <select name="choice" id="id_choice">
  65. <option value="1">ChoiceOption 1</option>
  66. <option value="2" selected="selected">ChoiceOption 2</option>
  67. <option value="3">ChoiceOption 3</option>
  68. </select><input type="hidden" name="initial-choice" value="2" id="initial-id_choice" /></p>
  69. <p><label for="id_choice_int">Choice int:</label> <select name="choice_int" id="id_choice_int">
  70. <option value="1">ChoiceOption 1</option>
  71. <option value="2" selected="selected">ChoiceOption 2</option>
  72. <option value="3">ChoiceOption 3</option>
  73. </select><input type="hidden" name="initial-choice_int" value="2" id="initial-id_choice_int" /></p>
  74. <p><label for="id_multi_choice">Multi choice:</label> <select multiple="multiple" name="multi_choice" id="id_multi_choice">
  75. <option value="1">ChoiceOption 1</option>
  76. <option value="2" selected="selected">ChoiceOption 2</option>
  77. <option value="3" selected="selected">ChoiceOption 3</option>
  78. </select><input type="hidden" name="initial-multi_choice" value="2" id="initial-id_multi_choice_0" />
  79. <input type="hidden" name="initial-multi_choice" value="3" id="initial-id_multi_choice_1" /> <span class="helptext"> Hold down "Control", or "Command" on a Mac, to select more than one.</span></p>
  80. <p><label for="id_multi_choice_int">Multi choice int:</label> <select multiple="multiple" name="multi_choice_int" id="id_multi_choice_int">
  81. <option value="1">ChoiceOption 1</option>
  82. <option value="2" selected="selected">ChoiceOption 2</option>
  83. <option value="3" selected="selected">ChoiceOption 3</option>
  84. </select><input type="hidden" name="initial-multi_choice_int" value="2" id="initial-id_multi_choice_int_0" />
  85. <input type="hidden" name="initial-multi_choice_int" value="3" id="initial-id_multi_choice_int_1" /> <span class="helptext"> Hold down "Control", or "Command" on a Mac, to select more than one.</span></p>""")
  86. class FormsModelTestCase(TestCase):
  87. def test_unicode_filename(self):
  88. # FileModel with unicode filename and data #########################
  89. f = FileForm(data={}, files={'file1': SimpleUploadedFile('?????????.txt', '???? ??????? ???? ??? ????????? ?? ??? ?')}, auto_id=False)
  90. self.assertTrue(f.is_valid())
  91. self.assertTrue('file1' in f.cleaned_data)
  92. m = FileModel.objects.create(file=f.cleaned_data['file1'])
  93. self.assertEqual(m.file.name, u'tests/\u6211\u96bb\u6c23\u588a\u8239\u88dd\u6eff\u6652\u9c54.txt')
  94. m.delete()
  95. def test_boundary_conditions(self):
  96. # Boundary conditions on a PostitiveIntegerField #########################
  97. class BoundaryForm(ModelForm):
  98. class Meta:
  99. model = BoundaryModel
  100. f = BoundaryForm({'positive_integer': 100})
  101. self.assertTrue(f.is_valid())
  102. f = BoundaryForm({'positive_integer': 0})
  103. self.assertTrue(f.is_valid())
  104. f = BoundaryForm({'positive_integer': -100})
  105. self.assertFalse(f.is_valid())
  106. def test_formfield_initial(self):
  107. # Formfield initial values ########
  108. # If the model has default values for some fields, they are used as the formfield
  109. # initial values.
  110. class DefaultsForm(ModelForm):
  111. class Meta:
  112. model = Defaults
  113. self.assertEqual(DefaultsForm().fields['name'].initial, u'class default value')
  114. self.assertEqual(DefaultsForm().fields['def_date'].initial, datetime.date(1980, 1, 1))
  115. self.assertEqual(DefaultsForm().fields['value'].initial, 42)
  116. r1 = DefaultsForm()['callable_default'].as_widget()
  117. r2 = DefaultsForm()['callable_default'].as_widget()
  118. self.assertNotEqual(r1, r2)
  119. # In a ModelForm that is passed an instance, the initial values come from the
  120. # instance's values, not the model's defaults.
  121. foo_instance = Defaults(name=u'instance value', def_date=datetime.date(1969, 4, 4), value=12)
  122. instance_form = DefaultsForm(instance=foo_instance)
  123. self.assertEqual(instance_form.initial['name'], u'instance value')
  124. self.assertEqual(instance_form.initial['def_date'], datetime.date(1969, 4, 4))
  125. self.assertEqual(instance_form.initial['value'], 12)
  126. from django.forms import CharField
  127. class ExcludingForm(ModelForm):
  128. name = CharField(max_length=255)
  129. class Meta:
  130. model = Defaults
  131. exclude = ['name', 'callable_default']
  132. f = ExcludingForm({'name': u'Hello', 'value': 99, 'def_date': datetime.date(1999, 3, 2)})
  133. self.assertTrue(f.is_valid())
  134. self.assertEqual(f.cleaned_data['name'], u'Hello')
  135. obj = f.save()
  136. self.assertEqual(obj.name, u'class default value')
  137. self.assertEqual(obj.value, 99)
  138. self.assertEqual(obj.def_date, datetime.date(1999, 3, 2))