1#!/usr/bin/env python3 2# 3# Copyright 2019 The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16"""Main entrypoint for all of atest's unittest.""" 17 18import os 19import sys 20import unittest 21from importlib import import_module 22 23 24def get_test_modules(): 25 """Returns a list of test modules. 26 27 Finds all the test files (*_test.py) and get their relative 28 path (internal/lib/utils_test.py) and translate it to an import path and 29 strip the py ext (internal.lib.utils_test). 30 31 Returns: 32 List of strings (the testable module import path). 33 """ 34 testable_modules = [] 35 base_path = os.path.dirname(os.path.realpath(__file__)) 36 37 for dirpath, _, files in os.walk(base_path): 38 for f in files: 39 if f.endswith("_test.py"): 40 # Now transform it into a relative import path. 41 full_file_path = os.path.join(dirpath, f) 42 rel_file_path = os.path.relpath(full_file_path, base_path) 43 rel_file_path, _ = os.path.splitext(rel_file_path) 44 rel_file_path = rel_file_path.replace(os.sep, ".") 45 testable_modules.append(rel_file_path) 46 47 return testable_modules 48 49 50def main(_): 51 """Main unittest entry. 52 53 Args: 54 argv: A list of system arguments. (unused) 55 56 Returns: 57 0 if success. None-zero if fails. 58 """ 59 # Force remove syspath related to atest to make sure the env is clean. 60 # These tests need to run in isolation (to find bugs like b/132086641) 61 # so we scrub out all atest modules. 62 for path in sys.path: 63 if path.endswith('/atest'): 64 sys.path.remove(path) 65 test_modules = get_test_modules() 66 for mod in test_modules: 67 import_module(mod) 68 69 loader = unittest.defaultTestLoader 70 test_suite = loader.loadTestsFromNames(test_modules) 71 runner = unittest.TextTestRunner(verbosity=2) 72 result = runner.run(test_suite) 73 sys.exit(not result.wasSuccessful()) 74 75 76if __name__ == '__main__': 77 main(sys.argv[1:]) 78