diff options
Diffstat (limited to 'archaeological_files/tests.py')
| -rw-r--r-- | archaeological_files/tests.py | 18 | 
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 | 
