0L<.^ d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z m
Z
d dl m Z d dl
mZmZmZmZmZ d dlmZmZmZmZmZ d d lmZmZ d dl Z eee gef Z d
Z ed Z G d d
Z! d dedede"dede de deegef fdZ#ede defd Z$ed dedede fd Z%ed dedede fd Z& G d d e' Z(d Z) ee)e!j* e!j# e#e(g dS )! N)contextmanager)partial)
get_referrers
get_referents)setitemgetitem)
ModuleType)AnyTypeVarCallableDictTuple)resolve not_thereResolvedclassmethod_type
class_type)wrapextend_docstringc D t | | ot || S N)
isinstance)xy
descriptors f/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/testfixtures/replace.pynot_same_descriptorr s$ a $ $ FZ:-F-F)FF Rc e Zd ZdZd ZdefdZ ddeded e d
ed e
d e d
efdZ ddeded e d
ed e
d e d
dfdZ
d e ded
dfdZd e de fdZd deded e d
dfdZd dededed
dfdZd dZd Zd Zd ZdS ) Replacerz
These are used to manage the mocking out of objects so that units
of code can be tested without having to rely on their normal
dependencies.
c i | _ d S r ) originalsselfs r __init__zReplacer.__init__ s
:<r resolvedc 0 |t u rk|j t u r, t |j |j n# t $ r Y nw xY w|j t u r! |j |j = d S # t $ r Y d S w xY wd S | |j |j | d S r ) r settersetattrdelattr containernameAttributeErrorr KeyError)r% r' values r _replacezReplacer._replace# s I '))H .
>>>>% D')) *8=999 DD *)
OOH .
u E E E E Es! 4
A A
A" "
A0/A0TNtarget replacementstrictr, accessorr- returnc ||t d t |t r"|t d t || }nt }||}|pt |dd }|t d |I t } ||| }nY# t $ r Y nMt $ r t
} |||t }Y n)w xY w ||| }n# t t f$ r Y nw xY w|r4|t u s+||u s' ||| } || urt |d| d| t ||t u rt nt || }|j
t d |j t u r|rt d |j z |}
t |j t$ r|j |j j v r |j j |j |_ t) |j |t* rt+ | }
n*t) |j |t, rt- | }
| ||
t1 | } | | j vr ||f| j | < |S )
M
Replace the specified target with the supplied replacement.
Nz-accessor is not used unless name is specifiedz0name cannot be specified when target is a string__name__z2name must be specified when target is not a stringz
resolved to z , expected z%target must contain at least one dot!zOriginal %r not found) TypeErrorr strr r getattrr r/ r. AssertionErrorr r r* r)
ValueErrorfoundr- r, type__dict__r classmethod staticmethodr1 idr# ) r% r2 r3 r4 r, r5 r- r' r? expectedreplacement_to_usekeys r __call__zReplacer.__call__2 s
Y & &6 & !88=!HII I ( h($ / /
? } 2 ;;;!)!3!; LL
?%0%=%=""$X^[,OO
?%1+%>%>"
h 2333jj dn $ $"(("2DN3 s* =B
C "CC C C'&C'c & | |||||| dS )r8 N r% r2 r3 r4 r, r5 r- s r replacezReplacer.replace{ s$
V[&)XtDDDDDr c ~ | t j |t d|t u rt nt | dS )a
This method provides a convenient way of ensuring an environment variable
in :any:`os.environ` is set to a particular value.
If you wish to ensure that an environment variable is *not* present,
then use :any:`not_there` as the ``replacement``.
F)r- r5 r4 r3 N)osenvironr r r; )r% r- r3 s r
in_environzReplacer.in_environ sP
RZdWU&1Y&>&>C DTDT V V V V V Vr break_on_staticc " t | D ]~}|rt |t rd |fc S t |t rJd|v rF| | |u r/t | D ] }t |t
r|d fc c S dS )NrA )NN)r r rC dictgetr@ )r% attributer- rQ referrerr, s r _find_containerzReplacer._find_container s %i00 3 3H
3:h #E#E
3X~%%%Hd++
3
h0F0F<<%%22%28%<%< 3 3 %i66 3#,d? 2 2 2 2 23zr rU c t | st d |pt |dd }d}t |t r*t | D ]}t |t r|}n8| ||d \ }}|| ||d \ }}|t d|d| | ||t | dS )
au
This method provides a convenient way to replace methods, static methods and class
methods on their classes.
If the attribute being replaced has a ``__name__`` that differs from the attribute
name on the class, such as that returned by poorly implemented decorators, then
``name`` must be used to provide the correct name.
zattribute must be callabler9 NT)rQ Fz could not find container of z using name r- r5 r3 ) callabler: r< r r r r rW r. )r% rU r3 r- r, referred
staticmethod__s r on_classzReplacer.on_class s "" : 899 9;wy*d;; i!1 2 2 ` )) 4 4
)
) h
33 ) (I
) (,';';It]a';'b'b $I} ( # 3 3M4Y^ 3 _ _ 1 !a !a!aY]!a!abb b YTGMMMMMMr modulec r |pt |j j }|j } | ||t | dS )a
This method provides a convenient way to replace targets that are module globals,
particularly functions or other objects with a ``__name__`` attribute.
If an object has been imported into a module other than the one where it has been
defined, then ``module`` should be used to specify the module where you would
like the replacement to occur.
rY N)r
__module__r? r9 r< )r% r2 r3 r_ r, r- s r in_modulezReplacer.in_module sB >gf&7 8 8 > YTGMMMMMMr c t | j D ]+\ }\ }}| ||j | j |= ,dS )z
Restore all the original objects that have been replaced by
calls to the :meth:`replace` method of this :class:`Replacer`.
N)tupler# itemsr1 r? )r% id_r2 originals r restorezReplacer.restore s`
(-T^-A-A-C-C'D'D $ $ #C#&( MM(HN 3 3 3s## $ $r c | S r rJ r$ s r __enter__zReplacer.__enter__ s r c . | d S r )rh )r% r@ r0 tracebacks r __exit__zReplacer.__exit__ s r c | j r:t j dd | j D z d S d S )Nz;Replacer deleted without being restored, originals left: %rc i | ]\ }}|| S rJ rJ ).0kvs r
z$Replacer.__del__.. s 'R'R'RA!'R'R'Rr )r# warningswarnvaluesr$ s r __del__zReplacer.__del__ s` >
M%'R'R$.:O:O:Q:Q'R'R'RS
r TNNNr )r6 N)r9 ra __qualname____doc__r& r r1 r
r boolAccessorr; rH rL rP rW r r^ r rb rh rj rm rw rJ r r r! r! s' = = =
F
F
F
F
F DHOSG Gs G GD G G2:GILGXYG G G GR EINRE Ec E ET E E19EHKEW[E E E E Vs V V V V V V s T N N( N NC NSW N N N N6 N N N# Nz NUY N N N N$ $ $ $ r r! Tr2 r3 r4 r, r5 r- r6 c x t }t t |j | ||||| |j S )zY
A decorator to replace a target object for the duration of a test
function.
)r! r r rH rh )r2 r3 r4 r, r5 r- rs r rL rL s=
A
FKHdSS r c # K t 5 }| | | dV ddd dS # 1 swxY w Y dS )zW
This context manager provides a quick way to use :meth:`Replacer.in_environ`.
N)r! rP )r- r3 r~ s r replace_in_environr s
q T;'''
s
9= =rU c # K t 5 }| | || dV ddd dS # 1 swxY w Y dS )zU
This context manager provides a quick way to use :meth:`Replacer.on_class`.
N)r! r^ )rU r3 r- r~ s r replace_on_classr s
q
9k4000
:>>r_ c # K t 5 }| | || dV ddd dS # 1 swxY w Y dS )zV
This context manager provides a quick way to use :meth:`Replacer.in_module`.
N)r! rb )r2 r3 r_ r~ s r replace_in_moduler s
q FK000
r c
J e Zd ZdZ ddededededed ef d
Z d efd Z
d
Z dS )ReplacezU
A context manager that uses a :class:`Replacer` to replace a single target.
TNr2 r3 r4 r, r5 r- c || _ || _ || _ || _ || _ || _ t
| _ d S r )r2 r3 r4 r, r5 r- r! _replacerrK s r r& zReplace.__init__ s> & '"*
!r r6 c r | | j | j | j | j | j | j S r )r r2 r3 r4 r, r5 r- r$ s r rj zReplace.__enter__ s4 ~~ K)4;
W[W`
r c 8 | j d S r )r rh )r% exc_typeexc_valexc_tbs r rm zReplace.__exit__ s r rx ) r9 ra ry rz r
r r{ r| r; r& rj rm rJ r r r r s
?CJN
$
$
$,-
$7;
$
$-5
$DG
$
$
$
$
1
! ! ! ! !r r a
:param target:
This must be one of the following:
- A string containing the dotted-path to the object to be replaced, in which case it will be
resolved the the object to be replaced.
This path may specify a module in a package, an attribute of a module, or any attribute of
something contained within a module.
- The container of the object to be replaced, in which case ``name`` must be specified.
- The object to be replaced, in which case ``container`` must be specified.
``name`` must also be specified if it cannot be obtained from the ``__name__`` attribute
of the object to be replaced.
:param replacement: The object to use as a replacement.
:param strict: When `True`, an exception will be raised if an
attempt is made to replace an object that does
not exist or if the object that is obtained using the ``accessor`` to
access the ``name`` from the ``container`` is not identical to the ``target``.
:param container:
The container of the object from which ``target`` can be accessed using either
:func:`getattr` or :func:`~operator.getitem`.
:param accessor:
Either :func:`getattr` or :func:`~operator.getitem`. If not supplied, this will be inferred
preferring :func:`~operator.getitem` over :func:`getattr`.
:param name:
The name used to access the ``target`` from the ``container`` using the ``accessor``.
If required but not specified, the ``__name__`` attribute of the ``target`` will be used.
rx r )+rN
contextlibr functoolsr gcr r operatorr r typesr typingr
r r r
r testfixtures.resolver r r r r testfixtures.utilsr r rt r; r| r r r! r{ rL r r r objectr replace_params_docrH rJ r r r s % % % % % % + + + + + + + + % % % % % % % % 6 6 6 6 6 6 6 6 6 6 6 6 6 6 [ [ [ [ [ [ [ [ [ [ [ [ [ [ 5 5 5 5 5 5 5 5 S#JO $G G G GCLL} } } } } } } }B 7;FJ "% /3 )1 @C xj("# S s s # c Z ! ! ! ! !f ! ! !4# L ##X%5wHJ J J J Jr