| #!/usr/bin/env python3 |
| # |
| # 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 mock |
| import shutil |
| import tempfile |
| import unittest |
| |
| from acts import keys |
| from acts import signals |
| from acts import test_runner |
| |
| import acts_android_device_test |
| import mock_controller |
| |
| |
| class ActsTestRunnerTest(unittest.TestCase): |
| """This test class has unit tests for the implementation of everything |
| under acts.test_runner. |
| """ |
| |
| def setUp(self): |
| self.tmp_dir = tempfile.mkdtemp() |
| self.base_mock_test_config = { |
| "testbed": { |
| "name": "SampleTestBed", |
| }, |
| "logpath": self.tmp_dir, |
| "cli_args": None, |
| "testpaths": ["./"], |
| "icecream": 42, |
| "extra_param": "haha" |
| } |
| self.mock_run_list = [('SampleTest', None)] |
| |
| def tearDown(self): |
| shutil.rmtree(self.tmp_dir) |
| |
| def test_run_twice(self): |
| """Verifies that: |
| 1. Repeated run works properly. |
| 2. The original configuration is not altered if a test controller |
| module modifies configuration. |
| """ |
| mock_test_config = dict(self.base_mock_test_config) |
| tb_key = keys.Config.key_testbed.value |
| mock_ctrlr_config_name = mock_controller.ACTS_CONTROLLER_CONFIG_NAME |
| my_config = [{ |
| "serial": "xxxx", |
| "magic": "Magic1" |
| }, { |
| "serial": "xxxx", |
| "magic": "Magic2" |
| }] |
| mock_test_config[tb_key][mock_ctrlr_config_name] = my_config |
| tr = test_runner.TestRunner(mock_test_config, |
| [('IntegrationTest', None)]) |
| tr.run() |
| self.assertTrue(mock_test_config[tb_key][mock_ctrlr_config_name][0]) |
| tr.run() |
| tr.stop() |
| results = tr.results.summary_dict() |
| self.assertEqual(results["Requested"], 2) |
| self.assertEqual(results["Executed"], 2) |
| self.assertEqual(results["Passed"], 2) |
| |
| @mock.patch( |
| 'acts.controllers.adb.AdbProxy', |
| return_value=acts_android_device_test.MockAdbProxy(1, return_value='')) |
| @mock.patch( |
| 'acts.controllers.fastboot.FastbootProxy', |
| return_value=acts_android_device_test.MockFastbootProxy(1)) |
| @mock.patch( |
| 'acts.controllers.android_device.list_adb_devices', return_value=["1"]) |
| @mock.patch( |
| 'acts.controllers.android_device.get_all_instances', |
| return_value=acts_android_device_test.get_mock_ads(1)) |
| @mock.patch( |
| 'acts.controllers.android_device.AndroidDevice.ensure_screen_on', |
| return_value=True) |
| @mock.patch( |
| 'acts.controllers.android_device.AndroidDevice.exit_setup_wizard', |
| return_value=True) |
| def test_run_two_test_classes(self, mock_exit_setup_wizard, |
| mock_ensure_screen_on, mock_get_all, |
| mock_list_adb, mock_fastboot, mock_adb): |
| """Verifies that running more than one test class in one test run works |
| properly. |
| |
| This requires using a built-in controller module. Using AndroidDevice |
| module since it has all the mocks needed already. |
| """ |
| mock_test_config = dict(self.base_mock_test_config) |
| tb_key = keys.Config.key_testbed.value |
| mock_ctrlr_config_name = mock_controller.ACTS_CONTROLLER_CONFIG_NAME |
| my_config = [{ |
| "serial": "xxxx", |
| "magic": "Magic1" |
| }, { |
| "serial": "xxxx", |
| "magic": "Magic2" |
| }] |
| mock_test_config[tb_key][mock_ctrlr_config_name] = my_config |
| mock_test_config[tb_key]["AndroidDevice"] = [{ |
| "serial": "1", |
| "skip_sl4a": True |
| }] |
| tr = test_runner.TestRunner(mock_test_config, |
| [('IntegrationTest', None), |
| ('IntegrationTest', None)]) |
| tr.run() |
| tr.stop() |
| results = tr.results.summary_dict() |
| self.assertEqual(results["Requested"], 2) |
| self.assertEqual(results["Executed"], 2) |
| self.assertEqual(results["Passed"], 2) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |