diff options
Diffstat (limited to 'crosperf/crosperf_unittest.py')
-rwxr-xr-x | crosperf/crosperf_unittest.py | 102 |
1 files changed, 51 insertions, 51 deletions
diff --git a/crosperf/crosperf_unittest.py b/crosperf/crosperf_unittest.py index 774159ff..7b52f2e0 100755 --- a/crosperf/crosperf_unittest.py +++ b/crosperf/crosperf_unittest.py @@ -1,14 +1,12 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- # -# Copyright (c) 2014 The Chromium OS Authors. All rights reserved. +# Copyright 2014 The ChromiumOS Authors # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Unittest for crosperf.""" -from __future__ import division -from __future__ import print_function import argparse import io @@ -17,8 +15,9 @@ import unittest import unittest.mock as mock import crosperf -import settings_factory import experiment_file +import settings_factory + EXPERIMENT_FILE_1 = """ board: x86-alex @@ -41,50 +40,51 @@ EXPERIMENT_FILE_1 = """ class CrosperfTest(unittest.TestCase): - """Crosperf test class.""" - - def setUp(self): - input_file = io.StringIO(EXPERIMENT_FILE_1) - self.exp_file = experiment_file.ExperimentFile(input_file) - - def testDryRun(self): - with tempfile.NamedTemporaryFile('w', encoding='utf-8') as f: - f.write(EXPERIMENT_FILE_1) - f.flush() - crosperf.Main(['', f.name, '--dry_run']) - - def testConvertOptionsToSettings(self): - parser = argparse.ArgumentParser() - parser.add_argument('-l', - '--log_dir', - dest='log_dir', - default='', - help='The log_dir, default is under ' - '<crosperf_logs>/logs') - crosperf.SetupParserOptions(parser) - argv = ['crosperf/crosperf.py', 'temp.exp', '--rerun=True'] - options, _ = parser.parse_known_args(argv) - settings = crosperf.ConvertOptionsToSettings(options) - self.assertIsNotNone(settings) - self.assertIsInstance(settings, settings_factory.GlobalSettings) - self.assertEqual(len(settings.fields), 40) - self.assertTrue(settings.GetField('rerun')) - argv = ['crosperf/crosperf.py', 'temp.exp'] - options, _ = parser.parse_known_args(argv) - settings = crosperf.ConvertOptionsToSettings(options) - self.assertFalse(settings.GetField('rerun')) - - def testExceptionPrintTraceback(self): - """Test the main function can print traceback in exception.""" - - def mock_RunCrosperf(*_args, **_kwargs): - return 10 / 0 - - with mock.patch('crosperf.RunCrosperf', new=mock_RunCrosperf): - with self.assertRaises(ZeroDivisionError) as context: - crosperf.Main([]) - self.assertEqual('division by zero', str(context.exception)) - - -if __name__ == '__main__': - unittest.main() + """Crosperf test class.""" + + def setUp(self): + input_file = io.StringIO(EXPERIMENT_FILE_1) + self.exp_file = experiment_file.ExperimentFile(input_file) + + def testDryRun(self): + with tempfile.NamedTemporaryFile("w", encoding="utf-8") as f: + f.write(EXPERIMENT_FILE_1) + f.flush() + crosperf.Main(["", f.name, "--dry_run"]) + + def testConvertOptionsToSettings(self): + parser = argparse.ArgumentParser() + parser.add_argument( + "-l", + "--log_dir", + dest="log_dir", + default="", + help="The log_dir, default is under " "<crosperf_logs>/logs", + ) + crosperf.SetupParserOptions(parser) + argv = ["crosperf/crosperf.py", "temp.exp", "--rerun=True"] + options, _ = parser.parse_known_args(argv) + settings = crosperf.ConvertOptionsToSettings(options) + self.assertIsNotNone(settings) + self.assertIsInstance(settings, settings_factory.GlobalSettings) + self.assertEqual(len(settings.fields), 40) + self.assertTrue(settings.GetField("rerun")) + argv = ["crosperf/crosperf.py", "temp.exp"] + options, _ = parser.parse_known_args(argv) + settings = crosperf.ConvertOptionsToSettings(options) + self.assertFalse(settings.GetField("rerun")) + + def testExceptionPrintTraceback(self): + """Test the main function can print traceback in exception.""" + + def mock_RunCrosperf(*_args, **_kwargs): + return 10 / 0 + + with mock.patch("crosperf.RunCrosperf", new=mock_RunCrosperf): + with self.assertRaises(ZeroDivisionError) as context: + crosperf.Main([]) + self.assertEqual("division by zero", str(context.exception)) + + +if __name__ == "__main__": + unittest.main() |