From: Radek Czajka <radekczajka@nowoczesnapolska.org.pl>
Date: Fri, 19 Sep 2014 14:24:47 +0000 (+0200)
Subject: Add mock Search class for development.
X-Git-Url: https://git.mdrn.pl/wolnelektury.git/commitdiff_plain/01ba89e7caab3e386df4b1a99a93b98a237a0457?ds=sidebyside

Add mock Search class for development.
---

diff --git a/apps/search/index.py b/apps/search/index.py
index 7f8bf9f6a..31417ca87 100644
--- a/apps/search/index.py
+++ b/apps/search/index.py
@@ -940,3 +940,7 @@ class Search(SolrIndex):
         for f in filters:
             query = query.query(f)
         return query
+
+
+if getattr(settings, 'SEARCH_MOCK', False):
+    from .mock_search import Search
diff --git a/apps/search/mock_search.py b/apps/search/mock_search.py
new file mode 100644
index 000000000..253c277a8
--- /dev/null
+++ b/apps/search/mock_search.py
@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from mock import Mock
+from catalogue.models import Book, Tag
+from random import randint, choice
+
+
+class Search(Mock):
+    """
+    Search mock for development without setting up Solr.
+
+    Instead of connecting to an actual search server, it returns
+    some random results for any query.
+    """
+    class MockIndex(Mock):
+        def analyze(*args, **kwargs):
+            return []
+
+    index = MockIndex()
+
+    @staticmethod
+    def _find_some_books(snippets=False, query_terms=None, max_results=20):
+        from .index import SearchResult
+
+        qs = Book.objects.order_by('?')
+        if snippets:
+            qs = qs.exclude(fragments=None)
+        results = []
+        for book in qs[:randint(1, max_results)]:
+            doc = {
+                'score': randint(0, 100),
+                'book_id': book.pk,
+                'published_date': randint(1000, 1920),
+                }
+            if snippets:
+                fragment = book.fragments.order_by('?')[0]
+                doc.update({
+                    'header_type': choice(['strofa', 'akap']),
+                    'header_index': randint(100, 200),
+                    'header_span': randint(100, 200),
+                    'fragment_anchor': fragment.anchor,
+                    'snippets_position': randint(100, 200),
+                    'snippets_length': randint(100, 200),
+                    'snippets_revision': randint(1, 100),
+                    'themes_pl': fragment.tags.filter(category='theme'),
+                })
+            res = SearchResult(doc, how_found='mock', query_terms=query_terms)
+            if snippets:
+                res.snippets = [fragment.short_text]
+            results.append(res)
+        return results
+
+    def search_phrase(self, searched, field='text', book=False,
+            filters=None, snippets=False):
+        return self._find_some_books(snippets)
+
+    def search_some(self, searched, fields, book=True, filters=None, snippets=True, query_terms=None):
+        return self._find_some_books(snippets, query_terms)
+
+    def search_books(self, query, filters=None, max_results=10):
+        return self._find_some_books(snippets, max_results=max_results)
+
+    def search_everywhere(self, searched, query_terms=None):
+        return []
+
+    def hint_tags(self, query, pdcounter=True, prefix=True):
+        return Tag.objects.exclude(category='set').order_by('?')[:randint(1, 10)]
+
+    def hint_books(self, prefix):
+        return Book.objects.order_by('?')[:randint(1, 10)]
diff --git a/apps/search/views.py b/apps/search/views.py
index 514dc6e1f..a50897199 100644
--- a/apps/search/views.py
+++ b/apps/search/views.py
@@ -59,10 +59,11 @@ def did_you_mean(query, tokens):
     # return query
 
 
+@cache.never_cache
 def hint(request):
     prefix = request.GET.get('term', '')
     if len(prefix) < 2:
-        return JsonResponse([])
+        return JsonResponse([], safe=False)
 
     prefix = remove_query_syntax_chars(prefix)
 
@@ -106,9 +107,10 @@ def hint(request):
         return HttpResponse("%s(%s);" % (callback, json.dumps(data)),
                             content_type="application/json; charset=utf-8")
     else:
-        return JsonResponse(data)
+        return JsonResponse(data, safe=False)
 
 
+@cache.never_cache
 def main(request):
     results = {}