• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2015 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4"""
5Module to be included by the main test executor file within each test area
6of MBIM compliance suite. This harness is responsible for loading the
7required test within an area and invoking it.
8
9"""
10
11import imp
12import os
13
14import common
15from autotest_lib.client.bin import test
16from autotest_lib.client.cros.cellular.mbim_compliance import mbim_errors
17
18
19class MbimTestRunner(test.test):
20    """
21    Main test class which hooks up the MBIM compliance suite to autotest.
22    It invokes the MBIM tests within an area of the MBIM compliance test suite.
23
24    """
25    _TEST_AREA_FOLDER = None
26
27    version = 1
28
29    def run_once(self, subtest_name, **kwargs):
30        """
31        Method invoked by autotest framework to start a test from the
32        corresponding control file.
33
34        @param subtest_name: Name of the compliance test to be invoked.
35                The test has to be in the same folder as the control file and
36                should have file_name and class_name as |subtest_name|.
37        @param kwargs: Optional arguments which are passed to the actual test
38                being run.
39
40        """
41        module_name = os.path.join(self._TEST_AREA_FOLDER, subtest_name + ".py")
42        try:
43            test_module = imp.load_source(subtest_name, module_name)
44        except ImportError:
45            mbim_errors.log_and_raise(mbim_errors.MBIMComplianceFrameworkError,
46                                      'Test module %s not found', module_name)
47        try:
48            test_class = getattr(test_module, subtest_name)
49        except AttributeError:
50            mbim_errors.log_and_raise(mbim_errors.MBIMComplianceFrameworkError,
51                                      'Test class %s not found', subtest_name)
52        test = test_class()
53        test.run_test(**kwargs)
54