[Android] Add --repeat for gtests and instrumentation tests.

build/android/pylib/results/json_results.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import json
 
 from pylib.base import base_test_result
 
 
-def GenerateResultsDict(test_run_result):
-  """Create a results dict from |test_run_result| suitable for writing to JSON.
+def GenerateResultsDict(test_run_results):
+  """Create a results dict from |test_run_results| suitable for writing to JSON.
   Args:
-    test_run_result: a base_test_result.TestRunResults object.
+    test_run_results: a list of base_test_result.TestRunResults objects.
   Returns:
     A results dict that mirrors the one generated by
       base/test/launcher/test_results_tracker.cc:SaveSummaryAsJSON.
   """
   # Example json output.
   # {
   #   "global_tags": [],
   #   "all_tests": [
   #     "test1",
   #     "test2",
@@ skipping 13 matching lines @@
   #       "test2": [
   #         {
   #           "status": "FAILURE",
   #           "elapsed_time_ms": 12,
   #           "output_snippet": "",
   #           "output_snippet_base64": "",
   #           "losless_snippet": "",
   #         },
   #       ],
   #     },
+  #     {
+  #       "test1": [
+  #         {
+  #           "status": "SUCCESS",
+  #           "elapsed_time_ms": 1,
+  #           "output_snippet": "",
+  #           "output_snippet_base64": "",
+  #           "losless_snippet": "",
+  #         },
+  #       ],
+  #       "test2": [
+  #         {
+  #           "status": "FAILURE",
+  #           "elapsed_time_ms": 12,
+  #           "output_snippet": "",
+  #           "output_snippet_base64": "",
+  #           "losless_snippet": "",
+  #         },
+  #       ],
+  #     },
+  #     ...
   #   ],
   # }
 
-  assert isinstance(test_run_result, base_test_result.TestRunResults)
-
   def status_as_string(s):
     if s == base_test_result.ResultType.PASS:
       return 'SUCCESS'
     elif s == base_test_result.ResultType.SKIP:
       return 'SKIPPED'
     elif s == base_test_result.ResultType.FAIL:
       return 'FAILURE'
     elif s == base_test_result.ResultType.CRASH:
       return 'CRASH'
     elif s == base_test_result.ResultType.TIMEOUT:
       return 'TIMEOUT'
     elif s == base_test_result.ResultType.UNKNOWN:
       return 'UNKNOWN'
 
-  def generate_iteration_data(t):
-    return {
-      t.GetName(): [
-        {
-          'status': status_as_string(t.GetType()),
-          'elapsed_time_ms': t.GetDuration(),
-          'output_snippet': '',
-          'losless_snippet': '',
-          'output_snippet_base64:': '',
-        }
-      ]
+  all_tests = set()
+  per_iteration_data = []
+  for test_run_result in test_run_results:
+    iteration_data = {
+      t.GetName(): [{
+        'status': status_as_string(t.GetType()),
+        'elapsed_time_ms': t.GetDuration(),
+        'output_snippet': '',
+        'losless_snippet': '',
+        'output_snippet_base64:': '',
+      }]
+      for t in test_run_result.GetAll()
     }
-
-  all_tests_tuple, per_iteration_data_tuple = zip(
-      *[(t.GetName(), generate_iteration_data(t))
-        for t in test_run_result.GetAll()])
+    all_tests = all_tests.union(set(iteration_data.iterkeys()))
+    per_iteration_data.append(iteration_data)
 
   return {
     'global_tags': [],
-    'all_tests': list(all_tests_tuple),
+    'all_tests': sorted(list(all_tests)),
     # TODO(jbudorick): Add support for disabled tests within base_test_result.
     'disabled_tests': [],
-    'per_iteration_data': list(per_iteration_data_tuple),
+    'per_iteration_data': per_iteration_data,
   }
 
 
 def GenerateJsonResultsFile(test_run_result, file_path):
   """Write |test_run_result| to JSON.
 
   This emulates the format of the JSON emitted by
   base/test/launcher/test_results_tracker.cc:SaveSummaryAsJSON.
 
   Args:
@@ skipping 30 matching lines @@
   testsuite_runs = json_results['per_iteration_data']
   for testsuite_run in testsuite_runs:
     for test, test_runs in testsuite_run.iteritems():
       results_list.extend(
           [base_test_result.BaseTestResult(test,
                                            string_as_status(tr['status']),
                                            duration=tr['elapsed_time_ms'])
           for tr in test_runs])
   return results_list