135 lines
4.4 KiB
Python
135 lines
4.4 KiB
Python
"""
|
|
safely patch object attributes, dicts and environment variables.
|
|
|
|
Usage
|
|
----------------
|
|
|
|
Use the `monkeypatch funcarg`_ to safely modify or delete environment
|
|
variables, object attributes or dictionary values. For example, if you want
|
|
to set the environment variable ``ENV1`` and patch the
|
|
``os.path.abspath`` function to return a particular value during a test
|
|
function execution you can write it down like this:
|
|
|
|
.. sourcecode:: python
|
|
|
|
def test_mytest(monkeypatch):
|
|
monkeypatch.setenv('ENV1', 'myval')
|
|
monkeypatch.setattr(os.path, 'abspath', lambda x: '/')
|
|
... # your test code
|
|
|
|
The function argument will do the modifications and memorize the
|
|
old state. After the test function finished execution all
|
|
modifications will be reverted. See the `monkeypatch blog post`_
|
|
for an extensive discussion.
|
|
|
|
To add to a possibly existing environment parameter you
|
|
can use this example:
|
|
|
|
.. sourcecode:: python
|
|
|
|
def test_mypath_finding(monkeypatch):
|
|
monkeypatch.setenv('PATH', 'x/y', prepend=":")
|
|
# x/y will be at the beginning of $PATH
|
|
|
|
calling "undo" finalization explicitely
|
|
-----------------------------------------
|
|
|
|
Usually at the end of function execution py.test will invoke
|
|
a teardown hook which undoes the changes. If you cannot wait
|
|
that long you can also call finalization explicitely::
|
|
|
|
monkeypatch.undo()
|
|
|
|
This will undo previous changes. This call consumes the
|
|
undo stack. Calling it a second time has no effect.
|
|
Within a test you can continue to use the monkeypatch
|
|
object, however.
|
|
|
|
.. _`monkeypatch blog post`: http://tetamap.wordpress.com/2009/03/03/monkeypatching-in-unit-tests-done-right/
|
|
"""
|
|
|
|
import py, os, sys
|
|
|
|
def pytest_funcarg__monkeypatch(request):
|
|
"""The returned ``monkeypatch`` funcarg provides these
|
|
helper methods to modify objects, dictionaries or os.environ::
|
|
|
|
monkeypatch.setattr(obj, name, value)
|
|
monkeypatch.delattr(obj, name, raising=True)
|
|
monkeypatch.setitem(mapping, name, value)
|
|
monkeypatch.delitem(obj, name, raising=True)
|
|
monkeypatch.setenv(name, value, prepend=False)
|
|
monkeypatch.delenv(name, value, raising=True)
|
|
monkeypatch.syspath_prepend(path)
|
|
|
|
All modifications will be undone when the requesting
|
|
test function finished its execution. For the ``del``
|
|
methods the ``raising`` parameter determines if a
|
|
KeyError or AttributeError will be raised if the
|
|
deletion has no target.
|
|
"""
|
|
monkeypatch = MonkeyPatch()
|
|
request.addfinalizer(monkeypatch.undo)
|
|
return monkeypatch
|
|
|
|
notset = object()
|
|
|
|
class MonkeyPatch:
|
|
def __init__(self):
|
|
self._setattr = []
|
|
self._setitem = []
|
|
|
|
def setattr(self, obj, name, value):
|
|
self._setattr.insert(0, (obj, name, getattr(obj, name, notset)))
|
|
setattr(obj, name, value)
|
|
|
|
def delattr(self, obj, name, raising=True):
|
|
if not hasattr(obj, name):
|
|
if raising:
|
|
raise AttributeError(name)
|
|
else:
|
|
self._setattr.insert(0, (obj, name, getattr(obj, name, notset)))
|
|
delattr(obj, name)
|
|
|
|
def setitem(self, dic, name, value):
|
|
self._setitem.insert(0, (dic, name, dic.get(name, notset)))
|
|
dic[name] = value
|
|
|
|
def delitem(self, dic, name, raising=True):
|
|
if name not in dic:
|
|
if raising:
|
|
raise KeyError(name)
|
|
else:
|
|
self._setitem.insert(0, (dic, name, dic.get(name, notset)))
|
|
del dic[name]
|
|
|
|
def setenv(self, name, value, prepend=None):
|
|
value = str(value)
|
|
if prepend and name in os.environ:
|
|
value = value + prepend + os.environ[name]
|
|
self.setitem(os.environ, name, value)
|
|
|
|
def delenv(self, name, raising=True):
|
|
self.delitem(os.environ, name, raising=raising)
|
|
|
|
def syspath_prepend(self, path):
|
|
if not hasattr(self, '_savesyspath'):
|
|
self._savesyspath = sys.path[:]
|
|
sys.path.insert(0, str(path))
|
|
|
|
def undo(self):
|
|
for obj, name, value in self._setattr:
|
|
if value is not notset:
|
|
setattr(obj, name, value)
|
|
else:
|
|
delattr(obj, name)
|
|
self._setattr[:] = []
|
|
for dictionary, name, value in self._setitem:
|
|
if value is notset:
|
|
del dictionary[name]
|
|
else:
|
|
dictionary[name] = value
|
|
self._setitem[:] = []
|
|
if hasattr(self, '_savesyspath'):
|
|
sys.path[:] = self._savesyspath
|