Test: Adiciona VerifyCEP

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

diff --git a/test/test_municipios.py b/test/test_municipios.py
index b98961c..5b6482f 100644 (file)
--- a/test/test_municipios.py
+++ b/test/test_municipios.py
@@ -6,6 +6,27 @@ import os
 myPath = os.path.dirname(os.path.abspath(__file__))
 sys.path.insert(0, myPath + '/../src')
 
+from municipios import Municipios
+
+class TestMunicipios(unittest.TestCase):
+
+    def setUp(self):
+        self.municipios = Municipios()
+
+        self.municipios.carregar_estado('MG')
+        self.belo_horizonte = self.municipios.get_municipio('4123')
+
+        self.municipios.carregar_estado('SP')
+        self.sao_paulo = self.municipios.get_municipio('7107')
+
+    def test_VerifyCEP(self):
+        self.assertTrue(self.municipios.verify_cep(self.belo_horizonte, '31230120'))
+        self.assertFalse(self.municipios.verify_cep(self.belo_horizonte, '05020000'))
+
+        self.assertTrue(self.municipios.verify_cep(self.sao_paulo, '05020000'))
+        self.assertFalse(self.municipios.verify_cep(self.sao_paulo, '31230120'))
+
+
 if __name__ == '__main__':
     unittest.main()