forked from rufuspollock-okfn/bibserver
/
test_resultmanager.py
44 lines (33 loc) · 1.34 KB
/
test_resultmanager.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from nose.tools import assert_equal
from test.base import *
import bibserver.resultmanager
import bibserver.setconfig
import bibserver.dao
class TestResultManager:
@classmethod
def setup_class(cls):
recdict = fixtures['records'][0]
cls.record = dao.Record.upsert(recdict)
@classmethod
def teardown_class(cls):
conn, db = dao.get_conn()
conn.delete_index(TESTDB)
def test_01(self):
config = bibserver.setconfig.Configuration(bibserver.config.config)
facet_fields = config.get_default_args()['facet_field']
results = bibserver.dao.Record.query('tolstoy',
facet_fields=facet_fields)
args = None
manager = bibserver.resultmanager.ResultManager(results, config, args)
assert_equal(manager.numFound(), 1)
prev = manager.get_previous(0)
assert_equal(prev, [])
assert_equal(manager.page_size(), 10)
# commented out this test because ordering comes from ES anyway
# this will probably not be needed after refactoring resultmanager
#out = manager.get_ordered_facets('collection')
#assert_equal(out, [('great',1), ('novels',1)])
recorddicts = manager.set()
print recorddicts
out = manager.get_str(recorddicts[0], 'title')
assert recorddicts[0]['title'] in out, out