from django.contrib.localflavor.ch.forms import (CHZipCodeField, CHPhoneNumberField, CHIdentityCardNumberField, CHStateSelect) from django.test import SimpleTestCase class CHLocalFlavorTests(SimpleTestCase): def test_CHStateSelect(self): f = CHStateSelect() out = u'''''' self.assertHTMLEqual(f.render('state', 'AG'), out) def test_CHZipCodeField(self): error_format = [u'Enter a zip code in the format XXXX.'] valid = { '1234': '1234', '0000': '0000', } invalid = { '800x': error_format, '80 00': error_format, } self.assertFieldOutput(CHZipCodeField, valid, invalid) def test_CHPhoneNumberField(self): error_format = [u'Phone numbers must be in 0XX XXX XX XX format.'] valid = { '012 345 67 89': '012 345 67 89', '0123456789': '012 345 67 89', } invalid = { '01234567890': error_format, '1234567890': error_format, } self.assertFieldOutput(CHPhoneNumberField, valid, invalid) def test_CHIdentityCardNumberField(self): error_format = [u'Enter a valid Swiss identity or passport card number in X1234567<0 or 1234567890 format.'] valid = { 'C1234567<0': 'C1234567<0', '2123456700': '2123456700', } invalid = { 'C1234567<1': error_format, '2123456701': error_format, } self.assertFieldOutput(CHIdentityCardNumberField, valid, invalid)