diff options
Diffstat (limited to 'tests/mobly/test_runner_test.py')
-rwxr-xr-x | tests/mobly/test_runner_test.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/tests/mobly/test_runner_test.py b/tests/mobly/test_runner_test.py index efddc4c..0343e0d 100755 --- a/tests/mobly/test_runner_test.py +++ b/tests/mobly/test_runner_test.py @@ -374,20 +374,17 @@ class TestRunnerTest(unittest.TestCase): mock_cls_instance = mock.MagicMock() mock_test_class.return_value = mock_cls_instance test_runner._print_test_names(mock_test_class) - mock_cls_instance.setup_generated_tests.assert_called_once() - mock_cls_instance.get_existing_test_names.assert_called_once() - mock_cls_instance._controller_manager.unregister_controllers.assert_called_once( - ) + mock_cls_instance._pre_run.assert_called_once() + mock_cls_instance._clean_up.assert_called_once() def test_print_test_names_with_exception(self): mock_test_class = mock.MagicMock() mock_cls_instance = mock.MagicMock() mock_test_class.return_value = mock_cls_instance test_runner._print_test_names(mock_test_class) - mock_cls_instance.setup_generated_tests.side_effect = Exception( + mock_cls_instance._pre_run.side_effect = Exception( 'Something went wrong.') - mock_cls_instance._controller_manager.unregister_controllers.assert_called_once( - ) + mock_cls_instance._clean_up.assert_called_once() if __name__ == "__main__": |