def test_lista_materias_anexadas_ciclo(): tipo_materia = baker.make(TipoMateriaLegislativa, descricao="Tipo_Teste") regime_tramitacao = baker.make(RegimeTramitacao, descricao="Regime_Teste") materia_principal = baker.make(MateriaLegislativa, numero=20, ano=2018, data_apresentacao="2018-01-04", regime_tramitacao=regime_tramitacao, tipo=tipo_materia) materia_anexada = baker.make(MateriaLegislativa, numero=21, ano=2019, data_apresentacao="2019-05-04", regime_tramitacao=regime_tramitacao, tipo=tipo_materia) baker.make(Anexada, materia_principal=materia_principal, materia_anexada=materia_anexada, data_anexacao="2019-05-11") baker.make(Anexada, materia_principal=materia_anexada, materia_anexada=materia_principal, data_anexacao="2020-11-05") lista = lista_anexados(materia_principal) assert len(lista) == 1 assert lista[0] == materia_anexada
def test_lista_documentos_anexados(): tipo_documento = mommy.make(TipoDocumentoAdministrativo, descricao="Tipo_Teste") documento_principal = mommy.make(DocumentoAdministrativo, numero=20, ano=2018, data="2018-01-04", tipo=tipo_documento) documento_anexado = mommy.make(DocumentoAdministrativo, numero=21, ano=2019, data="2019-05-04", tipo=tipo_documento) documento_anexado_anexado = mommy.make(DocumentoAdministrativo, numero=22, ano=2020, data="2020-01-05", tipo=tipo_documento) mommy.make(Anexado, documento_principal=documento_principal, documento_anexado=documento_anexado, data_anexacao="2019-05-11") mommy.make(Anexado, documento_principal=documento_anexado, documento_anexado=documento_anexado_anexado, data_anexacao="2020-11-05") lista = lista_anexados(documento_principal, False) assert len(lista) == 2 assert lista[0] == documento_anexado assert lista[1] == documento_anexado_anexado