| # Copyright 2014 The Chromium OS 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 logging |
| |
| from autotest_lib.client.bin import utils |
| from autotest_lib.client.common_lib.cros import chrome, tpm_utils |
| from telemetry.core.platform import cros_interface |
| |
| |
| def _ExecuteOobeCmd(browser, cmd): |
| logging.info('Invoking ' + cmd) |
| oobe = browser.oobe |
| oobe.WaitForJavaScriptExpression('typeof Oobe !== \'undefined\'', 10) |
| oobe.ExecuteJavaScript(cmd) |
| |
| |
| def SwitchToRemora(browser): |
| """Switch to Remora enrollment. |
| |
| @param browser: telemetry browser object. |
| """ |
| _cri = cros_interface.CrOSInterface() |
| pid = _cri.GetChromePid() |
| # This will restart the browser. |
| chrome.Chrome.did_browser_crash( |
| lambda: _ExecuteOobeCmd(browser, |
| 'Oobe.remoraRequisitionForTesting();')) |
| utils.poll_for_condition(lambda: pid != _cri.GetChromePid(), timeout=60) |
| utils.poll_for_condition(lambda: browser.oobe_exists, timeout=30) |
| |
| _ExecuteOobeCmd(browser, 'Oobe.skipToLoginForTesting();') |
| tpm_utils.SaveTPMPassword() |
| |
| |
| def FinishEnrollment(oobe): |
| """Wait for enrollment to finish and dismiss the last enrollment screen. |
| |
| @param oobe: telemetry oobe object. |
| """ |
| oobe.WaitForJavaScriptExpression( |
| "document.getElementById('oauth-enrollment').className." |
| "search('oauth-enroll-state-success') != -1", 30) |
| oobe.EvaluateJavaScript('Oobe.enterpriseEnrollmentDone();') |
| |
| |
| def RemoraEnrollment(browser, user_id, password): |
| """Enterprise login for a Remora device. |
| |
| @param browser: telemetry browser object. |
| @param user_id: login credentials user_id. |
| @param password: login credentials password. |
| """ |
| SwitchToRemora(browser) |
| browser.oobe.NavigateGaiaLogin(user_id, password) |
| FinishEnrollment(browser.oobe) |