old search tests not so suitable for sunbunt

diff --git a/apps/search/index.py b/apps/search/index.py
index ea1a6c5..484340a 100644 (file)
--- a/apps/search/index.py
+++ b/apps/search/index.py
@@ -914,10 +914,15 @@ class Search(SolrIndex):
         Searches for Book objects using query
         """
         bks = []
         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:
         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
  
             except catalogue.models.Book.DoesNotExist: pass
         return bks
  

diff --git a/apps/search/tests/index.py b/apps/search/tests/index.py
index 12f6f7d..3e3d290 100644 (file)
--- a/apps/search/tests/index.py
+++ b/apps/search/tests/index.py
@@ -18,7 +18,8 @@ class BookSearchTests(WLTestCase):
         WLTestCase.setUp(self)
 
         index = Index()
         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):
         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.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
     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):
     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 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
+