From 8a02c9aafa6573e27bf42d50bfa985ccf6cccb0b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Rodrigues?= Date: Wed, 27 May 2020 15:29:52 -0300 Subject: [PATCH] Fix #3168 - lib migration (#3169) --- requirements/test-requirements.txt | 2 +- sapl/audiencia/tests/test_audiencia.py | 14 +- sapl/base/tests/test_base.py | 4 +- sapl/base/tests/test_view_base.py | 150 +++++++++--------- sapl/comissoes/tests/test_comissoes.py | 20 +-- sapl/compilacao/tests/test_compilacao.py | 22 +-- .../tests/test_tipo_texto_articulado_form.py | 4 +- sapl/crud/tests/test_base.py | 12 +- sapl/materia/tests/test_materia.py | 106 ++++++------- sapl/materia/tests/test_materia_form.py | 12 +- sapl/norma/tests/test_norma.py | 20 +-- sapl/parlamentares/tests/test_mandato.py | 8 +- .../parlamentares/tests/test_parlamentares.py | 38 ++--- sapl/protocoloadm/tests/test_protocoloadm.py | 104 ++++++------ sapl/sessao/tests/test_sessao.py | 80 +++++----- sapl/sessao/tests/test_sessao_view.py | 30 ++-- sapl/test_general.py | 4 +- 17 files changed, 315 insertions(+), 315 deletions(-) diff --git a/requirements/test-requirements.txt b/requirements/test-requirements.txt index 5599b6d80..1a4ccd563 100644 --- a/requirements/test-requirements.txt +++ b/requirements/test-requirements.txt @@ -3,7 +3,7 @@ coverage==4.1 django-webtest==1.7.8 flake8==2.6.2 isort==4.2.5 -model-mommy==1.2.6 +model-bakery==1.1.0 pep8==1.7.0 pytest==2.9.2 pytest-cov==2.3.0 diff --git a/sapl/audiencia/tests/test_audiencia.py b/sapl/audiencia/tests/test_audiencia.py index b7c2e26a8..d217bf460 100644 --- a/sapl/audiencia/tests/test_audiencia.py +++ b/sapl/audiencia/tests/test_audiencia.py @@ -1,6 +1,6 @@ import pytest import datetime -from model_mommy import mommy +from model_bakery import baker from django.utils.translation import ugettext as _ from sapl.audiencia import forms @@ -11,7 +11,7 @@ from sapl.materia.models import MateriaLegislativa, TipoMateriaLegislativa @pytest.mark.django_db(transaction=False) def test_tipo_audiencia_publica_model(): - mommy.make(TipoAudienciaPublica, + baker.make(TipoAudienciaPublica, nome='Teste_Nome_Tipo_Audiencia_Publica', tipo='A') @@ -22,7 +22,7 @@ def test_tipo_audiencia_publica_model(): @pytest.mark.django_db(transaction=False) def test_audiencia_publica_model(): - mommy.make(AudienciaPublica, + baker.make(AudienciaPublica, numero=1, nome='Teste_Nome_Audiencia_Publica', tema='Teste_Tema_Audiencia_Publica', @@ -43,14 +43,14 @@ def test_audiencia_publica_model(): @pytest.mark.django_db(transaction=False) def test_anexo_audiencia_publica_model(): - audiencia = mommy.make(AudienciaPublica, + audiencia = baker.make(AudienciaPublica, numero=2, nome='Nome_Audiencia_Publica', tema='Tema_Audiencia_Publica', data='2017-04-22', hora_inicio='17:04') - mommy.make(AnexoAudienciaPublica, + baker.make(AnexoAudienciaPublica, audiencia=audiencia) anexo_audiencia_publica = AnexoAudienciaPublica.objects.first() @@ -76,9 +76,9 @@ def test_valida_campos_obrigatorios_audiencia_form(): @pytest.mark.django_db(transaction=False) def test_audiencia_form_hora_invalida(): - tipo_materia = mommy.make(TipoMateriaLegislativa) + tipo_materia = baker.make(TipoMateriaLegislativa) - tipo = mommy.make(TipoAudienciaPublica) + tipo = baker.make(TipoAudienciaPublica) form = forms.AudienciaForm(data={'nome': 'Nome da Audiencia', 'tema': 'Tema da Audiencia', diff --git a/sapl/base/tests/test_base.py b/sapl/base/tests/test_base.py index 296a241af..952de6f97 100644 --- a/sapl/base/tests/test_base.py +++ b/sapl/base/tests/test_base.py @@ -1,12 +1,12 @@ import pytest -from model_mommy import mommy +from model_bakery import baker from sapl.base.models import CasaLegislativa @pytest.mark.django_db(transaction=False) def test_casa_legislativa_model(): - mommy.make(CasaLegislativa, + baker.make(CasaLegislativa, nome='Teste_Nome_Casa_Legislativa', sigla='TSCL', endereco='Teste_Endereço_Casa_Legislativa', diff --git a/sapl/base/tests/test_view_base.py b/sapl/base/tests/test_view_base.py index 75748edd0..1020da2f4 100644 --- a/sapl/base/tests/test_view_base.py +++ b/sapl/base/tests/test_view_base.py @@ -1,9 +1,9 @@ import pytest -from model_mommy import mommy +from model_bakery import baker import datetime from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker from sapl.base.models import Autor, TipoAutor from sapl.comissoes.models import Comissao, TipoComissao @@ -26,26 +26,26 @@ from sapl.base.views import (protocolos_duplicados, protocolos_com_materias, @pytest.mark.django_db(transaction=False) def test_lista_protocolos_com_materias(): - mommy.make( + baker.make( Protocolo, numero=15, ano=2035 ) - mommy.make( + baker.make( Protocolo, numero=33, ano=2035 ) - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Materia_Teste" ) - regime_tramitacao = mommy.make( + regime_tramitacao = baker.make( RegimeTramitacao, descricao="Regime_Tramitacao_Teste" ) - mommy.make( + baker.make( MateriaLegislativa, numero=16, ano=2035, @@ -54,7 +54,7 @@ def test_lista_protocolos_com_materias(): tipo=tipo_materia, numero_protocolo=15 ) - mommy.make( + baker.make( MateriaLegislativa, numero=17, ano=2035, @@ -74,21 +74,21 @@ def test_lista_protocolos_com_materias(): @pytest.mark.django_db(transaction=False) def test_lista_materias_protocolo_inexistente(): - protocolo_a = mommy.make( + protocolo_a = baker.make( Protocolo, numero=15, ano=2031 ) - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Materia_Teste" ) - regime_tramitacao = mommy.make( + regime_tramitacao = baker.make( RegimeTramitacao, descricao="Regime_Tramitacao_Teste" ) - mommy.make( + baker.make( MateriaLegislativa, numero=16, ano=2031, @@ -97,7 +97,7 @@ def test_lista_materias_protocolo_inexistente(): tipo=tipo_materia, numero_protocolo=15 ) - materia = mommy.make( + materia = baker.make( MateriaLegislativa, numero=17, ano=2031, @@ -115,13 +115,13 @@ def test_lista_materias_protocolo_inexistente(): @pytest.mark.django_db(transaction=False) def test_lista_mandatos_sem_data_inicio(): - parlamentar = mommy.make( + parlamentar = baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste", nome_parlamentar="Nome_Parlamentar_Teste", sexo='M' ) - legislatura = mommy.make( + legislatura = baker.make( Legislatura, numero=1, data_inicio='2015-05-02', @@ -129,12 +129,12 @@ def test_lista_mandatos_sem_data_inicio(): data_eleicao='2015-02-05' ) - mandato_a = mommy.make( + mandato_a = baker.make( Mandato, parlamentar=parlamentar, legislatura=legislatura ) - mommy.make( + baker.make( Mandato, parlamentar=parlamentar, legislatura=legislatura, @@ -149,19 +149,19 @@ def test_lista_mandatos_sem_data_inicio(): @pytest.mark.django_db(transaction=False) def test_lista_parlamentares_duplicados(): - mommy.make( + baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste", nome_parlamentar="Nome_Parlamentar_Teste", sexo='M' ) - mommy.make( + baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste", nome_parlamentar="Nome_Parlamentar_Teste", sexo='M' ) - mommy.make( + baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste-1", nome_parlamentar="Nome_Parlamentar_Teste-1", @@ -180,47 +180,47 @@ def test_lista_parlamentares_duplicados(): @pytest.mark.django_db(transaction=False) def test_lista_parlamentares_mandatos_intersecao(): - legislatura = mommy.make( + legislatura = baker.make( Legislatura, numero=1, data_inicio='2017-07-04', data_fim='2170-05-01', data_eleicao='2017-04-07' ) - parlamentar_a = mommy.make( + parlamentar_a = baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste", nome_parlamentar="Nome_Parlamentar_Teste", sexo='M' ) - parlamentar_b = mommy.make( + parlamentar_b = baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste-1", nome_parlamentar="Nome_Parlamentar_Teste-1", sexo='M' ) - mandato_a = mommy.make( + mandato_a = baker.make( Mandato, parlamentar=parlamentar_a, legislatura=legislatura, data_inicio_mandato='2017-07-08', data_fim_mandato='2018-01-07' ) - mandato_b = mommy.make( + mandato_b = baker.make( Mandato, parlamentar=parlamentar_a, legislatura=legislatura, data_inicio_mandato='2017-07-09' ) - mommy.make( + baker.make( Mandato, parlamentar=parlamentar_b, legislatura=legislatura, data_inicio_mandato='2017-11-17', data_fim_mandato='2018-08-02' ) - mommy.make( + baker.make( Mandato, parlamentar=parlamentar_b, legislatura=legislatura, @@ -235,46 +235,46 @@ def test_lista_parlamentares_mandatos_intersecao(): @pytest.mark.django_db(transaction=False) def test_lista_parlamentares_filiacoes_intersecao(): - partido = mommy.make( + partido = baker.make( Partido, sigla="ST", nome="Nome_Partido_Teste" ) - parlamentar_a = mommy.make( + parlamentar_a = baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste", nome_parlamentar="Nome_Parlamentar_Teste", sexo='M' ) - parlamentar_b = mommy.make( + parlamentar_b = baker.make( Parlamentar, nome_completo="Nome_Completo_Parlamentar_Teste-1", nome_parlamentar="Nome_Parlamentar_Teste-1", sexo='M' ) - filiacao_a = mommy.make( + filiacao_a = baker.make( Filiacao, parlamentar=parlamentar_a, partido=partido, data='2018-02-02', data_desfiliacao='2019-08-01' ) - filiacao_b = mommy.make( + filiacao_b = baker.make( Filiacao, parlamentar=parlamentar_a, partido=partido, data='2018-02-23', data_desfiliacao='2020-02-04' ) - mommy.make( + baker.make( Filiacao, parlamentar=parlamentar_b, partido=partido, data='2018-02-07', data_desfiliacao='2018-02-27' ) - mommy.make( + baker.make( Filiacao, parlamentar=parlamentar_b, partido=partido, @@ -289,22 +289,22 @@ def test_lista_parlamentares_filiacoes_intersecao(): @pytest.mark.django_db(transaction=False) def test_lista_autores_duplicados(): - tipo_autor = mommy.make( + tipo_autor = baker.make( TipoAutor, descricao="Tipo_Autor_Teste" ) - mommy.make( + baker.make( Autor, tipo=tipo_autor, nome="Nome_Autor_Teste" ) - mommy.make( + baker.make( Autor, tipo=tipo_autor, nome="Nome_Autor_Teste" ) - mommy.make( + baker.make( Autor, tipo=tipo_autor, nome="Nome_Autor_Teste-1" @@ -319,16 +319,16 @@ def test_lista_autores_duplicados(): @pytest.mark.django_db(transaction=False) def test_lista_bancada_comissao_autor_externo(): - tipo_autor = mommy.make( + tipo_autor = baker.make( TipoAutor, descricao="Tipo_Autor_Teste" ) - tipo_autor_externo = mommy.make( + tipo_autor_externo = baker.make( TipoAutor, descricao="Externo" ) - legislatura = mommy.make( + legislatura = baker.make( Legislatura, numero=1, data_inicio='2012-01-03', @@ -336,7 +336,7 @@ def test_lista_bancada_comissao_autor_externo(): data_eleicao='2011-10-04' ) - bancada_a = mommy.make( + bancada_a = baker.make( Bancada, legislatura=legislatura, nome="Bancada_Teste", @@ -347,7 +347,7 @@ def test_lista_bancada_comissao_autor_externo(): tipo=tipo_autor ) - bancada_b = mommy.make( + bancada_b = baker.make( Bancada, legislatura=legislatura, nome="Bancada_Teste-1", @@ -358,14 +358,14 @@ def test_lista_bancada_comissao_autor_externo(): tipo=tipo_autor_externo ) - tipo_comissao = mommy.make( + tipo_comissao = baker.make( TipoComissao, nome="Tipo_Comissao_Teste", natureza='T', sigla="TCT" ) - comissao_a = mommy.make( + comissao_a = baker.make( Comissao, nome="Comissao_Teste", sigla="CT", @@ -376,7 +376,7 @@ def test_lista_bancada_comissao_autor_externo(): tipo=tipo_autor ) - comissao_b = mommy.make( + comissao_b = baker.make( Comissao, nome="Comissao_Teste-1", sigla="CT1", @@ -399,48 +399,48 @@ def test_lista_bancada_comissao_autor_externo(): @pytest.mark.django_db(transaction=False) def test_lista_anexados_ciclicas(): ## DocumentoAdministrativo - tipo_documento = mommy.make( + tipo_documento = baker.make( TipoDocumentoAdministrativo, sigla="TT", descricao="Tipo_Teste" ) - documento_a = mommy.make( + documento_a = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=26, ano=2019, data='2019-05-15', ) - documento_b = mommy.make( + documento_b = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=27, ano=2019, data='2019-05-16', ) - documento_c = mommy.make( + documento_c = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=28, ano=2019, data='2019-05-17', ) - documento_a1 = mommy.make( + documento_a1 = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=29, ano=2019, data='2019-05-18', ) - documento_b1 = mommy.make( + documento_b1 = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=30, ano=2019, data='2019-05-19', ) - documento_c1 = mommy.make( + documento_c1 = baker.make( DocumentoAdministrativo, tipo=tipo_documento, numero=31, @@ -448,43 +448,43 @@ def test_lista_anexados_ciclicas(): data='2019-05-20', ) - mommy.make( + baker.make( Anexado, documento_principal=documento_a, documento_anexado=documento_b, data_anexacao='2019-05-21' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_a, documento_anexado=documento_c, data_anexacao='2019-05-22' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_b, documento_anexado=documento_c, data_anexacao='2019-05-23' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_a1, documento_anexado=documento_b1, data_anexacao='2019-05-24' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_a1, documento_anexado=documento_c1, data_anexacao='2019-05-25' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_b1, documento_anexado=documento_c1, data_anexacao='2019-05-26' ) - mommy.make( + baker.make( Anexado, documento_principal=documento_c1, documento_anexado=documento_b1, @@ -494,16 +494,16 @@ def test_lista_anexados_ciclicas(): lista_documento_ciclicos = anexados_ciclicos(False) ## Matéria - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Teste" ) - regime_tramitacao = mommy.make( + regime_tramitacao = baker.make( RegimeTramitacao, descricao="Regime_Teste" ) - materia_a = mommy.make( + materia_a = baker.make( MateriaLegislativa, numero=20, ano=2018, @@ -511,7 +511,7 @@ def test_lista_anexados_ciclicas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_b = mommy.make( + materia_b = baker.make( MateriaLegislativa, numero=21, ano=2019, @@ -519,7 +519,7 @@ def test_lista_anexados_ciclicas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_c = mommy.make( + materia_c = baker.make( MateriaLegislativa, numero=22, ano=2019, @@ -527,7 +527,7 @@ def test_lista_anexados_ciclicas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_a1 = mommy.make( + materia_a1 = baker.make( MateriaLegislativa, numero=23, ano=2018, @@ -535,7 +535,7 @@ def test_lista_anexados_ciclicas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_b1 = mommy.make( + materia_b1 = baker.make( MateriaLegislativa, numero=24, ano=2019, @@ -543,7 +543,7 @@ def test_lista_anexados_ciclicas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_c1 = mommy.make( + materia_c1 = baker.make( MateriaLegislativa, numero=25, ano=2019, @@ -552,43 +552,43 @@ def test_lista_anexados_ciclicas(): tipo=tipo_materia ) - mommy.make( + baker.make( Anexada, materia_principal=materia_a, materia_anexada=materia_b, data_anexacao='2019-05-11' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_a, materia_anexada=materia_c, data_anexacao='2019-05-12' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_b, materia_anexada=materia_c, data_anexacao='2019-05-13' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_a1, materia_anexada=materia_b1, data_anexacao='2019-05-11' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_a1, materia_anexada=materia_c1, data_anexacao='2019-05-12' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_b1, materia_anexada=materia_c1, data_anexacao='2019-05-13' ) - mommy.make( + baker.make( Anexada, materia_principal=materia_c1, materia_anexada=materia_b1, diff --git a/sapl/comissoes/tests/test_comissoes.py b/sapl/comissoes/tests/test_comissoes.py index 274e36adb..705ff8a57 100644 --- a/sapl/comissoes/tests/test_comissoes.py +++ b/sapl/comissoes/tests/test_comissoes.py @@ -1,7 +1,7 @@ import pytest from django.core.urlresolvers import reverse from django.utils.translation import ugettext as _ -from model_mommy import mommy +from model_bakery import baker from sapl.comissoes.models import Comissao, Composicao, Periodo from sapl.comissoes.models import TipoComissao, Reuniao @@ -10,18 +10,18 @@ from sapl.comissoes import forms def make_composicao(comissao): - periodo = mommy.make(Periodo, + periodo = baker.make(Periodo, data_inicio='2016-01-01', data_fim='2016-12-31') - mommy.make(Composicao, + baker.make(Composicao, periodo=periodo, comissao=comissao) return Composicao.objects.first() def make_comissao(): - tipo = mommy.make(TipoComissao) - mommy.make(Comissao, + tipo = baker.make(TipoComissao) + baker.make(Comissao, tipo=tipo, nome='Comissão Teste', sigla='CT', @@ -30,15 +30,15 @@ def make_comissao(): def make_filiacao(): - partido = mommy.make(Partido, + partido = baker.make(Partido, nome='Partido Meu', sigla='PM') - parlamentar = mommy.make(Parlamentar, + parlamentar = baker.make(Parlamentar, nome_parlamentar='Eduardo', nome_completo='Eduardo', sexo='M', ativo=True) - mommy.make(Filiacao, + baker.make(Filiacao, data='2016-03-22', parlamentar=parlamentar, partido=partido) @@ -48,7 +48,7 @@ def make_filiacao(): @pytest.mark.django_db(transaction=False) def test_tipo_comissao_model(): - mommy.make(TipoComissao, + baker.make(TipoComissao, nome='Teste_Nome_Tipo_Comissao', natureza='T', sigla='TSTC') @@ -80,7 +80,7 @@ def test_incluir_parlamentar_errors(admin_client): @pytest.mark.django_db(transaction=False) def test_incluir_comissao_submit(admin_client): - tipo = mommy.make(TipoComissao, + tipo = baker.make(TipoComissao, sigla='T', nome='Teste') diff --git a/sapl/compilacao/tests/test_compilacao.py b/sapl/compilacao/tests/test_compilacao.py index 4eacb8cb1..fec382392 100644 --- a/sapl/compilacao/tests/test_compilacao.py +++ b/sapl/compilacao/tests/test_compilacao.py @@ -1,5 +1,5 @@ import pytest -from model_mommy import mommy +from model_bakery import baker from sapl.compilacao.models import PerfilEstruturalTextoArticulado from sapl.compilacao.models import TipoTextoArticulado @@ -10,7 +10,7 @@ from sapl.compilacao.models import TipoDispositivoRelationship @pytest.mark.django_db(transaction=False) def test_perfil_estrutural_texto_articulado_model(): - perfil_estrutural_texto_articulado = mommy.make( + perfil_estrutural_texto_articulado = baker.make( PerfilEstruturalTextoArticulado, nome='Teste_Nome_Perfil', sigla='TSPETA') @@ -21,7 +21,7 @@ def test_perfil_estrutural_texto_articulado_model(): @pytest.mark.django_db(transaction=False) def test_tipo_texto_articulado_model(): - tipo_texto_articulado = mommy.make( + tipo_texto_articulado = baker.make( TipoTextoArticulado, sigla='TTP', descricao='T_Desc_Tipo_Texto_Articulado' @@ -33,7 +33,7 @@ def test_tipo_texto_articulado_model(): @pytest.mark.django_db(transaction=False) def test_texto_articulado_model(): - texto_articulado = mommy.make( + texto_articulado = baker.make( TextoArticulado, ementa='Teste_Ementa_Texto_Articulado', numero='12345678', @@ -47,7 +47,7 @@ def test_texto_articulado_model(): @pytest.mark.django_db(transaction=False) def test_tipo_nota_model(): - tipo_nota = mommy.make( + tipo_nota = baker.make( TipoNota, sigla='TTN', nome='Teste_Nome_Tipo_Nota' @@ -59,7 +59,7 @@ def test_tipo_nota_model(): @pytest.mark.django_db(transaction=False) def test_tipo_vide_model(): - tipo_vide = mommy.make( + tipo_vide = baker.make( TipoVide, sigla='TTV', nome='Teste_Nome_Tipo_Vide' @@ -71,7 +71,7 @@ def test_tipo_vide_model(): @pytest.mark.django_db(transaction=False) def test_tipo_dispositivo_model(): - tipo_dispositivo = mommy.make( + tipo_dispositivo = baker.make( TipoDispositivo, nome='Teste_Nome_Tipo_Dispositivo', rotulo_ordinal=0 @@ -83,24 +83,24 @@ def test_tipo_dispositivo_model(): @pytest.mark.django_db(transaction=False) def test_tipo_dispositivo_relationship_model(): - tipo_dispositivo_pai = mommy.make( + tipo_dispositivo_pai = baker.make( TipoDispositivo, nome='Tipo_Dispositivo_Pai', rotulo_ordinal=0 ) - t_dispositivo_filho = mommy.make( + t_dispositivo_filho = baker.make( TipoDispositivo, nome='Tipo_Dispositivo_Filho', rotulo_ordinal=0 ) - p_e_texto_articulado = mommy.make( + p_e_texto_articulado = baker.make( PerfilEstruturalTextoArticulado, nome='Teste_Nome_Perfil', sigla='TSPETA') - tipo_dispositivo_relationship = mommy.make( + tipo_dispositivo_relationship = baker.make( TipoDispositivoRelationship, pai=tipo_dispositivo_pai, filho_permitido=t_dispositivo_filho, diff --git a/sapl/compilacao/tests/test_tipo_texto_articulado_form.py b/sapl/compilacao/tests/test_tipo_texto_articulado_form.py index 6c64adc78..1377966ff 100644 --- a/sapl/compilacao/tests/test_tipo_texto_articulado_form.py +++ b/sapl/compilacao/tests/test_tipo_texto_articulado_form.py @@ -1,6 +1,6 @@ import pytest from django.utils.translation import ugettext as _ -from model_mommy import mommy +from model_bakery import baker from sapl.compilacao import forms from sapl.compilacao.models import PerfilEstruturalTextoArticulado, TipoNota @@ -41,7 +41,7 @@ def test_valida_campos_obrigatorios_nota_form(): @pytest.mark.django_db(transaction=False) def test_nota_form_invalido(): - tipo = mommy.make(TipoNota) + tipo = baker.make(TipoNota) form = forms.NotaForm(data={'titulo': 'titulo', 'texto': 'teste', diff --git a/sapl/crud/tests/test_base.py b/sapl/crud/tests/test_base.py index 8e1d10965..89e485fae 100644 --- a/sapl/crud/tests/test_base.py +++ b/sapl/crud/tests/test_base.py @@ -1,6 +1,6 @@ import pytest from django.core.urlresolvers import reverse -from model_mommy import mommy +from model_bakery import baker from sapl.crud.base import (CrispyLayoutFormMixin, CrudListView, from_to, get_field_display, make_pagination) @@ -55,7 +55,7 @@ def test_make_pagination(index, num_pages, result): def test_get_field_display(): - stub = mommy.prepare(Country, is_cold=True) + stub = baker.prepare(Country, is_cold=True) assert get_field_display(stub, 'name')[1] == stub.name assert get_field_display(stub, 'continent')[1] == str(stub.continent) # must return choice display, not the value @@ -88,7 +88,7 @@ def test_layout_fieldnames(_layout, result): def test_layout_detail_fieldsets(): - stub = mommy.make(Country, + stub = baker.make(Country, name='Brazil', continent__name='South America', is_cold=False) @@ -191,7 +191,7 @@ def test_flux_list_paginate_detail( population, is_cold = i, i % 2 == 0 entries_labels.append([ name, continent, str(population), 'Yes' if is_cold else 'No']) - mommy.make(Country, + baker.make(Country, name=name, continent__name=continent, population=population, @@ -252,7 +252,7 @@ def test_flux_list_paginate_detail( def test_flux_list_create_detail(app, cancel, make_invalid_submit): # to have a couple an option for continent field - stub_continent = mommy.make(Continent) + stub_continent = baker.make(Continent) res = app.get('/country/') @@ -304,7 +304,7 @@ def test_flux_list_create_detail(app, cancel, make_invalid_submit): def get_detail_page(app): - stub = mommy.make(Country, name='Country Stub') + stub = baker.make(Country, name='Country Stub') res = app.get('/country/%s' % stub.id) # on detail page assert_on_detail_page(res, stub.name) diff --git a/sapl/materia/tests/test_materia.py b/sapl/materia/tests/test_materia.py index c6a4f542d..e9c7ae2db 100644 --- a/sapl/materia/tests/test_materia.py +++ b/sapl/materia/tests/test_materia.py @@ -4,7 +4,7 @@ from django.contrib.contenttypes.models import ContentType from django.core.files.uploadedfile import SimpleUploadedFile from django.core.urlresolvers import reverse from django.db.models import Max -from model_mommy import mommy +from model_bakery import baker import pytest from sapl.base.models import Autor, TipoAutor, AppConfig @@ -25,15 +25,15 @@ from sapl.utils import models_with_gr_for_model, lista_anexados @pytest.mark.django_db(transaction=False) def test_lista_materias_anexadas(): - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Teste" ) - regime_tramitacao = mommy.make( + regime_tramitacao = baker.make( RegimeTramitacao, descricao="Regime_Teste" ) - materia_principal = mommy.make( + materia_principal = baker.make( MateriaLegislativa, numero=20, ano=2018, @@ -41,7 +41,7 @@ def test_lista_materias_anexadas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_anexada = mommy.make( + materia_anexada = baker.make( MateriaLegislativa, numero=21, ano=2019, @@ -49,7 +49,7 @@ def test_lista_materias_anexadas(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_anexada_anexada = mommy.make( + materia_anexada_anexada = baker.make( MateriaLegislativa, numero=22, ano=2020, @@ -58,13 +58,13 @@ def test_lista_materias_anexadas(): tipo=tipo_materia ) - mommy.make( + baker.make( Anexada, materia_principal=materia_principal, materia_anexada=materia_anexada, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexada, materia_principal=materia_anexada, materia_anexada=materia_anexada_anexada, @@ -80,15 +80,15 @@ def test_lista_materias_anexadas(): @pytest.mark.django_db(transaction=False) def test_lista_materias_anexadas_ciclo(): - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Teste" ) - regime_tramitacao = mommy.make( + regime_tramitacao = baker.make( RegimeTramitacao, descricao="Regime_Teste" ) - materia_principal = mommy.make( + materia_principal = baker.make( MateriaLegislativa, numero=20, ano=2018, @@ -96,7 +96,7 @@ def test_lista_materias_anexadas_ciclo(): regime_tramitacao=regime_tramitacao, tipo=tipo_materia ) - materia_anexada = mommy.make( + materia_anexada = baker.make( MateriaLegislativa, numero=21, ano=2019, @@ -105,13 +105,13 @@ def test_lista_materias_anexadas_ciclo(): tipo=tipo_materia ) - mommy.make( + baker.make( Anexada, materia_principal=materia_principal, materia_anexada=materia_anexada, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexada, materia_principal=materia_anexada, materia_anexada=materia_principal, @@ -126,8 +126,8 @@ def test_lista_materias_anexadas_ciclo(): @pytest.mark.django_db(transaction=False) def make_unidade_tramitacao(descricao): # Cria uma comissão para ser a unidade de tramitação - tipo_comissao = mommy.make(TipoComissao) - comissao = mommy.make(Comissao, + tipo_comissao = baker.make(TipoComissao) + comissao = baker.make(Comissao, tipo=tipo_comissao, nome=descricao, sigla='T', @@ -138,7 +138,7 @@ def make_unidade_tramitacao(descricao): assert comissao.nome == descricao # Cria a unidade - unidade = mommy.make(UnidadeTramitacao, comissao=comissao) + unidade = baker.make(UnidadeTramitacao, comissao=comissao) assert unidade.comissao == comissao return unidade @@ -147,10 +147,10 @@ def make_unidade_tramitacao(descricao): @pytest.mark.django_db(transaction=False) def make_norma(): # Cria um novo tipo de Norma - tipo = mommy.make(TipoNormaJuridica, + tipo = baker.make(TipoNormaJuridica, sigla='T1', descricao='Teste_Tipo_Norma') - mommy.make(NormaJuridica, + baker.make(NormaJuridica, tipo=tipo, numero=1, ano=2016, @@ -169,13 +169,13 @@ def make_norma(): @pytest.mark.django_db(transaction=False) def make_materia_principal(): - regime_tramitacao = mommy.make(RegimeTramitacao, descricao='Teste_Regime') + regime_tramitacao = baker.make(RegimeTramitacao, descricao='Teste_Regime') # Cria a matéria principal - tipo = mommy.make(TipoMateriaLegislativa, + tipo = baker.make(TipoMateriaLegislativa, sigla='T1', descricao='Teste_MateriaLegislativa') - mommy.make(MateriaLegislativa, + baker.make(MateriaLegislativa, tipo=tipo, numero='165', ano='2002', @@ -195,11 +195,11 @@ def test_materia_anexada_submit(admin_client): materia_principal = make_materia_principal() # Cria a matéria que será anexada - tipo_anexada = mommy.make(TipoMateriaLegislativa, + tipo_anexada = baker.make(TipoMateriaLegislativa, sigla='T2', descricao='Teste_2') - regime_tramitacao = mommy.make(RegimeTramitacao, descricao='Teste_Regime') - mommy.make(MateriaLegislativa, + regime_tramitacao = baker.make(RegimeTramitacao, descricao='Teste_Regime') + baker.make(MateriaLegislativa, tipo=tipo_anexada, numero='32', ano='2004', @@ -230,10 +230,10 @@ def test_materia_anexada_submit(admin_client): def test_autoria_submit(admin_client): materia_principal = make_materia_principal() # Cria um tipo de Autor - tipo_autor = mommy.make(TipoAutor, descricao='Teste Tipo_Autor') + tipo_autor = baker.make(TipoAutor, descricao='Teste Tipo_Autor') # Cria um Autor - autor = mommy.make( + autor = baker.make( Autor, tipo=tipo_autor, nome='Autor Teste') @@ -261,8 +261,8 @@ def test_despacho_inicial_submit(admin_client): materia_principal = make_materia_principal() # Cria uma comissão - tipo_comissao = mommy.make(TipoComissao) - comissao = mommy.make(Comissao, + tipo_comissao = baker.make(TipoComissao) + comissao = baker.make(Comissao, tipo=tipo_comissao, nome='Teste', ativa=True, @@ -312,16 +312,16 @@ def test_documento_acessorio_submit(admin_client): materia_principal = make_materia_principal() # Cria um tipo de Autor - tipo_autor = mommy.make(TipoAutor, descricao='Teste Tipo_Autor') + tipo_autor = baker.make(TipoAutor, descricao='Teste Tipo_Autor') # Cria um Autor - autor = mommy.make( + autor = baker.make( Autor, tipo=tipo_autor, nome='Autor Teste') # Cria um tipo de documento - tipo = mommy.make(TipoDocumento, + tipo = baker.make(TipoDocumento, descricao='Teste') # Testa POST @@ -373,7 +373,7 @@ def test_legislacao_citada_submit(admin_client): def test_tramitacao_submit(admin_client): materia_principal = make_materia_principal() # Cria status para tramitação - status_tramitacao = mommy.make(StatusTramitacao, + status_tramitacao = baker.make(StatusTramitacao, indicador='F', sigla='ST', descricao='Status_Teste') @@ -541,10 +541,10 @@ def test_form_errors_relatoria(admin_client): @pytest.mark.django_db(transaction=False) def test_proposicao_submit(admin_client): - tipo_autor = mommy.make(TipoAutor, descricao='Teste Tipo_Autor') + tipo_autor = baker.make(TipoAutor, descricao='Teste Tipo_Autor') user = get_user_model().objects.filter(is_active=True)[0] - autor = mommy.make( + autor = baker.make( Autor, user=user, tipo=tipo_autor, @@ -557,11 +557,11 @@ def test_proposicao_submit(admin_client): *models_with_gr_for_model(TipoProposicao)) for pk, mct in enumerate(mcts): - tipo_conteudo_related = mommy.make(mct, pk=pk + 1) + tipo_conteudo_related = baker.make(mct, pk=pk + 1) response = admin_client.post( reverse('sapl.materia:proposicao_create'), - {'tipo': mommy.make( + {'tipo': baker.make( TipoProposicao, pk=3, tipo_conteudo_related=tipo_conteudo_related).pk, 'descricao': 'Teste proposição', @@ -586,11 +586,11 @@ def test_proposicao_submit(admin_client): @pytest.mark.django_db(transaction=False) def test_form_errors_proposicao(admin_client): - tipo_autor = mommy.make(TipoAutor, descricao='Teste Tipo_Autor') + tipo_autor = baker.make(TipoAutor, descricao='Teste Tipo_Autor') user = get_user_model().objects.filter(is_active=True)[0] - autor = mommy.make( + autor = baker.make( Autor, user=user, tipo=tipo_autor, @@ -616,13 +616,13 @@ def test_form_errors_proposicao(admin_client): def test_numeracao_materia_legislativa_por_legislatura(admin_client): # Criar Legislaturas - legislatura1 = mommy.make(Legislatura, + legislatura1 = baker.make(Legislatura, data_inicio='2014-01-01', data_fim='2018-12-31', numero=20, data_eleicao='2013-10-15' ) - legislatura2 = mommy.make(Legislatura, + legislatura2 = baker.make(Legislatura, data_inicio='2009-01-01', data_fim='2013-12-31', numero=21, @@ -630,9 +630,9 @@ def test_numeracao_materia_legislativa_por_legislatura(admin_client): ) # Cria uma materia na legislatura1 - tipo_materia = mommy.make(TipoMateriaLegislativa, + tipo_materia = baker.make(TipoMateriaLegislativa, id=1, sequencia_numeracao='L') - materia = mommy.make(MateriaLegislativa, + materia = baker.make(MateriaLegislativa, tipo=tipo_materia, ano=2017, numero=1, @@ -660,9 +660,9 @@ def test_numeracao_materia_legislativa_por_legislatura(admin_client): def test_numeracao_materia_legislativa_por_ano(admin_client): # Cria uma materia - tipo_materia = mommy.make(TipoMateriaLegislativa, + tipo_materia = baker.make(TipoMateriaLegislativa, id=1, sequencia_numeracao='A') - materia = mommy.make(MateriaLegislativa, + materia = baker.make(MateriaLegislativa, tipo=tipo_materia, ano=2017, numero=1 @@ -687,38 +687,38 @@ def test_numeracao_materia_legislativa_por_ano(admin_client): @pytest.mark.django_db(transaction=False) def test_tramitacoes_materias_anexadas(admin_client): - config = mommy.make(AppConfig, tramitacao_materia=True) + config = baker.make(AppConfig, tramitacao_materia=True) - tipo_materia = mommy.make( + tipo_materia = baker.make( TipoMateriaLegislativa, descricao="Tipo_Teste" ) - materia_principal = mommy.make( + materia_principal = baker.make( MateriaLegislativa, ano=2018, data_apresentacao="2018-01-04", tipo=tipo_materia ) - materia_anexada = mommy.make( + materia_anexada = baker.make( MateriaLegislativa, ano=2019, data_apresentacao="2019-05-04", tipo=tipo_materia ) - materia_anexada_anexada = mommy.make( + materia_anexada_anexada = baker.make( MateriaLegislativa, ano=2020, data_apresentacao="2020-01-05", tipo=tipo_materia ) - mommy.make( + baker.make( Anexada, materia_principal=materia_principal, materia_anexada=materia_anexada, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexada, materia_principal=materia_anexada, materia_anexada=materia_anexada_anexada, @@ -730,7 +730,7 @@ def test_tramitacoes_materias_anexadas(admin_client): unidade_tramitacao_destino_1 = make_unidade_tramitacao(descricao="Teste 2") unidade_tramitacao_destino_2 = make_unidade_tramitacao(descricao="Teste 3") - status = mommy.make( + status = baker.make( StatusTramitacao, indicador='R') diff --git a/sapl/materia/tests/test_materia_form.py b/sapl/materia/tests/test_materia_form.py index 6392ca05b..6e1d7b3ba 100644 --- a/sapl/materia/tests/test_materia_form.py +++ b/sapl/materia/tests/test_materia_form.py @@ -1,6 +1,6 @@ import pytest from django.utils.translation import ugettext as _ -from model_mommy import mommy +from model_bakery import baker from sapl.comissoes.models import Comissao, TipoComissao from sapl.materia import forms @@ -24,7 +24,7 @@ def test_valida_campos_obrigatorios_ficha_pesquisa_form(): @pytest.mark.django_db(transaction=False) def test_ficha_pesquisa_form_datas_invalidas(): - tipo = mommy.make(TipoMateriaLegislativa) + tipo = baker.make(TipoMateriaLegislativa) form = forms.FichaPesquisaForm(data={'tipo_materia': str(tipo.pk), 'data_inicial': '10/11/2017', @@ -37,7 +37,7 @@ def test_ficha_pesquisa_form_datas_invalidas(): @pytest.mark.django_db(transaction=False) def test_ficha_pesquisa_form_invalido(): - tipo = mommy.make(TipoMateriaLegislativa) + tipo = baker.make(TipoMateriaLegislativa) form = forms.FichaPesquisaForm(data={'tipo_materia': str(tipo.pk), 'data_inicial': '10/11/2017', @@ -62,7 +62,7 @@ def test_valida_campos_obrigatorios_ficha_seleciona_form(): @pytest.mark.django_db(transaction=False) def test_ficha_seleciona_form_valido(): - materia = mommy.make(MateriaLegislativa) + materia = baker.make(MateriaLegislativa) form = forms.FichaSelecionaForm(data={'materia': str(materia.pk)}) @@ -178,8 +178,8 @@ def test_valida_campos_obrigatorios_devolver_proposicao_form(): @pytest.mark.django_db(transaction=False) def test_valida_campos_obrigatorios_relatoria_form(): - tipo_comissao = mommy.make(TipoComissao) - comissao = mommy.make(Comissao, + tipo_comissao = baker.make(TipoComissao) + comissao = baker.make(Comissao, tipo=tipo_comissao, nome='Comissao Teste', sigla='T', diff --git a/sapl/norma/tests/test_norma.py b/sapl/norma/tests/test_norma.py index 277ed47ed..efbab6531 100644 --- a/sapl/norma/tests/test_norma.py +++ b/sapl/norma/tests/test_norma.py @@ -1,6 +1,6 @@ from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker import pytest from sapl.base.models import AppConfig @@ -13,10 +13,10 @@ from sapl.norma.models import NormaJuridica, TipoNormaJuridica @pytest.mark.django_db(transaction=False) def test_incluir_norma_submit(admin_client): # Cria um tipo de norma - tipo = mommy.make(TipoNormaJuridica, + tipo = baker.make(TipoNormaJuridica, sigla='T', descricao='Teste') - config = mommy.make(AppConfig) + config = baker.make(AppConfig) # Testa POST response = admin_client.post(reverse('sapl.norma:normajuridica_create'), @@ -78,12 +78,12 @@ def test_norma_form_invalida(): @pytest.mark.django_db(transaction=False) def test_norma_juridica_materia_inexistente(): - tipo = mommy.make(TipoNormaJuridica) - tipo_materia = mommy.make(TipoMateriaLegislativa, descricao='VETO') + tipo = baker.make(TipoNormaJuridica) + tipo_materia = baker.make(TipoMateriaLegislativa, descricao='VETO') # cria uma matéria qualquer em 2017 pois, no teste, o campo ano_materia # está vazio - materia = mommy.make(MateriaLegislativa, + materia = baker.make(MateriaLegislativa, tipo=tipo_materia, ano=2017, numero=1, @@ -109,9 +109,9 @@ def test_norma_juridica_materia_inexistente(): @pytest.mark.django_db(transaction=False) def test_norma_juridica_materia_existente(): - tipo = mommy.make(TipoNormaJuridica) - tipo_materia = mommy.make(TipoMateriaLegislativa) - mommy.make(MateriaLegislativa, + tipo = baker.make(TipoNormaJuridica) + tipo_materia = baker.make(TipoMateriaLegislativa) + baker.make(MateriaLegislativa, numero=2, ano=2017, tipo=tipo_materia) @@ -147,7 +147,7 @@ def test_norma_relacionada_form_campos_obrigatorios(): @pytest.mark.django_db(transaction=False) def test_norma_pesquisa_form_datas_invalidas(): - tipo = mommy.make(TipoNormaJuridica) + tipo = baker.make(TipoNormaJuridica) form = NormaPesquisaSimplesForm(data={'tipo_norma': str(tipo.pk), 'data_inicial': '10/11/2017', diff --git a/sapl/parlamentares/tests/test_mandato.py b/sapl/parlamentares/tests/test_mandato.py index cc027da1f..5151f9702 100644 --- a/sapl/parlamentares/tests/test_mandato.py +++ b/sapl/parlamentares/tests/test_mandato.py @@ -1,7 +1,7 @@ from datetime import datetime import pytest -from model_mommy import mommy +from model_bakery import baker from sapl.parlamentares.models import Filiacao, Legislatura, Mandato @@ -13,12 +13,12 @@ def data(valor): def test_filiacoes(): - legislatura = mommy.make(Legislatura, + legislatura = baker.make(Legislatura, data_inicio=data('2001-01-01'), data_fim=data('2001-12-31'), ) - mandato = mommy.make(Mandato, legislatura=legislatura) - f1_fora, f2, f3, f4 = [mommy.make(Filiacao, + mandato = baker.make(Mandato, legislatura=legislatura) + f1_fora, f2, f3, f4 = [baker.make(Filiacao, parlamentar=mandato.parlamentar, data=ini, data_desfiliacao=fim) diff --git a/sapl/parlamentares/tests/test_parlamentares.py b/sapl/parlamentares/tests/test_parlamentares.py index 8ebd443be..3cb2b7d90 100644 --- a/sapl/parlamentares/tests/test_parlamentares.py +++ b/sapl/parlamentares/tests/test_parlamentares.py @@ -1,7 +1,7 @@ import pytest from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker from sapl.parlamentares.forms import FrenteForm, LegislaturaForm, MandatoForm from sapl.parlamentares.models import (Dependente, Filiacao, Legislatura, @@ -42,8 +42,8 @@ def test_incluir_parlamentar_errors(admin_client): @pytest.mark.django_db(transaction=False) def test_filiacao_submit(admin_client): - mommy.make(Parlamentar, pk=14) - mommy.make(Partido, pk=32) + baker.make(Parlamentar, pk=14) + baker.make(Partido, pk=32) admin_client.post(reverse('sapl.parlamentares:filiacao_create', kwargs={'pk': 14}), @@ -58,9 +58,9 @@ def test_filiacao_submit(admin_client): @pytest.mark.django_db(transaction=False) def test_dependente_submit(admin_client): - mommy.make(Parlamentar, pk=14) - mommy.make(Partido, pk=32) - mommy.make(TipoDependente, pk=3) + baker.make(Parlamentar, pk=14) + baker.make(Partido, pk=32) + baker.make(TipoDependente, pk=3) admin_client.post(reverse('sapl.parlamentares:dependente_create', kwargs={'pk': 14}), @@ -77,7 +77,7 @@ def test_dependente_submit(admin_client): @pytest.mark.django_db(transaction=False) def test_form_errors_dependente(admin_client): - mommy.make(Parlamentar, pk=14) + baker.make(Parlamentar, pk=14) response = admin_client.post( reverse('sapl.parlamentares:dependente_create', kwargs={'pk': 14}), @@ -94,7 +94,7 @@ def test_form_errors_dependente(admin_client): @pytest.mark.django_db(transaction=False) def test_form_errors_filiacao(admin_client): - mommy.make(Parlamentar, pk=14) + baker.make(Parlamentar, pk=14) response = admin_client.post(reverse('sapl.parlamentares:filiacao_create', kwargs={'pk': 14}), @@ -110,8 +110,8 @@ def test_form_errors_filiacao(admin_client): @pytest.mark.django_db(transaction=False) def test_mandato_submit(admin_client): - mommy.make(Parlamentar, pk=14) - mommy.make(Legislatura, pk=5) + baker.make(Parlamentar, pk=14) + baker.make(Legislatura, pk=5) admin_client.post(reverse('sapl.parlamentares:mandato_create', kwargs={'pk': 14}), @@ -134,7 +134,7 @@ def test_mandato_submit(admin_client): @pytest.mark.django_db(transaction=False) def test_form_errors_mandato(admin_client): - mommy.make(Parlamentar, pk=14) + baker.make(Parlamentar, pk=14) response = admin_client.post(reverse('sapl.parlamentares:mandato_create', kwargs={'pk': 14}), {'legislatura': '', @@ -158,8 +158,8 @@ def test_mandato_form_invalido(): @pytest.mark.django_db(transaction=False) def test_mandato_form_duplicado(): - parlamentar = mommy.make(Parlamentar, pk=1) - legislatura = mommy.make(Legislatura, pk=1) + parlamentar = baker.make(Parlamentar, pk=1) + legislatura = baker.make(Legislatura, pk=1) Mandato.objects.create(parlamentar=parlamentar, legislatura=legislatura, @@ -184,8 +184,8 @@ def test_mandato_form_duplicado(): @pytest.mark.django_db(transaction=False) def test_mandato_form_datas_invalidas(): - parlamentar = mommy.make(Parlamentar, pk=1) - legislatura = mommy.make(Legislatura, pk=1, + parlamentar = baker.make(Parlamentar, pk=1) + legislatura = baker.make(Legislatura, pk=1, data_inicio='2017-01-01', data_fim='2021-12-31') @@ -275,7 +275,7 @@ def test_legislatura_form_numeros_invalidos(): assert legislatura_form.is_valid() - legislatura = mommy.make(Legislatura, pk=1, + legislatura = baker.make(Legislatura, pk=1, numero=5, data_inicio='2017-02-01', data_fim='2021-12-31', @@ -306,13 +306,13 @@ def test_legislatura_form_numeros_invalidos(): assert legislatura_form.is_valid() - legislatura = mommy.make(Legislatura, pk=2, + legislatura = baker.make(Legislatura, pk=2, numero=1, data_inicio='2002-02-01', data_fim='2005-12-31', data_eleicao='2001-11-01') - legislatura2 = mommy.make(Legislatura, pk=3, + legislatura2 = baker.make(Legislatura, pk=3, numero=3, data_inicio='2008-02-01', data_fim='2011-12-31', @@ -344,7 +344,7 @@ def test_valida_campos_obrigatorios_frente_form(): @pytest.mark.django_db(transaction=False) def test_frente_form_valido(): - parlamentares = mommy.make(Parlamentar) + parlamentares = baker.make(Parlamentar) form = FrenteForm(data={'nome': 'Nome da Frente', 'parlamentar': str(parlamentares.pk), diff --git a/sapl/protocoloadm/tests/test_protocoloadm.py b/sapl/protocoloadm/tests/test_protocoloadm.py index 631c56243..1e0ab091b 100644 --- a/sapl/protocoloadm/tests/test_protocoloadm.py +++ b/sapl/protocoloadm/tests/test_protocoloadm.py @@ -4,7 +4,7 @@ from django.core.urlresolvers import reverse from django.utils import timezone from django.utils.encoding import force_text from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker from urllib.parse import urlencode import pytest @@ -34,7 +34,7 @@ def test_anular_protocolo_acessivel(admin_client): @pytest.mark.django_db(transaction=False) def test_anular_protocolo_submit(admin_client): - mommy.make(Protocolo, numero='76', ano='2016', anulado=False) + baker.make(Protocolo, numero='76', ano='2016', anulado=False) # TODO: setar usuario e IP response = admin_client.post(reverse('sapl.protocoloadm:anular_protocolo'), @@ -69,7 +69,7 @@ def test_form_anular_protocolo_inexistente(): @pytest.mark.django_db(transaction=False) def test_form_anular_protocolo_valido(): - mommy.make(Protocolo, numero='1', ano='2016', anulado=False) + baker.make(Protocolo, numero='1', ano='2016', anulado=False) form = AnularProtocoloAdmForm({'numero': '1', 'ano': '2016', 'justificativa_anulacao': 'TESTE'}) @@ -79,7 +79,7 @@ def test_form_anular_protocolo_valido(): @pytest.mark.django_db(transaction=False) def test_form_anular_protocolo_anulado(): - mommy.make(Protocolo, numero='1', ano='2016', anulado=True) + baker.make(Protocolo, numero='1', ano='2016', anulado=True) form = AnularProtocoloAdmForm({'numero': '1', 'ano': '2016', 'justificativa_anulacao': 'TESTE'}) @@ -89,7 +89,7 @@ def test_form_anular_protocolo_anulado(): @pytest.mark.django_db(transaction=False) def test_form_anular_protocolo_campos_obrigatorios(): - mommy.make(Protocolo, numero='1', ano='2016', anulado=False) + baker.make(Protocolo, numero='1', ano='2016', anulado=False) # TODO: generalizar para diminuir o tamanho deste método @@ -127,27 +127,27 @@ def test_form_anular_protocolo_campos_obrigatorios(): @pytest.mark.django_db(transaction=False) def test_create_tramitacao(admin_client): - tipo_doc = mommy.make( + tipo_doc = baker.make( TipoDocumentoAdministrativo, descricao='Teste Tipo_DocAdm') - documento_adm = mommy.make( + documento_adm = baker.make( DocumentoAdministrativo, tipo=tipo_doc) - unidade_tramitacao_local_1 = mommy.make( + unidade_tramitacao_local_1 = baker.make( UnidadeTramitacao, pk=1) - unidade_tramitacao_destino_1 = mommy.make( + unidade_tramitacao_destino_1 = baker.make( UnidadeTramitacao, pk=2) - unidade_tramitacao_destino_2 = mommy.make( + unidade_tramitacao_destino_2 = baker.make( UnidadeTramitacao, pk=3) - status = mommy.make( + status = baker.make( StatusTramitacaoAdministrativo) - tramitacao = mommy.make( + tramitacao = baker.make( TramitacaoAdministrativo, unidade_tramitacao_local=unidade_tramitacao_local_1, unidade_tramitacao_destino=unidade_tramitacao_destino_1, @@ -307,7 +307,7 @@ def test_anular_protocolo_form_anula_protocolo_inexistente(): @pytest.mark.django_db(transaction=False) def test_anular_protocolo_form_anula_protocolo_anulado(): - mommy.make(Protocolo, numero=1, ano=2017, anulado=True) + baker.make(Protocolo, numero=1, ano=2017, anulado=True) form = AnularProtocoloAdmForm(data={'numero': '1', 'ano': '2017', @@ -322,15 +322,15 @@ def test_anular_protocolo_form_anula_protocolo_anulado(): @pytest.mark.django_db(transaction=False) def test_anular_protocolo_form_anula_protocolo_com_doc_vinculado(): - tipo_materia = mommy.make(TipoMateriaLegislativa) + tipo_materia = baker.make(TipoMateriaLegislativa) - mommy.make(Protocolo, + baker.make(Protocolo, numero=1, ano=2017, tipo_materia=tipo_materia, anulado=False) - mommy.make(MateriaLegislativa, + baker.make(MateriaLegislativa, ano=2017, numero_protocolo=1) @@ -345,15 +345,15 @@ def test_anular_protocolo_form_anula_protocolo_com_doc_vinculado(): [_("Protocolo 1/2017 não pode ser removido pois existem " "documentos vinculados a ele.")] - tipo_documento = mommy.make(TipoDocumentoAdministrativo) + tipo_documento = baker.make(TipoDocumentoAdministrativo) - protocolo_documento = mommy.make(Protocolo, + protocolo_documento = baker.make(Protocolo, numero=2, ano=2017, tipo_documento=tipo_documento, anulado=False) - mommy.make(DocumentoAdministrativo, + baker.make(DocumentoAdministrativo, protocolo=protocolo_documento) form = AnularProtocoloAdmForm(data={'numero': '2', @@ -387,8 +387,8 @@ def test_documento_administrativo_invalido(): @pytest.mark.django_db(transaction=False) def test_documento_administrativo_protocolo_inexistente(): - tipo = mommy.make(TipoDocumentoAdministrativo) - protocolo = mommy.make(Protocolo, + tipo = baker.make(TipoDocumentoAdministrativo) + protocolo = baker.make(Protocolo, ano=2017, numero=10, anulado=False, @@ -413,7 +413,7 @@ def test_documento_administrativo_protocolo_inexistente(): @pytest.mark.django_db(transaction=False) def test_protocolo_documento_form_invalido(): - config = mommy.make(AppConfig) + config = baker.make(AppConfig) form = ProtocoloDocumentForm( data={}, @@ -440,7 +440,7 @@ def test_protocolo_documento_form_invalido(): @pytest.mark.django_db(transaction=False) def test_protocolo_materia_invalido(): - config = mommy.make(AppConfig) + config = baker.make(AppConfig) form = ProtocoloMateriaForm(data={}, initial={ @@ -466,25 +466,25 @@ def test_protocolo_materia_invalido(): @pytest.mark.django_db(transaction=False) def test_lista_documentos_anexados(): - tipo_documento = mommy.make( + tipo_documento = baker.make( TipoDocumentoAdministrativo, descricao="Tipo_Teste" ) - documento_principal = mommy.make( + documento_principal = baker.make( DocumentoAdministrativo, numero=20, ano=2018, data="2018-01-04", tipo=tipo_documento ) - documento_anexado = mommy.make( + documento_anexado = baker.make( DocumentoAdministrativo, numero=21, ano=2019, data="2019-05-04", tipo=tipo_documento ) - documento_anexado_anexado = mommy.make( + documento_anexado_anexado = baker.make( DocumentoAdministrativo, numero=22, ano=2020, @@ -492,13 +492,13 @@ def test_lista_documentos_anexados(): tipo=tipo_documento ) - mommy.make( + baker.make( Anexado, documento_principal=documento_principal, documento_anexado=documento_anexado, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexado, documento_principal=documento_anexado, documento_anexado=documento_anexado_anexado, @@ -515,8 +515,8 @@ def test_lista_documentos_anexados(): @pytest.mark.django_db(transaction=False) def make_unidade_tramitacao(descricao): # Cria uma comissão para ser a unidade de tramitação - tipo_comissao = mommy.make(TipoComissao) - comissao = mommy.make(Comissao, + tipo_comissao = baker.make(TipoComissao) + comissao = baker.make(Comissao, tipo=tipo_comissao, nome=descricao, sigla='T', @@ -527,7 +527,7 @@ def make_unidade_tramitacao(descricao): assert comissao.nome == descricao # Cria a unidade - unidade = mommy.make(UnidadeTramitacao, comissao=comissao) + unidade = baker.make(UnidadeTramitacao, comissao=comissao) assert unidade.comissao == comissao return unidade @@ -536,27 +536,27 @@ def make_unidade_tramitacao(descricao): @pytest.mark.django_db(transaction=False) def test_tramitacoes_documentos_anexados(admin_client): - config = mommy.make(AppConfig, tramitacao_documento=True) + config = baker.make(AppConfig, tramitacao_documento=True) - tipo_documento = mommy.make( + tipo_documento = baker.make( TipoDocumentoAdministrativo, descricao="Tipo_Teste" ) - documento_principal = mommy.make( + documento_principal = baker.make( DocumentoAdministrativo, numero=20, ano=2018, data="2018-01-04", tipo=tipo_documento ) - documento_anexado = mommy.make( + documento_anexado = baker.make( DocumentoAdministrativo, numero=21, ano=2019, data="2019-05-04", tipo=tipo_documento ) - documento_anexado_anexado = mommy.make( + documento_anexado_anexado = baker.make( DocumentoAdministrativo, numero=22, ano=2020, @@ -564,13 +564,13 @@ def test_tramitacoes_documentos_anexados(admin_client): tipo=tipo_documento ) - mommy.make( + baker.make( Anexado, documento_principal=documento_principal, documento_anexado=documento_anexado, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexado, documento_principal=documento_anexado, documento_anexado=documento_anexado_anexado, @@ -582,7 +582,7 @@ def test_tramitacoes_documentos_anexados(admin_client): unidade_tramitacao_destino_1 = make_unidade_tramitacao(descricao="Teste 2") unidade_tramitacao_destino_2 = make_unidade_tramitacao(descricao="Teste 3") - status = mommy.make( + status = baker.make( StatusTramitacaoAdministrativo, indicador='R') @@ -808,11 +808,11 @@ def test_tramitacoes_documentos_anexados(admin_client): @pytest.mark.django_db(transaction=False) def test_tramitacao_lote_documentos_form(admin_client): - tipo_documento = mommy.make( + tipo_documento = baker.make( TipoDocumentoAdministrativo, descricao="Tipo_Teste" ) - documento = mommy.make( + documento = baker.make( DocumentoAdministrativo, numero=20, ano=2018, @@ -823,7 +823,7 @@ def test_tramitacao_lote_documentos_form(admin_client): unidade_tramitacao_local_1 = make_unidade_tramitacao(descricao="Teste 1") unidade_tramitacao_destino_1 = make_unidade_tramitacao(descricao="Teste 2") - status = mommy.make( + status = baker.make( StatusTramitacaoAdministrativo, indicador='R') @@ -897,27 +897,27 @@ def test_tramitacao_lote_documentos_form(admin_client): @pytest.mark.django_db(transaction=False) def test_tramitacao_lote_documentos_views(admin_client): - config = mommy.make(AppConfig, tramitacao_documento=True) + config = baker.make(AppConfig, tramitacao_documento=True) - tipo_documento = mommy.make( + tipo_documento = baker.make( TipoDocumentoAdministrativo, descricao="Tipo_Teste" ) - documento_principal = mommy.make( + documento_principal = baker.make( DocumentoAdministrativo, numero=20, ano=2018, data="2018-01-04", tipo=tipo_documento ) - documento_anexado = mommy.make( + documento_anexado = baker.make( DocumentoAdministrativo, numero=21, ano=2019, data="2019-05-04", tipo=tipo_documento ) - documento_anexado_anexado = mommy.make( + documento_anexado_anexado = baker.make( DocumentoAdministrativo, numero=22, ano=2020, @@ -925,7 +925,7 @@ def test_tramitacao_lote_documentos_views(admin_client): tipo=tipo_documento ) - documento_sem_anexados = mommy.make( + documento_sem_anexados = baker.make( DocumentoAdministrativo, numero=23, ano=2020, @@ -933,13 +933,13 @@ def test_tramitacao_lote_documentos_views(admin_client): tipo=tipo_documento ) - mommy.make( + baker.make( Anexado, documento_principal=documento_principal, documento_anexado=documento_anexado, data_anexacao="2019-05-11" ) - mommy.make( + baker.make( Anexado, documento_principal=documento_anexado, documento_anexado=documento_anexado_anexado, @@ -951,7 +951,7 @@ def test_tramitacao_lote_documentos_views(admin_client): unidade_tramitacao_destino_2 = make_unidade_tramitacao(descricao="Teste 3") unidade_tramitacao_destino_3 = make_unidade_tramitacao(descricao="Teste 4") - status = mommy.make( + status = baker.make( StatusTramitacaoAdministrativo, indicador='R') diff --git a/sapl/sessao/tests/test_sessao.py b/sapl/sessao/tests/test_sessao.py index 3ed1e47f6..0f42465e5 100644 --- a/sapl/sessao/tests/test_sessao.py +++ b/sapl/sessao/tests/test_sessao.py @@ -2,7 +2,7 @@ import pytest from datetime import datetime from django.core.exceptions import ValidationError from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker from sapl.materia.models import MateriaLegislativa, TipoMateriaLegislativa from sapl.parlamentares.models import Legislatura, Parlamentar, Partido,SessaoLegislativa @@ -33,9 +33,9 @@ def test_valida_campos_obrigatorios_sessao_plenaria_form(): @pytest.mark.django_db(transaction=False) def test_sessao_plenaria_form_valido(): - legislatura = mommy.make(Legislatura) - sessao = mommy.make(SessaoLegislativa) - tipo = mommy.make(TipoSessaoPlenaria) + legislatura = baker.make(Legislatura) + sessao = baker.make(SessaoLegislativa) + tipo = baker.make(TipoSessaoPlenaria) form = forms.SessaoPlenariaForm(data={'legislatura': str(legislatura.pk), 'numero': '1', @@ -51,10 +51,10 @@ def test_sessao_plenaria_form_valido(): @pytest.mark.django_db(transaction=False) def test_numero_duplicado_sessao_plenaria_form(): - legislatura = mommy.make(Legislatura) - sessao = mommy.make(SessaoLegislativa) - tipo = mommy.make(TipoSessaoPlenaria) - sessao_plenaria = mommy.make(SessaoPlenaria, + legislatura = baker.make(Legislatura) + sessao = baker.make(SessaoLegislativa) + tipo = baker.make(TipoSessaoPlenaria) + sessao_plenaria = baker.make(SessaoPlenaria, legislatura=legislatura, sessao_legislativa=sessao, tipo=tipo, @@ -97,11 +97,11 @@ def data(valor): @pytest.mark.django_db(transaction=False) def test_bancada_form_valido(): - legislatura = mommy.make(Legislatura, + legislatura = baker.make(Legislatura, data_inicio=data('2017-11-10'), data_fim=data('2017-12-31'), ) - partido = mommy.make(Partido) + partido = baker.make(Partido) form = forms.BancadaForm(data={'legislatura': str(legislatura.pk), 'nome': 'Nome da Bancada', @@ -116,11 +116,11 @@ def test_bancada_form_valido(): @pytest.mark.django_db(transaction=False) def test_bancada_form_datas_invalidas(): - legislatura = mommy.make(Legislatura, + legislatura = baker.make(Legislatura, data_inicio=data('2017-11-10'), data_fim=data('2017-12-31'), ) - partido = mommy.make(Partido) + partido = baker.make(Partido) form = forms.BancadaForm(data={'legislatura': str(legislatura.pk), 'nome': 'Nome da Bancada', @@ -133,12 +133,12 @@ def test_bancada_form_datas_invalidas(): @pytest.mark.django_db(transaction=False) def test_expediente_materia_form_valido(): - tipo_materia = mommy.make(TipoMateriaLegislativa) - materia = mommy.make(MateriaLegislativa, tipo=tipo_materia) + tipo_materia = baker.make(TipoMateriaLegislativa) + materia = baker.make(MateriaLegislativa, tipo=tipo_materia) - sessao = mommy.make(SessaoPlenaria) + sessao = baker.make(SessaoPlenaria) - instance = mommy.make(ExpedienteMateria, sessao_plenaria=sessao, + instance = baker.make(ExpedienteMateria, sessao_plenaria=sessao, materia=materia) form = forms.ExpedienteMateriaForm(data={'data_ordem': '28/12/2009', @@ -157,10 +157,10 @@ def test_expediente_materia_form_valido(): def test_registro_votacao_tem_ordem_xor_expediente(): def registro_votacao_com(ordem, expediente): - return mommy.make(RegistroVotacao, ordem=ordem, expediente=expediente) + return baker.make(RegistroVotacao, ordem=ordem, expediente=expediente) - ordem = mommy.make(OrdemDia) - expediente = mommy.make(ExpedienteMateria) + ordem = baker.make(OrdemDia) + expediente = baker.make(ExpedienteMateria) # a validação funciona com exatamente um dos campos preenchido registro_votacao_com(ordem, None).full_clean() @@ -175,38 +175,38 @@ def test_registro_votacao_tem_ordem_xor_expediente(): registro_votacao_com(ordem, expediente).full_clean() def create_sessao_plenaria(): - legislatura = mommy.make(Legislatura) - sessao = mommy.make(SessaoLegislativa) - tipo = mommy.make(TipoSessaoPlenaria) - return mommy.make(SessaoPlenaria, + legislatura = baker.make(Legislatura) + sessao = baker.make(SessaoLegislativa) + tipo = baker.make(TipoSessaoPlenaria) + return baker.make(SessaoPlenaria, legislatura=legislatura, sessao_legislativa=sessao, tipo=tipo, numero=1) def create_materia_legislativa(): - tipo_materia = mommy.make(TipoMateriaLegislativa) - return mommy.make(MateriaLegislativa, tipo=tipo_materia) + tipo_materia = baker.make(TipoMateriaLegislativa) + return baker.make(MateriaLegislativa, tipo=tipo_materia) @pytest.mark.django_db(transaction=False) def test_delete_sessao_plenaria_cascade_registro_votacao_ordemdia(): materia = create_materia_legislativa() sessao_plenaria = create_sessao_plenaria() - ordem = mommy.make(OrdemDia, + ordem = baker.make(OrdemDia, sessao_plenaria=sessao_plenaria, materia=materia, tipo_votacao='2') - tipo_resultado_votacao = mommy.make(TipoResultadoVotacao, + tipo_resultado_votacao = baker.make(TipoResultadoVotacao, nome='ok', natureza="A") - registro = mommy.make(RegistroVotacao, + registro = baker.make(RegistroVotacao, tipo_resultado_votacao=tipo_resultado_votacao, materia=materia, ordem=ordem) - presenca = mommy.make(PresencaOrdemDia, + presenca = baker.make(PresencaOrdemDia, sessao_plenaria=sessao_plenaria) - parlamentar = mommy.make(Parlamentar) - voto_parlamentar = mommy.make(VotoParlamentar, + parlamentar = baker.make(Parlamentar) + voto_parlamentar = baker.make(VotoParlamentar, votacao=registro, parlamentar=parlamentar, ordem=ordem) @@ -234,21 +234,21 @@ def test_delete_sessao_plenaria_cascade_registro_votacao_ordemdia(): def test_delete_sessao_plenaria_cascade_registro_votacao_expediente(): materia = create_materia_legislativa() sessao_plenaria = create_sessao_plenaria() - expediente = mommy.make(ExpedienteMateria, + expediente = baker.make(ExpedienteMateria, sessao_plenaria=sessao_plenaria, materia=materia, tipo_votacao='2') - tipo_resultado_votacao = mommy.make(TipoResultadoVotacao, + tipo_resultado_votacao = baker.make(TipoResultadoVotacao, nome='ok', natureza="A") - registro = mommy.make(RegistroVotacao, + registro = baker.make(RegistroVotacao, tipo_resultado_votacao=tipo_resultado_votacao, materia=materia, expediente=expediente) - presenca = mommy.make(SessaoPlenariaPresenca, + presenca = baker.make(SessaoPlenariaPresenca, sessao_plenaria=sessao_plenaria) - parlamentar = mommy.make(Parlamentar) - voto_parlamentar = mommy.make(VotoParlamentar, + parlamentar = baker.make(Parlamentar) + voto_parlamentar = baker.make(VotoParlamentar, votacao=registro, parlamentar=parlamentar, expediente=expediente) @@ -274,7 +274,7 @@ def test_delete_sessao_plenaria_cascade_registro_votacao_expediente(): @pytest.mark.django_db(transaction=False) def test_delete_sessao_plenaria_cascade_integrante_mesa(): sessao_plenaria = create_sessao_plenaria() - mesa = mommy.make(IntegranteMesa,sessao_plenaria=sessao_plenaria) + mesa = baker.make(IntegranteMesa,sessao_plenaria=sessao_plenaria) sessao_plenaria.delete() mesa_filter = IntegranteMesa.objects.filter(sessao_plenaria=sessao_plenaria).exists() assert not mesa_filter @@ -282,7 +282,7 @@ def test_delete_sessao_plenaria_cascade_integrante_mesa(): @pytest.mark.django_db(transaction=False) def test_delete_sessao_plenaria_cascade_expedientesessao(): sessao_plenaria = create_sessao_plenaria() - expediente_sessao = mommy.make(ExpedienteSessao, sessao_plenaria=sessao_plenaria) + expediente_sessao = baker.make(ExpedienteSessao, sessao_plenaria=sessao_plenaria) sessao_plenaria.delete() expediente_sessao_filter = ExpedienteSessao.objects.filter(sessao_plenaria=sessao_plenaria).exists() assert not expediente_sessao_filter @@ -290,7 +290,7 @@ def test_delete_sessao_plenaria_cascade_expedientesessao(): @pytest.mark.django_db(transaction=False) def test_delete_sessao_plenaria_cascade_orador(): sessao_plenaria = create_sessao_plenaria() - expediente_sessao = mommy.make(Orador, sessao_plenaria=sessao_plenaria) + expediente_sessao = baker.make(Orador, sessao_plenaria=sessao_plenaria) sessao_plenaria.delete() orador_filter = Orador.objects.filter(sessao_plenaria=sessao_plenaria).exists() assert not orador_filter diff --git a/sapl/sessao/tests/test_sessao_view.py b/sapl/sessao/tests/test_sessao_view.py index 18ac82340..322d7fd5f 100644 --- a/sapl/sessao/tests/test_sessao_view.py +++ b/sapl/sessao/tests/test_sessao_view.py @@ -1,7 +1,7 @@ import pytest from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ -from model_mommy import mommy +from model_bakery import baker from sapl.parlamentares.models import Legislatura, SessaoLegislativa from sapl.sessao.models import (SessaoPlenaria, TipoSessaoPlenaria, @@ -23,9 +23,9 @@ from sapl.sessao.views import (get_identificacao_basica, get_conteudo_multimidia @pytest.mark.django_db(transaction=False) def test_incluir_sessao_plenaria_submit(admin_client): - legislatura = mommy.make(Legislatura) - sessao = mommy.make(SessaoLegislativa) - tipo = mommy.make(TipoSessaoPlenaria, id=1) + legislatura = baker.make(Legislatura) + sessao = baker.make(SessaoLegislativa) + tipo = baker.make(TipoSessaoPlenaria, id=1) response = admin_client.post(reverse('sapl.sessao:sessaoplenaria_create'), {'legislatura': str(legislatura.pk), @@ -65,9 +65,9 @@ def test_incluir_sessao_errors(admin_client): @pytest.mark.django_db(transaction=False) class TestResumoView(): def setup(self): - self.sessao_plenaria = mommy.make(SessaoPlenaria) - self.parlamentar = mommy.make(Parlamentar) - self.cargo_mesa = mommy.make(CargoMesa) + self.sessao_plenaria = baker.make(SessaoPlenaria) + self.parlamentar = baker.make(Parlamentar) + self.cargo_mesa = baker.make(CargoMesa) self.integrante_mesa = IntegranteMesa(sessao_plenaria=self.sessao_plenaria, parlamentar=self.parlamentar, @@ -108,16 +108,16 @@ class TestResumoView(): }]} def test_get_presenca_sessao(self): - justificativa = mommy.make(JustificativaAusencia,sessao_plenaria=self.sessao_plenaria) - presenca = mommy.make(SessaoPlenariaPresenca,sessao_plenaria=self.sessao_plenaria) + justificativa = baker.make(JustificativaAusencia,sessao_plenaria=self.sessao_plenaria) + presenca = baker.make(SessaoPlenariaPresenca,sessao_plenaria=self.sessao_plenaria) resposta_presenca = get_presenca_sessao(self.sessao_plenaria) assert resposta_presenca['presenca_sessao'] == [presenca.parlamentar] assert resposta_presenca['justificativa_ausencia'][0] == justificativa def test_get_expedientes(self): - tipo_expediente = mommy.make(TipoExpediente) - expediente = mommy.make(ExpedienteSessao,sessao_plenaria=self.sessao_plenaria,tipo=tipo_expediente) + tipo_expediente = baker.make(TipoExpediente) + expediente = baker.make(ExpedienteSessao,sessao_plenaria=self.sessao_plenaria,tipo=tipo_expediente) resposta_expediente = get_expedientes(self.sessao_plenaria) @@ -130,9 +130,9 @@ class TestResumoView(): pass def test_get_oradores_explicacoes_pessoais(self): - parlamentar = mommy.make(Parlamentar) - partido_sigla = mommy.make(Filiacao, parlamentar=parlamentar) - orador = mommy.make(Orador,sessao_plenaria=self.sessao_plenaria,parlamentar=parlamentar) + parlamentar = baker.make(Parlamentar) + partido_sigla = baker.make(Filiacao, parlamentar=parlamentar) + orador = baker.make(Orador,sessao_plenaria=self.sessao_plenaria,parlamentar=parlamentar) resultado_get_oradores = get_oradores_explicacoes_pessoais(self.sessao_plenaria) @@ -143,7 +143,7 @@ class TestResumoView(): }] def test_get_ocorrencias_da_sessao(self): - ocorrencia = mommy.make(OcorrenciaSessao, sessao_plenaria=self.sessao_plenaria) + ocorrencia = baker.make(OcorrenciaSessao, sessao_plenaria=self.sessao_plenaria) resultado_get_ocorrencia = get_ocorrencias_da_sessao(self.sessao_plenaria) assert resultado_get_ocorrencia['ocorrencias_da_sessao'][0] == ocorrencia \ No newline at end of file diff --git a/sapl/test_general.py b/sapl/test_general.py index ef0c3a087..b4eba2236 100644 --- a/sapl/test_general.py +++ b/sapl/test_general.py @@ -1,7 +1,7 @@ from django.apps import apps from django.db.models import CharField, TextField from django.db.models.fields import BooleanField -from model_mommy import mommy +from model_bakery import baker import pytest from .settings import SAPL_APPS @@ -29,7 +29,7 @@ def test_str_sanity(): # __str__ semantics is not considered and should be tested separetely for app in sapl_appconfs: for model in app.get_models(): - obj = mommy.prepare(model) + obj = baker.prepare(model) try: str(obj) except Exception as exc: