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)