def test_reduce_results(self):
"""Verify that root nodes, subparts, and subject groups get removed"""
results = {
'total_hits':
5,
'results': [{
'label': ['111'],
'text': 'Root'
}, {
'label': ['111', 'Subpart'],
'text': 'Empty Part'
}, {
'label': ['111', 'Subpart', 'A'],
'text': 'Subpart A'
}, {
'label': ['111', 'Subjgrp', 'ABC'],
'text': 'Subject group'
}, {
'label': ['111', '12', 'a'],
'text': '111.12(a)'
}]
}
PartialSearch().reduce_results(results, 0)
self.assertEqual(results['results'], [{
'label': ['111', '12', 'a'],
'text': '111.12(a)'
}])
self.assertEqual(results['total_hits'], 1)
def test_add_prev_next(self):
view = PartialSearch()
context = {'results': {'total_hits': 77}}
view.add_prev_next(0, context)
self.assertFalse('previous' in context)
self.assertEqual(context['next'], {'page': 1, 'length': 10})
del context['next']
view.add_prev_next(5, context)
self.assertEqual(context['previous'], {'page': 4, 'length': 10})
self.assertEqual(context['next'], {'page': 6, 'length': 10})
del context['previous']
del context['next']
view.add_prev_next(6, context)
self.assertEqual(context['previous'], {'page': 5, 'length': 10})
self.assertEqual(context['next'], {'page': 7, 'length': 7})
del context['previous']
del context['next']
view.add_prev_next(7, context)
self.assertEqual(context['previous'], {'page': 6, 'length': 10})
self.assertFalse('next' in context)
