B >cU@sddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlZddlmZddlmZedZejdddkr&ejjZne ZGddde Z!GdddZ"e e e"ddfdddZ#e$e#Z%ej$dde#Z&ej$dde#Z'ej$d de#Z(ej$d!de#Z)gZ*iZ+e d"e fe e dd#d$d%Z,e e dd&d'd(Z-e e dd&d)d*Z.e e dd&d+d,Z/e e dd&d-d.Z0e e dd&d/d0Z1e e dd&d1d2Z2e e dd&d3d4Z3e e dd&d5d6Z4e e dd&d7d8Z5e e dd&d9d:Z6e e dd&d;d<Z7e e dd&d=d>Z8e e dd&d?d@Z9e e dd&dAdBZ:e ddCdDdEZ;ddFdGdHZdS)NN)Any)Callable)cast)Dict) Generator)Iterable)List)Mapping)Optional)overload)Tuple)Type)TypeVar)Union)get_mock_module)parse_ini_boolean_T)c@seZdZdZdS)PytestMockWarningz3Base class for all warnings emitted by pytest-mock.N)__name__ __module__ __qualname____doc__rr1lib/python3.7/site-packages/pytest_mock/plugin.pyr$src@seZdZdZeddddZdddeeddd d Zdd d d Ze j j ddddZ e ee j j dddZdeee j j dddZdeeedddZGdddZdS) MockerFixturez Fixture that provides the same interface to functions in the mock module, ensuring that they are uninstalled at the end of each test. N)configreturncCsg|_t||_}||j||_|j|_|j|_|j|_|j|_|j |_ t |dr^|j |_ |j |_ |j |_ |j|_|j|_|j|_|j|_t |dr|j|_dS)N AsyncMockseal)_patches_and_mocksr mock_module_PatcherpatchMock MagicMockNonCallableMockZNonCallableMagicMockZ PropertyMockhasattrr!ZcallANYDEFAULTZcreate_autospecsentinelZ mock_openr")selfrr$rrr__init__.s&   zMockerFixture.__init__F) return_value side_effect)r0r1r cCsft|dr|j|jf}n|jf}x@|jD]6\}}t|ds.wrapperc sVd_d_y||IdH}Wn*tk rJ}z |_Wdd}~XYnX|_|S)N)rArBrC)rDrErFrG)rHrIrr async_wrappersz(MockerFixture.spy..async_wrapper)r1autospecN)getattrinspectZisclassr3Zgetattr_static classmethod staticmethodZismethodZ isfunctionasyncioZiscoroutinefunction functoolsupdate_wrapperr&objectrArB)r.r?r@rLrJrKwrappedr)rHrIrspyss     zMockerFixture.spy)r@r cCsttjj|jjdd|dS)z Create a stub method. It accepts any arguments. Ideal to register to callbacks in tests. :param name: the constructed stub's name as used in repr :return: Stub object. c_sdS)Nr)rDrErrrz$MockerFixture.stub..)specr@)runittestr;r(r$)r.r@rrrstubszMockerFixture.stubcCstt|jjdd|dS)z Create a async stub method. It accepts any arguments. Ideal to register to callbacks in tests. :param name: the constructed stub's name as used in repr :return: Stub object. c_sdS)Nr)rDrErrrrWrXz*MockerFixture.async_stub..)rYr@)r AsyncMockTyper$r!)r.r@rrr async_stubszMockerFixture.async_stubc @s~eZdZdZeZddZeeeee j j dddZ edddddfee eeeeeeeeeee j j d d d Zedddddfeje ejeejeeejeejejee j j d d d Zdejeejeeejeejeejeee e j j fdddZdeeeefe feeeefeeeeffeeedddZede deejeeejeejdee j j d ddZed e eeejeeejeejdeed ddZee deejeeejeejegefeed ddZed!e deejeeejeejegefeed ddZedddddfe ejeejeeejeejeegefeed ddZdS)"zMockerFixture._Patcherz Object to provide the same interface as mock.patch, mock.patch.object, etc. We need this indirection to keep the same API of the mock package. cCs||_||_dS)N)_Patcher__patches_and_mocksr$)r.Zpatches_and_mocksr$rrrr/szMockerFixture._Patcher.__init__) mock_funcwarn_on_mock_enterrDrEr csb|||}|}|j||ft|dr^t|dr^|r^tjdkrJdndfdd|j_|S)zPatches something by calling the given function from the mock module, registering the patch to stop it later and returns the mock object resulting from the mock call. r2 __enter__)rcstjdtdS)Na0Mocks returned by pytest-mock do not need to be used as context managers. The mocker fixture automatically undoes mocking at the end of a test. This warning can be ignored if it was triggered by mocking a context manager. https://pytest-mock.readthedocs.io/en/latest/remarks.html#usage-as-context-manager) stacklevel)warningswarnrr)depthrrrWsz5MockerFixture._Patcher._start_patch..)startr^appendr*sys version_inforar1)r.r_r`rDrEr4Zmockedr)rhr _start_patchs   z#MockerFixture._Patcher._start_patchNF) target attributenewrYcreatespec_setrL new_callablerEr c Ks>||jkr|jj}|j|jjjd||f||||||d| S)zAPI to mock.patch.objectT)rprYrqrrrLrs)r,r$rmr&rT) r.rnrorprYrqrrrLrsrErrrrTs zMockerFixture._Patcher.objectc Ks>||jkr|jj}|j|jjjd||f||||||d| S)zThis is equivalent to mock.patch.object except that the returned mock does not issue a warning when used as a context manager.F)rprYrqrrrLrs)r,r$rmr&rT) r.rnrorprYrqrrrLrsrErrrcontext_managers z&MockerFixture._Patcher.context_manager)rnrYrqrrrLrsrEr cKs(|j|jjjd|f|||||d|S)zAPI to mock.patch.multipleT)rYrqrrrLrs)rmr$r&multiple)r.rnrYrqrrrLrsrErrrru%s zMockerFixture._Patcher.multipler)in_dictvaluesr9rEr cKs"|j|jjjd|f||d|S)zAPI to mock.patch.dictT)rwr9)rmr$r&dict)r.rvrwr9rErrrrx<szMockerFixture._Patcher.dict.) rnrprYrqrrrLrsrEr c KsdS)Nr) r.rnrprYrqrrrLrsrErrr__call__Ms zMockerFixture._Patcher.__call__c KsdS)Nr) r.rnrprYrqrrrLrsrErrrry[s c KsdS)Nr) r.rnrprYrqrrrLrsrErrrryis c KsdS)Nr) r.rnrprYrqrrrLrsrErrrryws c Ks:||jkr|jj}|j|jjd|f||||||d|S)zAPI to mock.patchT)rprYrqrrrLrs)r,r$rmr&) r.rnrprYrqrrrLrsrErrrrys )NFNNN)rF)......).....).....)rrrrrTr,r/rboolrZr;r(rmstrr builtinsrtrrurr rr rxr ryrrrrrrr%s ,(4 ""   2r%)N)N)rrrrrr/rzr6r:rZr;r(r8rTr{rVr r[r\r]r%rrrrr(s   5  r) pytestconfigr ccst|}|V|dS)z Return an object that has the same interface to the `mock` module, but takes care of automatically undoing all patches after each test method. N)rr:)r}resultrrr_mockersrclass)ZscopemodulepackageZsession.)__wrapped_mock_method__rDrEr c Os0d}y|||dStk r*}zt|ddrsv                 | 5