def test_on_progress(self):
progress_callback = mock.MagicMock(name="progress_callback")
request = self.request(task=self.task, progress_callback=progress_callback)
request.on_progress("event_data", 0.0)
request.on_progress("event_data", 1.0)
expected_calls = [mock.call(self.task, "event_data", 0.0), mock.call(self.task, "event_data", 1.0)]
self.assertEqual(progress_callback.mock_calls, expected_calls)
def test_on_progress(self):
progress_callback = mock.MagicMock(name='progress_callback')
request = self.request(task=self.task,
progress_callback=progress_callback)
request.on_progress('event_data', 0.0)
request.on_progress('event_data', 1.0)
expected_calls = [
mock.call(self.task, 'event_data', 0.0),
mock.call(self.task, 'event_data', 1.0)
]
self.assertEqual(progress_callback.mock_calls, expected_calls)
def test_wait_for_any(self):
fs = [futures.Future(), futures.Future()]
ex = self.executor()
ex.wait_for_any(fs)
expected_calls = [mock.call(fs, None)]
self.assertEqual(self.wait_for_any_mock.mock_calls, expected_calls)
def test_wait_for_any(self):
fs = [futures.Future(), futures.Future()]
ex = self.executor()
ex.wait_for_any(fs)
expected_calls = [mock.call(fs, timeout=None)]
self.assertEqual(self.wait_for_any_mock.mock_calls, expected_calls)
def test_wait_for_any_with_timeout(self):
timeout = 30
fs = [futures.Future(), futures.Future()]
ex = self.executor()
ex.wait_for_any(fs, timeout)
main_mock_calls = [mock.call(fs, timeout)]
self.assertEqual(self.wait_for_any_mock.mock_calls, main_mock_calls)
def test_wait_for_any_with_timeout(self):
timeout = 30
fs = [futures.Future(), futures.Future()]
ex = self.executor()
ex.wait_for_any(fs, timeout)
master_mock_calls = [mock.call(fs, timeout=timeout)]
self.assertEqual(self.wait_for_any_mock.mock_calls, master_mock_calls)
