Current File : //usr/local/lib/python3.8/unittest/test/testmock/__pycache__/testwith.cpython-38.pyc
U
qa/ @ 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 ;/usr/local/lib/python3.8/unittest/test/testmock/testwith.pyr
s 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 s> t dt tj | ttjd W 5 Q R X | 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 sZ | t8 tdt tj | ttjd t W 5 Q R X W 5 Q R X | ttj d S r ) assertRaisesr
r r r r r r r r r r r test_with_statement_exception s
z&WithTest.test_with_statement_exceptionc C sJ t dt &}| t|d | t|td W 5 Q R X | 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 sP G dd dt }|j}t |d | |j|d W 5 Q R X | |j| d S )Nc @ s e Zd ZdZdS )z6WithTest.test_patch_object_with_statement.<locals>.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 ddR tdt :}tdt "}| t|d | t|d W 5 Q R X W 5 Q R X W 5 Q R X | 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 sL t dt tj$}| t|d | |tjd W 5 Q R X | 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 s\ t }t |_t |_d|j_|}| ||jj W 5 Q R X |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 sP t }| t | dd W 5 Q R X W 5 Q R X |j | |jj d S )Nr! )r r TypeErrorr- r0 r r. Zcalled)r r1 r r r $test_context_manager_with_magic_mockO s
z-WithTest.test_context_manager_with_magic_mockc
C sr t dt tjH}| t|d t dt }| t|d W 5 Q R X | t|d W 5 Q R X | 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 sl t dt :}| t|d t dt }| t|d W 5 Q R X W 5 Q R X | 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 5 Q R X | |i | t: t |ddi | |ddi tdW 5 Q R X W 5 Q R X | |i d S )NabZKonrad)r dictr r NameError)r r! r r r test_dict_context_managerr s z"WithTest.test_dict_context_managerc
C s G dd d}| }t j|dddN}t j|ddd}| W 5 Q R X | |jd | |jd | W 5 Q R X | |jd d S )Nc @ s e Zd Zdd ZdS )z5WithTest.test_double_patch_instance_method.<locals>.Cc S s d S )Nr r r r r f z7WithTest.test_double_patch_instance_method.<locals>.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, r3 r6 r7 r8 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 sF t }tdt |dd}| || td W 5 Q R X |d d S N%s.openTZcreater! )r r r assertIsopenassert_called_once_withr r1 Zpatchedr r r test_mock_open s
zTestMockOpen.test_mock_openc
C s t }|j}tdt |dd" td}| W 5 Q R X W 5 Q R X tdt t t d d d g}| |j
| | || d S rE ) r r/ r r rI readr r- r. r
mock_callsrH )r r1 handler> expected_callsr r r test_mock_open_context_manager s
z+TestMockOpen.test_mock_open_context_managerc C s t }tdt |dd> td}| W 5 Q R X td}| W 5 Q R X W 5 Q R X 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 r1 r> rP r r r -test_mock_open_context_manager_multiple_times s
z:TestMockOpen.test_mock_open_context_manager_multiple_timesc C sN t }t| tdt |dd}| || td W 5 Q R X |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 sH t dd}tdt |dd td}| }W 5 Q R X | |d d S )Nr! Z read_datarF TrG rR r r r rI rM r r r1 hresultr r r test_read_data s
zTestMockOpen.test_read_datac C s t dd}tdt |dd& td}| }| }| }W 5 Q R X | |d | |d | |d | | d
t d d}tdt |dd td}| }W 5 Q R X | |d | | d
d S ) N foo
bar
baz
rU rF TrG rR foo
bar
baz
r! r r r rI readliner )r r1 rX line1line2line3rY r r r test_readline_data s
z TestMockOpen.test_readline_datac C s t dd}tdt |dd td}dd |D }W 5 Q R X | |d d
| |d d | |d
d | | d | t t| W 5 Q R X d S )Nr[ rU rF TrG rR c S s g | ]}|qS r r .0lr r r
<listcomp> s z6TestMockOpen.test_dunder_iter_data.<locals>.<listcomp>r r\ rA r] r^ r_ ) r r r rI r ra r
StopIterationnext)r r1 rX linesr r r test_dunder_iter_data s
z"TestMockOpen.test_dunder_iter_datac C s t dd}tdt |dd, td}t|}t|}dd |D }W 5 Q R X | |d | |d
| |d d | | d
d S )Nr[ rU rF TrG rR c S s g | ]}|qS r r rf r r r ri s z/TestMockOpen.test_next_data.<locals>.<listcomp>r\ r] r r^ r_ )r r r rI rl r ra )r r1 rX rb rc rm r r r test_next_data s
zTestMockOpen.test_next_datac C s t dd}tdt |dd td}| }W 5 Q R X | |ddd g t d
d}tdt |dd td}| }W 5 Q R X | |ddd g d S ) Nr[ rU rF TrG rR r\ r] r^ foo
bar
bazZbazr r r rI readlinesr rW r r r test_readlines_data s
z TestMockOpen.test_readlines_datac
C sV t dd}tdt |dd$ tdd}| }W 5 Q R X W 5 Q R X | |d d S )N rU rF TrG abcrbrV )r r1 r> rY r r r test_read_bytes s
z TestMockOpen.test_read_bytesc
C s~ t dd}tdt |dd4 tdd }| }| }| }W 5 Q R X W 5 Q R X | |d | |d | |d
d S Ns abc
def
ghi
rU rF TrG ru rv s abc
s def
s ghi
r` )r r2 r> rb rc rd r r r test_readline_bytes s
z TestMockOpen.test_readline_bytesc
C s\ t dd}tdt |dd$ tdd}| }W 5 Q R X W 5 Q R X | |dd d
g d S rx rq )r r2 r> rY r r r test_readlines_bytes s
z!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_datar1 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 5 Q R X | |d | |dd g t dd}tdt |dd td}| }| }W 5 Q R X | |d | |d
d S ) Nr[ rU rF TrG rR r\ r] r^ zbar
baz
)r r r rI ra rr r rM )r r1 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 r/ ra rr r )r r1 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 ry rz 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 <module> s ~ M
Mini Shell