def test_fetch_concerned_records(self):
from invenio.bibrank_selfcites_task import fetch_concerned_records, \
store_last_updated, \
get_bibrankmethod_lastupdate
name = 'selfcites'
old_date = datetime(year=1900, month=1, day=1).strftime("%Y-%m-%d %H:%M:%S")
try:
original_date = get_bibrankmethod_lastupdate(name)
except IndexError:
original_date = old_date
store_last_updated(name, old_date)
self.assert_(fetch_concerned_records('selfcites'))
future_date = datetime.now() + timedelta(days=1)
store_last_updated(name, future_date)
self.assert_(not fetch_concerned_records('selfcites'))
# Restore value in db
store_last_updated(name, original_date)
def test_fetch_concerned_records_recids(self):
from invenio.bibrank_selfcites_task import fetch_concerned_records
ids_param = ((1, 3), (5, 10))
recids, end_date = fetch_concerned_records('selfcites', ids_param)
self.assertEqual(recids, intbitset([1, 2, 3, 5, 6, 7, 8, 9, 10]))
self.assertEqual(end_date, None)
def test_fetch_concerned_records_recids(self):
from invenio.bibrank_selfcites_task import fetch_concerned_records
ids_param = ((1, 3), (5, 10))
recids, end_date = fetch_concerned_records('selfcites', ids_param)
self.assertEqual(recids, intbitset([1, 2, 3, 5, 6, 7, 8, 9, 10]))
self.assertEqual(end_date, None)
