Add local results details pages.

build/android/pylib/results/presentation/test_results_presentation.py

 #! /usr/bin/env python
 #
 # Copyright 2017 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 argparse
 import collections
 import json
 import tempfile
@@ skipping 256 matching lines @@
 def feedback_url(result_details_link):
   url_args = urllib.urlencode([
       ('labels', 'Pri-2,Type-Bug,Restrict-View-Google'),
       ('summary', 'Result Details Feedback:'),
       ('components', 'Test>Android'),
       ('comment', 'Please check out: %s' % result_details_link)])
   return 'https://bugs.chromium.org/p/chromium/issues/entry?%s' % url_args
 
 
 def results_to_html(results_dict, cs_base_url, bucket, test_name,
-                    builder_name, build_number):
-  """Convert list of test results into html format."""
+                    builder_name, build_number, local_output):
+  """Convert list of test results into html format.
 
-  test_rows_header, test_rows = create_test_table(results_dict, cs_base_url,
-                                                  test_name)
+  Args:
+    local_output: Whether this results file is uploaded to Google Storage or
+        just a local file.
+  """
+  test_rows_header, test_rows = create_test_table(
+      results_dict, cs_base_url, test_name)
   suite_rows_header, suite_rows, suite_row_footer = create_suite_table(
       results_dict)
 
   suite_table_values = {
     'table_id': 'suite-table',
     'table_headers': suite_rows_header,
     'table_row_blocks': suite_rows,
     'table_footer': suite_row_footer,
   }
 
   test_table_values = {
     'table_id': 'test-table',
     'table_headers': test_rows_header,
     'table_row_blocks': test_rows,
   }
 
   main_template = JINJA_ENVIRONMENT.get_template(
       os.path.join('template', 'main.html'))
-  dest = google_storage_helper.unique_name(
-      '%s_%s_%s' % (test_name, builder_name, build_number))
 
-  result_details_link = google_storage_helper.get_url_link(
-      dest, '%s/html' % bucket)
-
-  return (main_template.render(  #  pylint: disable=no-member
-      {'tb_values': [suite_table_values, test_table_values],
-       'feedback_url': feedback_url(result_details_link)
-      }), dest, result_details_link)
+  if local_output:
+    html_render = main_template.render(  #  pylint: disable=no-member
+        {
+          'tb_values': [suite_table_values, test_table_values]
+        })
+    return html_render
+  else:
+    dest = google_storage_helper.unique_name(
+        '%s_%s_%s' % (test_name, builder_name, build_number))
+    result_details_link = google_storage_helper.get_url_link(
+        dest, '%s/html' % bucket)
+    html_render = main_template.render(  #  pylint: disable=no-member
+        {
+          'tb_values': [suite_table_values, test_table_values],
+          'feedback_url': feedback_url(result_details_link),
+        })
+    return (html_render, dest, result_details_link)
 
 
-def result_details(json_path, cs_base_url, bucket, test_name,
-                   builder_name, build_number):
-  """Get result details from json path and then convert results to html."""
+def result_details(json_path, test_name, cs_base_url, bucket=None,
+                   builder_name=None, build_number=None, local_output=False):
+  """Get result details from json path and then convert results to html.
+
+  Args:
+    local_output: Whether this results file is uploaded to Google Storage or
+        just a local file.
+  """
 
   with open(json_path) as json_file:
     json_object = json.loads(json_file.read())
 
   if not 'per_iteration_data' in json_object:
     return 'Error: json file missing per_iteration_data.'
 
   results_dict = collections.defaultdict(list)
   for testsuite_run in json_object['per_iteration_data']:
     for test, test_runs in testsuite_run.iteritems():
       results_dict[test].extend(test_runs)
-  return results_to_html(results_dict, cs_base_url, bucket,
-                         test_name, builder_name, build_number)
+  return results_to_html(results_dict, cs_base_url, bucket, test_name,
+                         builder_name, build_number, local_output)
 
 
 def upload_to_google_bucket(html, bucket, dest):
   with tempfile.NamedTemporaryFile(suffix='.html') as temp_file:
     temp_file.write(html)
     temp_file.flush()
     return google_storage_helper.upload(
         name=dest,
         filepath=temp_file.name,
         bucket='%s/html' % bucket,
         content_type='text/html',
         authenticated_link=True)
 
 
 def main():
   parser = argparse.ArgumentParser()
   parser.add_argument('--json-file', help='Path of json file.')
   parser.add_argument('--cs-base-url', help='Base url for code search.',
                       default='http://cs.chromium.org')
   parser.add_argument('--bucket', help='Google storage bucket.', required=True)
   parser.add_argument('--builder-name', help='Builder name.')
   parser.add_argument('--build-number', help='Build number.')
   parser.add_argument('--test-name', help='The name of the test.',
                       required=True)
   parser.add_argument(
       '-o', '--output-json',
-      help='(Swarming Merge Script API)'
-           ' Output JSON file to create.')
+      help='(Swarming Merge Script API) '
+           'Output JSON file to create.')
   parser.add_argument(
       '--build-properties',
       help='(Swarming Merge Script API) '
            'Build property JSON file provided by recipes.')
   parser.add_argument(
       '--summary-json',
-      help='(Swarming Merge Script API)'
-           ' Summary of shard state running on swarming.'
-           ' (Output of the swarming.py collect'
-           ' --task-summary-json=XXX command.)')
+      help='(Swarming Merge Script API) '
+           'Summary of shard state running on swarming. '
+           '(Output of the swarming.py collect '
+           '--task-summary-json=XXX command.)')
   parser.add_argument(
       'positional', nargs='*',
       help='output.json from shards.')
 
   args = parser.parse_args()
 
   if ((args.build_properties is None) ==
          (args.build_number is None or args.builder_name is None)):
     raise parser.error('Exactly one of build_perperties or '
                        '(build_number or builder_name) should be given.')
@@ skipping 28 matching lines @@
       raise parser.error('More than 1 json file specified.')
     json_file = args.positional[0]
   elif args.json_file:
     json_file = args.json_file
 
   if not os.path.exists(json_file):
     raise IOError('--json-file %s not found.' % json_file)
 
   # Link to result details presentation page is a part of the page.
   result_html_string, dest, result_details_link = result_details(
-      json_file, args.cs_base_url, args.bucket,
-      args.test_name, builder_name, build_number)
+      json_file, args.test_name, args.cs_base_url, args.bucket,
+      builder_name, build_number)
 
   result_details_link_2 = upload_to_google_bucket(
       result_html_string.encode('UTF-8'),
       args.bucket, dest)
-
   assert result_details_link == result_details_link_2, (
       'Result details link do not match. The link returned by get_url_link'
       ' should be the same as that returned by upload.')
 
   if args.output_json:
     with open(json_file) as original_json_file:
       json_object = json.load(original_json_file)
       json_object['links'] = {'result_details': result_details_link}
       with open(args.output_json, 'w') as f:
         json.dump(json_object, f)
   else:
     print result_details_link
 
 if __name__ == '__main__':
   sys.exit(main())