+++ /dev/null
-# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
-#
-from unittest.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()
-
- def search_words(self, words, fields, required=None, book=True, picture=False):
- from .index import SearchResult
-
- max_results = 20
-
- if picture: return []
-
- qs = Book.objects.filter(findable=True).order_by('?')
- results = []
- for book in qs[:randint(1, max_results)]:
- doc = {
- 'score': randint(0, 100),
- 'book_id': book.pk,
- 'published_date': randint(1000, 1920),
- }
- res = SearchResult(doc, how_found='mock', query_terms=words)
- results.append(res)
- return results
-