def test_fetch_events_explicit_several_batches(self):
all_fetched_events = []
self.events = range(0, 9)
batch_size = 2
events_fetcher = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=batch_size)
for i in range(0, 4):
events_batch = events_fetcher._fetch_events_batch()
self.assertEqual(len(events_batch), batch_size)
all_fetched_events.extend(events_batch)
remaining_events_batch = events_fetcher._fetch_events_batch()
self.assertEqual(len(remaining_events_batch), 1)
all_fetched_events.extend(remaining_events_batch)
self.assertEqual(self.events, all_fetched_events)
def test_fetch_events_explicit_several_batches(self):
all_fetched_events = []
self.events = range(0, 9)
batch_size = 2
events_fetcher = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=batch_size)
for i in range(0, 4):
events_batch = events_fetcher._fetch_events_batch()
self.assertEqual(len(events_batch), batch_size)
all_fetched_events.extend(events_batch)
remaining_events_batch = events_fetcher._fetch_events_batch()
self.assertEqual(len(remaining_events_batch), 1)
all_fetched_events.extend(remaining_events_batch)
self.assertEqual(self.events, all_fetched_events)
def test_fetch_events_explicit_single_batch(self):
self.events = range(0, 10)
events_fetcher = ExecutionEventsFetcher(self.client, 'execution_id',
batch_size=100)
batch_events = events_fetcher._fetch_events_batch()
self.assertListEqual(self.events, batch_events)
def test_fetch_events_explicit_single_batch(self):
self.events = range(0, 10)
events_fetcher = ExecutionEventsFetcher(self.client, 'execution_id',
batch_size=100)
batch_events = events_fetcher._fetch_events_batch()
self.assertListEqual(self.events, batch_events)
