| #!/usr/bin/env python3 |
| # |
| # Copyright 2018, 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. |
| |
| """Unittests for result_reporter.""" |
| |
| |
| from io import StringIO |
| import sys |
| import unittest |
| from unittest import mock |
| from unittest.mock import patch |
| |
| from atest import arg_parser |
| from atest import atest_configs |
| from atest import result_reporter |
| from atest.test_finders import test_info |
| from atest.test_runners import test_runner_base |
| |
| |
| RESULT_PASSED_TEST = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='someClassName#sostName', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_PASSED_TEST_MODULE_2 = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule2', |
| test_name='someClassName#sostName', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_PASSED_TEST_RUNNER_2_NO_MODULE = test_runner_base.TestResult( |
| runner_name='someTestRunner2', |
| group_name=None, |
| test_name='someClassName#sostName', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_FAILED_TEST = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='someClassName2#sestName2', |
| status=test_runner_base.FAILED_STATUS, |
| details='someTrace', |
| test_count=1, |
| test_time='', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_RUN_FAILURE = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='someClassName#sostName', |
| status=test_runner_base.ERROR_STATUS, |
| details='someRunFailureReason', |
| test_count=1, |
| test_time='', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_INVOCATION_FAILURE = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name=None, |
| test_name=None, |
| status=test_runner_base.ERROR_STATUS, |
| details='someInvocationFailureReason', |
| test_count=1, |
| test_time='', |
| runner_total=None, |
| group_total=None, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_IGNORED_TEST = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='someClassName#sostName', |
| status=test_runner_base.IGNORED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_ASSUMPTION_FAILED_TEST = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='someClassName#sostName', |
| status=test_runner_base.ASSUMPTION_FAILED, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| ADDITIONAL_INFO_PERF01_TEST01 = { |
| 'repetition_index': '0', |
| 'cpu_time': '10001.10001', |
| 'name': 'perfName01', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1001', |
| 'run_name': 'perfName01', |
| 'real_time': '11001.11001', |
| } |
| |
| RESULT_PERF01_TEST01 = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='somePerfClass01#perfName01', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info=ADDITIONAL_INFO_PERF01_TEST01, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_PERF01_TEST02 = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='somePerfClass01#perfName02', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={ |
| 'repetition_index': '0', |
| 'cpu_time': '10002.10002', |
| 'name': 'perfName02', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1002', |
| 'run_name': 'perfName02', |
| 'real_time': '11002.11002', |
| }, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_PERF01_TEST03_NO_CPU_TIME = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='somePerfClass01#perfName03', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={ |
| 'repetition_index': '0', |
| 'name': 'perfName03', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1003', |
| 'run_name': 'perfName03', |
| 'real_time': '11003.11003', |
| }, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| RESULT_PERF02_TEST01 = test_runner_base.TestResult( |
| runner_name='someTestRunner', |
| group_name='someTestModule', |
| test_name='somePerfClass02#perfName11', |
| status=test_runner_base.PASSED_STATUS, |
| details=None, |
| test_count=1, |
| test_time='(10ms)', |
| runner_total=None, |
| group_total=2, |
| additional_info={ |
| 'repetition_index': '0', |
| 'cpu_time': '20001.20001', |
| 'name': 'perfName11', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '2001', |
| 'run_name': 'perfName11', |
| 'real_time': '21001.21001', |
| }, |
| test_run_name='com.android.UnitTests', |
| ) |
| |
| |
| # pylint: disable=protected-access |
| # pylint: disable=invalid-name |
| class ResultReporterUnittests(unittest.TestCase): |
| """Unit tests for result_reporter.py""" |
| |
| def setUp(self): |
| self.rr = result_reporter.ResultReporter() |
| |
| def tearDown(self): |
| mock.patch.stopall() |
| |
| @mock.patch.object(result_reporter.ResultReporter, '_print_group_title') |
| @mock.patch.object(result_reporter.ResultReporter, '_update_stats') |
| @mock.patch.object(result_reporter.ResultReporter, '_print_result') |
| def test_process_test_result(self, mock_print, mock_update, mock_title): |
| """Test process_test_result method.""" |
| # Passed Test |
| self.assertTrue('someTestRunner' not in self.rr.runners) |
| self.rr.process_test_result(RESULT_PASSED_TEST) |
| self.assertTrue('someTestRunner' in self.rr.runners) |
| group = self.rr.runners['someTestRunner'].get('someTestModule') |
| self.assertIsNotNone(group) |
| mock_title.assert_called_with(RESULT_PASSED_TEST) |
| mock_update.assert_called_with(RESULT_PASSED_TEST, group) |
| mock_print.assert_called_with(RESULT_PASSED_TEST) |
| # Failed Test |
| mock_title.reset_mock() |
| self.rr.process_test_result(RESULT_FAILED_TEST) |
| mock_title.assert_not_called() |
| mock_update.assert_called_with(RESULT_FAILED_TEST, group) |
| mock_print.assert_called_with(RESULT_FAILED_TEST) |
| # Test with new Group |
| mock_title.reset_mock() |
| self.rr.process_test_result(RESULT_PASSED_TEST_MODULE_2) |
| self.assertTrue('someTestModule2' in self.rr.runners['someTestRunner']) |
| mock_title.assert_called_with(RESULT_PASSED_TEST_MODULE_2) |
| # Test with new Runner |
| mock_title.reset_mock() |
| self.rr.process_test_result(RESULT_PASSED_TEST_RUNNER_2_NO_MODULE) |
| self.assertTrue('someTestRunner2' in self.rr.runners) |
| mock_title.assert_called_with(RESULT_PASSED_TEST_RUNNER_2_NO_MODULE) |
| |
| def test_print_result_run_name(self): |
| """Test print run name function in print_result method.""" |
| try: |
| rr = result_reporter.ResultReporter() |
| capture_output = StringIO() |
| sys.stdout = capture_output |
| run_name = 'com.android.UnitTests' |
| rr._print_result( |
| test_runner_base.TestResult( |
| runner_name='runner_name', |
| group_name='someTestModule', |
| test_name='someClassName#someTestName', |
| status=test_runner_base.FAILED_STATUS, |
| details='someTrace', |
| test_count=2, |
| test_time='(2h44m36.402s)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name=run_name, |
| ) |
| ) |
| # Make sure run name in the first line. |
| capture_output_str = capture_output.getvalue().strip() |
| self.assertTrue(run_name in capture_output_str.split('\n')[0]) |
| run_name2 = 'com.android.UnitTests2' |
| capture_output = StringIO() |
| sys.stdout = capture_output |
| rr._print_result( |
| test_runner_base.TestResult( |
| runner_name='runner_name', |
| group_name='someTestModule', |
| test_name='someClassName#someTestName', |
| status=test_runner_base.FAILED_STATUS, |
| details='someTrace', |
| test_count=2, |
| test_time='(2h43m36.402s)', |
| runner_total=None, |
| group_total=2, |
| additional_info={}, |
| test_run_name=run_name2, |
| ) |
| ) |
| # Make sure run name in the first line. |
| capture_output_str = capture_output.getvalue().strip() |
| self.assertTrue(run_name2 in capture_output_str.split('\n')[0]) |
| finally: |
| sys.stdout = sys.__stdout__ |
| |
| def test_register_unsupported_runner(self): |
| """Test register_unsupported_runner method.""" |
| self.rr.register_unsupported_runner('NotSupported') |
| runner = self.rr.runners['NotSupported'] |
| self.assertIsNotNone(runner) |
| self.assertEqual(runner, result_reporter.UNSUPPORTED_FLAG) |
| |
| def test_update_stats_passed(self): |
| """Test _update_stats method.""" |
| # Passed Test |
| group = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST, group) |
| self.assertEqual(self.rr.run_stats.passed, 1) |
| self.assertEqual(self.rr.run_stats.failed, 0) |
| self.assertEqual(self.rr.run_stats.run_errors, False) |
| self.assertEqual(self.rr.failed_tests, []) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 0) |
| self.assertEqual(group.ignored, 0) |
| self.assertEqual(group.run_errors, False) |
| # Passed Test New Group |
| group2 = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST_MODULE_2, group2) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 0) |
| self.assertEqual(self.rr.run_stats.run_errors, False) |
| self.assertEqual(self.rr.failed_tests, []) |
| self.assertEqual(group2.passed, 1) |
| self.assertEqual(group2.failed, 0) |
| self.assertEqual(group.ignored, 0) |
| self.assertEqual(group2.run_errors, False) |
| |
| def test_update_stats_failed(self): |
| """Test _update_stats method.""" |
| # Passed Test |
| group = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST, group) |
| # Passed Test New Group |
| group2 = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST_MODULE_2, group2) |
| # Failed Test Old Group |
| self.rr._update_stats(RESULT_FAILED_TEST, group) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 1) |
| self.assertEqual(self.rr.run_stats.run_errors, False) |
| self.assertEqual(self.rr.failed_tests, [RESULT_FAILED_TEST.test_name]) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 1) |
| self.assertEqual(group.ignored, 0) |
| self.assertEqual(group.total, 2) |
| self.assertEqual(group2.total, 1) |
| self.assertEqual(group.run_errors, False) |
| # Test Run Failure |
| self.rr._update_stats(RESULT_RUN_FAILURE, group) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 1) |
| self.assertEqual(self.rr.run_stats.run_errors, True) |
| self.assertEqual(self.rr.failed_tests, [RESULT_FAILED_TEST.test_name]) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 1) |
| self.assertEqual(group.ignored, 0) |
| self.assertEqual(group.run_errors, True) |
| self.assertEqual(group2.run_errors, False) |
| # Invocation Failure |
| self.rr._update_stats(RESULT_INVOCATION_FAILURE, group) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 1) |
| self.assertEqual(self.rr.run_stats.run_errors, True) |
| self.assertEqual(self.rr.failed_tests, [RESULT_FAILED_TEST.test_name]) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 1) |
| self.assertEqual(group.ignored, 0) |
| self.assertEqual(group.run_errors, True) |
| |
| def test_update_stats_ignored_and_assumption_failure(self): |
| """Test _update_stats method.""" |
| # Passed Test |
| group = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST, group) |
| # Passed Test New Group |
| group2 = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PASSED_TEST_MODULE_2, group2) |
| # Failed Test Old Group |
| self.rr._update_stats(RESULT_FAILED_TEST, group) |
| # Test Run Failure |
| self.rr._update_stats(RESULT_RUN_FAILURE, group) |
| # Invocation Failure |
| self.rr._update_stats(RESULT_INVOCATION_FAILURE, group) |
| # Ignored Test |
| self.rr._update_stats(RESULT_IGNORED_TEST, group) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 1) |
| self.assertEqual(self.rr.run_stats.run_errors, True) |
| self.assertEqual(self.rr.failed_tests, [RESULT_FAILED_TEST.test_name]) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 1) |
| self.assertEqual(group.ignored, 1) |
| self.assertEqual(group.run_errors, True) |
| # 2nd Ignored Test |
| self.rr._update_stats(RESULT_IGNORED_TEST, group) |
| self.assertEqual(self.rr.run_stats.passed, 2) |
| self.assertEqual(self.rr.run_stats.failed, 1) |
| self.assertEqual(self.rr.run_stats.run_errors, True) |
| self.assertEqual(self.rr.failed_tests, [RESULT_FAILED_TEST.test_name]) |
| self.assertEqual(group.passed, 1) |
| self.assertEqual(group.failed, 1) |
| self.assertEqual(group.ignored, 2) |
| self.assertEqual(group.run_errors, True) |
| |
| # Assumption_Failure test |
| self.rr._update_stats(RESULT_ASSUMPTION_FAILED_TEST, group) |
| self.assertEqual(group.assumption_failed, 1) |
| # 2nd Assumption_Failure test |
| self.rr._update_stats(RESULT_ASSUMPTION_FAILED_TEST, group) |
| self.assertEqual(group.assumption_failed, 2) |
| |
| @patch.object( |
| atest_configs, |
| 'GLOBAL_ARGS', |
| arg_parser.create_atest_arg_parser().parse_args([]), |
| ) |
| def test_print_summary_ret_val(self): |
| """Test print_summary method's return value.""" |
| # PASS Case |
| self.rr.process_test_result(RESULT_PASSED_TEST) |
| self.assertEqual(0, self.rr.print_summary()) |
| # PASS Case + Fail Case |
| self.rr.process_test_result(RESULT_FAILED_TEST) |
| self.assertNotEqual(0, self.rr.print_summary()) |
| # PASS Case + Fail Case + PASS Case |
| self.rr.process_test_result(RESULT_PASSED_TEST_MODULE_2) |
| self.assertNotEqual(0, self.rr.print_summary()) |
| |
| @patch.object( |
| atest_configs, |
| 'GLOBAL_ARGS', |
| arg_parser.create_atest_arg_parser().parse_args([]), |
| ) |
| def test_print_summary_ret_val_err_stat(self): |
| """Test print_summary method's return value.""" |
| # PASS Case |
| self.rr.process_test_result(RESULT_PASSED_TEST) |
| self.assertEqual(0, self.rr.print_summary()) |
| # PASS Case + Fail Case |
| self.rr.process_test_result(RESULT_RUN_FAILURE) |
| self.assertNotEqual(0, self.rr.print_summary()) |
| # PASS Case + Fail Case + PASS Case |
| self.rr.process_test_result(RESULT_PASSED_TEST_MODULE_2) |
| self.assertNotEqual(0, self.rr.print_summary()) |
| |
| def test_collect_tests_only_no_throw(self): |
| rr = result_reporter.ResultReporter(collect_only=True) |
| rr.process_test_result(RESULT_PASSED_TEST) |
| |
| self.assertEqual(0, self.rr.print_collect_tests()) |
| |
| def test_update_perf_info(self): |
| """Test update_perf_info method.""" |
| group = result_reporter.RunStat() |
| # 1. Test PerfInfo after RESULT_PERF01_TEST01 |
| # _update_stats() will call _update_perf_info() |
| self.rr._update_stats(RESULT_PERF01_TEST01, group) |
| correct_perf_info = [] |
| trim_perf01_test01 = { |
| 'repetition_index': '0', |
| 'cpu_time': '10001.10001', |
| 'name': 'perfName01', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1001', |
| 'run_name': 'perfName01', |
| 'real_time': '11001.11001', |
| 'test_name': 'somePerfClass01#perfName01', |
| } |
| correct_perf_info.append(trim_perf01_test01) |
| self.assertEqual(self.rr.run_stats.perf_info.perf_info, correct_perf_info) |
| # 2. Test PerfInfo after RESULT_PERF01_TEST01 |
| self.rr._update_stats(RESULT_PERF01_TEST02, group) |
| trim_perf01_test02 = { |
| 'repetition_index': '0', |
| 'cpu_time': '10002.10002', |
| 'name': 'perfName02', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1002', |
| 'run_name': 'perfName02', |
| 'real_time': '11002.11002', |
| 'test_name': 'somePerfClass01#perfName02', |
| } |
| correct_perf_info.append(trim_perf01_test02) |
| self.assertEqual(self.rr.run_stats.perf_info.perf_info, correct_perf_info) |
| # 3. Test PerfInfo after RESULT_PERF02_TEST01 |
| self.rr._update_stats(RESULT_PERF02_TEST01, group) |
| trim_perf02_test01 = { |
| 'repetition_index': '0', |
| 'cpu_time': '20001.20001', |
| 'name': 'perfName11', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '2001', |
| 'run_name': 'perfName11', |
| 'real_time': '21001.21001', |
| 'test_name': 'somePerfClass02#perfName11', |
| } |
| correct_perf_info.append(trim_perf02_test01) |
| self.assertEqual(self.rr.run_stats.perf_info.perf_info, correct_perf_info) |
| # 4. Test PerfInfo after RESULT_PERF01_TEST03_NO_CPU_TIME |
| self.rr._update_stats(RESULT_PERF01_TEST03_NO_CPU_TIME, group) |
| # Nothing added since RESULT_PERF01_TEST03_NO_CPU_TIME lack of cpu_time |
| self.assertEqual(self.rr.run_stats.perf_info.perf_info, correct_perf_info) |
| |
| def test_classify_perf_info(self): |
| """Test _classify_perf_info method.""" |
| group = result_reporter.RunStat() |
| self.rr._update_stats(RESULT_PERF01_TEST01, group) |
| self.rr._update_stats(RESULT_PERF01_TEST02, group) |
| self.rr._update_stats(RESULT_PERF02_TEST01, group) |
| # trim the time form 10001.10001 to 10001 |
| trim_perf01_test01 = { |
| 'repetition_index': '0', |
| 'cpu_time': '10001.10001', |
| 'name': 'perfName01', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1001', |
| 'run_name': 'perfName01', |
| 'real_time': '11001.11001', |
| 'test_name': 'somePerfClass01#perfName01', |
| } |
| trim_perf01_test02 = { |
| 'repetition_index': '0', |
| 'cpu_time': '10002.10002', |
| 'name': 'perfName02', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '1002', |
| 'run_name': 'perfName02', |
| 'real_time': '11002.11002', |
| 'test_name': 'somePerfClass01#perfName02', |
| } |
| trim_perf02_test01 = { |
| 'repetition_index': '0', |
| 'cpu_time': '20001.20001', |
| 'name': 'perfName11', |
| 'repetitions': '0', |
| 'run_type': 'iteration', |
| 'label': '2123', |
| 'threads': '1', |
| 'time_unit': 'ns', |
| 'iterations': '2001', |
| 'run_name': 'perfName11', |
| 'real_time': '21001.21001', |
| 'test_name': 'somePerfClass02#perfName11', |
| } |
| correct_classify_perf_info = { |
| 'somePerfClass01': [trim_perf01_test01, trim_perf01_test02], |
| 'somePerfClass02': [trim_perf02_test01], |
| } |
| classify_perf_info, max_len = ( |
| self.rr.run_stats.perf_info._classify_perf_info() |
| ) |
| correct_max_len = { |
| 'real_time': 11, |
| 'cpu_time': 11, |
| 'name': 10, |
| 'iterations': 9, |
| 'time_unit': 2, |
| } |
| self.assertEqual(max_len, correct_max_len) |
| self.assertEqual(classify_perf_info, correct_classify_perf_info) |
| |
| def test_print_perf_test_metrics_perf_tests_print_attempted(self): |
| test_infos = [ |
| test_info.TestInfo( |
| 'some_module', |
| 'TestRunner', |
| set(), |
| compatibility_suites=['performance-tests'], |
| ) |
| ] |
| sut = result_reporter.ResultReporter(test_infos=test_infos) |
| |
| is_print_attempted = sut._print_perf_test_metrics() |
| |
| self.assertTrue(is_print_attempted) |
| |
| def test_print_perf_test_metrics_not_perf_tests_print__not_attempted(self): |
| test_infos = [ |
| test_info.TestInfo( |
| 'some_module', |
| 'TestRunner', |
| set(), |
| compatibility_suites=['not-perf-test'], |
| ) |
| ] |
| sut = result_reporter.ResultReporter(test_infos=test_infos) |
| |
| is_print_attempted = sut._print_perf_test_metrics() |
| |
| self.assertFalse(is_print_attempted) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
