with testdata.patch() context manager

[Jaymon]

it would be cool to allow a context manager through patch so you can easily patch something and then restore the patched code to the original at the end

Likewise, it would be cool if there was an easy way for patch to handle the builtins, like if you specify something for the module and it isn't there then you check builtins

[Jaymon]

I just had to do something like this:

input_name = "raw_input" if is_py2 else "input"
orig_input = getattr(builtins, input_name)

class MockInput(object):
    def __init__(self, v):
        self.v = v
    def __call__(self, *args, **kwargs):
        return self.v


 setattr(builtins, input_name, MockInput("Yes"))

# restore patched
setattr(builtins, input_name, orig_input)

because testdata.patch isn't flexible enough

[Jaymon]

python mocking raw input in unittests

search:
python test prompt