`{)ddlZddlmZddlmZddlmZmZmZm Z m Z m Z e j Z e jZGddeZGddejZGd d ejZed krejdSdS) N)catch_warnings) is_instance) MagicMockMockpatchsentinel mock_opencallceZdZdS)SampleExceptionN)__name__ __module__ __qualname__F/opt/alt/python311/lib64/python3.11/unittest/test/testmock/testwith.pyr r srr cVeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S)WithTestctdtztj5|t tjddddn #1swxYwY|t tjdSN %s.something unpatched)rr r Something2 assertEqual something Somethingselfs rtest_with_statementzWithTest.test_with_statements >H,h.A B B J J   Y(;[ I I I J J J J J J J J J J J J J J J H$677777s'AAAcz|t5tdtztj5|ttjdt#1swxYwY dddn #1swxYwY|ttjdSr) assertRaisesr rr rrrrrrs rtest_with_statement_exceptionz&WithTest.test_with_statement_exceptions    / / ( (~0(2EFF ( (  H,?MMM%''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( H$677777s.#B 5A33A7 7B :A7 ;B  BBc8tdtz5}|t|d|t |t ddddn #1swxYwY|ttjdS)Nrrzpatching wrong type) rr rr assertTruerrrrrmock_somethings rtest_with_statement_aszWithTest.test_with_statement_as s >H, - - 3   Y D D D OOK BB1 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 H$677777sAA**A.1A.cGddt}|j}tj|d5||j|ddddn #1swxYwY||j|dS)NceZdZdZdS)6WithTest.test_patch_object_with_statement..FoofooN)r rrrrrrFoor*)sIIIrr,rr)objectrrassertNotEqualr)rr,originals r test_patch_object_with_statementz)WithTest.test_patch_object_with_statement(s     &   = \#{ + + F F    x E E E F F F F F F F F F F F F F F F 11111sAA Actd5tdtz5}tdtz5}|t|d|t |ddddn #1swxYwYdddn #1swxYwYdddn #1swxYwY|tt j|t t jdS)NT)recordr%s.something_elser) rrr rrsomething_elserr SomethingElserr&mock_something_elses rtest_with_statement_nestedz#WithTest.test_with_statement_nested1s 4 ( ( ( . .~011 .^UK^aiKiEjEj .oB  NKHHH  1D!,... . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . H$6777 )?@@@@@sXB4B9B: BB B B B B4B! !B4$B! %B44B8;B8c>tdtztj5}|t |d||tjddddn #1swxYwY|t tjdS)Nrrz wrong patchrr rPatchedrrrr%s rtest_with_statement_specifiedz&WithTest.test_with_statement_specified<s >H,h.> ? ? N>   Y D D D   ^X-=} M M M N N N N N N N N N N N N N N N H$677777s>A--A14A1cdt}t|_t|_d|j_|5}|||jjdddn #1swxYwY|j|jddddS)NF)r __enter____exit__ return_valuerassert_called_with)rmockms rtestContextManagerMockingz"WithTest.testContextManagerMockingCsvv %* "  =Q   Q ; < < < = = = = = = = = = = = = = = = ))+++ ((tT:::::s!A00A47A4c4t}|t5|5ddzdddn #1swxYwYdddn #1swxYwY|j||jjdS)Nr+)rr! TypeErrorr>rAr$r?called)rrBs r$test_context_manager_with_magic_mockz-WithTest.test_context_manager_with_magic_mockOs {{   y ) )                                    ))+++  ,-----s1A> AA AA AAActdtztj5}|t |dtdtz5}|t |ddddn #1swxYwY|t |ddddn #1swxYwY|t tjddS)Nrrzrestored with wrong instancez not restoredr:)rr& mock_agains r"test_with_statement_same_attributez+WithTest.test_with_statement_same_attributeYsU >H,h.> ? ? =>   Y D D D~011 EZ  J DDD E E E E E E E E E E E E E E E   Y; = = =  = = = = = = = = = = = = = = = H$6GGGGGs54B3B4 B3B B3B B33B7:B7ctdtz5}|t|dtdtz5}|t|ddddn #1swxYwYdddn #1swxYwY|tt j|tt jdS)Nrrr3)rr rrr4rrr5r6s rtest_with_statement_imbricatedz'WithTest.test_with_statement_imbricatedfs@ >H, - - .   Y D D D*X566 .:M  1D!,... . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . H$6777 )?@@@@@s54B  A5) B 5A9 9B <A9 =B  BBci}tj|ddi5||ddidddn #1swxYwY||i|t5tj|ddi5||ddit d#1swxYwY dddn #1swxYwY||idS)NabKonrad)rdictrr! NameError)rr+s rtest_dict_context_managerz"WithTest.test_dict_context_managerrs Zc3Z ( ( . .   S3* - - - . . . . . . . . . . . . . . . b!!!   y ) ) * *C#s,, * *  sCj111))) * * * * * * * * * * * * * * * * * * * * * * * * b!!!!!s??AA:C(B::B> >CB> CCCcGdd}|}tj|d5}tj|d5}|dddn #1swxYwY||jd||jd|dddn #1swxYwY||jddS)NceZdZdZdS)5WithTest.test_double_patch_instance_method..CcdS)Nrrs rfz7WithTest.test_double_patch_instance_method..C.fsrN)r rrrZrrrCrXs     rr[rZr)rr-rZr call_count)rr[cpatch1patch2s r!test_double_patch_instance_methodz*WithTest.test_double_patch_instance_methodsl         ACC \!S ! ! Va%%                   V. 2 2 2   V. 2 2 2 CCEEE                 *A.....s6CA$ C$A( (C+A( ,A CC  C N)r rrrr"r'r0r8r<rDrIrLrNrUrarrrrrs888 888888222AAA888 ; ; ;... H H H A A A " " " / / / / /rrcheZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdS) TestMockOpenct}tdtz|d5}|||t ddddn #1swxYwY|ddSN%s.openTcreater+)r rr assertIsopenassert_called_once_withrrBpatcheds rtest_mock_openzTestMockOpen.test_mock_opens{{ 9x'd ; ; ; w MM'4 ( ( ( KKK                $$U+++++s&AA"AcLt}|j}tdtz|d5t d5}|dddn #1swxYwYdddn #1swxYwYt dt t t dddg}| |j || ||dSre) r r@rr rjreadr r>r?r mock_callsri)rrBhandlerZexpected_callss rtest_mock_open_context_managerz+TestMockOpen.test_mock_open_context_managersh{{" 9x'd ; ; ;  e                               u++tvv'7'7'9'9466;;==&&//$d;;= .999 a     s5A8A! A8!A% %A8(A% )A88A<?A<c dt}tdtz|d5td5}|dddn #1swxYwYtd5}|dddn #1swxYwYdddn #1swxYwYt dt t t dddt dt t t dddg}||j |dS)NrfTrgr+bar) r rr rjrpr r>r?rrq)rrBrZrss r-test_mock_open_context_manager_multiple_timesz:TestMockOpen.test_mock_open_context_manager_multiple_timess{{ 9x'd ; ; ;  e                e                                KK))++TVV[[]] FFOOD$ - - KK))++TVV[[]] FFOOD$ - - / .99999sXB,A B,A B,!A "B,4B B,B B,B B,,B03B0ct}t|tdtz|d5}|||t ddddn #1swxYwY|ddSre)rr rr rirjrkrls rtest_explicit_mockzTestMockOpen.test_explicit_mocks{{$ 9x'd ; ; ; w MM'4 ( ( ( KKK                $$U+++++s&A**A.1A.ctd}tdtz|d5td}|}dddn #1swxYwY||ddS)Nr+ read_datarfTrgrvr rr rjrprrrBhresults rtest_read_datazTestMockOpen.test_read_datas5))) 9x'd ; ; ;  U AVVXXF                '''''s$AA"Ac:td}tdtz|d5td}|}|}|}dddn #1swxYwY||d||d||d ||d td }tdtz|d5td}|}dddn #1swxYwY||d ||d dS) N foo bar baz r{rfTrgrvfoo bar baz r+r rr rjreadliner)rrBrline1line2line3rs rtest_readline_datazTestMockOpen.test_readline_datas#4555 9x'd ; ; ; ! !U AJJLLEJJLLEJJLLE  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ((( ((( ((( r***5))) 9x'd ; ; ; " "U AZZ\\F " " " " " " " " " " " " " " " ''' r*****s$A BB B"$EEEc<td}tdtz|d5td}d|D}dddn #1swxYwY||dd ||d d ||d d ||d|t5t|ddddS#1swxYwYdS)Nrr{rfTrgrvcg|]}|Srr.0ls r z6TestMockOpen.test_dunder_iter_data..NNN1QNNNrrrr\rrr) r rr rjrrr! StopIterationnext)rrBrliness rtest_dunder_iter_dataz"TestMockOpen.test_dunder_iter_datas#4555 9x'd ; ; ; # #U A"NNNNE # # # # # # # # # # # # # # # q7+++ q7+++ q7+++ r***   } - -   GGG                  s#AAA4DDDctd}tdtz|d5td}t |}t |}d|D}dddn #1swxYwY||d||d ||d d ||d dS) Nrr{rfTrgrvcg|]}|Srrrs rrz/TestMockOpen.test_next_data..rrrrrrr)r rr rjrrr)rrBrrrrs rtest_next_datazTestMockOpen.test_next_datas #4555 9x'd ; ; ; # #U AGGEGGE"NNNNE  # # # # # # # # # # # # # # # ((( ((( q7+++ r*****s:A11A58A5ctd}tdtz|d5td}|}dddn #1swxYwY||gdtd}tdtz|d5td}|}dddn #1swxYwY||gd dS) Nrr{rfTrgrv)rrr foo bar baz)rrbazr rr rj readlinesrr~s rtest_readlines_dataz TestMockOpen.test_readlines_datasw#4555 9x'd ; ; ; # #U A[[]]F # # # # # # # # # # # # # # # !rs"######666666KKKKKKKKKKKKKKKK  ('&&&&i&&&{/{/{/{/{/x {/{/{/|I3I3I3I3I38$I3I3I3X zHMOOOOOr