class ModelCMSPCase(TestCase): """Caso de teste de métodos que usam objetos model no XmlCMSP""" def setUp(self): casa = GeradorCasaLegislativa().gerar_cmsp() self.xmlCMSP = XmlCMSP(casa, True) type(self).preencher_banco(casa) @staticmethod def preencher_banco(casa): partido = models.Partido(nome="PTest", numero="1") partido.save() parlamentar = models.Parlamentar( id_parlamentar="1", nome="Teste_vereador", partido=partido, casa_legislativa=casa) parlamentar.save() def test_vereador_sem_partido(self): xml_vereador = etree.fromstring( "<Vereador Partido=\"nao tem partido\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEquals( partido, models.Partido.objects.get(nome=models.SEM_PARTIDO)) def test_vereador_com_partido(self): xml_vereador = etree.fromstring("<Vereador Partido=\"PTest\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEquals(partido, models.Partido.objects.get(nome="PTest")) # def test_retorna_vereador_existente(self): # TODO: 2 parlamentares identicos podem ser cadastrados se 1 # ja existir no banco # xml_vereador = etree.fromstring("<Vereador IDParlamentar=\"1\" # NomeParlamentar=\"Teste_vereador\"/>") # parlamentar = self.xmlCMSP.votante(xml_vereador) # self.assertEquals(parlamentar,models.Parlamentar.objects.get( # nome = "Teste_vereador")) def test_salva_vereador_inexistente(self): xml_vereador = etree.fromstring("<Vereador IDParlamentar=\"999\" \ Nome=\"Nao_consta\" Partido=\"PN\"/>") parlamentar = self.xmlCMSP.vereador(xml_vereador) self.assertEquals( parlamentar, models.Parlamentar.objects.get(id_parlamentar=999))
class ModelCMSPCase(TestCase): """Caso de teste de métodos que usam objetos model no XmlCMSP""" def setUp(self): casa = GeradorCasaLegislativa().gerar_cmsp() self.xmlCMSP = XmlCMSP(casa) type(self).preencher_banco(casa) @staticmethod def preencher_banco(casa): partido = models.Partido(nome="PTest", numero="1") partido.save() parlamentar = models.Parlamentar(id_parlamentar="1", nome="Teste_vereador", partido=partido, casa_legislativa=casa) parlamentar.save() def test_vereador_sem_partido(self): xml_vereador = etree.fromstring( "<Vereador Partido=\"nao tem partido\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEquals(partido, models.Partido.objects.get(nome=models.SEM_PARTIDO)) def test_vereador_com_partido(self): xml_vereador = etree.fromstring("<Vereador Partido=\"PTest\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEquals(partido, models.Partido.objects.get(nome="PTest")) def test_retorna_vereador_existente(self): xml_vereador = etree.fromstring( "<Vereador IDParlamentar=\"2\" Nome=\"Seu Vereador\" Partido=\"PTest\"/>" ) parlamentar = self.xmlCMSP.vereador(xml_vereador) self.assertEquals(parlamentar.nome, 'Seu Vereador') self.assertEquals(parlamentar.partido.nome, 'PTest') self.assertEquals(parlamentar.id_parlamentar, '2') def test_salva_vereador_inexistente(self): xml_vereador = etree.fromstring("<Vereador IDParlamentar=\"999\" \ Nome=\"Nao_consta\" Partido=\"PN\"/>") parlamentar = self.xmlCMSP.vereador(xml_vereador) self.assertEquals(parlamentar, models.Parlamentar.objects.get(id_parlamentar=999))
class ModelCMSPCase(TestCase): """Caso de teste de métodos que usam objetos model no XmlCMSP""" def setUp(self): casa = GeradorCasaLegislativa().gerar_cmsp() self.xmlCMSP = XmlCMSP(casa) type(self).preencher_banco(casa) @staticmethod def preencher_banco(casa): partido = models.Partido(nome="PTest", numero="1") partido.save() parlamentar = models.Parlamentar( id_parlamentar="1", nome="Teste_vereador", partido=partido, casa_legislativa=casa) parlamentar.save() def test_vereador_sem_partido(self): xml_vereador = etree.fromstring( "<Vereador Partido=\"nao tem partido\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEqual( partido, models.Partido.objects.get(nome=models.SEM_PARTIDO)) def test_vereador_com_partido(self): xml_vereador = etree.fromstring("<Vereador Partido=\"PTest\"/>") partido = self.xmlCMSP.partido(xml_vereador) self.assertEqual(partido, models.Partido.objects.get(nome="PTest")) def test_retorna_vereador_existente(self): xml_vereador = etree.fromstring("<Vereador IDParlamentar=\"2\" Nome=\"Seu Vereador\" Partido=\"PTest\"/>") parlamentar = self.xmlCMSP.vereador(xml_vereador) self.assertEqual(parlamentar.nome, 'Seu Vereador') self.assertEqual(parlamentar.partido.nome, 'PTest') self.assertEqual(parlamentar.id_parlamentar, '2') def test_salva_vereador_inexistente(self): xml_vereador = etree.fromstring("<Vereador IDParlamentar=\"999\" \ Nome=\"Nao_consta\" Partido=\"PN\"/>") parlamentar = self.xmlCMSP.vereador(xml_vereador) self.assertEqual( parlamentar, models.Parlamentar.objects.get(id_parlamentar=999))