apps/search/tests/index.py

   1 # -*- coding: utf-8 -*-
   2 from django.conf import settings
   3 from django.test.utils import override_settings
   4 from catalogue.test_utils import WLTestCase, get_fixture
   5 from os import path
   6 import tempfile
   7 from catalogue.models import Book, Tag
   8 from search import Index, Search, SearchResult
   9 import catalogue
  10 import opds
  11
  12
  13 @override_settings(
  14     SEARCH_INDEX = tempfile.mkdtemp(prefix='djangotest_search_'),
  15 )
  16 class BookSearchTests(WLTestCase):
  17     def setUp(self):
  18         WLTestCase.setUp(self)
  19
  20         index = Index()
  21         index.index.delete_all()
  22         index.index.commit()
  23
  24         with self.settings(NO_SEARCH_INDEX=False):
  25             self.do_doktora = Book.from_xml_file(
  26                 get_fixture('do-doktora.xml', opds))
  27             self.do_anusie = Book.from_xml_file(
  28                 get_fixture('fraszka-do-anusie.xml', catalogue))
  29
  30         self.search = Search()
  31
  32     def test_search_perfect_book_author(self):
  33         books = self.search.search_books(self.search.index.query(authors=u"sęp szarzyński"))
  34         assert len(books) == 1
  35         assert books[0].book_id == self.book.id
  36
  37     def test_search_perfect_book_title(self):
  38         books = self.search.search_books(self.search.index.query(u"fraszka anusie"))
  39         assert len(books) == 1
  40         assert books[0].book_id == self.book.id
  41
  42     def test_search_perfect_parts(self):
  43         books = self.search.search_phrase(u"Jakoż hamować")
  44         assert len(books) == 2
  45         for b in books:
  46             b.book_id == self.book.id
  47         a = SearchResult.aggregate(books)
  48         # just one fragment hit.
  49         assert len(a[0].hits) == 1
  50
  51     def test_search_perfect_author_title(self):
  52         books = self.search.search_books(self.search.index.query(authors=u"szarzyński anusie"))
  53         assert books == []
  54
  55