151 lines
5.4 KiB
Python
151 lines
5.4 KiB
Python
# 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 json
|
|
import logging
|
|
|
|
from autotest_lib.client.bin import test
|
|
from autotest_lib.client.bin import utils
|
|
from autotest_lib.client.common_lib import error
|
|
from autotest_lib.client.common_lib.cros import chrome
|
|
from autotest_lib.client.common_lib.cros import policy
|
|
from autotest_lib.client.cros import cryptohome
|
|
from autotest_lib.client.cros.enterprise import enterprise_fake_dmserver
|
|
from autotest_lib.client.cros.power import power_status
|
|
|
|
|
|
class enterprise_PowerManagement(test.test):
|
|
"""Verify the power management policy setting."""
|
|
version = 1
|
|
|
|
def initialize(self, percent_initial_charge_min=10):
|
|
"""
|
|
Setup local variables and init the fake DM server
|
|
|
|
@param percent_initial_charge_min: Minimum percentage of battery
|
|
required for the test to run.
|
|
|
|
"""
|
|
# Username and password for the fake dm server can be anything
|
|
# they are not used to authenticate against GAIA.
|
|
self.username = 'fake-user@managedchrome.com'
|
|
self.password = 'fakepassword'
|
|
|
|
self._power_status = power_status.get_status()
|
|
if not self._power_status.on_ac():
|
|
# Ensure that the battery has some charge.
|
|
self._power_status.assert_battery_state(percent_initial_charge_min)
|
|
logging.info("Device power type is %s", self._power_type)
|
|
|
|
# Note: FakeDMServer requires policy protos to be installed.
|
|
policy.install_protobufs(self.autodir, self.job)
|
|
self.fake_dm_server = enterprise_fake_dmserver.FakeDMServer()
|
|
self.fake_dm_server.start(self.tmpdir, self.debugdir)
|
|
|
|
def cleanup(self):
|
|
"""Close out anything used by this test."""
|
|
self.fake_dm_server.stop()
|
|
|
|
@property
|
|
def _power_type(self):
|
|
"""
|
|
Returns appropriate power type based on whether DUT is on AC or not.
|
|
|
|
@returns string of power type.
|
|
|
|
"""
|
|
if self._power_status.on_ac():
|
|
return "AC"
|
|
|
|
return "Battery"
|
|
|
|
def _setup_lock_policy(self):
|
|
"""Setup policy to lock screen in 10 seconds of idle time."""
|
|
self._screen_lock_delay = 10
|
|
screen_lock_policy = '{ "%s": %d }' % (self._power_type,
|
|
self._screen_lock_delay * 1000)
|
|
policy_blob = """{
|
|
"google/chromeos/user": {
|
|
"mandatory": {
|
|
"ScreenLockDelays": %s
|
|
}
|
|
},
|
|
"managed_users": [ "*" ],
|
|
"policy_user": "%s",
|
|
"current_key_index": 0,
|
|
"invalidation_source": 16,
|
|
"invalidation_name": "test_policy"
|
|
}""" % (json.dumps(screen_lock_policy), self.username)
|
|
|
|
self.fake_dm_server.setup_policy(policy_blob)
|
|
|
|
def _setup_logout_policy(self):
|
|
"""Setup policy to logout in 10 seconds of idle time."""
|
|
self._screen_logout_delay = 10
|
|
idle_settings_policy = '''{
|
|
"%s": {
|
|
"Delays": {
|
|
"ScreenDim": 2000,
|
|
"ScreenOff": 3000,
|
|
"IdleWarning": 4000,
|
|
"Idle": %d
|
|
},
|
|
"IdleAction": "Logout"
|
|
}
|
|
}''' % (self._power_type, self._screen_logout_delay * 1000)
|
|
|
|
policy_blob = """{
|
|
"google/chromeos/user": {
|
|
"mandatory": {
|
|
"PowerManagementIdleSettings": %s
|
|
}
|
|
},
|
|
"managed_users": [ "*" ],
|
|
"policy_user": "%s",
|
|
"current_key_index": 0,
|
|
"invalidation_source": 16,
|
|
"invalidation_name": "test_policy"
|
|
}""" % (json.dumps(idle_settings_policy), self.username)
|
|
|
|
self.fake_dm_server.setup_policy(policy_blob)
|
|
|
|
def _create_chrome(self):
|
|
"""
|
|
Create an instance of chrome.
|
|
|
|
@returns a telemetry browser instance.
|
|
|
|
"""
|
|
extra_browser_args = '--device-management-url=%s ' % (
|
|
self.fake_dm_server.server_url)
|
|
return chrome.Chrome(
|
|
extra_browser_args=extra_browser_args,
|
|
autotest_ext=True,
|
|
disable_gaia_services=False,
|
|
gaia_login=False,
|
|
username=self.username,
|
|
password=self.password,
|
|
expect_policy_fetch=True)
|
|
|
|
def run_once(self):
|
|
"""Run the power management policy tests."""
|
|
self._setup_lock_policy()
|
|
with self._create_chrome() as cr:
|
|
utils.poll_for_condition(
|
|
lambda: cr.login_status['isScreenLocked'],
|
|
exception=error.TestFail('User is not locked'),
|
|
timeout=self._screen_lock_delay * 2,
|
|
sleep_interval=1,
|
|
desc='Expects to find Chrome locked.')
|
|
|
|
self._setup_logout_policy()
|
|
with self._create_chrome() as cr:
|
|
utils.poll_for_condition(
|
|
lambda: not cryptohome.is_vault_mounted(user=self.username,
|
|
allow_fail=True),
|
|
exception=error.TestFail('User is not logged out'),
|
|
timeout=self._screen_logout_delay*2,
|
|
sleep_interval=1,
|
|
desc='Expects to find user logged out.')
|