Update librarian.

[wolnelektury.git] / apps / opds / tests / __init__.py

diff --git a/apps/opds/tests/__init__.py b/apps/opds/tests/__init__.py
index d6e9ca0..8d51b7b 100755 (executable)
--- a/apps/opds/tests/__init__.py
+++ b/apps/opds/tests/__init__.py
@@ -1,4 +1,7 @@
 # -*- coding: utf-8 -*-
 # -*- 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 lxml import etree
 from django.core.files.base import ContentFile
 import catalogue
 from lxml import etree
 from django.core.files.base import ContentFile
 import catalogue


@@ -6,6 +9,7 @@ from catalogue.test_utils import (BookInfoStub, PersonStub, info_args,
         WLTestCase, get_fixture)
 from catalogue.models import Book
 from librarian import WLURI, XMLNamespace
         WLTestCase, get_fixture)
 from catalogue.models import Book
 from librarian import WLURI, XMLNamespace

+from search import Index, Search

 
 AtomNS = XMLNamespace("http://www.w3.org/2005/Atom")
 
 
 AtomNS = XMLNamespace("http://www.w3.org/2005/Atom")
 


@@ -14,6 +18,10 @@ class OpdsSearchTests(WLTestCase):
     """Tests search feed in OPDS.."""
     def setUp(self):
         WLTestCase.setUp(self)
     """Tests search feed in OPDS.."""
     def setUp(self):
         WLTestCase.setUp(self)

+        index = Index()
+        index.index.delete_all()
+        index.index.commit()
+

         with self.settings(NO_SEARCH_INDEX=False):
             self.do_doktora = Book.from_xml_file(
                 get_fixture('do-doktora.xml'))
         with self.settings(NO_SEARCH_INDEX=False):
             self.do_doktora = Book.from_xml_file(
                 get_fixture('do-doktora.xml'))


@@ -40,7 +48,7 @@ class OpdsSearchTests(WLTestCase):
         both = set([self.do_doktora, self.do_anusie])
         self.assert_finds('title=fraszka', both)
         self.assert_finds('title=fraszka', both)
         both = set([self.do_doktora, self.do_anusie])
         self.assert_finds('title=fraszka', both)
         self.assert_finds('title=fraszka', both)

-        self.assert_finds('q=title:fraszka', [self.do_doktora])
+        self.assert_finds('q=title:doktora', [self.do_doktora])

 
     def test_opds_search_author(self):
         """Search by author."""
 
     def test_opds_search_author(self):
         """Search by author."""