diff --git a/tests/plano_trabalho/participante_outra_lotacao.py b/tests/plano_trabalho/participante_outra_lotacao.py index f8d6331..fd17762 100644 --- a/tests/plano_trabalho/participante_outra_lotacao.py +++ b/tests/plano_trabalho/participante_outra_lotacao.py @@ -35,7 +35,10 @@ def test_create_plano_trabalho_participante_outra_unidade(self, input_part): ) assert response.status_code == status.HTTP_201_CREATED - self.assert_equal_plano_trabalho(response.json(), self.input_pt) + response_data = response.json() + self.assert_equal_plano_trabalho(response_data, self.input_pt) + # verifica se a resposta veio com o cod_unidade_executora esperado + assert response_data['cod_unidade_executora'] == 101 # Consulta API para conferir se a criação foi persistida response = self.get_plano_trabalho( @@ -44,5 +47,9 @@ def test_create_plano_trabalho_participante_outra_unidade(self, input_part): ) assert response.status_code == status.HTTP_200_OK + response_data = response.json() + # verifica se os dados gravados são iguais aos informados - self.assert_equal_plano_trabalho(response.json(), self.input_pt) + self.assert_equal_plano_trabalho(response_data, self.input_pt) + # verifica o cod_unidade_executora + assert response_data['cod_unidade_executora'] == 101