blob: 0593bc9002aa827663cea5036fe076b9c1d4dd09 [file] [log] [blame]
# Copyright (c) 2013 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.
"""Provides utility class for stopping and restarting services"""
import logging
from autotest_lib.client.bin import utils
from autotest_lib.client.common_lib import error
class ServiceStopper(object):
"""Class to manage CrOS services.
Public attributes:
services_to_stop: list of services that should be stopped
Public constants:
POWER_DRAW_SERVICES: list of services that influence power test in
unpredictable/undesirable manners.
Public methods:
stop_sevices: stop running system services.
restore_services: restore services that were previously stopped.
Private attributes:
_services_stopped: list of services that were successfully stopped
POWER_DRAW_SERVICES = ['powerd', 'update-engine', 'bluetoothd']
def __init__(self, services_to_stop=[]):
"""Initialize instance of class.
By Default sets an empty list of services.
self.services_to_stop = services_to_stop
self._services_stopped = []
def stop_services(self):
"""Turn off managed services."""
for service in self.services_to_stop:
cmd = 'status %s' % service
out = utils.system_output(cmd, ignore_status=True)
is_stopped = 'start/running' not in out
if is_stopped:
utils.system('stop %s' % service)
except error.CmdError as e:
logging.warning('Error stopping service %s. %s',
service, str(e))
def restore_services(self):
"""Restore services that were stopped."""
for service in reversed(self._services_stopped):
utils.system('start %s' % service, ignore_status=True)
self._services_stopped = []