server/cros/audio/audio_test.py

# Copyright 2015 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 import error
from autotest_lib.server import site_utils
from autotest_lib.server import test
from autotest_lib.site_utils import lxc


class AudioTest(test.test):
    """Base class for audio tests.

    AudioTest provides a common warmup() function for the collection
    of audio tests.
    It is not mandatory to use this base class for audio tests, it is for
    convenience only.

    """

    def warmup(self):
        """Warmup for the test before executing main logic of the test."""
        # test.test is an old-style class.
        test.test.warmup(self)
        self.install_sox()
        self.check_sox_installed()


    def install_sox(self):
        """Install sox command on autotest drone."""
        try:
            lxc.install_package('sox')
        except error.ContainerError:
            logging.info('Can not install sox outside of container.')


    def check_sox_installed(self):
        """Checks if sox is installed.

        @raises: error.TestError if sox is not installed.

        """
        try:
            utils.run('sox --help')
            logging.info('Found sox executable.')
        except error.CmdError:
            error_message = 'sox command is not installed.'
            if site_utils.is_inside_chroot():
                error_message += ' sudo emerge sox to install sox in chroot'
            raise error.TestError(error_message)