Merge pull request #258 from czpython/feature/search-tests

adds search tests

aldryn_newsblog/search_indexes.py

@@ -3,6 +3,8 @@
 
 from django.conf import settings
 
+from haystack.constants import DEFAULT_ALIAS
+
 from aldryn_search.utils import get_index_base
 
 from .models import Article
@@ -17,6 +19,11 @@ class ArticleIndex(get_index_base()):
     def get_title(self, obj):
         return obj.title
 
+    def get_url(self, obj):
+        using = getattr(self, '_backend_alias', DEFAULT_ALIAS)
+        language = self.get_current_language(using=using, obj=obj)
+        return obj.get_absolute_url(language)
+
     def get_description(self, obj):
         return obj.lead_in
 

aldryn_newsblog/tests/__init__.py

@@ -100,6 +100,8 @@ def create_article(self, content=None, **kwargs):
         fields.update(kwargs)
 
         article = Article.objects.create(**fields)
+        # save again to calculate article search_data.
+        article.save()
 
         if content:
             api.add_plugin(article.content, 'TextPlugin',

aldryn_newsblog/tests/test_search.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+from django.utils.timezone import now
+from django.utils.translation import activate
+
+from aldryn_newsblog.models import Article
+from aldryn_newsblog.search_indexes import ArticleIndex
+
+from . import NewsBlogTestCase
+
+
+class ArticleIndexingTests(NewsBlogTestCase):
+
+    def get_index(self):
+        from haystack.constants import DEFAULT_ALIAS
+
+        index = ArticleIndex()
+        index._backend_alias = DEFAULT_ALIAS
+        return index
+
+    def test_article_is_indexed_using_prepare(self):
+        activate(self.language)
+
+        lead_in = 'Hello! this text will be searchable.'
+
+        article = self.create_article(lead_in=lead_in)
+
+        index = self.get_index()
+
+        data = index.prepare(article)
+
+        count = data['text'].count(lead_in)
+
+        self.assertTrue(count != 0, "Couldn't find %s in text" % lead_in)
+
+    def test_translated_article_is_indexed_using_prepare(self):
+        activate(self.language)
+
+        lead_in = 'Hello! this text will be searchable.'
+
+        # create english article
+        article = self.create_article(lead_in=lead_in)
+
+        # create german translation for article
+        article.set_current_language('de')
+        article.title = '%s [de]' % self.rand_str()
+        article.save()
+
+        index = self.get_index()
+
+        data = index.prepare(article)
+
+        self.assertEquals(data['language'], 'de')
+        self.assertEquals(data['url'], article.get_absolute_url('de'))

test_requirements/base.txt

@@ -3,6 +3,7 @@ coverage>=3.7.1
 dj-database-url
 django-classy-tags
 django-mptt<0.7.0  # rq.filter: <0.7.0
+django-haystack
 django-sekizai
 django-treebeard>=2.0
 djangocms-admin-style

test_settings.py

@@ -96,6 +96,7 @@
     # that were previously used, which reveals this issue.
     #
     'PARLER_ENABLE_CACHING': False,
+    'ALDRYN_SEARCH_DEFAULT_LANGUAGE': 'en',
     'HAYSTACK_CONNECTIONS': {
         'default': {
             'ENGINE': 'haystack.backends.simple_backend.SimpleEngine',