1 files changed, 33 insertions, 37 deletions
diff --git a/archaeological_files/tests.py b/archaeological_files/tests.py
index b23c91a95..8a1069d51 100644
--- a/archaeological_files/tests.py
+++ b/archaeological_files/tests.py
@@ -25,48 +25,26 @@ from django.contrib.auth.models import User
 from django.core.urlresolvers import reverse
 from django.test.client import Client
 
-from ishtar_common.tests import TestCase, COMMON_FIXTURES
+from ishtar_common.tests import TestCase, COMMON_FIXTURES, create_superuser
 
-from ishtar_common.models import PersonType, Town, IshtarSiteProfile
+from ishtar_common.models import Town, IshtarSiteProfile
 from archaeological_files import models
-from archaeological_operations.models import Parcel, ParcelOwner
-from archaeological_operations.tests import OperationInitTest, \
+from archaeological_operations.models import Parcel, ParcelOwner, ActType, \
+    AdministrativeAct
+from archaeological_operations.tests import OperationInitTest, FileInit, \
     FILE_TOWNS_FIXTURES
 
 
-class FileInit(object):
-    def login_as_superuser(self):
-        self.client.login(username='username', password='tralala')
-
-    def create_file(self):
-        self.extra_models, self.model_list = {}, []
-        self.user, created = User.objects.get_or_create(username='username',
-                                                        is_superuser=True)
-        self.user.set_password('tralala')
-        self.user.save()
-        self.o_user, created = User.objects.get_or_create(username='ousername')
-        person_type, created = PersonType.objects.get_or_create(
-            label=u'Test ' u'person type', txt_idx='test_person',
-            available=True)
-        self.extra_models['person_type'] = person_type
-        self.model_list.append(person_type)
-
-        person = models.Person(surname='Surname', name='Name',
-                               history_modifier=self.o_user)
-        person.save()
-        self.extra_models['person'] = person
-        self.model_list.append(person)
-
-        file_type, created = models.FileType.objects.get_or_create(
-            label=u'Test file type', txt_idx='test_file', available=True)
-        self.extra_models['file_type'] = file_type
-        self.model_list.append(file_type)
-
-        dct = {'year': 2010, 'numeric_reference': 1000, 'file_type': file_type,
-               'internal_reference': u'UNIT_testÉ ?', 'in_charge': person,
-               'history_modifier': self.o_user, 'total_surface': 10000}
-        self.item = self.model(**dct)
-        self.item.save()
+def create_administrativact(user, fle):
+    act_type, created = ActType.objects.get_or_create(
+        txt_idx='act_type_F', intented_to='F')
+    dct = {'history_modifier': user,
+           'act_type': act_type,
+           'associated_file': fle,
+           'signature_date': datetime.date(2017, 07, 10),
+           'index': 22}
+    adminact, created = AdministrativeAct.objects.get_or_create(**dct)
+    return [act_type], [adminact]
 
 
 class FileTest(TestCase, FileInit):
@@ -293,3 +271,21 @@ class FileOperationTest(TestCase, OperationInitTest, FileInit):
         q = ParcelOwner.objects.filter(parcel__operation=self.operation,
                                        parcel__parcel_number='42')
         self.assertEqual(q.count(), 1)
+
+
+class DashboardTest(TestCase, FileInit):
+    fixtures = FILE_TOWNS_FIXTURES
+    model = models.File
+
+    def setUp(self):
+        self.username, self.password, self.user = create_superuser()
+        IshtarSiteProfile.objects.create()
+        self.create_file()
+
+    def test_dashboard(self):
+        url = 'dashboard-file'
+        c = Client()
+        c.login(username=self.username, password=self.password)
+
+        response = c.get(reverse(url))
+        self.assertEqual(response.status_code, 200)