[rlz,cros] RLZ glue for ChromeOS.

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
 run pydoc on this file.
@@ skipping 4509 matching lines @@
       A dictionary with the following keys:
 
       'next_screen': The title of the next OOBE screen as a string.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = { 'command': 'AcceptOOBENetworkScreen' }
     return self._GetResultFromJSONRequest(cmd_dict, windex=None)
 
-  def AcceptOOBEEula(self, accepted, usage_stats_reporting=False):
+  def AcceptOOBEEula(self, accepted, usage_stats_reporting=False,
+                     rlz_enabled=False):
     """Accepts OOBE EULA and advances to the next screen.
 
     Assumes that we're already at the OOBE EULA screen.
 
     Args:
       accepted: Boolean indicating whether the EULA should be accepted.
       usage_stats_reporting: Boolean indicating whether UMA should be enabled.
+      rlz_enabled: Boolean indicating whether RLZ should be enabled.
 
     Returns:
       A dictionary with the following keys:
 
       'next_screen': The title of the next OOBE screen as a string.
 
     Raises:
       pyauto_errors.JSONInterfaceError if the automation call returns an error.
     """
     cmd_dict = { 'command': 'AcceptOOBEEula',
                  'accepted': accepted,
-                 'usage_stats_reporting': usage_stats_reporting }
+                 'usage_stats_reporting': usage_stats_reporting,
+                 'rlz_enabled': rlz_enabled }
     return self._GetResultFromJSONRequest(cmd_dict, windex=None)
 
   def CancelOOBEUpdate(self):
     """Skips update on OOBE and advances to the next screen.
 
     Returns:
       A dictionary with the following keys:
 
       'next_screen': The title of the next OOBE screen as a string.
 
@@ skipping 1968 matching lines @@
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()