| # Copyright 2019 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. |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.cros.enterprise import enterprise_policy_base |
| from autotest_lib.client.cros.input_playback import keyboard |
| |
| from telemetry.core import exceptions |
| |
| |
| class policy_NewTabPageLocation( |
| enterprise_policy_base.EnterprisePolicyTest): |
| """ |
| Tests the NewTabPageLocation policy in Chrome OS. |
| |
| If the NewTabPageLocation policy is set, when a NewTab is opened, |
| the page configured page will be directly loaded. |
| |
| """ |
| version = 1 |
| |
| def _homepage_check(self, case_value): |
| """ |
| Open a new tab and checks the proper page is opened. |
| |
| @param case_value: policy value for this case. |
| |
| """ |
| self.keyboard.press_key('ctrl+t') |
| |
| # Try to get a policy from the current tab. If it works, that means the |
| # policy page is open, and the newtab policy set it. Default (not set) |
| # would be chrome://newtab, and this try would not work. |
| try: |
| self._get_policy_stats_shown(self.cr.browser.tabs[-1], |
| 'NewTabPageLocation') |
| if not case_value: |
| raise error.TestFail( |
| 'NewTabPageLocation was set when it should not be!') |
| |
| except exceptions.EvaluateException: |
| if case_value: |
| raise error.TestFail( |
| 'NewTabPageLocation was not set when it should be!') |
| |
| if not case_value: |
| url = self.cr.browser.tabs[-1].GetUrl() |
| if url != 'chrome://newtab/': |
| raise error.TestFail( |
| 'NewTab was not "chrome://newtab/" instead got {}' |
| .format(url)) |
| |
| def run_once(self, case): |
| """ |
| Setup and run the test configured for the specified test case. |
| |
| @param case: Name of the test case to run. |
| |
| """ |
| self.keyboard = keyboard.Keyboard() |
| TEST_CASES = {'Set': 'chrome://policy', |
| 'NotSet': None} |
| |
| case_value = TEST_CASES[case] |
| policy_setting = {'NewTabPageLocation': case_value} |
| self.setup_case(user_policies=policy_setting) |
| self._homepage_check(case_value) |