Fix error during report generation

[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 -*-
+# 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
@@ -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
+from search import Index, Search
 
 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)
+        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'))
@@ -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)
-        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."""