2*V[) @ s d d l m Z d d l Z d d l m Z d d l m Z m Z m Z m
Z
m Z m Z e
j
Z e
j Z Gd d d e j Z Gd d d e j Z e d k r e j d S)
)catch_warningsN) is_instance) MagicMockMockpatchsentinel mock_opencallc @ s e Z d Z d d Z d d Z d d Z d d Z d d
Z d d Z d
d Z d d Z
d d Z d d Z d d Z
d S)WithTestc C sK t d t t j | j t t j d Wd QRX| j t t j d S)Nz %s.something unpatched)r __name__r
Something2 assertEqual something Something)self r /testwith.pytest_with_statement s z WithTest.test_with_statementc C s} yD t d t t j ' | j t t j d t d Wd QRXWn t k
rX Yn X| j d | j t t j d S)Nz %s.somethingr powzpatch swallowed exception) r r r r
r r ExceptionZfailr )r r r r test_with_statement_exception s
z&WithTest.test_with_statement_exceptionc C s] t d t 3 } | j t | d | j t | t d Wd QRX| j t t j d S)Nz %s.somethingr 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 sf Gd d d t } | j } t j | d | j | j | d Wd QRX| j | j | d S)Nc @ s e Z d Z d Z d S)z6WithTest.test_patch_object_with_statement.<locals>.FoofooN)r
__module__ __qualname__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 d d _ t d t G } t d t - } | j t | d | j t | d Wd QRXWd QRXWd QRX| j t t j | j t t j d S)NrecordTz %s.somethingz%s.something_elser ) r r r r r something_elser r
SomethingElse)r r mock_something_elser r r test_with_statement_nested7 s z#WithTest.test_with_statement_nestedc C s` t d t t j 0 } | j t | d | j | t j d Wd QRX| j t t j d S)Nz %s.somethingr z wrong patch)r r r Patchedr r r )r r r r r test_with_statement_specifiedC s z&WithTest.test_with_statement_specifiedc C sz t } t | _ t | _ d | j _ | } | j | | j j Wd QRX| j j | j j d d d d S)NF)r __enter____exit__ return_valuer assert_called_with)r mockmr r r testContextManagerMockingJ s
z"WithTest.testContextManagerMockingc C sZ t } | j t |
d d Wd QRXWd QRX| j j | j | j j d S)Nr )r assertRaises TypeErrorr( r+ r r) Zcalled)r r, r r r $test_context_manager_with_magic_mockV s
z-WithTest.test_context_manager_with_magic_mockc C s t d t t j Z } | j t | d t d t } | j t | d Wd QRX| j t | d Wd QRX| j t t j d d S)Nz %s.somethingr z restored with wrong instancez not restored)r r r r& r r r )r r Z
mock_againr r r "test_with_statement_same_attribute` s z+WithTest.test_with_statement_same_attributec C s t d t G } | j t | d t d t } | j t | d Wd QRXWd QRX| j t t j | j t t j d S)Nz %s.somethingr z%s.something_else)r r r r r" r r r# )r r r$ r r r test_with_statement_imbricatedm s z'WithTest.test_with_statement_imbricatedc C s i } t j | d d i | j | d d i Wd QRX| j | i | j t G t j | d d i ' | j | d d i t d Wd QRXWd QRX| j | i d S)NabZKonrad)r dictr r0 NameError)r r r r r test_dict_context_managery s z"WithTest.test_dict_context_managerN)r r r r r r r r% r' r. r2 r3 r4 r9 r r r r r
s
r
c @ s e Z d Z d d Z d d Z d d Z d d Z d d
Z d d Z d
d Z d d Z
d d Z d d Z d d Z
d d Z d d Z d S) TestMockOpenc
C sW t } t d t | d d ! } | j | | t d Wd QRX| j d d S)Nz%s.opencreateTr )r r r assertIsopenassert_called_once_with)r r, patchedr r r test_mock_open s
zTestMockOpen.test_mock_openc C s t } | j } t d t | d d % t d } | j Wd QRXWd QRXt d t j t j t j d d d g } | j | j
| | j | | d S)Nz%s.openr; Tr ) r r* r r r= readr r( r) r
mock_callsr< )r r, handlefexpected_callsr r r test_mock_open_context_manager s !z+TestMockOpen.test_mock_open_context_managerc C s t } t d t | d d E t d } | j Wd QRXt d } | j Wd QRXWd QRXt d t j t j t j d d d t d t j t j t j d d d g } | j | j | d S)Nz%s.openr; Tr bar)
r r r r= rA r r( r) r rB )r r, rD rE r r r -test_mock_open_context_manager_multiple_times s !!z:TestMockOpen.test_mock_open_context_manager_multiple_timesc
C sa t } t | t d t | d d ! } | j | | t d Wd QRX| j d d S)Nz%s.openr; Tr )r r r r r< r= r> )r r, r? r r r test_explicit_mock s
z TestMockOpen.test_explicit_mockc
C s\ t d d } t d t | d d t d } | j } Wd QRX| j | d d S)N read_datar z%s.openr; TrG )r r r r= rA r )r r, hresultr r r test_read_data s
zTestMockOpen.test_read_datac C s t d d } t d t | d d 5 t d } | j } | j } | j } Wd QRX| j | d | j | d | j | d t d d
} t d t | d d t d } | j } Wd QRX| j | d
d S) NrJ z foo
bar
baz
z%s.openr; TrG zfoo
zbar
zbaz
r )r r r r= readliner )r r, rK line1line2line3rL r r r test_readline_data s z TestMockOpen.test_readline_datac C s t d d } t d t | d d t d } | j } Wd QRX| j | d d d g t d d
} t d t | d d t d } | j } Wd QRX| j | d d d g d S) NrJ z foo
bar
baz
z%s.openr; TrG zfoo
zbar
zbaz
z foo
bar
bazZbaz)r r r r= readlinesr )r r, rK rL r r r test_readlines_data s z TestMockOpen.test_readlines_datac C si t d d } t d t | d d * t d d } | j } Wd QRXWd QRX| j | d d S)NrJ s z%s.openr; Tabcrb)r r r r= rA r )r r, rD rL r r r test_read_bytes s
z TestMockOpen.test_read_bytesc C s t d d } t d t | d d B t d d + } | j } | j } | j } Wd QRXWd QRX| j | d | j | d | j | d
d S) NrJ s abc
def
ghi
z%s.openr; TrU rV s abc
s def
s ghi
)r r r r= rN r )r r- rD rO rP rQ r r r test_readline_bytes s z TestMockOpen.test_readline_bytesc C sr t d d } t d t | d d * t d d } | j } Wd QRXWd QRX| j | d d d
g d S) NrJ s abc
def
ghi
z%s.openr; TrU rV s abc
s def
s ghi
)r r r r= rS r )r r- rD rL r r r test_readlines_bytes s
z!TestMockOpen.test_readlines_bytesc C s5 d } t d | } | j | j d | d S)Nz foo
bar
bazrJ
)r r rA )r Z some_datar, r r r !test_mock_open_read_with_argument s z.TestMockOpen.test_mock_open_read_with_argumentc C s t d d } t d t | d d ) t d } | j } | j } Wd QRX| j | d | j | d d g t d d } t d t | d d ) t d } | j } | j } Wd QRX| j | d | j | d
d S) NrJ z foo
bar
baz
z%s.openr; TrG zfoo
zbar
zbaz
zbar
baz
)r r r r= rN rS r rA )r r, rK rO restr r r test_interleaved_reads s z#TestMockOpen.test_interleaved_readsc C s t d d } | } d | j _ d | j _ d g | j _ | j | j d | j | j d | j | j d g | j | j d | j | j d d S)NrJ r rG )r rA r* rN rS r )r r, rC r r r test_overriding_return_values s z*TestMockOpen.test_overriding_return_valuesN)r r r r@ rF rH rI rM rR rT rW rX rY r[ r] r^ r r r r r: s r: __main__)warningsr Z unittest2ZunittestZmock.tests.supportr r, r r r r r r r r r# r" ZTestCaser
r: r mainr r r r <module> s . v