from django.contrib.localflavor.ar.forms import (ARProvinceSelect,

    ARPostalCodeField, ARDNIField, ARCUITField)

from utils import LocalFlavorTestCase

class ARLocalFlavorTests(LocalFlavorTestCase):

    def test_ARProvinceSelect(self):

        f = ARProvinceSelect()

        out = u'''<select name="provincias">

<option value="B">Buenos Aires</option>

<option value="K">Catamarca</option>

<option value="H">Chaco</option>

<option value="U">Chubut</option>

<option value="C">Ciudad Aut\xf3noma de Buenos Aires</option>

<option value="X">C\xf3rdoba</option>

<option value="W">Corrientes</option>

<option value="E">Entre R\xedos</option>

<option value="P">Formosa</option>

<option value="Y">Jujuy</option>

<option value="L">La Pampa</option>

<option value="F">La Rioja</option>

<option value="M">Mendoza</option>

<option value="N">Misiones</option>

<option value="Q">Neuqu\xe9n</option>

<option value="R">R\xedo Negro</option>

<option value="A" selected="selected">Salta</option>

<option value="J">San Juan</option>

<option value="D">San Luis</option>

<option value="Z">Santa Cruz</option>

<option value="S">Santa Fe</option>

<option value="G">Santiago del Estero</option>

<option value="V">Tierra del Fuego, Ant\xe1rtida e Islas del Atl\xe1ntico Sur</option>

<option value="T">Tucum\xe1n</option>

</select>'''

        self.assertEqual(f.render('provincias', 'A'), out)

    def test_ARPostalCodeField(self):

        error_format = [u'Enter a postal code in the format NNNN or ANNNNAAA.']

        error_atmost = [u'Ensure this value has at most 8 characters (it has 9).']

        error_atleast = [u'Ensure this value has at least 4 characters (it has 3).']

        valid = {

            '5000': '5000',

            'C1064AAB': 'C1064AAB',

            'c1064AAB': 'C1064AAB',

            'C1064aab': 'C1064AAB',

            '4400': '4400',

            u'C1064AAB': 'C1064AAB',

        }

        invalid = {

            'C1064AABB': error_atmost + error_format,

            'C1064AA': error_format,

            'C1064AB': error_format,

            '106AAB': error_format,

            '500': error_atleast + error_format,

            '5PPP': error_format,

        }

        self.assertFieldOutput(ARPostalCodeField, valid, invalid)

    def test_ARDNIField(self):

        error_length = [u'This field requires 7 or 8 digits.']

        error_digitsonly = [u'This field requires only numbers.']

        valid = {

            '20123456': '20123456',

            '20.123.456': '20123456',

            u'20123456': '20123456',

            u'20.123.456': '20123456',

            '20.123456': '20123456',

            '9123456': '9123456',

            '9.123.456': '9123456',

        }

        invalid = {

            '101234566': error_length,

            'W0123456': error_digitsonly,

            '10,123,456': error_digitsonly,

        }

        self.assertFieldOutput(ARDNIField, valid, invalid)

    def test_ARCUITField(self):

        error_format = [u'Enter a valid CUIT in XX-XXXXXXXX-X or XXXXXXXXXXXX format.']

        error_invalid = [u'Invalid CUIT.']

        valid = {

            '20-10123456-9': '20-10123456-9',

            u'20-10123456-9': '20-10123456-9',

            '27-10345678-4': '27-10345678-4',

            '20101234569': '20-10123456-9',

            '27103456784': '27-10345678-4',

        }

        invalid = {

            '2-10123456-9': error_format,

            '210123456-9': error_format,

            '20-10123456': error_format,

            '20-10123456-': error_format,

             '20-10123456-5': error_invalid,

            '2-10123456-9': error_format,

             '27-10345678-1': error_invalid,

             u'27-10345678-1': error_invalid,

        }

        self.assertFieldOutput(ARCUITField, valid, invalid)