From: Marcin Koziej <marcin.koziej@nowoczesnapolska.org.pl>
Date: Fri, 26 Oct 2012 10:35:51 +0000 (+0200)
Subject: old search tests not so suitable for sunbunt
X-Git-Url: https://git.mdrn.pl/wolnelektury.git/commitdiff_plain/b9c23cfd467a8674f9c7efa2abe4ab8e1e8a3c63?ds=inline

old search tests not so suitable for sunbunt
---

diff --git a/apps/search/index.py b/apps/search/index.py
index ea1a6c581..484340a87 100644
--- a/apps/search/index.py
+++ b/apps/search/index.py
@@ -914,10 +914,15 @@ class Search(SolrIndex):
         Searches for Book objects using query
         """
         bks = []
+        bks_found = set()
+        query = query.query(is_book=True)
         res = self.apply_filters(query, filters).field_limit(['book_id'])
         for r in res:
             try:
-                bks.append(catalogue.models.Book.objects.get(id=r['book_id']))
+                bid = r['book_id']
+                if not bid in bks_found:
+                    bks.append(catalogue.models.Book.objects.get(id=bid))
+                    bks_found.add(bid)
             except catalogue.models.Book.DoesNotExist: pass
         return bks
  
diff --git a/apps/search/tests/index.py b/apps/search/tests/index.py
index 12f6f7d1c..3e3d290e2 100644
--- a/apps/search/tests/index.py
+++ b/apps/search/tests/index.py
@@ -18,7 +18,8 @@ class BookSearchTests(WLTestCase):
         WLTestCase.setUp(self)
 
         index = Index()
-        index.index.delete_all()
+        self.search = Search()
+        index.delete_query(self.search.index.query(uid="*"))
         index.index.commit()
 
         with self.settings(NO_SEARCH_INDEX=False):
@@ -27,29 +28,24 @@ class BookSearchTests(WLTestCase):
             self.do_anusie = Book.from_xml_file(
                 get_fixture('fraszka-do-anusie.xml', catalogue))
 
-        self.search = Search()
-
     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].book_id == self.book.id
+        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(u"fraszka anusie"))
+        books = self.search.search_books(self.search.index.query(title=u"fraszka do anusie"))
         assert len(books) == 1
-        assert books[0].book_id == self.book.id
-
-    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
-
-    def test_search_perfect_author_title(self):
-        books = self.search.search_books(self.search.index.query(authors=u"szarzyński anusie"))
-        assert books == []
-
-        
+        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
+