from django.test import SimpleTestCase

from localflavor.cn.forms import CNIDCardField, CNPostCodeField, CNProvinceSelect


class CNLocalFlavorTests(SimpleTestCase):
    def test_CNProvinceSelect(self):
        f = CNProvinceSelect()
        correct_output = '''<select name="provinces">
<option value="anhui">\u5b89\u5fbd</option>
<option value="beijing">\u5317\u4eac</option>
<option value="chongqing">\u91cd\u5e86</option>
<option value="fujian">\u798f\u5efa</option>
<option value="gansu">\u7518\u8083</option>
<option value="guangdong">\u5e7f\u4e1c</option>
<option value="guangxi">\u5e7f\u897f\u58ee\u65cf\u81ea\u6cbb\u533a</option>
<option value="guizhou">\u8d35\u5dde</option>
<option value="hainan">\u6d77\u5357</option>
<option value="hebei">\u6cb3\u5317</option>
<option value="heilongjiang">\u9ed1\u9f99\u6c5f</option>
<option value="henan">\u6cb3\u5357</option>
<option value="hongkong">\u9999\u6e2f</option>
<option value="hubei" selected="selected">\u6e56\u5317</option>
<option value="hunan">\u6e56\u5357</option>
<option value="jiangsu">\u6c5f\u82cf</option>
<option value="jiangxi">\u6c5f\u897f</option>
<option value="jilin">\u5409\u6797</option>
<option value="liaoning">\u8fbd\u5b81</option>
<option value="macao">\u6fb3\u95e8</option>
<option value="neimongol">\u5185\u8499\u53e4\u81ea\u6cbb\u533a</option>
<option value="ningxia">\u5b81\u590f\u56de\u65cf\u81ea\u6cbb\u533a</option>
<option value="qinghai">\u9752\u6d77</option>
<option value="shaanxi">\u9655\u897f</option>
<option value="shandong">\u5c71\u4e1c</option>
<option value="shanghai">\u4e0a\u6d77</option>
<option value="shanxi">\u5c71\u897f</option>
<option value="sichuan">\u56db\u5ddd</option>
<option value="taiwan">\u53f0\u6e7e</option>
<option value="tianjin">\u5929\u6d25</option>
<option value="xinjiang">\u65b0\u7586\u7ef4\u543e\u5c14\u81ea\u6cbb\u533a</option>
<option value="xizang">\u897f\u85cf\u81ea\u6cbb\u533a</option>
<option value="yunnan">\u4e91\u5357</option>
<option value="zhejiang">\u6d59\u6c5f</option>
</select>'''
        self.assertHTMLEqual(f.render('provinces', 'hubei'), correct_output)

    def test_CNPostCodeField(self):
        error_format = ['Enter a post code in the format XXXXXX.']
        valid = {
            '091209': '091209'
        }
        invalid = {
            '09120': error_format,
            '09120916': error_format
        }
        self.assertFieldOutput(CNPostCodeField, valid, invalid)

    def test_CNIDCardField(self):
        valid = {
            # A valid 1st generation ID Card Number.
            '110101491001001': '110101491001001',
            # A valid 2nd generation ID Card number.
            '11010119491001001X': '11010119491001001X',
            # Another valid 2nd gen ID Number with a case change
            '11010119491001001x': '11010119491001001X'
        }

        wrong_format = ['ID Card Number consists of 15 or 18 digits.']
        wrong_location = ['Invalid ID Card Number: Wrong location code']
        wrong_bday = ['Invalid ID Card Number: Wrong birthdate']
        wrong_checksum = ['Invalid ID Card Number: Wrong checksum']

        invalid = {
            'abcdefghijklmnop': wrong_format,
            '1010101010101010': wrong_format,
            '010101491001001': wrong_location,  # 1st gen, 01 is invalid
            '110101491041001': wrong_bday,  # 1st gen. There wasn't day 41
            '92010119491001001X': wrong_location,  # 2nd gen, 92 is invalid
            '91010119491301001X': wrong_bday,
            # 2nd gen, 19491301 is invalid date
            '910101194910010014': wrong_checksum  # 2nd gen
        }
        self.assertFieldOutput(CNIDCardField, valid, invalid)
