# -*- coding: utf-8 -*- import datetime from django.conf import settings from django.forms import * from django.forms.extras import SelectDateWidget from django.forms.util import ErrorList from django.utils import translation from django.utils import unittest from django.utils.encoding import force_unicode from django.utils.encoding import smart_unicode from error_messages import AssertFormErrorsMixin class GetDate(Form): mydate = DateField(widget=SelectDateWidget) class FormsExtraTestCase(unittest.TestCase, AssertFormErrorsMixin): ############### # Extra stuff # ############### # The forms library comes with some extra, higher-level Field and Widget def test_selectdate(self): w = SelectDateWidget(years=('2007','2008','2009','2010','2011','2012','2013','2014','2015','2016')) self.assertEqual(w.render('mydate', ''), """ """) self.assertEqual(w.render('mydate', None), w.render('mydate', '')) self.assertEqual(w.render('mydate', '2010-04-15'), """ """) # Accepts a datetime or a string: self.assertEqual(w.render('mydate', datetime.date(2010, 4, 15)), w.render('mydate', '2010-04-15')) # Invalid dates still render the failed date: self.assertEqual(w.render('mydate', '2010-02-31'), """ """) # Using a SelectDateWidget in a form: w = SelectDateWidget(years=('2007','2008','2009','2010','2011','2012','2013','2014','2015','2016'), required=False) self.assertEqual(w.render('mydate', ''), """ """) self.assertEqual(w.render('mydate', '2010-04-15'), """ """) a = GetDate({'mydate_month':'4', 'mydate_day':'1', 'mydate_year':'2008'}) self.assertTrue(a.is_valid()) self.assertEqual(a.cleaned_data['mydate'], datetime.date(2008, 4, 1)) # As with any widget that implements get_value_from_datadict, # we must be prepared to accept the input from the "as_hidden" # rendering as well. self.assertEqual(a['mydate'].as_hidden(), '') b = GetDate({'mydate':'2008-4-1'}) self.assertTrue(b.is_valid()) self.assertEqual(b.cleaned_data['mydate'], datetime.date(2008, 4, 1)) # Invalid dates shouldn't be allowed c = GetDate({'mydate_month':'2', 'mydate_day':'31', 'mydate_year':'2010'}) self.assertFalse(c.is_valid()) self.assertEqual(c.errors, {'mydate': [u'Enter a valid date.']}) # label tag is correctly associated with month dropdown d = GetDate({'mydate_month':'1', 'mydate_day':'1', 'mydate_year':'2010'}) self.assertTrue('' in d.as_p()) def test_multiwidget(self): # MultiWidget and MultiValueField ############################################# # MultiWidgets are widgets composed of other widgets. They are usually # combined with MultiValueFields - a field that is composed of other fields. # MulitWidgets can themselved be composed of other MultiWidgets. # SplitDateTimeWidget is one example of a MultiWidget. class ComplexMultiWidget(MultiWidget): def __init__(self, attrs=None): widgets = ( TextInput(), SelectMultiple(choices=(('J', 'John'), ('P', 'Paul'), ('G', 'George'), ('R', 'Ringo'))), SplitDateTimeWidget(), ) super(ComplexMultiWidget, self).__init__(widgets, attrs) def decompress(self, value): if value: data = value.split(',') return [data[0], data[1], datetime.datetime.strptime(data[2], "%Y-%m-%d %H:%M:%S")] return [None, None, None] def format_output(self, rendered_widgets): return u'\n'.join(rendered_widgets) w = ComplexMultiWidget() self.assertEqual(w.render('name', 'some text,JP,2007-04-25 06:24:00'), """ """) class ComplexField(MultiValueField): def __init__(self, required=True, widget=None, label=None, initial=None): fields = ( CharField(), MultipleChoiceField(choices=(('J', 'John'), ('P', 'Paul'), ('G', 'George'), ('R', 'Ringo'))), SplitDateTimeField() ) super(ComplexField, self).__init__(fields, required, widget, label, initial) def compress(self, data_list): if data_list: return '%s,%s,%s' % (data_list[0],''.join(data_list[1]),data_list[2]) return None f = ComplexField(widget=w) self.assertEqual(f.clean(['some text', ['J','P'], ['2007-04-25','6:24:00']]), u'some text,JP,2007-04-25 06:24:00') self.assertFormErrors([u'Select a valid choice. X is not one of the available choices.'], f.clean, ['some text',['X'], ['2007-04-25','6:24:00']]) # If insufficient data is provided, None is substituted self.assertFormErrors([u'This field is required.'], f.clean, ['some text',['JP']]) class ComplexFieldForm(Form): field1 = ComplexField(widget=w) f = ComplexFieldForm() self.assertEqual(f.as_table(), """Field1: """) f = ComplexFieldForm({'field1_0':'some text','field1_1':['J','P'], 'field1_2_0':'2007-04-25', 'field1_2_1':'06:24:00'}) self.assertEqual(f.as_table(), """Field1: """) self.assertEqual(f.cleaned_data['field1'], u'some text,JP,2007-04-25 06:24:00') def test_ipaddress(self): f = IPAddressField() self.assertFormErrors([u'This field is required.'], f.clean, '') self.assertFormErrors([u'This field is required.'], f.clean, None) self.assertEqual(f.clean('127.0.0.1'), u'127.0.0.1') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '256.125.1.5') f = IPAddressField(required=False) self.assertEqual(f.clean(''), u'') self.assertEqual(f.clean(None), u'') self.assertEqual(f.clean('127.0.0.1'), u'127.0.0.1') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '256.125.1.5') def test_generic_ipaddress_invalid_arguments(self): self.assertRaises(ValueError, GenericIPAddressField, protocol="hamster") self.assertRaises(ValueError, GenericIPAddressField, protocol="ipv4", unpack_ipv4=True) def test_generic_ipaddress_as_generic(self): # The edge cases of the IPv6 validation code are not deeply tested # here, they are covered in the tests for django.utils.ipv6 f = GenericIPAddressField() self.assertFormErrors([u'This field is required.'], f.clean, '') self.assertFormErrors([u'This field is required.'], f.clean, None) self.assertEqual(f.clean('127.0.0.1'), u'127.0.0.1') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '256.125.1.5') self.assertEqual(f.clean('fe80::223:6cff:fe8a:2e8a'), u'fe80::223:6cff:fe8a:2e8a') self.assertEqual(f.clean('2a02::223:6cff:fe8a:2e8a'), u'2a02::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '12345:2:3:4') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1::2:3::4') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, 'foo::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1::2:3:4:5:6:7:8') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1:2') def test_generic_ipaddress_as_ipv4_only(self): f = GenericIPAddressField(protocol="IPv4") self.assertFormErrors([u'This field is required.'], f.clean, '') self.assertFormErrors([u'This field is required.'], f.clean, None) self.assertEqual(f.clean('127.0.0.1'), u'127.0.0.1') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '256.125.1.5') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, 'fe80::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv4 address.'], f.clean, '2a02::223:6cff:fe8a:2e8a') def test_generic_ipaddress_as_ipv6_only(self): f = GenericIPAddressField(protocol="IPv6") self.assertFormErrors([u'This field is required.'], f.clean, '') self.assertFormErrors([u'This field is required.'], f.clean, None) self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '127.0.0.1') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '256.125.1.5') self.assertEqual(f.clean('fe80::223:6cff:fe8a:2e8a'), u'fe80::223:6cff:fe8a:2e8a') self.assertEqual(f.clean('2a02::223:6cff:fe8a:2e8a'), u'2a02::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '12345:2:3:4') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '1::2:3::4') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, 'foo::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '1::2:3:4:5:6:7:8') self.assertFormErrors([u'Enter a valid IPv6 address.'], f.clean, '1:2') def test_generic_ipaddress_as_generic_not_required(self): f = GenericIPAddressField(required=False) self.assertEqual(f.clean(''), u'') self.assertEqual(f.clean(None), u'') self.assertEqual(f.clean('127.0.0.1'), u'127.0.0.1') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, 'foo') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '127.0.0.') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1.2.3.4.5') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '256.125.1.5') self.assertEqual(f.clean('fe80::223:6cff:fe8a:2e8a'), u'fe80::223:6cff:fe8a:2e8a') self.assertEqual(f.clean('2a02::223:6cff:fe8a:2e8a'), u'2a02::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '12345:2:3:4') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1::2:3::4') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, 'foo::223:6cff:fe8a:2e8a') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1::2:3:4:5:6:7:8') self.assertFormErrors([u'Enter a valid IPv4 or IPv6 address.'], f.clean, '1:2') def test_generic_ipaddress_normalization(self): # Test the normalising code f = GenericIPAddressField() self.assertEqual(f.clean('::ffff:0a0a:0a0a'), u'::ffff:10.10.10.10') self.assertEqual(f.clean('::ffff:10.10.10.10'), u'::ffff:10.10.10.10') self.assertEqual(f.clean('2001:000:a:0000:0:fe:fe:beef'), u'2001:0:a::fe:fe:beef') self.assertEqual(f.clean('2001::a:0000:0:fe:fe:beef'), u'2001:0:a::fe:fe:beef') f = GenericIPAddressField(unpack_ipv4=True) self.assertEqual(f.clean('::ffff:0a0a:0a0a'), u'10.10.10.10') def test_smart_unicode(self): class Test: def __str__(self): return 'ŠĐĆŽćžšđ' class TestU: def __str__(self): return 'Foo' def __unicode__(self): return u'\u0160\u0110\u0106\u017d\u0107\u017e\u0161\u0111' self.assertEqual(smart_unicode(Test()), u'\u0160\u0110\u0106\u017d\u0107\u017e\u0161\u0111') self.assertEqual(smart_unicode(TestU()), u'\u0160\u0110\u0106\u017d\u0107\u017e\u0161\u0111') self.assertEqual(smart_unicode(1), u'1') self.assertEqual(smart_unicode('foo'), u'foo') def test_accessing_clean(self): class UserForm(Form): username = CharField(max_length=10) password = CharField(widget=PasswordInput) def clean(self): data = self.cleaned_data if not self.errors: data['username'] = data['username'].lower() return data f = UserForm({'username': 'SirRobin', 'password': 'blue'}) self.assertTrue(f.is_valid()) self.assertEqual(f.cleaned_data['username'], u'sirrobin') def test_overriding_errorlist(self): class DivErrorList(ErrorList): def __unicode__(self): return self.as_divs() def as_divs(self): if not self: return u'' return u'

' % ''.join([u'

' % force_unicode(e) for e in self]) class CommentForm(Form): name = CharField(max_length=50, required=False) email = EmailField() comment = CharField() data = dict(email='invalid') f = CommentForm(data, auto_id=False, error_class=DivErrorList) self.assertEqual(f.as_p(), """

Name:

Enter a valid e-mail address.

Email:

This field is required.

Comment:

""") def test_multipart_encoded_form(self): class FormWithoutFile(Form): username = CharField() class FormWithFile(Form): username = CharField() file = FileField() class FormWithImage(Form): image = ImageField() self.assertFalse(FormWithoutFile().is_multipart()) self.assertTrue(FormWithFile().is_multipart()) self.assertTrue(FormWithImage().is_multipart()) class FormsExtraL10NTestCase(unittest.TestCase): def setUp(self): super(FormsExtraL10NTestCase, self).setUp() self.old_use_l10n = getattr(settings, 'USE_L10N', False) settings.USE_L10N = True translation.activate('nl') def tearDown(self): translation.deactivate() settings.USE_L10N = self.old_use_l10n super(FormsExtraL10NTestCase, self).tearDown() def test_l10n(self): w = SelectDateWidget(years=('2007','2008','2009','2010','2011','2012','2013','2014','2015','2016'), required=False) self.assertEqual(w.value_from_datadict({'date_year': '2010', 'date_month': '8', 'date_day': '13'}, {}, 'date'), '13-08-2010') self.assertEqual(w.render('date', '13-08-2010'), """ """) # Years before 1900 work w = SelectDateWidget(years=('1899',)) self.assertEqual(w.value_from_datadict({'date_year': '1899', 'date_month': '8', 'date_day': '13'}, {}, 'date'), '13-08-1899') def test_l10n_invalid_date_in(self): # Invalid dates shouldn't be allowed a = GetDate({'mydate_month':'2', 'mydate_day':'31', 'mydate_year':'2010'}) self.assertFalse(a.is_valid()) # 'Geef een geldige datum op.' = 'Enter a valid date.' self.assertEqual(a.errors, {'mydate': [u'Geef een geldige datum op.']}) def test_form_label_association(self): # label tag is correctly associated with first rendered dropdown a = GetDate({'mydate_month':'1', 'mydate_day':'1', 'mydate_year':'2010'}) self.assertTrue('' in a.as_p())