| # 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. |
| import time |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.common_lib import utils |
| from autotest_lib.client.common_lib.cros import retry |
| |
| from autotest_lib.client.cros.enterprise import enterprise_policy_base |
| |
| from py_utils import TimeoutException |
| |
| |
| class policy_DeviceWilcoDtcAllowed( |
| enterprise_policy_base.EnterprisePolicyTest): |
| """ |
| Test for the DeviceWilcoDtcAllowed policy. |
| |
| If the policy is set to True then Wilco daemons should be running. |
| If the policy is set to False/None then Wilco daemons should not be |
| running. |
| |
| """ |
| version = 1 |
| _POLICY = 'DeviceWilcoDtcAllowed' |
| |
| @retry.retry(TimeoutException, timeout_min=5, delay_sec=10) |
| def _run_setup_case(self, case): |
| self.setup_case(device_policies={self._POLICY: case}, enroll=True, |
| extra_chrome_flags=['--user-always-affiliated']) |
| |
| def _update_policy_value(self, value): |
| self.fake_dm_server.setup_policy(self._make_json_blob( |
| device_policies={'DeviceWilcoDtcAllowed': value})) |
| policy_tab = self.navigate_to_url(self.CHROME_POLICY_PAGE) |
| reload_button = "document.querySelector('button#reload-policies')" |
| policy_tab.ExecuteJavaScript("%s.click()" % reload_button) |
| utils.poll_for_condition( |
| lambda: self._get_policy_value_from_new_tab( |
| 'DeviceWilcoDtcAllowed') == value, |
| exception=error.TestFail('Policy value not updated.'), |
| timeout=10, |
| sleep_interval=1) |
| |
| def _check_status(self, expected_status): |
| result = utils.run('status wilco_dtc') |
| |
| if expected_status: |
| if 'process' not in result.stdout: |
| raise error.TestFail( |
| 'Wilco daemons are not running and they should be.') |
| else: |
| if 'process' in result.stdout: |
| raise error.TestFail( |
| 'Wilco daemons are running and they should not be.') |
| |
| |
| def run_once(self, case): |
| """ |
| Entry point of this test. |
| |
| @param case: True, False, or None for the value of the policy. |
| |
| """ |
| self._run_setup_case(case) |
| |
| if case is True: |
| self._check_status(case) |
| self._update_policy_value(False) |
| # This sleep is needed to give status some time to update. |
| time.sleep(2) |
| self._check_status(False) |
| |
| if case is False: |
| self._check_status(case) |
| self._update_policy_value(True) |
| # This sleep is needed to give status some time to update. |
| time.sleep(2) |
| self._check_status(True) |
| |
| if case is None: |
| self._check_status(None) |