acts/framework/tests/acts_unittest_suite.py

#!/usr/bin/env python3.4
#
#   Copyright 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 sys
import unittest

import acts_adb_test
import acts_android_device_test
import acts_asserts_test
import acts_base_class_test
import acts_records_test
import acts_test_runner_test

def compile_suite():
    test_classes_to_run = [
        acts_adb_test.ActsAdbTest,
        acts_asserts_test.ActsAssertsTest,
        acts_base_class_test.ActsBaseClassTest,
        acts_test_runner_test.ActsTestRunnerTest,
        acts_android_device_test.ActsAndroidDeviceTest,
        acts_records_test.ActsRecordsTest
    ]

    loader = unittest.TestLoader()

    suites_list = []
    for test_class in test_classes_to_run:
        suite = loader.loadTestsFromTestCase(test_class)
        suites_list.append(suite)

    big_suite = unittest.TestSuite(suites_list)
    return big_suite

if __name__ == "__main__":
    # This is the entry point for running all ACTS unit tests.
    runner = unittest.TextTestRunner()
    results = runner.run(compile_suite())
    sys.exit(not results.wasSuccessful())