diff options
Diffstat (limited to 'llvm_tools/test_helpers.py')
-rw-r--r-- | llvm_tools/test_helpers.py | 38 |
1 files changed, 18 insertions, 20 deletions
diff --git a/llvm_tools/test_helpers.py b/llvm_tools/test_helpers.py index 67d88d9f..ea47a695 100644 --- a/llvm_tools/test_helpers.py +++ b/llvm_tools/test_helpers.py @@ -1,25 +1,22 @@ -# -*- coding: utf-8 -*- # Copyright 2019 The ChromiumOS Authors # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """Helper functions for unit testing.""" - -from contextlib import contextmanager +import contextlib import json import os -from tempfile import mkstemp +import tempfile -class ArgsOutputTest(object): +class ArgsOutputTest: """Testing class to simulate a argument parser object.""" def __init__(self, svn_option="google3"): self.chroot_path = "/abs/path/to/chroot" self.last_tested = "/abs/path/to/last_tested_file.json" self.llvm_version = svn_option - self.verbose = False self.extra_change_lists = None self.options = ["latest-toolchain"] self.builders = ["some-builder"] @@ -30,21 +27,22 @@ def CallCountsToMockFunctions(mock_function): """A decorator that passes a call count to the function it decorates. Examples: - @CallCountsToMockFunctions - def foo(call_count): - return call_count - ... - ... - [foo(), foo(), foo()] - [0, 1, 2] + @CallCountsToMockFunctions + def foo(call_count): + return call_count + ... + ... + [foo(), foo(), foo()] + [0, 1, 2] """ counter = [0] def Result(*args, **kwargs): - # For some values of `counter`, the mock function would simulate raising - # an exception, so let the test case catch the exception via - # `unittest.TestCase.assertRaises()` and to also handle recursive functions. + # For some values of `counter`, the mock function would simulate + # raising an exception, so let the test case catch the exception via + # `unittest.TestCase.assertRaises()` and to also handle recursive + # functions. prev_counter = counter[0] counter[0] += 1 @@ -59,8 +57,8 @@ def WritePrettyJsonFile(file_name, json_object): """Writes the contents of the file to the json object. Args: - file_name: The file that has contents to be used for the json object. - json_object: The json object to write to. + file_name: The file that has contents to be used for the json object. + json_object: The json object to write to. """ json.dump(file_name, json_object, indent=4, separators=(",", ": ")) @@ -72,11 +70,11 @@ def CreateTemporaryJsonFile(): return CreateTemporaryFile(suffix=".json") -@contextmanager +@contextlib.contextmanager def CreateTemporaryFile(suffix=""): """Makes a temporary file.""" - fd, temp_file_path = mkstemp(suffix=suffix) + fd, temp_file_path = tempfile.mkstemp(suffix=suffix) os.close(fd) |