wip

sapl/protocoloadm/tests/test_protocoloadm.py

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, date
 
 import pytest
 from django.core.urlresolvers import reverse
@@ -145,7 +145,7 @@ def test_create_tramitacao(admin_client):
         unidade_tramitacao_destino=unidade_tramitacao_destino_1,
         status=status,
         documento=documento_adm,
-        data_tramitacao=datetime.date(2016, 8, 21))
+        data_tramitacao=date(2016, 8, 21))
 
     response = admin_client.post(
         reverse(
@@ -155,7 +155,7 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_local_1.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date(2016, 8, 21)},
+         'data_tramitacao': date(2016, 8, 21)},
         follow=True)
 
     msg = force_text(_('A origem da nova tramitação deve ser igual ao '
@@ -173,7 +173,7 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_destino_2.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date(2016, 8, 20)},
+         'data_tramitacao': date(2016, 8, 20)},
         follow=True)
 
     msg = _('A data da nova tramitação deve ser ' +
@@ -191,7 +191,7 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_destino_2.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date.today() + datetime.timedelta(
+         'data_tramitacao': date.today() + datetime.timedelta(
              days=1)},
         follow=True)
 
@@ -210,8 +210,8 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_destino_2.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date(2016, 8, 21),
-         'data_encaminhamento': datetime.date(2016, 8, 20)},
+         'data_tramitacao': date(2016, 8, 21),
+         'data_encaminhamento': date(2016, 8, 20)},
         follow=True)
 
     msg = force_text(_('A data de encaminhamento deve ser ' +
@@ -229,8 +229,8 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_destino_2.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date(2016, 8, 21),
-         'data_fim_prazo': datetime.date(2016, 8, 20)},
+         'data_tramitacao': date(2016, 8, 21),
+         'data_fim_prazo': date(2016, 8, 20)},
         follow=True)
 
     msg = _('A data fim de prazo deve ser ' +
@@ -248,12 +248,12 @@ def test_create_tramitacao(admin_client):
          'unidade_tramitacao_destino': unidade_tramitacao_destino_2.pk,
          'documento': documento_adm.pk,
          'status': status.pk,
-         'data_tramitacao': datetime.date(2016, 8, 21)},
+         'data_tramitacao': date(2016, 8, 21)},
         follow=True)
 
     tramitacao = TramitacaoAdministrativo.objects.last()
     # Verifica se a tramitacao que obedece as regras de negócios é criada
-    assert tramitacao.data_tramitacao == datetime.date(2016, 8, 21)
+    assert tramitacao.data_tramitacao == date(2016, 8, 21)
 
 
 @pytest.mark.django_db(transaction=False)