Adiciona teste para formulario de inclusão de norma juridica

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

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