diff --git a/norma/test_norma.py b/norma/test_norma.py new file mode 100644 index 000000000..640d49630 --- /dev/null +++ b/norma/test_norma.py @@ -0,0 +1,52 @@ +import pytest +from django.core.urlresolvers import reverse +from model_mommy import mommy + +from .models import NormaJuridica, TipoNormaJuridica + + +@pytest.mark.django_db(transaction=False) +def test_incluir_norma_submit(client): + # Cria um tipo de norma + tipo = mommy.make(TipoNormaJuridica, + sigla='T', + descricao='Teste') + + # Testa POST + response = client.post(reverse('norma_incluir'), + {'tipo': tipo.pk, + 'numero': '1', + 'ano': '2016', + 'data': '2016-03-22', + 'esfera_federacao': 'E', + 'ementa': 'Teste_Ementa', + 'salvar': 'salvar'}, + follow=True) + assert response.status_code == 200 + + # Verifica se a matéria foi anexada corretamente + norma = NormaJuridica.objects.first() + assert norma.numero == 1 + assert norma.ano == 2016 + assert norma.tipo == tipo + + +@pytest.mark.django_db(transaction=False) +def test_incluir_norma_errors(client): + + response = client.post(reverse('norma_incluir'), + {'salvar': 'salvar'}, + follow=True) + + assert (response.context_data['form'].errors['tipo'] == + ['Este campo é obrigatório.']) + assert (response.context_data['form'].errors['numero'] == + ['Este campo é obrigatório.']) + assert (response.context_data['form'].errors['ano'] == + ['Este campo é obrigatório.']) + assert (response.context_data['form'].errors['data'] == + ['Este campo é obrigatório.']) + assert (response.context_data['form'].errors['esfera_federacao'] == + ['Este campo é obrigatório.']) + assert (response.context_data['form'].errors['ementa'] == + ['Este campo é obrigatório.']) diff --git a/norma/views.py b/norma/views.py index 8f97224f1..19007a898 100644 --- a/norma/views.py +++ b/norma/views.py @@ -115,6 +115,9 @@ class NormaIncluirView(CreateView): form_class = NormaJuridicaForm success_url = reverse_lazy('normajuridica:list') + def get_success_url(self): + return reverse_lazy('normajuridica:list') + def form_valid(self, form): norma = form.save(commit=False) norma.timestamp = datetime.now()