def test_main_grpc(self): def load_and_run( manifest, swarming_server, is_grpc, cost_usd_hour, start, json_file, run_isolated_flags, bot_file, auth_params_file): self.assertEqual('foo', manifest) self.assertEqual('http://localhost', swarming_server) self.assertTrue(is_grpc) self.assertEqual(3600., cost_usd_hour) self.assertEqual(time.time(), start) self.assertEqual('task_summary.json', json_file) self.assertEqual(['--min-free-space', '1'], run_isolated_flags) self.assertEqual('/path/to/bot-file', bot_file) self.assertEqual('/path/to/auth-params-file', auth_params_file) self.mock(task_runner, 'load_and_run', load_and_run) cmd = [ '--swarming-server', 'http://localhost', '--in-file', 'foo', '--out-file', 'task_summary.json', '--cost-usd-hour', '3600', '--start', str(time.time()), '--bot-file', '/path/to/bot-file', '--auth-params-file', '/path/to/auth-params-file', '--is-grpc', '--', '--min-free-space', '1', ] self.assertEqual(0, task_runner.main(cmd))
def test_main_reboot(self): def load_and_run(manifest, swarming_server, cost_usd_hour, start): self.assertEqual('foo', manifest) self.assertEqual('http://localhost', swarming_server.url) self.assertEqual(3600., cost_usd_hour) self.assertEqual(time.time(), start) return False self.mock(task_runner, 'load_and_run', load_and_run) cmd = [ '--swarming-server', 'http://localhost', '--file', 'foo', '--cost-usd-hour', '3600', '--start', str(time.time()), ] self.assertEqual(task_runner.TASK_FAILED, task_runner.main(cmd))
def test_main(self): def load_and_run( manifest, swarming_server, cost_usd_hour, start, json_file): self.assertEqual('foo', manifest) self.assertEqual('http://localhost', swarming_server.url) self.assertEqual(3600., cost_usd_hour) self.assertEqual(time.time(), start) self.assertEqual('task_summary.json', json_file) return True self.mock(task_runner, 'load_and_run', load_and_run) cmd = [ '--swarming-server', 'http://localhost', '--file', 'foo', '--cost-usd-hour', '3600', '--start', str(time.time()), '--json-file', 'task_summary.json', ] self.assertEqual(0, task_runner.main(cmd))
def test_main_reboot(self): def load_and_run( manifest, swarming_server, cost_usd_hour, start, json_file, min_free_space): self.assertEqual('foo', manifest) self.assertEqual('http://localhost', swarming_server) self.assertEqual(3600., cost_usd_hour) self.assertEqual(time.time(), start) self.assertEqual('task_summary.json', json_file) self.assertEqual(1, min_free_space) self.mock(task_runner, 'load_and_run', load_and_run) cmd = [ '--swarming-server', 'http://localhost', '--in-file', 'foo', '--out-file', 'task_summary.json', '--cost-usd-hour', '3600', '--start', str(time.time()), '--min-free-space', '1', ] self.assertEqual(0, task_runner.main(cmd))
def test_main_reboot(self): def load_and_run(manifest, swarming_server, cost_usd_hour, start, json_file): self.assertEqual('foo', manifest) self.assertEqual('http://localhost', swarming_server.url) self.assertEqual(3600., cost_usd_hour) self.assertEqual(time.time(), start) self.assertEqual('task_summary.json', json_file) return False self.mock(task_runner, 'load_and_run', load_and_run) cmd = [ '--swarming-server', 'http://localhost', '--file', 'foo', '--cost-usd-hour', '3600', '--start', str(time.time()), '--json-file', 'task_summary.json', ] self.assertEqual(task_runner.TASK_FAILED, task_runner.main(cmd))
def CMDtask_runner(args): """Internal command to run a swarming task.""" # TODO(maruel): rename function. logging_utils.prepare_logging('task_runner.log') import task_runner return task_runner.main(args)
def CMDtask_runner(args): """Internal command to run a swarming task.""" logging_utils.prepare_logging('task_runner.log') import task_runner return task_runner.main(args)