X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/124258c72bb9e69e6336de195f3a3b3e74509b95..bf3f18ce9313f4929ff327fbf12803157de171cb:/apps/search/tests/index.py?ds=sidebyside diff --git a/apps/search/tests/index.py b/apps/search/tests/index.py index 5b37cf05a..b8d44726f 100644 --- a/apps/search/tests/index.py +++ b/apps/search/tests/index.py @@ -1,34 +1,56 @@ -from __future__ import with_statement - -from search import Index, Search, IndexStore -from catalogue import models -from catalogue.test_utils import WLTestCase -from lucene import PolishAnalyzer, Version -#from nose.tools import raises +# -*- 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 unittest import skipIf +from django.conf import settings +from django.test.utils import override_settings +from catalogue.test_utils import WLTestCase, get_fixture from os import path - - - - +import tempfile +from catalogue.models import Book, Tag +from search.index import Index, Search, SearchResult +import catalogue +import opds + + +@override_settings( + SEARCH_INDEX = tempfile.mkdtemp(prefix='djangotest_search_'), +) +@skipIf(getattr(settings, 'NO_SEARCH_INDEX', False), + u'Requires search server and NO_SEARCH_INDEX=False.') class BookSearchTests(WLTestCase): def setUp(self): WLTestCase.setUp(self) - txt = path.join(path.dirname(__file__), 'files/fraszka-do-anusie.xml') - self.book = models.Book.from_xml_file(txt) - - search = Index() #PolishAnalyzer(Version.LUCENE_34)) - with search: - search.index_book(self.book) - print "index: %s" % search - - def test_search(self): - search = Search() - bks,_= search.search("wolne") - self.assertEqual(len(bks), 1) - self.assertEqual(bks[0].id, 1) - - bks,_= search.search("technical_editors: sutkowska") - self.assertEqual(len(bks), 1) - self.assertEqual(bks[0].id, 1) - + index = Index() + self.search = Search() + index.delete_query(self.search.index.query(uid="*")) + index.index.commit() + + self.do_doktora = Book.from_xml_file( + get_fixture('do-doktora.xml', opds)) + self.do_anusie = Book.from_xml_file( + get_fixture('fraszka-do-anusie.xml', catalogue)) + + def test_search_perfect_book_author(self): + books = self.search.search_books(self.search.index.query(authors=u"sęp szarzyński")) + assert len(books) == 1 + assert books[0].id == self.do_anusie.id + + # here we lack slop functionality as well + def test_search_perfect_book_title(self): + books = self.search.search_books(self.search.index.query(title=u"fraszka do anusie")) + assert len(books) == 1 + assert books[0].id == self.do_anusie.id + + # TODO: Add slop option to sunburnt + # def test_search_perfect_parts(self): + # books = self.search.search_phrase(u"Jakoż hamować") + # assert len(books) == 2 + # for b in books: + # b.book_id == self.book.id + # a = SearchResult.aggregate(books) + # # just one fragment hit. + # assert len(a[0].hits) == 1 +