Test: ValidarCPF

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

diff --git a/test/test_contribuinte.py b/test/test_contribuinte.py
index c417acc..6048350 100644 (file)
--- a/test/test_contribuinte.py
+++ b/test/test_contribuinte.py
@@ -5,5 +5,18 @@ import os
 import dirs
 dirs.default_irpf_dir = dirs.IRPFTestDir()
 
+from contribuinte import Contribuinte
+
+class TestContribuinte(unittest.TestCase):
+
+    def setUp(self):
+        self.cpf = '34278161174'
+        self.cpf_invalido = '34278161175'
+
+        self.contribuinte = Contribuinte(self.cpf)
+
+    def test_ValidarCPF(self):
+        self.assertTrue(self.contribuinte._validate_cpf(self.cpf))
+        self.assertFalse(self.contribuinte._validate_cpf(self.cpf_invalido))
 
 # vim:tabstop=4:expandtab:smartindent:fenc=utf8