| Index: third_party/google-endpoints/apitools/gen/test_utils.py
|
| diff --git a/third_party/google-endpoints/apitools/gen/test_utils.py b/third_party/google-endpoints/apitools/gen/test_utils.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..92934e7643b1e9672e13515945713b96996fba0b
|
| --- /dev/null
|
| +++ b/third_party/google-endpoints/apitools/gen/test_utils.py
|
| @@ -0,0 +1,55 @@
|
| +#
|
| +# Copyright 2015 Google Inc.
|
| +#
|
| +# 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.
|
| +
|
| +"""Various utilities used in tests."""
|
| +
|
| +import contextlib
|
| +import os
|
| +import tempfile
|
| +import shutil
|
| +import sys
|
| +
|
| +import six
|
| +import unittest2
|
| +
|
| +
|
| +RunOnlyOnPython27 = unittest2.skipUnless(
|
| + sys.version_info[:2] == (2, 7), 'Only runs in Python 2.7')
|
| +
|
| +
|
| +@contextlib.contextmanager
|
| +def TempDir(change_to=False):
|
| + if change_to:
|
| + original_dir = os.getcwd()
|
| + path = tempfile.mkdtemp()
|
| + try:
|
| + if change_to:
|
| + os.chdir(path)
|
| + yield path
|
| + finally:
|
| + if change_to:
|
| + os.chdir(original_dir)
|
| + shutil.rmtree(path)
|
| +
|
| +
|
| +@contextlib.contextmanager
|
| +def CaptureOutput():
|
| + new_stdout, new_stderr = six.StringIO(), six.StringIO()
|
| + old_stdout, old_stderr = sys.stdout, sys.stderr
|
| + try:
|
| + sys.stdout, sys.stderr = new_stdout, new_stderr
|
| + yield new_stdout, new_stderr
|
| + finally:
|
| + sys.stdout, sys.stderr = old_stdout, old_stderr
|
|
|