o v&ib@sUddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlZddlmZddlmZedZejjZeejjejjejj fZ!Gddde"Z#e GdddZ$e GdddZ%GdddZ&dedee&ddffddZ'e(e'Z)ej(dd e'Z*ej(d!d e'Z+ej(d"d e'Z,ej(d#d e'Z-gZ.e/ee0d$<iZ1e2e3efe0d%<d&ed'efd(ed)eddfd*d+Z4d&ed'efd(ed)eddfd,d-Z5d(ed)eddfd.d/Z6d(ed)eddfd0d1Z7d(ed)eddfd2d3Z8d(ed)eddfd4d5Z9d(ed)eddfd6d7Z:d(ed)eddfd8d9Z;d(ed)eddfd:d;Zd?Z>d(ed)eddfd@dAZ?d(ed)eddfdBdCZ@d(ed)eddfdDdEZAd(ed)eddfdFdGZBd(ed)eddfdHdIZCdJeddfdKdLZDdTdMdNZEdOeddfdPdQZFdJeddfdRdSZGdS)UN) Generator)Iterable)Iterator)Mapping) dataclass)field)Any)Callable)Optional)TypeVar)Union)cast)overload)get_mock_module)parse_ini_boolean_Tc@seZdZdZdS)PytestMockWarningz3Base class for all warnings emitted by pytest-mock.N)__name__ __module__ __qualname____doc__rrX/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pytest_mock/plugin.pyr$src@s&eZdZUeed<dZeeed<dS) MockCacheItemmockNpatch)rrrMockType__annotations__rr rrrrrr(s rc@s~eZdZUdZeedZeeed<de defddZ de de defd d Z de dd fd d Z dddZdeefddZd S) MockCachezJ Cache MagicMock and Patcher instances so we can undo them later. )default_factorycacherreturncCs&|jD] }|j|ur|Sqtd)Nz"This mock object is not registered)r!r ValueErrorselfr mock_itemrrr_find6s  zMockCache._findkwargscKs$|jtdd|i||jdS)Nrr)r!appendr)r%rr(rrradd<s z MockCache.addNcCs*||}|jr |j|j|dSN)r'rstopr!remover$rrrr.@s  zMockCache.removecCs2t|jD] }|jdur|jq|jdSr,)reversedr!rr-clear)r%r&rrrr0Fs   zMockCache.clearcCs t|jSr,)iterr!r%rrr__iter__Ls zMockCache.__iter__r"N)rrrrrlistr!rrrr'rr+r.r0rr3rrrrr.s  rc @seZdZdZdeddfddZ d#ded ed ed edef d d ZddddededdfddZ d$ddZ de j j ddfddZ d%dedededefddZd&deede j j fddZd&deedefdd ZGd!d"d"ZdS)' MockerFixturez Fixture that provides the same interface to functions in the mock module, ensuring that they are uninstalled at the end of each test. configr"NcCst|_t||_}||j||_|j|_|j|_|j|_|j |_ |j |_ t |dr0|j |_ |j |_ |j|_|j|_|j|_|j|_t |drO|j|_dSdS)N AsyncMockseal)r _mock_cacher mock_module_PatcherrMock MagicMockNonCallableMockNonCallableMagicMock PropertyMockhasattrr8callANYDEFAULTsentinel mock_openr9)r%r7r;rrr__init__Vs$   zMockerFixture.__init__Fspecspec_setinstancer(cKs(|jj|||fi|}|j||Sr,)r;create_autospecr:r+)r%rIrJrKr(mrrrrLjs zMockerFixture.create_autospec return_value side_effectrOrPcCst|dr |j|jf}n|jf}|jD]1}t|jdsqt|jdr&g|j_t|jdr0d|j_t|j|r?|jj||dq|jqdS)z Call reset_mock() on all patchers started by this fixture. :param bool return_value: Reset the return_value of mocks. :param bool side_effect: Reset the side_effect of mocks. r8 reset_mockspy_return_listspy_return_iterNrN) rBr=r8r:rrRrS isinstancerQ)r%rOrPsupports_reset_mock_with_argsr&rrrresetallss       zMockerFixture.resetallcCs|jdS)zi Stop all patchers started by this fixture. Can be safely called multiple times. N)r:r0r2rrrstopallszMockerFixture.stopallrcCs|j|dS)zp Stops a previous patch or spy call by passing the ``MagicMock`` object returned by it. N)r:r.)r%rrrrr-szMockerFixture.stopobjnameduplicate_iteratorscst||fdd}fdd}tr t|}nt|}tp/t}|jj||||dd_ d_ g_ d_ S)ak Create a spy of method. It will run method normally, but it is now possible to use `mock` call features with it, like call count. :param obj: An object. :param name: A method in object. :param duplicate_iterators: Whether to keep a copy of the returned iterator in `spy_return_iter`. :return: Spy object. c sd_d_z |i|}Wnty}z|_d}~wwr3t|tr3t|d\}}|_nd_|_j ||S)N) spy_return spy_exception BaseExceptionrTr itertoolsteerSrRr*)argsr(reduplicated_iteratorrZmethodspy_objrrwrappers  z"MockerFixture.spy..wrapperc s^d_d_z |i|IdH}Wnty#}z|_d}~ww|_j||Sr,)r\r]r^rRr*)rar(rbrc)rfrgrr async_wrappers z(MockerFixture.spy..async_wrapper)rPautospecN) getattrinspectiscoroutinefunction functoolsupdate_wrapperismethod isfunctionrobjectr\rSrRr])r%rXrYrZrhriwrappedrjrrerspys   zMockerFixture.spycCsttjj|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_dSr,rrar(rrrz$MockerFixture.stub..rIrY)r unittestrr>r;r%rYrrrstubszMockerFixture.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_rur,rrvrrrrwrxz*MockerFixture.async_stub..ry)r AsyncMockTyper;r8r{rrr async_stubszMockerFixture.async_stubc@s*eZdZdZeZddZdedededede f d d Z ed d d d d fd ede dede edede ede ededede fddZed d d d d fd e jde de jde e jdede e jde e jde jdede fddZ d(d e jde e jdede e jde e jde e jdedee e ffddZ  d)deeeefe fdeeeefeeeeffdededef dd Ze ! ! ! ! ! !d*d e dd de e jdede e jde e jdd dede fd"d#Ze ! ! ! ! !d+d e dede e jdede e jde e jdd dedefd$d#Zed e dd de e jdede e jde e jdegefdedefd%d#Ze ! ! ! ! !d+d e dd de e jdede e jde e jdegefdedefd&d#Zed d d d d fd e de jde e jdede e jde e jde egefdedefd'd#Zd S),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||_||_dSr,)_Patcher__mock_cacher;)r% mock_cacher;rrrrHs zMockerFixture._Patcher.__init__ mock_funcwarn_on_mock_enterrar(r"cOsN||i|}|}|jj||dt|dr%t|dr%|r%dd|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. )rrrQ __enter__cSstjdtddS)Na.Mocks 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/usage.html#usage-as-context-manager) stacklevel)warningswarnrrrrrrws z5MockerFixture._Patcher._start_patch..)startrr+rBrrP)r%rrrar(pmockedrrr _start_patchs  z#MockerFixture._Patcher._start_patchNFtarget attributenewrIcreaterJrj new_callablec K>||jur |jj}|j|jjjd||f||||||d| S)zAPI to mock.patch.objectTrrIrrJrjrrEr;rrrr r%rrrrIrrJrjrr(rrrrrs"  zMockerFixture._Patcher.objectc Kr)zThis is equivalent to mock.patch.object except that the returned mock does not issue a warning when used as a context manager.Frrrrrrcontext_manager7s"  z&MockerFixture._Patcher.context_managercKs(|j|jjjd|f|||||d|S)zAPI to mock.patch.multipleT)rIrrJrjr)rr;rmultiple)r%rrIrrJrjrr(rrrrUs  zMockerFixture._Patcher.multiplerin_dictvaluesr0cKs"|j|jjjd|f||d|S)zAPI to mock.patch.dictT)rr0)rr;rdict)r%rrr0r(rrrrlszMockerFixture._Patcher.dict.c Krur,r r%rrrIrrJrjrr(rrr__call__} zMockerFixture._Patcher.__call__c Krur,rrrrrrrc Krur,rrrrrrrc Krur,rrrrrrs c Ks:||jur |jj}|j|jjd|f||||||d|S)zAPI to mock.patchTr)rEr;rrrrrrrs  )NFNNN)rF)......).....)rrrrrrrErHrboolrrstrr builtinsrrrr rrtuplerrrr rrrrr<s    !   !                 r<)FFr4)Fr,)rrrrrrHrrrLrVrWrzrr>r-rrrrtr r|r}r~r<rrrrr6PsL    <  r6 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)r6rW)rresultrrr_mockers rclass)scopemodulepackagesession_mock_module_patches_mock_module_originals__wrapped_mock_method__.rar(c Os&d}z ||i|WdSty}zzt|ddrt|}nf|d}t|}|jdur|j\}}d} z ||ddks>JWntyY} z| dt| 7} WYd} ~ nd} ~ wwz||ksaJWnty|} z| dt| 7} WYd} ~ nd} ~ ww| r|d| 7}t|}d|_|d}~ww) NT_mock_introspection_appliedrr Args: Kwargs: pytest introspection follows: )AssertionErrorrkr call_argsr) rrar(__tracebackhide__rcmsg __mock_self actual_args actual_kwargs introspectione_argse_kwargsrrrassert_wrappers>     rcOsd}z ||i|WdSty}z|dd}t|dds"|r't|}n|d}t|}|jdurt|j}|d} d} ddlm} | || D]f\} } | durW| \}}nt}i}| durf| \}}}nt}i}z||ksrJWnty}z| d t|7} WYd}~nd}~wwz||ksJWqJty}z| d t|7} WYd}~qJd}~ww| r|d | 7}t|}d|_ |d}~ww) NT any_orderFrrrr) zip_longestrrr) rgetrkrcall_args_listr5r_rrr)rrar(rrcrrr actual_calls expect_callsrr actual_call expect_callrr_ expect_args expect_kwargsrrrrrassert_has_calls_wrappersV        rcO"d}ttdg|Ri|dS)NTassert_not_calledrrrar(rrrrwrap_assert_not_called2rcOr)NTassert_called_withrrrrrwrap_assert_called_with7rrcOr)NTassert_called_oncerrrrrwrap_assert_called_once<rrcOr)NTassert_called_once_withrrrrrwrap_assert_called_once_withArrcOr)NTassert_has_calls)rrrrrrwrap_assert_has_callsFs rcOr)NTassert_any_callrrrrrwrap_assert_any_callMrrcOr)NT assert_calledrrrrrwrap_assert_calledRrrcOr)NTassert_not_awaitedrrrrrwrap_assert_not_awaitedWrrcOr)NTassert_awaited_withrrrrrwrap_assert_awaited_with\rrcOr)NTassert_awaited_oncerrrrrwrap_assert_awaited_oncearrcOr)NTassert_awaited_once_withrrrrrwrap_assert_awaited_once_withfrrcOr)NTassert_has_awaitsrrrrrwrap_assert_has_awaitskrrcOr)NTassert_any_awaitrrrrrwrap_assert_any_awaitprrcOr)NTassert_awaitedrrrrrwrap_assert_awaitedurrr7c CstrdSt|}tttttttd}| D],\}}zt |j |}Wn t y+Yqw|t|<|j |j ||}|t|qt|drtttttttd}| D],\}}zt |j|}Wn t ykYqVw|t|<|j |j||}|t|qV|tdS)z Wrap assert methods of mock module so we can hide their traceback and add introspection information to specified argument asserts. N)rrrrrrrr8)rrrrrrr)rrrrrrrrritemsrkr?AttributeErrorrrrrrr*rBrrrrrrrr8 add_cleanupunwrap_assert_methods)r7r;wrappersrfrhoriginalpatcherasync_wrappersrrrwrap_assert_methodszsR       rc CsbtD]"}z|Wqty$}zt|dkrnWYd}~qd}~wwgtdd<tdS)Nz stop called on unstarted patcher)rr- RuntimeErrorrrr0)rrcrrrrs    rparsercCs$|jdddd|jdddddS)Nmock_traceback_monkeypatchzRMonkeypatch the mock library to improve reporting of the assert_called_... methodsTdefaultmock_use_standalone_modulezPUse standalone "mock" (from PyPI) instead of builtin "unittest.mock" on Python 3F)addini)rrrrpytest_addoptions rcCs8|jddd}t|dr|dkrt|dSdSdS)Nz--tbautorrnative) getoptionrgetinir)r7tbrrrpytest_configures  rr4)Hrrnrlr_ unittest.mockrzrcollections.abcrrrr dataclassesrrtypingrr r r r r rpytest_utilrrrrr8r}r>r@r UserWarningrrrr6rfixturemocker class_mocker module_mockerpackage_mockersession_mockerrr5rrrrrrrrrrrrrrrrrrrrrrrrrrrrs                !     0 5