# # Copyright (C) 2016 The Android Open Source Project # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # import os import logging import itertools from vts.runners.host import const from vts.testcases.kernel.ltp import ltp_configs from vts.testcases.kernel.ltp import ltp_enums from vts.testcases.kernel.ltp import test_case class TestCasesParser(object): """Load a ltp vts testcase definition file and parse it into a generator. Attributes: _data_path: string, the vts data path on host side _filter_func: function, a filter method that will emit exception if a test is filtered disabled_tests: list of string staging_tests: list of string """ def __init__(self, data_path, filter_func, disabled_tests, staging_tests): self._data_path = data_path self._filter_func = filter_func self._disabled_tests = disabled_tests self._staging_tests = staging_tests def ValidateDefinition(self, line): """Validate a tab delimited test case definition. Will check whether the given line of definition has three parts separated by tabs. It will also trim leading and ending white spaces for each part in returned tuple (if valid). Returns: A tuple in format (test suite, test name, test command) if definition is valid. None otherwise. """ items = [ item.strip() for item in line.split(ltp_enums.Delimiters.TESTCASE_DEFINITION) ] if not len(items) == 3 or not items: return None else: return items def Load(self, ltp_dir, n_bit, run_staging=False): """Read the definition file and yields a TestCase generator.""" run_scritp = self.GenerateLtpRunScript() for line in run_scritp: items = self.ValidateDefinition(line) if not items: continue testsuite, testname, command = items # Tests failed to build will have prefix "DISABLED_" if testname.startswith("DISABLED_"): logging.info("[Parser] Skipping test case {}-{}. Reason: " "not built".format(testsuite, testname)) continue # Some test cases contain semicolons in their commands, # and we replace them with && command = command.replace(';', '&&') testcase = test_case.TestCase( testsuite=testsuite, testname=testname, command=command) test_display_name = "{}_{}bit".format(str(testcase), n_bit) # Check runner's base_test filtering method try: self._filter_func(test_display_name) except: logging.info("[Parser] Skipping test case %s. Reason: " "filtered" % testcase.fullname) testcase.is_filtered = True testcase.note = "filtered" # For skipping tests that are not designed or ready for Android if test_display_name in self._disabled_tests: logging.info("[Parser] Skipping test case %s. Reason: " "disabled" % testcase.fullname) continue # For separating staging tests from stable tests if test_display_name in self._staging_tests: if not run_staging: # Skip staging tests in stable run continue else: testcase.is_staging = True testcase.note = "staging" else: if run_staging: # Skip stable tests in staging run continue logging.info("[Parser] Adding test case %s." % testcase.fullname) yield testcase def ReadCommentedTxt(self, filepath): '''Read a lines of a file that are not commented by #. Args: filepath: string, path of file to read Returns: A set of string representing non-commented lines in given file ''' if not filepath: logging.error('Invalid file path') return None with open(filepath, 'r') as f: lines_gen = (line.strip() for line in f) return set( line for line in lines_gen if line and not line.startswith('#')) def GenerateLtpTestCases(self, testsuite, disabled_tests_list): '''Generate test cases for each ltp test suite. Args: testsuite: string, test suite name Returns: A list of string ''' testsuite_script = os.path.join(self._data_path, ltp_configs.LTP_RUNTEST_DIR, testsuite) result = [] for line in open(testsuite_script, 'r'): line = line.strip() if not line or line.startswith('#'): continue testname = line.split()[0] testname_prefix = ('DISABLED_' if testname in disabled_tests_list else '') testname_modified = testname_prefix + testname result.append("\t".join([testsuite, testname_modified, line[len( testname):].strip()])) return result def GenerateLtpRunScript(self): '''Given a scenario group generate test case script. Args: scenario_group: string, file path of scanerio group file Returns: A list of string ''' disabled_tests_path = os.path.join( self._data_path, ltp_configs.LTP_DISABLED_BUILD_TESTS_CONFIG_PATH) disabled_tests_list = self.ReadCommentedTxt(disabled_tests_path) result = [] for testsuite in ltp_configs.TEST_SUITES: result.extend( self.GenerateLtpTestCases(testsuite, disabled_tests_list)) #TODO(yuexima): remove duplicate return result