Commit e4b6e934 authored by Takuto Ikuta's avatar Takuto Ikuta Committed by Chromium LUCI CQ

Revert "trigger_scripts: do not use use_swarming_go"

This reverts commit 18f22206.

Reason for revert: https://ci.chromium.org/p/chrome/builders/ci/chromeos-eve-chrome/12159

Original change's description:
> trigger_scripts: do not use use_swarming_go
>
> This is specified by default after https://crrev.com/c/2626865
>
> Bug: 1127205
> Change-Id: I6d1b64ab6c68283196cc5614bfd26a2e03b025be
> Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2627651
> Auto-Submit: Takuto Ikuta <tikuta@chromium.org>
> Reviewed-by: Kenneth Russell <kbr@chromium.org>
> Commit-Queue: Takuto Ikuta <tikuta@chromium.org>
> Cr-Commit-Position: refs/heads/master@{#843977}

TBR=kbr@chromium.org,tikuta@chromium.org,chromium-scoped@luci-project-accounts.iam.gserviceaccount.com,wenbinzhang@google.com

Change-Id: If016b45324dfcf43c70a35ebcf8d203af8a192c4
No-Presubmit: true
No-Tree-Checks: true
No-Try: true
Bug: 1127205
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2632546Reviewed-by: default avatarTakuto Ikuta <tikuta@chromium.org>
Commit-Queue: Takuto Ikuta <tikuta@chromium.org>
Cr-Commit-Position: refs/heads/master@{#843991}
parent 630506ce
...@@ -306,7 +306,10 @@ class BaseTestTriggerer(object): ...@@ -306,7 +306,10 @@ class BaseTestTriggerer(object):
suffix='.json') suffix='.json')
args_to_pass = self.modify_args(filtered_remaining_args, bot_index, args_to_pass = self.modify_args(filtered_remaining_args, bot_index,
shard_index, args.shards, json_temp) shard_index, args.shards, json_temp)
ret = self.run_swarming_go(args_to_pass, verbose) if args.use_swarming_go:
ret = self.run_swarming_go(args_to_pass, verbose)
else:
ret = self.run_swarming(args_to_pass, verbose)
if ret: if ret:
sys.stderr.write('Failed to trigger a task, aborting\n') sys.stderr.write('Failed to trigger a task, aborting\n')
return ret return ret
...@@ -321,9 +324,11 @@ class BaseTestTriggerer(object): ...@@ -321,9 +324,11 @@ class BaseTestTriggerer(object):
# which will be handled specially. # which will be handled specially.
merged_json['tasks'] = {} merged_json['tasks'] = {}
tasks = result_json['tasks'] tasks = result_json['tasks']
tasks = { if args.use_swarming_go:
task['request']['task_id']: task['request'] for task in tasks # TODO(crbug.com/1127205): remove this
} tasks = {
task['request']['task_id']: task['request'] for task in tasks
}
for k, v in tasks.items(): for k, v in tasks.items():
v['shard_index'] = shard_index v['shard_index'] = shard_index
merged_json['tasks'][k + ':%d:%d' % (shard_index, args.shards)] = v merged_json['tasks'][k + ':%d:%d' % (shard_index, args.shards)] = v
...@@ -356,7 +361,6 @@ class BaseTestTriggerer(object): ...@@ -356,7 +361,6 @@ class BaseTestTriggerer(object):
@staticmethod @staticmethod
def add_use_swarming_go_arg(parser): def add_use_swarming_go_arg(parser):
# TODO(crbug.com/1127205): remove this.
parser.add_argument( parser.add_argument(
'--use-swarming-go', '--use-swarming-go',
default=False, default=False,
......
...@@ -129,7 +129,9 @@ def main(): ...@@ -129,7 +129,9 @@ def main():
]) ])
new_args += additional_args[1:] new_args += additional_args[1:]
return triggerer.run_swarming_go(new_args, True) if args.use_swarming_go:
return triggerer.run_swarming_go(new_args, True)
return triggerer.run_swarming(new_args, True)
if __name__ == '__main__': if __name__ == '__main__':
......
...@@ -16,6 +16,7 @@ class Args(object): ...@@ -16,6 +16,7 @@ class Args(object):
self.dump_json = '' self.dump_json = ''
self.multiple_trigger_configs = None self.multiple_trigger_configs = None
self.multiple_dimension_script_verbose = False self.multiple_dimension_script_verbose = False
self.use_swarming_go = False
class FakeTriggerer(perf_device_trigger.PerfDeviceTriggerer): class FakeTriggerer(perf_device_trigger.PerfDeviceTriggerer):
...@@ -60,11 +61,13 @@ class UnitTest(unittest.TestCase): ...@@ -60,11 +61,13 @@ class UnitTest(unittest.TestCase):
def setup_and_trigger(self, def setup_and_trigger(self,
previous_task_assignment_map, previous_task_assignment_map,
alive_bots, alive_bots,
dead_bots): dead_bots,
use_swarming_go=False):
args = Args() args = Args()
args.shards = len(previous_task_assignment_map) args.shards = len(previous_task_assignment_map)
args.dump_json = 'output.json' args.dump_json = 'output.json'
args.multiple_dimension_script_verbose = True args.multiple_dimension_script_verbose = True
args.use_swarming_go = use_swarming_go
swarming_args = [ swarming_args = [
'trigger', 'trigger',
'--swarming', '--swarming',
...@@ -83,14 +86,14 @@ class UnitTest(unittest.TestCase): ...@@ -83,14 +86,14 @@ class UnitTest(unittest.TestCase):
triggerer = FakeTriggerer(args, swarming_args, triggerer = FakeTriggerer(args, swarming_args,
self.get_files(args.shards, previous_task_assignment_map, self.get_files(args.shards, previous_task_assignment_map,
alive_bots, dead_bots)) alive_bots, dead_bots, use_swarming_go=use_swarming_go))
triggerer.trigger_tasks( triggerer.trigger_tasks(
args, args,
swarming_args) swarming_args)
return triggerer return triggerer
def get_files(self, num_shards, previous_task_assignment_map, def get_files(self, num_shards, previous_task_assignment_map,
alive_bots, dead_bots): alive_bots, dead_bots, use_swarming_go):
files = {} files = {}
file_index = 0 file_index = 0
files['base_trigger_dimensions%d.json' % file_index] = ( files['base_trigger_dimensions%d.json' % file_index] = (
...@@ -110,13 +113,29 @@ class UnitTest(unittest.TestCase): ...@@ -110,13 +113,29 @@ class UnitTest(unittest.TestCase):
self.generate_last_task_to_shard_query_response(i, bot_id)) self.generate_last_task_to_shard_query_response(i, bot_id))
file_index = file_index + 1 file_index = file_index + 1
for i in xrange(num_shards): for i in xrange(num_shards):
task = { if use_swarming_go:
'tasks': [{ task = {
'tasks': [{
'request': {
'task_id': 'f%d' % i,
},
}],
}
else:
task = {
'base_task_name': 'webgl_conformance_tests',
'request': { 'request': {
'task_id': 'f%d' % i, 'expiration_secs': 3600,
'properties': {
'execution_timeout_secs': 3600,
},
},
'tasks': {
'webgl_conformance_tests on NVIDIA GPU on Windows': {
'task_id': 'f%d' % i,
},
}, },
}], }
}
files['base_trigger_dimensions%d.json' % file_index] = task files['base_trigger_dimensions%d.json' % file_index] = task
file_index = file_index + 1 file_index = file_index + 1
return files return files
...@@ -294,7 +313,8 @@ class UnitTest(unittest.TestCase): ...@@ -294,7 +313,8 @@ class UnitTest(unittest.TestCase):
triggerer = self.setup_and_trigger( triggerer = self.setup_and_trigger(
previous_task_assignment_map={0: 'build1'}, previous_task_assignment_map={0: 'build1'},
alive_bots=['build1'], alive_bots=['build1'],
dead_bots=[]) dead_bots=[],
use_swarming_go=True)
self.assertEquals(triggerer._triggered_with_swarming_go, 1) self.assertEquals(triggerer._triggered_with_swarming_go, 1)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment