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 |