Fix #3168 - lib migration (#3169)

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

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',

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',

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,

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')
 

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,

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',

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)

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')
 

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',

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',

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)

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),

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')
 

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

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

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: