Your IP : 216.73.216.213
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)� MagicMock�Mock�patch�sentinel� mock_open�callc @ 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 �D/opt/alt/python35/lib/python3.5/site-packages/mock/tests/testwith.py�test_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 � Exception�failr )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>.Foo�fooN)r �
__module__� __qualname__r r r r r �Foo/ s r r r )�objectr r �assertNotEqualr )r r �originalr 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)N�recordTz %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 �mock�mr 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, �called)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)N�a�bZKonrad)r �dictr r3 � 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* r1 r6 r7 r8 r= 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.open�createTr )r r r �assertIs�open�assert_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 rA �readr r+ r, r �
mock_callsr@ )r r/ �handle�f�expected_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 rA rE r r+ r, r rF )r r/ rH rI 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@ rA rB )r r/ rC 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? TrK )r r r rA rE r )r r/ �h�resultr 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) NrN z foo
bar
baz
z%s.openr? TrK zfoo
zbar
zbaz
r )r r r rA �readliner )r r/ rO �line1�line2�line3rP 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) NrN z foo
bar
baz
z%s.openr? TrK zfoo
zbar
zbaz
z foo
bar
bazZbaz)r r r rA � readlinesr )r r/ rO rP 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)NrN s �z%s.openr? T�abc�rb)r r r rA rE r )r r/ rH rP 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) NrN s abc
def
ghi
z%s.openr? TrY rZ s abc
s def
s ghi
)r r r rA rR r )r r0 rH rS rT rU 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) NrN s abc
def
ghi
z%s.openr? TrY rZ s abc
s def
s ghi
)r r r rA rW r )r r0 rH rP 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
bazrN �
)r r rE )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) NrN z foo
bar
baz
z%s.openr? TrK zfoo
zbar
zbaz
zbar
baz
)r r r rA rR rW r rE )r r/ rO rS �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)NrN r rK )r rE r- rR rW r )r r/ rG r r r � test_overriding_return_values s z*TestMockOpen.test_overriding_return_valuesN)r r r rD rJ rL rM rQ rV rX r[ r\ r] r_ ra rb r r r r r> � s r> �__main__)�warningsr Z unittest2�unittestZmock.tests.supportr r/ r r r r r r r r r& r% �TestCaser
r> r �mainr r r r �<module> s . v�