Add json test results format support for SwarmingIsolatedScriptTest

scripts/slave/recipe_modules/swarming/results_merger.py

Index: scripts/slave/recipe_modules/swarming/results_merger.py
diff --git a/scripts/slave/recipe_modules/swarming/results_merger.py b/scripts/slave/recipe_modules/swarming/results_merger.py
new file mode 100644
index 0000000000000000000000000000000000000000..be2a97b6671bfba007e22bacb5ce5980db253b70
--- /dev/null
+++ b/scripts/slave/recipe_modules/swarming/results_merger.py
@@ -0,0 +1,103 @@
+# Copyright 2016 The Chromium Authors. All rights reserved.

[Vadim Sh., 2016/10/04 01:05:29]

I suspect you will have hard time convincing recipe tests that this code is 100%
covered, because it will probably ignore code coverage provided by
results_merger_unittest.py.

[nednguyen, 2016/10/04 01:49:52]

I plan to add both expectation test & unittest.  Howver, if change this to be a
runnable script, does expectation test's coverage include the script's code?

[Sergiy Byelozyorov, 2016/10/04 08:54:52]

+1 for unit tests, because IMHO we have too many expectation tests (essentially,
integration tests) covering just a few lines of code. But like Vadim, I am not
sure if our coverage system takes unittests into account when computing
coverage. You can try running the CQ in dry run on this CL to check that.

+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+
+def merge_test_results(shard_results_list):
+  """ Merge list of results.
+
+  Args:
+    shard_results_list: list of results to merge. All the results must have the
+      same format. Supported format are simplified JSON format & Chromium JSON
+      test results format version 3 (see
+      https://www.chromium.org/developers/the-json-test-results-format)
+
+  Returns:
+    a
+  """
+  if 'seconds_since_epoch' in shard_results_list[0]:
+    return _merge_json_test_result_format(shard_results_list)
+  else:
+    return _merge_simplified_json_format(shard_results_list)
+
+
+def _merge_simplified_json_format(shard_results_list):
+  # This code is specialized to the "simplified" JSON format that used to be
+  # the standard for recipes.
+
+  # These are the only keys we pay attention to in the output JSON.
+  merged_results = {
+    'successes': [],
+    'failures': [],
+    'valid': True,
+  }
+
+  for result_json in shard_results_list:
+    for key in merged_results:
+      if key in result_json:
+        if isinstance(merged_results[key], list):
+          merged_results[key].extend(result_json[key])
+        elif isinstance(merged_results[key], bool):
+          merged_results[key] = merged_results[key] and result_json[key]
+        else:
+          raise Exception(
+            'Unknown key type ' + type(merged_results[key]) +
+            ' when handling key ' + key + '.')  # pragma: no cover
+  return merged_results
+
+def _merge_json_test_result_format(shard_results_list):
+  # This code is specialized to the Chromium JSON test results format version 3:
+  # https://www.chromium.org/developers/the-json-test-results-format
+
+  # These are required fields for the JSON test result format version 3.
+  merged_results = {
+    'tests': {},
+    'interrupted': False,
+    'path_delimiter': '',
+    'version': 3,
+    'seconds_since_epoch': float('inf'),
+    'num_failures_by_type': {
+    }
+  }
+  for result_json in shard_results_list:
+    if not ('tests' in result_json and
+            'interrupted' in result_json and
+            'path_delimiter' in result_json and
+            'version' in result_json and
+            'seconds_since_epoch' in result_json and
+            'num_failures_by_type' in result_json):
+      raise Exception('Invalid json test results')
+
+    # Traverse the result_json's test trie & merged_results's test tries in
+    # DFS order & add the n to merged['tests'].
+    curr_result_nodes_queue = [result_json['tests']]
+    merged_results_nodes_queue = [merged_results['tests']]
+    while curr_result_nodes_queue:
+      curr_node = curr_result_nodes_queue.pop()
+      merged_node = merged_results_nodes_queue.pop()
+      for k, v in curr_node.iteritems():
+        if k in merged_node:
+          curr_result_nodes_queue.append(v)
+          merged_results_nodes_queue.append(merged_node[k])
+        else:
+          merged_node[k] = v
+
+    # Update the rest of the fields for merged_results.
+    merged_results['interrupted'] |= result_json['interrupted']
+    if not merged_results['path_delimiter']:
+      merged_results['path_delimiter'] = result_json['path_delimiter']
+    elif merged_results['path_delimiter'] != result_json['path_delimiter']:
+      raise Exception(
+          'Incosistent path delimiter: %s %s' %
+          (merged_results['path_delimiter'],
+           result_json['path_delimiter']))
+    if result_json['version'] != 3:
+      raise Exception(
+          'Only version 3 of json test result format is supported')
+    merged_results['seconds_since_epoch'] = min(
+        merged_results['seconds_since_epoch'],
+        result_json['seconds_since_epoch'])
+    for result_type, count in result_json['num_failures_by_type'].iteritems():
+      merged_results['num_failures_by_type'].setdefault(result_type, 0)
+      merged_results['num_failures_by_type'][result_type] += count
+  return merged_results