def test_patch_file(capsys): p = Patcher() p.loads(""" from setuptools import setup setup(package='foobar') """) p.patch_run(function='setuptools.setup', handler=pyintercept.print_) out, _err = capsys.readouterr() assert out == "()\n{'package': 'foobar'}\n"
def test_patch_file(capsys): p = Patcher() p.loads(""" from setuptools import setup setup(package='foobar') """) p.patch_run(function='setuptools.setup', handler=pyintercept.print_) out, _err = capsys.readouterr() assert out == "()\n{'package': 'foobar'}\n"
def test_patch_local_function(capsys): def double(origfn, i): print(i * 3) p = Patcher() p.loads(""" def double(i): print(i * 2) double(1) double(2) """) p.patch_run(function='double', handler=double) out, _err = capsys.readouterr() assert out == '3\n6\n'
def test_patch_all_calls(capsys): p = Patcher() p.loads(""" def x(*args): pass x(1) def foo(**kwargs): pass foo(bar=2) """) p.patch(handler=pyintercept.print_) p.run() out, _err = capsys.readouterr() assert out == "(1,)\n{}\n()\n{'bar': 2}\n"
def test_patch_local_function(capsys): def double(origfn, i): print(i * 3) p = Patcher() p.loads(""" def double(i): print(i * 2) double(1) double(2) """) p.patch_run(function='double', handler=double) out, _err = capsys.readouterr() assert out == '3\n6\n'
def test_patch_all_calls(capsys): p = Patcher() p.loads(""" def x(*args): pass x(1) def foo(**kwargs): pass foo(bar=2) """) p.patch(handler=pyintercept.print_) p.run() out, _err = capsys.readouterr() assert out == "(1,)\n{}\n()\n{'bar': 2}\n"