:QgdZddlmZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZddlmZmZmZmZddlmZddlmZmZd Zdd Z dd Z!dd Z"dd Z#ddZ$ddZ% d d!dZ& d" d#dZ' d d$dZ( d d%dZ) d& d'dZ*d"d(dZ+ d d)dZ, d d*dZ-d+dZ.d,dZ/d-dZ0y).z!Utilities that ease unit-testing.) annotationsN)AnyListOptional)ANY MagicMockMock PropertyMockcallcreate_autospec mock_openpatch)CaptureFixtureFixtureRequestLogCaptureFixture MonkeyPatch)Element)elements_from_jsonelements_to_json)rrrrrr rr class_mock function_mockinitializer_mock instance_mock method_mock property_mockct|}|Jt|}t|}|J||k(sJtd||y)abRaises AssertionError if `elements -> JSON -> List[Element] -> JSON` are not equal. The procedure is: 1. Serialize `elements` to (original) JSON. 2. Deserialize that JSON to `List[Element]`. 3. Serialize that `List[Element]` to JSON. 3. Compare the original and round-tripped JSON, raise if they are different. N)textz JSON differs:)rr_diff)elements original_jsonround_tripped_elementsround_tripped_jsons Y/var/www/html/answerous/venv/lib/python3.12/site-packages/test_unstructured/unit_utils.pyassert_round_trips_through_JSONr$,sa%X.M  $$ $/]C)*@A  )) )  .+]1 .cNt|D]\}}|j||S)z5Updates the `id` attribute of each element to a hash.) enumerate id_to_hash)ridxelements r#assign_hash_idsr+Ds-!(+ W3 Or%c|jd}|jd}d}|djtjj ||zS)z~Diff of actual compared to expected. "+" indicates unexpected lines actual, "-" indicates lines missing from actual. T)keependszFdiff: '+': unexpected lines in actual, '-': lines missing from actual ) splitlinesjoindifflibDiffercompare)headingactualexpectedexpected_lines actual_liness r#rrKsW (($(7N$$d$3LWG RWWW^^-55lNST TTr%ctjtjjdz }||z }t |j S)zGResolve the absolute-path to `file_name` in the example-docs directory.z example-docs)pathlibPath__file__parentstrresolve) file_nameexample_docs_dir file_paths r#example_doc_pathrCVs@||H-44;;nL 9,I y  " ##r%cvtt|5}|jcdddS#1swYyxYw)z?Contents of example-doc `file_name` as text (decoded as utf-8).N)openrCread)r@fs r#example_doc_textrH]s0 y) *avvxs/8cH|rtjj|SdS)zWParse `datetime_str` to a datetime.datetime instance or None if `datetime_str` is None.N)dtdatetime fromisoformat) datetime_strs r#parse_optional_datetimerNcs6B2;; $ $\ 2LLr%c tt|fd|i|}|j|j|jS)a$Return mock patching class with qualified name `q_class_name`. The mock is autospec'ed based on the patched class unless the optional argument `autospec` is set to False. Any other keyword arguments are passed through to Mock(). Patch is reversed after calling test returns. autospecr addfinalizerstopstart)request q_class_namerPkwargs_patchs r#rrps6< =( =f =F % <<>r%c | |jn|}tj||fd|i|}|j|j|j S)zcReturn a mock for attribute `attr_name` on `cls`. Patch is reversed after pytest uses it. name) fixturenamerobjectrRrSrT)rUcls attr_namerZrWrXs r# cls_attr_mockr_~sM#',7  DD \\#y >t >v >F % <<>r%c tt|fd|i|}|j|j|jS)z|Return mock patching function with qualified name `q_function_name`. Patch is reversed after calling test returns. rPrQ)rUq_function_namerPrWrXs r#rrs6? @X @ @F % <<>r%c tj|df|dd|}|j|j|j S)zaReturn mock for __init__() method on `cls`. The patch is reversed after pytest uses it. __init__N)rP return_valuerr\rRrSrT)rUr]rPrWrXs r#rrs?\\#z ZH4 ZSY ZF % <<>r%c B||n |j}t|f||dd|S)abReturn a mock for an instance of `cls` that draws its spec from the class. The mock does not allow new attributes to be set on the instance. If `name` is missing or |None|, the name of the returned |Mock| instance is set to *request.fixturename*. Additional keyword arguments are passed through to the Mock() call that creates the mock. T)_namespec_setinstance)r[r )rUr]rZrhrWs r#rrs0#4)<)r%c tj||fd|i|}|j|j|j S)zfReturn mock for function `fn_name` on `std_mod`. The patch is reversed after pytest uses it. rPre)rUstd_modfn_namerPrWrXs r#stdlib_fn_mockrrs<\\'7 HX H HF % <<>r%c d|z}t|tfddi|}|j|j|j S)z?Return a mock for the builtin `open()` method in `module_name`.z%s.opencreateT)rr rRrSrT)rU module_namerWtargetrXs r# open_mockrwsB  $F 69; >t >v >F % <<>r%c tj||fdti|}|j|j|j S)zgA mock for property `prop_name` on class `cls`. Patch is reversed at the end of the test run. new_callable)rr\r rRrSrT)rUr] prop_namerWrXs r#rrs< \\#y N| Nv NF % <<>r%c pt|fi|}|j|j|jS)z}Return a mock patching the variable with qualified name `q_var_name`. Patch is reversed after calling test returns. rQ)rU q_var_namerWrXs r#var_mockr}s1 : ( (F % <<>r%)rz List[Element]returnNone)r list[Element]r~r)r4r>r5r>r6r>)r@r>r~r>)rMz Optional[str]r~zOptional[dt.datetime])T) rUrrVr>rPboolrWrr~r )N) rUrr]typer^r>rZ str | NonerWrr~r ) rUrrar>rPrrWrr~r ) rUrr]rrPrrWrr~r )NT) rUrr]rrZrrhrrWrr~r )rUrrZrrWrr~r ) rUrr]rrnr>rPrrWrr~r ) rUrrpztypes.ModuleTyperqr>rPrrWrr~r )rUrrur>rWrr~r ) rUrr]rrzr>rWrr~r )rUrr|r>rWrr~r )1__doc__ __future__rrKrJr1r:typestypingrrr unittest.mockrrr r r r r rpytestrrrrunstructured.documents.elementsrunstructured.staging.baserr__all__r$r+rrCrHrNrr_rrrrlrrrrwrr}rkr%r#rs"'" &&   RQ3J $0U$ MBF   +. :> QT   $         $EI   .1 =A TW   :>   "& 26 IL    X X X X X  X  X"%           (            r%