1#!/usr/bin/env python3 2# 3# Copyright 2020 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 plugin_lib's unittest.""" 17 18import logging 19import os 20import sys 21import unittest 22from importlib import import_module 23 24# Setup logging to be silent so unittests can pass through TF. 25logging.disable(logging.ERROR) 26 27def get_test_modules(): 28 """Returns a list of testable modules. 29 30 Finds all the test files (*_unittest.py) and gets their relative 31 paths (internal/lib/utils_test.py) and translate it to an import path and 32 strip the py ext (internal.lib.utils_test). 33 34 Returns: 35 A list of strings (the testable module import path). 36 """ 37 testable_modules = [] 38 package = os.path.dirname(os.path.realpath(__file__)) 39 base_path = os.path.dirname(package) 40 41 for dirpath, _, files in os.walk(package): 42 for _file in files: 43 if _file.endswith("_unittest.py"): 44 # Now transform it into a relative import path. 45 full_file_path = os.path.join(dirpath, _file) 46 rel_file_path = os.path.relpath(full_file_path, base_path) 47 rel_file_path, _ = os.path.splitext(rel_file_path) 48 rel_file_path = rel_file_path.replace(os.sep, ".") 49 testable_modules.append(rel_file_path) 50 51 return testable_modules 52 53 54def main(): 55 """Main unittest entry.""" 56 test_modules = get_test_modules() 57 for mod in test_modules: 58 import_module(mod) 59 60 loader = unittest.defaultTestLoader 61 test_suite = loader.loadTestsFromNames(test_modules) 62 runner = unittest.TextTestRunner(verbosity=2) 63 result = runner.run(test_suite) 64 sys.exit(not result.wasSuccessful()) 65 66 67if __name__ == '__main__': 68 main() 69