B ܇\+@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl Z ddl Z ddl mZddlmZed Zed d Zd d ZddZddZGdddeZeZGdddeZdS)z, monkeypatching and mocking functionality. )absolute_import)division)print_functionN)contextmanager)fixture)Pathz^No module named (.*)$ccst}|V|dS)aThe returned ``monkeypatch`` fixture provides these helper methods to modify objects, dictionaries or os.environ:: monkeypatch.setattr(obj, name, value, raising=True) 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, raising=True) monkeypatch.syspath_prepend(path) monkeypatch.chdir(path) All modifications will be undone after the requesting test function or fixture has finished. The ``raising`` parameter determines if a KeyError or AttributeError will be raised if the set/deletion operation has no target. N) MonkeyPatchundo)Zmpatchr 2lib/python3.7/site-packages/_pytest/monkeypatch.py monkeypatchsr c Cs|d}|d}t|}x|D]}|d|7}yt||}Wq"tk rTYnXq"y t|WnNtk r}z0t|d}||krntd||fWdd}~XYnXt|||}q"W|S)N.rzimport error in %s: %s)splitpop __import__getattrAttributeError ImportErrorstrannotated_getattr)namepartsZusedfoundpartexZexpectedr r r resolve-s&     "rcCs@yt||}Wn,tk r:tdt|j||fYnX|S)Nz#%r object at %s has no attribute %r)rrtype__name__)objrannr r r rJs rcCsTt|tjrd|kr"td|f|dd\}}t|}|rLt|||d||fS)Nr z+must be absolute import path string, not %r)r ) isinstancesix string_types TypeErrorrsplitrr)Z import_pathraisingmoduleattrtargetr r r derive_importpathTsr+c@seZdZddZdS)NotsetcCsdS)Nzr )selfr r r __repr___szNotset.__repr__N)r __module__ __qualname__r.r r r r r,^sr,c@seZdZdZddZeddZedfddZedfd d Z d d Z dd dZ ddZ dddZ dddZddZddZddZdS)rzj Object returned by the ``monkeypatch`` fixture keeping a record of setattr/item/env/syspath changes. cCsg|_g|_d|_d|_dS)N)_setattr_setitem_cwd _savesyspath)r-r r r __init__jszMonkeyPatch.__init__ccs t}z |VWd|XdS)a Context manager that returns a new :class:`MonkeyPatch` object which undoes any patching done inside the ``with`` block upon exit: .. code-block:: python import functools def test_partial(monkeypatch): with monkeypatch.context() as m: m.setattr(functools, "partial", 3) Useful in situations where it is desired to undo some patches before the test ends, such as mocking ``stdlib`` functions that might break pytest itself if mocked (for examples of this see `#3290 `_. N)rr )r-mr r r contextps zMonkeyPatch.contextTcCsd}ddl}|tkr:t|tjs(td|}t||\}}t||t}|rb|tkrbtd||f| |rz|j |t}|j |||ft|||dS)ai Set attribute value on target, memorizing the old value. By default raise AttributeError if the attribute did not exist. For convenience you can specify a string as ``target`` which will be interpreted as a dotted import path, with the last part being the attribute name. Example: ``monkeypatch.setattr("os.getcwd", lambda: "/")`` would set the ``getcwd`` function of the ``os`` module. The ``raising`` value determines if the setattr should fail if the attribute is not already present (defaults to True which means it will raise). TrNzcuse setattr(target, name, value) or setattr(target, value) with target being a dotted import stringz%r has no attribute %r)inspectnotsetr"r#r$r%r+rrisclass__dict__getr1appendsetattr)r-r*rvaluer'__tracebackhide__r8oldvalr r r r>s    zMonkeyPatch.setattrcCsd}ddl}|tkr6t|tjs(tdt||\}}t||sN|rt|n@t ||t}| |rr|j |t}|j |||ft||dS)a Delete attribute ``name`` from ``target``, by default raise AttributeError it the attribute did not previously exist. If no ``name`` is specified and ``target`` is a string it will be interpreted as a dotted import path with the last part being the attribute name. If ``raising`` is set to False, no exception will be raised if the attribute is missing. TrNzUuse delattr(target, name) or delattr(target) with target being a dotted import string)r8r9r"r#r$r%r+hasattrrrr:r;r<r1r=delattr)r-r*rr'r@r8rAr r r rCs      zMonkeyPatch.delattrcCs&|j||||tf|||<dS)z) Set dictionary entry ``name`` to value. N)r2r=r<r9)r-dicrr?r r r setitemszMonkeyPatch.setitemcCs:||kr|r6t|n |j||||tf||=dS)z Delete ``name`` from dict. Raise KeyError if it doesn't exist. If ``raising`` is set to False, no exception will be raised if the key is missing. N)KeyErrorr2r=r<r9)r-rDrr'r r r delitems  zMonkeyPatch.delitemcCs*tjr&t|ts&ttd|dS)zTOn Python 2, warn if the given environment variable name is not a native str (#4056)z,Environment variable name {!r} should be strN) r#ZPY2r"rwarningswarnpytest PytestWarningformat)r-rr r r _warn_if_env_name_is_not_strsz(MonkeyPatch._warn_if_env_name_is_not_strNcCsvt|ts8tjtdj||t|jdddt|}|rX|t j krX||t j |}| || t j ||dS)z Set environment variable ``name`` to ``value``. If ``prepend`` is a character, read the current environment variable value and prepend the ``value`` adjoined with the ``prepend`` character.zvValue of environment variable {name} type should be str, but got {value!r} (type: {type}); converted to str implicitly)rr?r) stacklevelN) r"rrHrIrJrKrLrrosenvironrMrE)r-rr?Zprependr r r setenvs  zMonkeyPatch.setenvcCs |||jtj||ddS)z Delete ``name`` from the environment. Raise KeyError if it does not exist. If ``raising`` is set to False, no exception will be raised if the environment variable is missing. )r'N)rMrGrPrQ)r-rr'r r r delenvs zMonkeyPatch.delenvcCs0|jdkrtjdd|_tjdt|dS)z< Prepend ``path`` to ``sys.path`` list of import locations. Nr)r4syspathinsertr)r-rUr r r syspath_prepends zMonkeyPatch.syspath_prependcCsP|jdkrt|_t|dr(|n$t|trBtt|n t|dS)z} Change the current working directory to the specified path. Path can be a string or a py.path.local object. Nchdir)r3rPgetcwdrBrXr"rr)r-rUr r r rX s     zMonkeyPatch.chdirc Csx8t|jD]*\}}}|tk r,t|||q t||q Wg|jdd<xJt|jD]<\}}}|tkry ||=Wqtk rYqXqT|||<qTWg|jdd<|jdk r|jtj dd<d|_|j dk rt |j d|_ dS)aE Undo previous changes. This call consumes the undo stack. Calling it a second time has no effect unless you do more monkeypatching after the undo call. There is generally no need to call `undo()`, since it is called automatically during tear-down. Note that the same `monkeypatch` fixture is used across a single test function invocation. If `monkeypatch` is used both by the test function itself and one of the test fixtures, calling `undo()` will undo all of the changes made in both functions. N) reversedr1r9r>rCr2rFr4rTrUr3rPrX)r-rrr?Z dictionaryr r r r s&     zMonkeyPatch.undo)T)N)T)rr/r0__doc__r5rr7r9r>rCrErGrMrRrSrWrXr r r r r rfs %"   r)r[Z __future__rrrrPrerTrH contextlibrr#rJZ_pytest.fixturesrZ_pytest.pathlibrcompileZRE_IMPORT_ERROR_NAMEr rrr+objectr,r9rr r r r s(