def test_new_events_after_fetched_all(self):
self.events = range(0, 10)
execution_events = ExecutionEventsFetcher(self.client, 'execution_id')
execution_events.fetch_events()
added = range(20, 25)
self.events.extend(range(20, 25))
events = execution_events.fetch_events()
self.assertListEqual(added, events)
def test_get_remaining_events_timeout(self):
self.events = range(0, 20)
execution_events = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=1,
timeout=3)
try:
execution_events.fetch_events(get_remaining_events=True)
self.fail()
except RuntimeError:
pass
def test_several_batches(self):
self.events = range(0, 10)
execution_events = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=2)
all_events = []
for i in range(0, 5):
events = execution_events.fetch_events()
self.assertEqual(2, len(events))
all_events.extend(events)
events = execution_events.fetch_events()
self.assertEqual(0, len(events))
self.assertListEqual(self.events, all_events)
def test_events_progress(self):
self.events = range(0, 5)
execution_events = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=100)
events = execution_events.fetch_events()
self.assertEqual(5, len(events))
self.events.extend(range(0, 10))
events = execution_events.fetch_events()
self.assertEqual(10, len(events))
self.events.extend(range(0, 5))
events = execution_events.fetch_events()
self.assertEqual(5, len(events))
def test_get_remaining_events_several_batches(self):
self.events = range(0, 10)
execution_events = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=2)
events = execution_events.fetch_events(get_remaining_events=True)
self.assertListEqual(self.events, events)
def test_no_events(self):
execution_events = ExecutionEventsFetcher(self.client,
'execution_id',
batch_size=2)
events = execution_events.fetch_events()
self.assertEqual(0, len(events))
def test_single_batch(self):
self.events = range(0, 10)
execution_events = ExecutionEventsFetcher(self.client, 'execution_id')
events = execution_events.fetch_events()
self.assertListEqual(self.events, events)
