To install, just run:
pip install redef
test_redef.py under your install path for examples.
Redef is intended to create lexically scoped variables, that when destroyed, undo mock behavior. It was inspired by a Perl module, Test::Resub
The best examples use
unittest, but it should work with any testing framework.
import unittest from redef import redef class MyClass: def lame_function(self): return "something I don't want" class MyTest(unittest.TestCase): def test1(self): myobj = MyClass() self.assertEqual(myobj.lame_function(), "something I don't want") want = 'something I want' with redef(MyClass, 'lame_function', lambda s: want) as rd_lf: self.assertEqual(myobj.lame_function(), want) # after test1, rd_lf gets deleted and resets def test2(self): myobj = MyClass() # test2 is uneffected by test1 self.assertEqual(myobj.lame_function(), "something I don't want")
This doesn't have to be a function, you can also redefine attributes directly on an object.
class MyClass: unpredictable = 'random string' my_global_object = MyClass() class MyTest(unittest.TestCase): def test3(self): rd_u = redef(my_global_object, 'unpredictable', 'unit testable string') # ... test something awesome! self.assertEqual(my_global_object.unpredictable, 'unit testable string') def test4(self): # hey, my_global_object is back to being unpredictable self.assertEqual(my_global_object.unpredictable, 'unpredictable')
If you don't want to rely on automatic scope you can explicitally call close on your redefed function. Calling del on the redefed function is the sam as calling close
There are other useful functions provided on the redef object itself:
- Class Redef:
- takes an object and a key. The rest of the arguments are kwargs:
- if the key doesn't exist in the object, an exception will be raised unless you provide kwargs must_exist
- value: if provided, this value will redefine the key in the object, otherwise you 'wiretap' the object
- must_call: if provided, when the Redef object is destroyed, it warns if this constraint is violated.
called(): Stores how many times a redef'd function was called.
method_args(): Stores the most recent
*argsto the redef'd function.
named_method_args(): Stores the most recent
**kwargsto the redef'd function.
named_method_argsback to the default state of
0, None, None
was_called(): Asks the redef object if there was ever a call. Regardless of calling the reset() function
never_called(): Asks the redef object if there was never a call. Regardless of calling the reset() function
Redef also provides a freebie static function:
redef(obj, key, value): Static constructor of a
obj: is a Class, Module, or Object you want to temporariliy change
key: The string name of the attribute you want to change
value: The new value. If the value is None, you only capture called, method_args, and named_method_args
- The None case won't redefine the key on the obj
These static functions were provided to show the usefulness of redef:
For example, you could capture stdout of a function call, and after capturing it,
sys.stdout goes back to normal:
- Has 2 variables you want:
- Has 2 variables you want:
Static Functions that return a CapturedOutput:
stdout_of(func, *args, **kwargs): Call a function and capture the stdout. Returns a
CapturedOutputobject that has the stdout and the return value of calling func.
stderr_of(func, *args, **kwargs): Call a function and capture the stderr. Returns a
CapturedOutputobject that has the stderr and the return value of calling func.
- A static function that creates a Redef object only for the purpose of capturing the method_args and called values. * The original functionality should remain the same.
Please ask any questions on github: http://github.com/joeheyming/redef/issues