Autocomplete: small fix and general tests

1 files changed, 17 insertions, 1 deletions

diff --git a/archaeological_files/tests.py b/archaeological_files/tests.py
index d78fa0e68..b8755ecbd 100644
--- a/archaeological_files/tests.py
+++ b/archaeological_files/tests.py
@@ -25,7 +25,8 @@ 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, create_superuser
+from ishtar_common.tests import TestCase, COMMON_FIXTURES, create_superuser, \
+    AutocompleteTestBase, AcItem
 
 from ishtar_common.models import Town, IshtarSiteProfile
 from archaeological_files import models
@@ -302,3 +303,18 @@ class DashboardTest(TestCase, FileInit):
 
         response = c.get(reverse(url))
         self.assertEqual(response.status_code, 200)
+
+
+class AutocompleteTest(AutocompleteTestBase, TestCase):
+    fixtures = FILE_TOWNS_FIXTURES
+    models = [
+        AcItem(models.File, 'autocomplete-file',
+               prepare_func="create_file"),
+    ]
+
+    def create_file(self, base_name):
+        item, __ = models.File.objects.get_or_create(
+            name=base_name,
+            file_type=models.FileType.objects.all()[0]
+        )
+        return item, None