/opt/imh-python/lib/python3.9/unittest/test/testmock/__pycache__
Edit: /opt/imh-python/lib/python3.9/unittest/test/testmock/__pycache__/testwith.cpython-39.pyc (12329B)
a
dh/ @ s d dl Z d dlmZ d dlmZ d dlmZmZmZm Z m
Z
m Z e j Z
e jZG dd deZG dd de jZG d d
d
e jZed kre dS ) N)catch_warnings) is_instance) MagicMockMockpatchsentinel mock_opencallc @ s e Zd ZdS )SampleExceptionN)__name__
__module__ __qualname__ r r @/opt/imh-python/lib/python3.9/unittest/test/testmock/testwith.pyr
r
c @ sl e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd ZdS )WithTestc C sR t dt tj | ttjd W d n1 s60 Y | ttj d S N %s.something unpatched)r r r
Something2 assertEqual something Somethingselfr r r test_with_statement s .z WithTest.test_with_statementc C s | tV tdt tj& | ttjd t W d n1 sH0 Y W d n1 sf0 Y | ttj d S r ) assertRaisesr
r r r r r r r r r r r test_with_statement_exception s
Bz&WithTest.test_with_statement_exceptionc C s^ t dt 0}| t|d | t|td W d n1 sB0 Y | ttj d S )Nr r zpatching wrong type) r r r r
assertTruer r r r r mock_somethingr r r test_with_statement_as s "z WithTest.test_with_statement_asc C sd G dd dt }|j}t |d | |j|d W d n1 sH0 Y | |j| d S )Nc @ s e Zd ZdZdS )z6WithTest.test_patch_object_with_statement.
.FoofooN)r r r
r r r r r Foo) s r# r r )objectr r ZassertNotEqualr )r r# Zoriginalr r r test_patch_object_with_statement( s
.z)WithTest.test_patch_object_with_statementc
C s t dd tdt X}tdt ,}| t|d | t|d W d n1 sX0 Y W d n1 sv0 Y W d n1 s0 Y | ttj | ttj d S )NT)recordr %s.something_elser ) r r r r r something_elser r
SomethingElser r Zmock_something_elser r r test_with_statement_nested1 s ^z#WithTest.test_with_statement_nestedc C s` t dt tj.}| t|d | |tjd W d n1 sD0 Y | ttj d S )Nr r z wrong patchr r r ZPatchedr r r r r r r test_with_statement_specified< s .z&WithTest.test_with_statement_specifiedc C sp t }t |_t |_d|j_| }| ||jj W d n1 sH0 Y |j |jd d d d S )NF)r __enter____exit__ return_valuer assert_called_with)r mockmr r r testContextManagerMockingC s .
z"WithTest.testContextManagerMockingc C sx t }| t< | dd W d n1 s40 Y W d n1 sR0 Y |j | |jj d S )Nr" )r r TypeErrorr. r1 r r/ Zcalled)r r2 r r r $test_context_manager_with_magic_mockO s D
z-WithTest.test_context_manager_with_magic_mockc C s t dt tjf}| t|d t dt }| t|d W d n1 sP0 Y | t|d W d n1 s|0 Y | ttjd d S )Nr r z restored with wrong instancez not restoredr, )r r Z
mock_againr r r "test_with_statement_same_attributeY s ,"z+WithTest.test_with_statement_same_attributec C s t dt X}| t|d t dt }| t|d W d n1 sL0 Y W d n1 sj0 Y | ttj | ttj d S )Nr r r' )r r r r r( r r r) r* r r r test_with_statement_imbricatedf s @z'WithTest.test_with_statement_imbricatedc C s i }t |ddi | |ddi W d n1 s:0 Y | |i | tX t |ddi( | |ddi tdW d n1 s0 Y W d n1 s0 Y | |i d S )NabZKonrad)r dictr r NameError)r r" r r r test_dict_context_managerr s . Dz"WithTest.test_dict_context_managerc C s G dd d}| }t j|dddl}t j|ddd}| W d n1 sT0 Y | |jd | |jd | W d n1 s0 Y | |jd d S )Nc @ s e Zd Zdd ZdS )z5WithTest.test_double_patch_instance_method..Cc S s d S )Nr r r r r f r z7WithTest.test_double_patch_instance_method..C.fN)r r r
r? r r r r C s r@ r? T)Zautospec r )r r$ r? r Z
call_count)r r@ cZpatch1Zpatch2r r r !test_double_patch_instance_method s &&z*WithTest.test_double_patch_instance_methodN)r r r
r r r! r% r+ r- r4 r7 r8 r9 r> rC r r r r r s
r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd S ) TestMockOpenc C sZ t }tdt |dd$}| || td W d n1 sB0 Y |d d S N%s.openTZcreater" )r r r assertIsopenassert_called_once_withr r2 Zpatchedr r r test_mock_open s
&zTestMockOpen.test_mock_openc C s t }|j}tdt |dd@ td}| W d n1 sF0 Y W d n1 sd0 Y tdt t t d d d g}| |j
| | || d S rE ) r r0 r r rI readr r. r/ r
mock_callsrH )r r2 handler? expected_callsr r r test_mock_open_context_manager s
Dz+TestMockOpen.test_mock_open_context_managerc C s t }tdt |ddp td}| W d n1 s@0 Y td}| W d n1 sp0 Y W d n1 s0 Y tdt t t d d d tdt t t d d d g}| |j | d S )NrF TrG r" bar)
r r r rI rM r r. r/ r rN )r r2 r? rP r r r -test_mock_open_context_manager_multiple_times s
&
Dz:TestMockOpen.test_mock_open_context_manager_multiple_timesc C sb t }t| tdt |dd$}| || td W d n1 sJ0 Y |d d S rE )r r r r rH rI rJ rK r r r test_explicit_mock s &z TestMockOpen.test_explicit_mockc C s\ t dd}tdt |dd td}| }W d n1 sB0 Y | |d d S )Nr" Z read_datarF TrG rR r r r rI rM r r r2 hresultr r r test_read_data s
&zTestMockOpen.test_read_datac C s t dd}tdt |dd0 td}| }| }| }W d n1 sR0 Y | |d | |d | |d | | d
t d d}tdt |dd td}| }W d n1 s0 Y | |d | | d
d S ) N foo
bar
baz
rU rF TrG rR foo
bar
baz
r" r r r rI readliner )r r2 rX line1line2line3rY r r r test_readline_data s
&
& z TestMockOpen.test_readline_datac C s t dd}tdt |dd& td}dd |D }W d n1 sH0 Y | |d d
| |d d | |d
d | | d | t t| W d n1 s0 Y d S )Nr[ rU rF TrG rR c S s g | ]}|qS r r .0lr r r
r z6TestMockOpen.test_dunder_iter_data..r r\ rA r] r^ r_ ) r r r rI r ra r
StopIterationnext)r r2 rX linesr r r test_dunder_iter_data s
, z"TestMockOpen.test_dunder_iter_datac C s t dd}tdt |dd6 td}t|}t|}dd |D }W d n1 sX0 Y | |d | |d
| |d d | | d
d S )Nr[ rU rF TrG rR c S s g | ]}|qS r r rf r r r ri r z/TestMockOpen.test_next_data..r\ r] r r^ r_ )r r r rI rl r ra )r r2 rX rb rc rm r r r test_next_data s
, zTestMockOpen.test_next_datac C s t dd}tdt |dd td}| }W d n1 sB0 Y | |g d t dd}tdt |dd td}| }W d n1 s0 Y | |g d d S )
Nr[ rU rF TrG rR )r\ r] r^ foo
bar
baz)r\ r] Zbazr r r rI readlinesr rW r r r test_readlines_data s
&
&z TestMockOpen.test_readlines_datac C s~ t dd}tdt |ddB tdd}| }W d n1 sF0 Y W d n1 sd0 Y | |d d S )N rU rF TrG abcrbrV )r r2 r? rY r r r test_read_bytes s
Dz TestMockOpen.test_read_bytesc C s t dd}tdt |ddR tdd(}| }| }| }W d n1 sV0 Y W d n1 st0 Y | |d | |d | |d
d S ) N abc
def
ghi
rU rF TrG ru rv abc
def
ghi
r` )r r3 r? rb rc rd r r r test_readline_bytes s
D z TestMockOpen.test_readline_bytesc C s t dd}tdt |ddB tdd}| }W d n1 sF0 Y W d n1 sd0 Y | |g d d S ) Nrx rU rF TrG ru rv )ry rz r{ rq )r r3 r? rY r r r test_readlines_bytes s
Dz!TestMockOpen.test_readlines_bytesc C s d}t |d}| | d|d d | | d|d d | }| |d|d d | |d|dd d S )Nrp rU
)r r rM )r Z some_datar2 r? r r r !test_mock_open_read_with_argument( s
z.TestMockOpen.test_mock_open_read_with_argumentc C s t dd}tdt |dd( td}| }| }W d n1 sJ0 Y | |d | |dd g t dd}tdt |dd( td}| }| }W d n1 s0 Y | |d | |d
d S ) Nr[ rU rF TrG rR r\ r] r^ zbar
baz
)r r r rI ra rr r rM )r r2 rX rb restr r r test_interleaved_reads5 s
&
& z#TestMockOpen.test_interleaved_readsc C s t dd}| }d|j_d|j_dg|j_| | d | | d | | dg | | d | | d d S )Nr" rU rR )r rM r0 ra rr r )r r2 rO r r r test_overriding_return_valuesI s
z*TestMockOpen.test_overriding_return_valuesN)r r r
rL rQ rS rT rZ re rn ro rs rw r| r} r r r r r r r rD s
rD __main__)Zunittestwarningsr Z unittest.test.testmock.supportr Z
unittest.mockr r r r r r r r r) r( Exceptionr
ZTestCaser rD r mainr r r r s ~ M