def test_indexes_library_datasets(self):
# prepare state.
DatasetFactory()
DatasetFactory()
search = Search(self.lib)
search.index_dataset = fudge.Fake().expects_call()
# testing
with fudge.patched_context(search, 'all_datasets', []):
search.index_datasets()
def test_tick_fn_gets_each_vid(self):
# prepare state.
DatasetFactory()
DatasetFactory()
search = Search(self.lib)
search.index_dataset = fudge.Fake().expects_call()
tick_f = fudge.Fake()\
.expects_call().with_args('datasets: 1 partitions: 0')\
.next_call().with_args('datasets: 2 partitions: 0')
# testing
with fudge.patched_context(search, 'all_datasets', []):
search.index_datasets(tick_f=tick_f)