def _test_fetch(zoo_keeper_client, enclosing_event, kubernetes_job_starter,
mock_lock_manager, some_data_set):
download_mgr = DownloadManager(
zoo_keeper_client,
kubernetes_job_starter,
mock_lock_manager,
data_set_to_path,
size_estimator=mock_size_estimator,
)
on_done = create_autospec(DownloadOnDone)
download_mgr.fetch(some_data_set, enclosing_event, on_done)
return on_done
def test_pass_through_estimator_error(
enclosing_event: BenchmarkEvent,
some_data_set: DownloadableContent,
failing_size_estimator: ContentSizeEstimator,
zoo_keeper_client: KazooClient,
kubernetes_job_starter: DownloadDispatcher,
mock_lock_manager: RWLockManager,
mock_lock: RWLock,
):
download_mgr = DownloadManager(zoo_keeper_client, kubernetes_job_starter,
mock_lock_manager, data_set_to_path,
failing_size_estimator)
on_done = create_autospec(DownloadOnDone)
download_mgr.fetch(some_data_set, enclosing_event, on_done)
_verify_failed_estimator(mock_lock, on_done, some_data_set)