commit 7d7200707d0718df785c3e016df693ce330a4696
author Chung-yih Wang <cywang@google.com> Tue Mar 01 09:45:43 2016 +0800
committer chrome-bot <chrome-bot@chromium.org> Fri Mar 11 15:17:24 2016 -0800
tree 784666feee8d387fc633465451fc78cfc99632ea
parent b00182f46caff82f8d3ab11dd2b31d7fb8baeef0

chrome.py: Append extra browser args for cheets platforms

BUG=b:27406867
TEST=None
CQ-DEPEND=CL:*250277

Change-Id: I9682c28951c25cc68b1a69f80119f051ea7ed12c
Signed-off-by: Chung-yih Wang <cywang@google.com>
Reviewed-on: https://chromium-review.googlesource.com/330983
Reviewed-by: Ben Cheng <bccheng@chromium.org>

client/common_lib/cros/chrome.py

diff --git a/client/common_lib/cros/chrome.py b/client/common_lib/cros/chrome.py
index 8bf6c55..87e71e1 100644
--- a/client/common_lib/cros/chrome.py
+++ b/client/common_lib/cros/chrome.py
@@ -17,6 +17,7 @@
     """Wrapper for creating a telemetry browser instance with extensions."""
 
 
+    CHEETS = '-cheets'
     BROWSER_TYPE_LOGIN = 'system'
     BROWSER_TYPE_GUEST = 'system-guest'
 
@@ -61,6 +62,12 @@
             extension_paths.append(self._autotest_ext_path)
 
         finder_options = browser_options.BrowserFinderOptions()
+        # Append cheets specific browser args
+        is_cheets_platform = utils.get_current_board().endswith(self.CHEETS)
+        if is_cheets_platform:
+            from autotest_lib.client.common_lib.cros import cheets
+            extra_browser_args = cheets.append_extra_args(extra_browser_args)
+            logged_in = True
         self._browser_type = (self.BROWSER_TYPE_LOGIN
                 if logged_in else self.BROWSER_TYPE_GUEST)
         finder_options.browser_type = self.browser_type
@@ -112,6 +119,8 @@
             try:
                 browser_to_create = browser_finder.FindBrowser(finder_options)
                 self._browser = browser_to_create.Create(finder_options)
+                if is_cheets_platform:
+                    cheets.post_processing_after_browser()
                 break
             except (exceptions.LoginException) as e:
                 logging.error('Timed out logging in, tries=%d, error=%s',