Enable installing test CA on Android & add test for it

telemetry/telemetry/internal/util/webpagereplay_go_server.py

 # 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.
 
 """Start and stop Web Page Replay."""
 
 from py_utils import atexit_with_log
 import logging
 import os
 import re
 import signal
 import subprocess
 import sys
 import tempfile
 import urllib
 
 from telemetry.core import util
 from telemetry.internal import forwarders
 from telemetry.internal.util import binary_manager
 
 import py_utils
 
+
 _WPR_DIR = os.path.abspath(os.path.join(
     util.GetCatapultDir(), 'web_page_replay_go'))
+_WPR_KEY_FILE = os.path.join(_WPR_DIR, 'wpr_key.pem')
+_WPR_CERT_FILE = os.path.join(_WPR_DIR, 'wpr_cert.pem')
+
 
 class ReplayError(Exception):
   """Catch-all exception for the module."""
   pass
 
 
 class ReplayNotFoundError(ReplayError):
   def __init__(self, label, path):
     super(ReplayNotFoundError, self).__init__()
     self.args = (label, path)
@@ skipping 76 matching lines @@
     assert ((android_device_id is None and adb_path is None) or
             (android_device_id is not None and adb_path is not None)), (
                 'android_device_id and adb_path must be both specified or '
                 'not specified')
     go_binary_path = cls._GetGoBinaryPath()
     if android_device_id is None:
       subprocess.check_call([go_binary_path, 'installroot'])
     else:
       subprocess.check_call(
           [go_binary_path, 'installroot',
+           '--https_cert_file=%s' % _WPR_CERT_FILE,
+           '--https_key_file=%s' % _WPR_KEY_FILE,
            '--android_device_id=%s' % android_device_id,
            '--adb_binary_path=%s' % adb_path])
 
   @classmethod
   def RemoveRootCertificate(cls, android_device_id=None, adb_path=None):
     """Remove installed root certificate on the host machine or on remote
     Android device.
 
       Args:
         android_device_id: a string id of the Android device.
         adb_path: path to adb binary to use for issuing commands to the device.
           This is specified iff android_device_id is specified.
     """
     assert ((android_device_id is None and adb_path is None) or
             (android_device_id is not None and adb_path is not None)), (
                 'android_device_id and adb_path must be both specified or '
                 'not specified')
     go_binary_path = cls._GetGoBinaryPath()
     if android_device_id is None:
       subprocess.check_call([go_binary_path, 'removeroot'])
     else:
       subprocess.check_call(
           [go_binary_path, 'removeroot',
+           '--https_cert_file=%s' % _WPR_CERT_FILE,
+           '--https_key_file=%s' % _WPR_KEY_FILE,
            '--android_device_id=%s' % android_device_id,
            '--adb_binary_path=%s' % adb_path])
 
   @property
   def http_port(self):
     if not self._IsStarted():
       return None
     return self._started_ports['http']
 
   @property
   def https_port(self):
     if not self._IsStarted():
       return None
     return self._started_ports['https']
 
   @staticmethod
   def _GetCommandLine(go_binary_path, http_port, https_port,
                       replay_options, archive_path):
     """Set WPR command-line options. Can be overridden if needed."""
     for option in replay_options:
       if option not in ['--record', '--replay', '--inject_scripts=']:
         raise ValueError("Invalid replay options %s" % replay_options)
     cmd_line = [go_binary_path]
     if '--record' in replay_options:
       cmd_line.append('record')
     else:
       cmd_line.append('replay')
-    key_file = os.path.join(_WPR_DIR, 'wpr_key.pem')
-    cert_file = os.path.join(_WPR_DIR, 'wpr_cert.pem')
     inject_script = os.path.join(_WPR_DIR, 'deterministic.js')
     cmd_line.extend([
         '--http_port=%s' % http_port,
         '--https_port=%s' % https_port,
-        '--https_key_file=%s' % key_file,
-        '--https_cert_file=%s' % cert_file])
+        '--https_key_file=%s' % _WPR_KEY_FILE,
+        '--https_cert_file=%s' % _WPR_CERT_FILE])
     if '--inject_scripts=' in replay_options:
       cmd_line.append('--inject_scripts=')
     else:
       cmd_line.append('--inject_scripts=%s' % inject_script)
     cmd_line.append(archive_path)
     return cmd_line
 
   def _CheckPath(self, label, path):
     if not os.path.exists(path):
       raise ReplayNotFoundError(label, path)
@@ skipping 193 matching lines @@
   On posix system, running this function before starting replay fixes a
   bug that shows up when Telemetry is run as a background command from a
   script. https://crbug.com/254572.
 
   Background: Signal masks on Linux are inherited from parent
   processes. If anything invoking us accidentally masks SIGINT
   (e.g. by putting a process in the background from a shell script),
   sending a SIGINT to the child will fail to terminate it.
   """
   signal.signal(signal.SIGINT, signal.SIG_DFL)