def build_job_scheduler_factory(self, master_config, job_graph):
output_stream_dir = master_config.output_stream_dir or self.working_dir
action_runner = actioncommand.create_action_runner_factory_from_config(
master_config.action_runner, )
return JobSchedulerFactory(
self.context,
output_stream_dir,
master_config.time_zone,
action_runner,
job_graph,
)
def setup_factory(self):
self.context = mock.Mock()
self.output_stream_dir = mock.Mock()
self.time_zone = mock.Mock()
self.action_runner = mock.create_autospec(
actioncommand.SubprocessActionRunnerFactory, )
self.factory = JobSchedulerFactory(
self.context,
self.output_stream_dir,
self.time_zone,
self.action_runner,
)