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)