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 � /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 � 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>.Foo�fooN)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)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) 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)N�a�bZKonrad)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.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 r= �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 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, �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) NrJ z foo
bar
baz
z%s.openr; TrG zfoo
zbar
zbaz
r )r r r r= �readliner )r r, rK �line1�line2�line3rL 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; T�abc�rb)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� | Name | Type | Size | Permission | Actions |
|---|---|---|---|---|
| __init__.cpython-35.opt-1.pyc | File | 95 B | 0644 |
|
| __init__.cpython-35.pyc | File | 95 B | 0644 |
|
| __main__.cpython-35.opt-1.pyc | File | 569 B | 0644 |
|
| __main__.cpython-35.pyc | File | 569 B | 0644 |
|
| support.cpython-35.opt-1.pyc | File | 1.17 KB | 0644 |
|
| support.cpython-35.pyc | File | 1.17 KB | 0644 |
|
| testcallable.cpython-35.opt-1.pyc | File | 5.98 KB | 0644 |
|
| testcallable.cpython-35.pyc | File | 5.98 KB | 0644 |
|
| testhelpers.cpython-35.opt-1.pyc | File | 36.11 KB | 0644 |
|
| testhelpers.cpython-35.pyc | File | 36.11 KB | 0644 |
|
| testmagicmethods.cpython-35.opt-1.pyc | File | 22.61 KB | 0644 |
|
| testmagicmethods.cpython-35.pyc | File | 22.61 KB | 0644 |
|
| testmock.cpython-35.opt-1.pyc | File | 49.25 KB | 0644 |
|
| testmock.cpython-35.pyc | File | 49.25 KB | 0644 |
|
| testpatch.cpython-35.opt-1.pyc | File | 71.85 KB | 0644 |
|
| testpatch.cpython-35.pyc | File | 71.85 KB | 0644 |
|
| testsentinel.cpython-35.opt-1.pyc | File | 1.38 KB | 0644 |
|
| testsentinel.cpython-35.pyc | File | 1.38 KB | 0644 |
|
| testwith.cpython-35.opt-1.pyc | File | 10.14 KB | 0644 |
|
| testwith.cpython-35.pyc | File | 10.14 KB | 0644 |
|