c2b9110cf71e32b0ecd1c2b3d0d369c729f4ae75
[wolnelektury.git] / apps / search / tests / index.py
1 from __future__ import with_statement
2
3 from search import Index, Search
4 from catalogue import models
5 from catalogue.test_utils import WLTestCase
6 from lucene import PolishAnalyzer, Version
7 #from nose.tools import raises
8 from os import path
9
10
11 class BookSearchTests(WLTestCase):
12     def setUp(self):
13         WLTestCase.setUp(self)
14
15         txt = path.join(path.dirname(__file__), 'files/fraszka-do-anusie.xml')
16         self.book = models.Book.from_xml_file(txt)
17
18         search = Index() #PolishAnalyzer(Version.LUCENE_34))
19         with search:
20             search.index_book(self.book)
21         print "index: %s" % search
22
23     def test_search(self):
24         search = Search()
25         bks,_= search.search("wolne")
26         self.assertEqual(len(bks), 1)
27         self.assertEqual(bks[0].id, 1)
28         
29         bks,_= search.search("technical_editors: sutkowska")
30         self.assertEqual(len(bks), 1)
31         self.assertEqual(bks[0].id, 1)
32