def test_set_last_position(self, mock_create_scanner): options = dict(apikey='fake', collection_name='collection', project_id='10804') hs_reader = HubstorageReader(dict(options=options), meta()) hs_reader.set_last_position(dict(last_key='resumekey')) self.assertEquals([mock.call.set_startafter('resumekey')], mock_create_scanner.return_value.mock_calls) self.assertEquals(dict(last_key='resumekey'), hs_reader.last_position)
def test_set_last_position(self, mock_create_scanner): options = dict(apikey='fake', collection_name='collection', project_id=10804) hs_reader = HubstorageReader(dict(options=options), meta()) hs_reader.set_last_position(dict(last_key='resumekey')) self.assertEquals([mock.call.set_startafter('resumekey')], mock_create_scanner.return_value.mock_calls) self.assertEquals(dict(last_key='resumekey'), hs_reader.last_position)