x/cloud-init
Code Issues Proposed changes

Add helpers for patching open and stdout/stderr.

Browse Source
This commit is contained in:
Daniel Watkins
2015-02-11 11:54:48 +00:00
parent e4e587c3c0
commit c3c7b26690
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/unittests/helpers.py
View File

@@ -254,6 +254,19 @@ class FilesystemMockingTestCase(ResourceUsingTestCase):
self.patched_funcs.enter_context(
mock.patch.object(mod, f, trap_func))
def patchOpen(self, new_root):
trap_func = retarget_many_wrapper(new_root, 1, open)
name = 'builtins.open' if PY3 else '__builtin__.open'
self.patched_funcs.enter_context(mock.patch(name, trap_func))
def patchStdoutAndStderr(self, stdout=None, stderr=None):
if stdout is not None:
self.patched_funcs.enter_context(
mock.patch.object(sys, 'stdout', stdout))
if stderr is not None:
self.patched_funcs.enter_context(
mock.patch.object(sys, 'stderr', stderr))
class HttprettyTestCase(TestCase):
# necessary as http_proxy gets in the way of httpretty