From 799d0e08dde2d0e4d73a7ff4ee51ec7d89f11e5a Mon Sep 17 00:00:00 2001 From: LeandroJatai Date: Tue, 14 Apr 2026 18:16:19 -0300 Subject: [PATCH] ajustes nos testes e no form mesa diretora --- sapl/parlamentares/forms.py | 7 +++-- sapl/parlamentares/tests/test_mesadiretora.py | 30 +++++++++++++++++-- 2 files changed, 32 insertions(+), 5 deletions(-) diff --git a/sapl/parlamentares/forms.py b/sapl/parlamentares/forms.py index f5bef2123..2775be5a1 100755 --- a/sapl/parlamentares/forms.py +++ b/sapl/parlamentares/forms.py @@ -821,8 +821,11 @@ class ComposicaoMesaForm(ModelForm): def __init__(self, *args, **kwargs): super(ComposicaoMesaForm, self).__init__(*args, **kwargs) - self.fields['parlamentar'].queryset = self.fields['parlamentar'].queryset.filter( - mandato__legislatura=self.initial['mesa_diretora'].legislatura) + mesa_diretora = self.initial.get('mesa_diretora', None) + + if mesa_diretora: + self.fields['parlamentar'].queryset = self.fields['parlamentar'].queryset.filter( + mandato__legislatura=mesa_diretora.legislatura) def clean(self): super(ComposicaoMesaForm, self).clean() diff --git a/sapl/parlamentares/tests/test_mesadiretora.py b/sapl/parlamentares/tests/test_mesadiretora.py index 9d0e19cee..ff663488d 100644 --- a/sapl/parlamentares/tests/test_mesadiretora.py +++ b/sapl/parlamentares/tests/test_mesadiretora.py @@ -22,7 +22,11 @@ def test_mesadiretora_form_invalido(): @pytest.mark.django_db(transaction=False) def test_mesadiretora_form_data_inicio_maior_que_data_fim(): - legislatura = baker.make('parlamentares.Legislatura') + legislatura = baker.make( + 'parlamentares.Legislatura', + data_inicio='2021-01-01', + data_fim='2024-12-31' + ) form = MesaDiretoraForm(data={ 'titulo': 'Mesa Diretora 2021-2022', @@ -38,7 +42,11 @@ def test_mesadiretora_form_data_inicio_maior_que_data_fim(): @pytest.mark.django_db(transaction=False) def test_mesadiretora_form_valido(): - legislatura = baker.make('parlamentares.Legislatura') + legislatura = baker.make( + 'parlamentares.Legislatura', + data_inicio='2021-01-01', + data_fim='2024-12-31' + ) form = MesaDiretoraForm(data={ 'titulo': 'Mesa Diretora 2021-2022', @@ -51,7 +59,11 @@ def test_mesadiretora_form_valido(): @pytest.mark.django_db(transaction=False) def test_mesadiretora_form_intersecao(): - legislatura = baker.make('parlamentares.Legislatura') + legislatura = baker.make( + 'parlamentares.Legislatura', + data_inicio='2021-01-01', + data_fim='2024-12-31' + ) mesa_existente = baker.make( 'parlamentares.MesaDiretora', legislatura=legislatura, @@ -108,11 +120,17 @@ def test_composicaomesa_form_valido(): parlamentar = baker.make('parlamentares.Parlamentar') cargo = baker.make('parlamentares.CargoMesa') mesa_diretora = baker.make('parlamentares.MesaDiretora') + mandato = baker.make( + 'parlamentares.Mandato', + parlamentar=parlamentar, + legislatura=mesa_diretora.legislatura) form = ComposicaoMesaForm(data={ 'parlamentar': parlamentar.id, 'cargo': cargo.id, 'mesa_diretora': mesa_diretora.id, + }, initial={ + 'mesa_diretora': mesa_diretora, }) assert form.is_valid() @@ -123,6 +141,10 @@ def test_composicaomesa_form_parlamentar_ocupando_cargo_na_mesma_mesa(): cargo1 = baker.make('parlamentares.CargoMesa') cargo2 = baker.make('parlamentares.CargoMesa') mesa_diretora = baker.make('parlamentares.MesaDiretora') + mandato = baker.make( + 'parlamentares.Mandato', + parlamentar=parlamentar, + legislatura=mesa_diretora.legislatura) # Cria uma composição de mesa existente com o mesmo parlamentar e mesa ComposicaoMesa.objects.create( @@ -135,6 +157,8 @@ def test_composicaomesa_form_parlamentar_ocupando_cargo_na_mesma_mesa(): 'parlamentar': parlamentar.id, 'cargo': cargo2.id, 'mesa_diretora': mesa_diretora.id, + }, initial={ + 'mesa_diretora': mesa_diretora, }) assert not form.is_valid()