Test: test_GetCampoContribuiente

[cascardo/irpf-gui.git] / test / test_contribuinte.py

diff --git a/test/test_contribuinte.py b/test/test_contribuinte.py
index 8af1570..ebd111d 100644 (file)
--- a/test/test_contribuinte.py
+++ b/test/test_contribuinte.py
@@ -11,6 +11,8 @@ class TestContribuinte(unittest.TestCase):
 
     def setUp(self):
         self.cpf = '34278161174'
+        self.cpf_normalizado = '342.781.611-74'
+
         self.cpf_invalido = '34278161175'
 
         self.contribuinte = Contribuinte(self.cpf)
@@ -20,6 +22,21 @@ class TestContribuinte(unittest.TestCase):
         self.assertTrue(self.contribuinte._validate_cpf(self.cpf), msg)
         self.assertFalse(self.contribuinte._validate_cpf(self.cpf_invalido), msg)
 
+    def test_NormalizeCPF(self):
+        msg = 'Normalização falhou'
+        self.assertEqual(self.contribuinte._normalize_cpf(self.cpf),
+                self.cpf_normalizado, msg)
+
+    def test_MinimizarCPF(self):
+        msg = 'Minimizar CPF falhou'
+        self.assertEqual(self.contribuinte._minimize_cpf(self.cpf_normalizado),
+                self.cpf, msg)
+
+    def test_GetCampoContribuiente(self):
+        self.assertEqual(self.contribuinte.get_campo_contribuinte('cep'),
+                '31230-120')
+        self.assertEqual(self.contribuinte.get_campo_contribuinte('nome'),
+                'MARIA DA SILVA DE SOUZA')
 
 
 # vim:tabstop=4:expandtab:smartindent:fenc=utf8