Your IP : 216.73.216.213
���]� � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l Z Wn e k
r~ d Z Yn Xy d d l Z Wn e k
r� d Z Yn Xd d l
m Z m Z d d l m
Z
m Z d d l m Z m Z m Z m Z d d l m Z d d l m Z d d d � Z d
d � Z Gd d
� d
� Z Gd d � d e j � Z Gd d � d � Z Gd d � d e j � Z Gd d � d � Z Gd d � d e j � Z Gd d � d e � Z Gd d � d e � Z! Gd d � d e � Z" Gd d � d e � Z# Gd d! � d! e � Z$ Gd" d# � d# e � Z% Gd$ d% � d% e � Z& Gd& d' � d' e � Z' Gd( d) � d) e � Z( Gd* d+ � d+ � Z) Gd, d- � d- e j � Z* Gd. d/ � d/ e j � Z+ e, d0 k r�e j- � d S)1� N)�BytesIO�StringIO)� FileInput� hook_encoded)�verbose�TESTFN� run_unittest�check_warnings)�unlink)�mock�wc C sK t t | � } t | | � } x | D] } | j | � q& W| j � | S)N)r �str�open�write�close)�i�lines�mode�name�f�line� r �3/opt/alt/python35/lib64/python3.5/test_fileinput.py�writeTmp$ s
r c G s% x | D] } | r t | � q Wd S)N)� safe_unlink)�namesr r r r �remove_tempfiles, s
r c @ sd e Z d Z d d � Z e d d � � Z d d � Z d d d � Z d d d � Z d
d � Z d S)�
LineReaderc C s
g | _ d S)N)�
_linesread)�selfr r r �__init__3 s zLineReader.__init__c C s& z | j d d � SWd g | _ Xd S)N)r )r r r r � linesread6 s zLineReader.linesreadc C s t | j d � � | _ | S)NT)�iter�
splitlines�it)r �filenamer r r r �openhook= s zLineReader.openhookNc C s& t | j d � } | j j | � | S)N� )�nextr$ r �append)r �sizer r r r �readlineA s zLineReader.readline� c C sZ g } d } xG | j � } | s% | S| j | � | t | � 7} | | k r | Sq Wd S)Nr )r+ r) �len)r Zhintr r* r r r r � readlinesF s
zLineReader.readlinesc C s d S)Nr )r r r r r R s zLineReader.close���)
�__name__�
__module__� __qualname__r �propertyr! r& r+ r. r r r r r r 1 s r c @ s. e Z d Z d d � Z d d d d � Z d S)�BufferSizesTestsc C s� x� d D]� \ } } d } } } } z� t d d d � t d � D� � } t d d d � t d � D� � } t d
d d � t d � D� � } t d
d d � t d � D� � } | j | | | | | | � Wd t | | | | � Xq Wd S)Nr r, � c S s g | ] } d | d � q S)zLine %s of file 1
r, r )�.0r r r r �
<listcomp>[ s z6BufferSizesTests.test_buffer_sizes.<locals>.<listcomp>� � c S s g | ] } d | d � q S)zLine %s of file 2
r, r )r6 r r r r r7 \ s �
� c S s g | ] } d | d � q S)zLine %s of file 3
r, r )r6 r r r r r7 ] s � � c S s g | ] } d | d � q S)zLine %s of file 4
r, r )r6 r r r r r7 ^ s �r r �r, r5 )r> r? )r �range�buffer_size_testr )r �round�bs�t1�t2�t3�t4r r r �test_buffer_sizesV s """" z"BufferSizesTests.test_buffer_sizesr c C s� t j d � } d | d } t r; t d | d | f � t d | | | | f d | � } t | � }
| j � | j t |
� d � | j |
d d
� | j |
d d � | j | j � d � | j | j
� | � t r� t d
| d | f � t d | | | | f d | � } d } x" | rE| d k rE| j � } q$W| j | j
� | � | j | j � d � | j | j � d � | j
| j � � | j
| j � � t r�t d | d | f � | j � | j | j � d � | j | j � d � | j � t r*t d | d | f � t d | | | | d f d | � } t j } zi t d � t _ t | � }
| j t |
� d � | j |
d d � | j | j
� d � | j � Wd | t _ Xt r�t d | d | f � t d | | | | f d | � } | j | j � d � | j | j
� d � | j � | j | j � d � | j | j
� d � t r�t d | d | f � t j }
zf t d | | | | f d d d | � } x. | D]& } | d d" � j � } t | � q�W| j � Wd |
t _ Xt d | | | | f d | � } xp | D]h } | j | d# d! � | j | d d$ � � } | j | d � | j t | j d � � | j � � q0W| j � d S)%NzLINE (\d+) OF FILE (\d+)r, � z %s. Simple iteration (bs=%s)r �files�bufsize� r= zLine 5 of file 1
r5 zLine 1 of file 4
z %s. Status variables (bs=%s)�xzLine 6 of file 2
� z%s. Nextfile (bs=%s)r9 zLine 1 of file 3
� z%s. Stdin (bs=%s)r; �-z Line 1 of stdin
Line 2 of stdin
�! � zLine 2 of stdin
z<stdin>z %s. Boundary conditions (bs=%s)z%s. Inplace (bs=%s)r< �inplace�
r/ r/ r/ )�re�compiler �printr �listr � assertEqualr- �linenor% r+ �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)r rD rE rF rG rC rB �pat�start�fir �sZ savestdinZ
savestdoutr �mr r r rA c s� !
!
$
!
'
!
)z!BufferSizesTests.buffer_size_testN)r0 r1 r2 rH rA r r r r r4 U s
r4 c @ s( e Z d Z d d � Z d d � Z d S)�UnconditionallyRaisec C s | | _ d | _ d S)NF)�exception_type�invoked)r rn r r r r � s z UnconditionallyRaise.__init__c O s d | _ | j � � d S)NT)ro rn )r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N)r0 r1 r2 r rr r r r r rm � s rm 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 d � Z d d � Z d d � Z d! d" � Z e j e e d# � d$ � d% d& � � Z d' d( � Z d) d* � Z d+ d, � Z d- S).�FileInputTestsc C sW d } } } } z*t d d g � } t d d g � } t d d g � } t d d g � } t d | | | | f � } | j � } | j | d � | j | j � d � | j | j � d � | j | j � | � | j � } | j | � | j | j � d � | j | j � d � | j | j � | � | j � Wd t | | | | � Xd S) Nr, r' r9 r; zThe only line there is.
r= rJ r )
r r r+ rY rZ r[ r% r\ r r )r rD rE rF rG rj r r r r �test_zero_byte_files� s&
z#FileInputTests.test_zero_byte_filesc C s� d } } z� t d d g � } t d d g � } t d | | f � } t | � } | j | d d d d d
d g � | j | j � d � | j | j � d
� Wd t | | � Xd S)Nr, zA
B
Cr9 zD
E
FrJ zA
zB
�CzD
zE
�Fr; rI )r r rX rY r[ rZ r )r rD rE rj r r r r �%test_files_that_dont_end_with_newline� s
"z4FileInputTests.test_files_that_dont_end_with_newlinec C s� d } } z� t d d g � } t d d g � } t d | | f � } | j | j � d � t | � } | j | j � d � | j � | j | j � d � t | � } | j | j � d � Wd t | | � Xd S)
Nr, zA
Br9 zC
DrJ r/ r/ r/ r/ ) r r rY �filenor( re r_ rX r )r rD rE rj r r r r � test_fileno� s
zFileInputTests.test_filenoc " C s� y t d d � } | j d � Wn t k
r4 Yn Xd } z� t d d g d d �} t d t f � � t d | d d � } Wd QRXt d t f � � t | � } Wd QRX| j | d
d d d
g � Wd t | � Xd S)Nr r z-FileInput should reject invalid mode argumentr, s A
B
C
D�wbr' rJ �UzA
zB
zC
�D) r �fail�
ValueErrorr r �DeprecationWarningrX rY r )r rj rD r r r r �test_opening_mode� s
z FileInputTests.test_opening_modec C sc t j d � �M } t d � | _ t d d g d d � } t | � } | j | d g � Wd QRXd S)Nz sys.stdins spam, bacon, sausage, and spamrJ rP r �rb)r Zpatchr �bufferr rX rY )r Zm_stdinrj r r r r �test_stdin_binary_mode s
z%FileInputTests.test_stdin_binary_modec C s} t j } zc t d � t _ | j t t j d � � t d d g d d � } t | � } | j | d g � Wd | t _ Xd S)Ns spam, bacon, sausage, and spamr� rJ rP r r� )r` ra r r\ �hasattrr rX rY )r Z
orig_stdinrj r r r r � test_detached_stdin_binary_mode s z.FileInputTests.test_detached_stdin_binary_modec C s� y, t d d d d d � � } | j d � Wn t k
r@ Yn Xy t d d � } | j d � Wn t k
ru Yn XGd d � d � } t d d
g � } | j t | � | � } t | g d | �� } | j � Wd QRX| j | j d � d S) NrS r, r& c S s d S)Nr )r rl r r r �<lambda>% s z7FileInputTests.test_file_opening_hook.<locals>.<lambda>zGFileInput should raise if both inplace and openhook arguments are givenz2FileInput should check openhook for being callablec @ s( e Z d Z d d � Z d d � Z d S)z=FileInputTests.test_file_opening_hook.<locals>.CustomOpenHookc S s
d | _ d S)NF)ro )r r r r r 1 s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c W s d | _ t | � S)NT)ro r )r rp r r r rr 3 s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__N)r0 r1 r2 r rr r r r r �CustomOpenHook0 s r� rT zopenhook not invoked) r r} r~ r �
addCleanupr r+ �
assertTruero )r rj r� �tZcustom_open_hookr r r �test_file_opening_hook" s"
z%FileInputTests.test_file_opening_hookc C s2 t t d � �2 } | j d � | j d d � | j d � Wd QRX| j t t � t d t d t d � � �� } yF | j | j � d � | j | j � d
� | j | j � d � Wn t k
r� | j
d � Yn X| j t � � t | � Wd QRX| j | j � d
� | j | j � d
� Wd QRXd S)Nrz s A
B
C
s
123456789
i� s �rJ r& �asciizA
zB
zC
zRead to end of filer' )
r r r r� r r r rY r+ �UnicodeDecodeErrorr} � assertRaisesrX )r r rj r r r �
test_readline> s"
z FileInputTests.test_readlinec C s� t t d � � } | j d � Wd QRX| j t t � t d t d d � �u } | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � Wd QRXd S)
Nrz s A
B
C
DrJ r r� s A
s B
s C
D� )r r r r� r r rY r+ )r r rj r r r �test_readline_binary_modeU s z(FileInputTests.test_readline_binary_modec C s� z� t d d g � } t d d g � } t d | | f � � } t | � } Wd QRX| j | d d d d d
d g � | j | j � d � | j | j � d
� | j | j f � Wd t | | � Xd S)Nr, zA
B
Cr9 zD
E
FrJ zA
zB
ru zD
zE
rv r; rI )r r rX rY r[ rZ �_filesr )r rD rE rj r r r r �test_context_managerb s "z#FileInputTests.test_context_managerc C ss za y5 t d d g � } t d | � �
} t � Wd QRXWn% t k
r_ | j | j f � Yn XWd t | � Xd S)Nr, r' rJ )r r �OSErrorrY r� r )r rD rj r r r �test_close_on_exceptiono s
z&FileInputTests.test_close_on_exceptionc
C s0 t d g � � } | j | j d � Wd QRXd S)NrJ rP )rP )r rY r� )r rj r r r �.test_empty_files_list_specified_to_constructory s z=FileInputTests.test_empty_files_list_specified_to_constructorc
C sy t d d d g � } | j t | � t d | g � �; } | d } | j | d � | d } | j | d � Wd QRXd S)Nr, zline1
zline2
rJ r )r r� r r rY )r r� rj Zretval1Zretval2r r r �test__getitem__} s
z FileInputTests.test__getitem__c C s| t d d d g � } | j t | � t d | g � �( } | j t � � } | d Wd QRXWd QRX| j | j j d � d S)Nr, zline1
zline2
rJ � accessing lines out of order)r� ) r r� r r r� � RuntimeErrorrY � exceptionrp )r r� rj �cmr r r �test__getitem__invalid_key� s z)FileInputTests.test__getitem__invalid_keyc C sv t d g � } | j t | � t d | g � �( } | j t � � } | d Wd QRXWd QRX| j | j j d � d S)Nr, rJ r �end of input reached)r� ) r r� r r r� �
IndexErrorrY r� rp )r r� rj r� r r r �test__getitem__eof� s z!FileInputTests.test__getitem__eofc C s� t j } t t � } ze t d d g � } | j t | � t d | g d d � �$ } t | � | t _ | j � Wd QRXWd | t _ X| j
| j d � d S)Nr, rT rJ rS Tzos.unlink() was not invoked) �osr
rm r� r r� r r r( r_ r� ro )r Zos_unlink_origZos_unlink_replacementr� rj r r r �%test_nextfile_oserror_deleting_backup� s
z4FileInputTests.test_nextfile_oserror_deleting_backupc C s� t j } t t � } z[ t d d g � } | j t | � t d | g d d � � } | t _ | j � Wd QRXWd | t _ X| j | j
d � d S)Nr, rT rJ rS Tzos.fstat() was not invoked) r� �fstatrm r� r r� r r r+ r� ro )r Z
os_fstat_origZos_fstat_replacementr� rj r r r �%test_readline_os_fstat_raises_OSError� s
z4FileInputTests.test_readline_os_fstat_raises_OSError�chmodzos.chmod does not existc C s� t j } t t � } z[ t d d g � } | j t | � t d | g d d � � } | t _ | j � Wd QRXWd | t _ X| j | j
d � d S)Nr, rT rJ rS Tzos.fstat() was not invoked) r� r� rm r� r r� r r r+ r� ro )r Z
os_chmod_origZos_chmod_replacementr� rj r r r �%test_readline_os_chmod_raises_OSError� s
z4FileInputTests.test_readline_os_chmod_raises_OSErrorc C s� Gd d � d t � } | � } t d d g � } | j t | � t d | g � �6 } | j } z | | _ | j � } Wd | | _ XWd QRX| j | j d � | j | d d � d S) Nc @ s( e Z d Z d d � Z d d � Z d S)zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S s t j | t � d S)N)rm r r~ )r r r r r � s zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | j � d S)N)rr )r r r r rx � s zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r0 r1 r2 r rx r r r r �FilenoRaisesValueError� s r� r, rT rJ z _file.fileno() was not invokedzfileno() should return -1r/ )
rm r r� r r �_filerx r� ro rY )r r� Z unconditionally_raise_ValueErrorr� rj Z file_backup�resultr r r �"test_fileno_when_ValueError_raised� s z1FileInputTests.test_fileno_when_ValueError_raisedc C s$ t � } t d d d g d | j � �� } | j | j g � | j | j � d � | j | j d g � | j | j � d � | j | j d g � | j | j � d � | j | j d d g � | j | j � d � | j | j d g � | j | j � d � | j | j g � Wd QRXd S)NrJ z line1
line2zline3
r& zline1
�line2r' )r r r& rY r! r+ )r �srcrj r r r �test_readline_buffering� s z&FileInputTests.test_readline_bufferingc C s t � } t d d d g d | j � �� } | j | j g � | j t | � d � | j | j d g � | j t | � d � | j | j d g � | j t | � d � | j | j d d g � | j t t | � | j | j d g � | j t t | � | j | j g � Wd QRXd S)NrJ z line1
line2zline3
r& zline1
r� r' )r r r& rY r! r( r� �
StopIteration)r r� rj r r r �test_iteration_buffering s z'FileInputTests.test_iteration_bufferingN) r0 r1 r2 rt rw ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr� r� r� r� r� r� r r r r rs � s*
( rs c @ s� e Z d Z d d d d d d 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)�
MockFileInputNFr' r �rc C sd | | _ | | _ | | _ | | _ | | _ | | _ d | _ t j d d � � | _ i | _
d S)Nc S s d S)Nr r r r r r r� s z(MockFileInput.__init__.<locals>.<lambda>) rJ rS �backuprK r r& r� � collections� defaultdict�invocation_counts�
return_values)r rJ rS r� rK r r& r r r r s zMockFileInput.__init__c C s | j d d 7<d S)Nr r, )r� )r r r r r s zMockFileInput.closec C s | j d d 7<| j d S)Nr_ r, )r� r� )r r r r r_ " s zMockFileInput.nextfilec C s | j d d 7<| j d S)Nr% r, )r� r� )r r r r r% & s zMockFileInput.filenamec C s | j d d 7<| j d S)NrZ r, )r� r� )r r r r rZ * s zMockFileInput.linenoc C s | j d d 7<| j d S)Nr[ r, )r� r� )r r r r r[ . s zMockFileInput.filelinenoc C s | j d d 7<| j d S)Nrx r, )r� r� )r r r r rx 2 s zMockFileInput.filenoc C s | j d d 7<| j d S)Nr] r, )r� r� )r r r r r] 6 s zMockFileInput.isfirstlinec C s | j d d 7<| j d S)Nr^ r, )r� r� )r r r r r^ : s zMockFileInput.isstdin) r0 r1 r2 r r r_ r% rZ r[ rx r] r^ r r r r r� s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� BaseFileInputGlobalMethodsTestc C s% t j | _ t j | _ t t _ d S)N)� fileinput�_state� _orig_stater �_orig_FileInputr� )r r r r �setUpB s z$BaseFileInputGlobalMethodsTest.setUpc C s | j t _ | j t _ d S)N)r� r� r r� r� )r r r r �tearDownG s z'BaseFileInputGlobalMethodsTest.tearDownc C sC | j | } | j | d | � t | j � } | j | d � d S)Nr, )r� rY r- )r Zmock_file_inputZ method_nameZ actual_countZactual_total_countr r r �assertExactlyOneInvocationK s
z9BaseFileInputGlobalMethodsTest.assertExactlyOneInvocationN)r0 r1 r2 r� r� 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 S)
�Test_fileinput_inputc C sq t � } t � | _ | t _ | j t � � } t j � Wd QRX| j d | j j
� | j | t j d � d S)N�input() already activezfileinput._state)r� ) r� �objectr� r� r� r� r� �inputrY r� rp �assertIs)r �instancer� r r r �1test_state_is_not_None_and_state_file_is_not_NoneV s zFTest_fileinput_input.test_state_is_not_None_and_state_file_is_not_Nonec C s) t � } d | _ | t _ | j � d S)N)r� r� r� r� �do_test_call_input)r r� r r r �-test_state_is_not_None_and_state_file_is_Nonec s zBTest_fileinput_input.test_state_is_not_None_and_state_file_is_Nonec C s d t _ | j � d S)N)r� r� r� )r r r r �test_state_is_Nonen s z'Test_fileinput_input.test_state_is_Nonec
C s t � } t � } t � } t � } t � } t � } t j d | d | d | d | d | d | � } | j | t j d � | j | | j d � | j | | j d � | j | | j d � | j | | j d � | j | | j d � | j | | j
d � d S)NrJ rS r� rK r r& zfileinput._state) r� r� r� r� r� rJ rS r� rK r r& )r rJ rS r� rK r r& r� r r r r� w s z'Test_fileinput_input.do_test_call_inputN)r0 r1 r2 r� r� r� r� r r r r r� S s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_closec C s' d t _ t j � | j t j � d S)N)r� r� r � assertIsNone)r r r r r� � s
z'Test_fileinput_close.test_state_is_Nonec C s@ t � } | t _ t j � | j | d � | j t j � d S)Nr )r� r� r� r r� r� )r r� r r r �test_state_is_not_None� s
z+Test_fileinput_close.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_nextfilec
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� r_ rY r� rp r� )r r� r r r r� � s
z*Test_fileinput_nextfile.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)Nr_ )r� r� r� r� r� r_ r� r� )r Znextfile_retvalr� �retvalr r r r� � s
z.Test_fileinput_nextfile.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� � s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_filenamec
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� r% rY r� rp r� )r r� r r r r� � s
z*Test_fileinput_filename.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)Nr% )r� r� r� r� r� r% r� r� )r Zfilename_retvalr� r� r r r r� � s
z.Test_fileinput_filename.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� � s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_linenoc
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� rZ rY r� rp r� )r r� r r r r� � s
z(Test_fileinput_lineno.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)NrZ )r� r� r� r� r� rZ r� r� )r Z
lineno_retvalr� r� r r r r� � s
z,Test_fileinput_lineno.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� � s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_filelinenoc
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� r[ rY r� rp r� )r r� r r r r� � s
z,Test_fileinput_filelineno.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)Nr[ )r� r� r� r� r� r[ r� r� )r Zfilelineno_retvalr� r� r r r r� s
z0Test_fileinput_filelineno.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� � s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_filenoc
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� rx rY r� rp r� )r r� r r r r� s
z(Test_fileinput_fileno.test_state_is_Nonec C st t � } t � } | | j d <| | _ | t _ t j � } | j | d � | j | | � | j t j | � d S)Nrx ) r� r� r� �
fileno_retvalr� r� rx r� r� )r r� r� r� r r r r� s
z,Test_fileinput_fileno.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_isfirstlinec
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� r] rY r� rp r� )r r� r r r r� 2 s
z-Test_fileinput_isfirstline.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)Nr] )r� r� r� r� r� r] r� r� )r Zisfirstline_retvalr� r� r r r r� < s
z1Test_fileinput_isfirstline.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� / s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_fileinput_isstdinc
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)N�no active input())r� ) r� r� r� r� r^ rY r� rp r� )r r� r r r r� M s
z)Test_fileinput_isstdin.test_state_is_Nonec C sk t � } t � } | | j d <| t _ t j � } | j | d � | j | | � | j t j | � d S)Nr^ )r� r� r� r� r� r^ r� r� )r Zisstdin_retvalr� r� r r r r� W s
z-Test_fileinput_isstdin.test_state_is_not_NoneN)r0 r1 r2 r� r� r r r r r� J s
r� c @ s( e Z d Z d d � Z d d � Z d S)�InvocationRecorderc C s
d | _ d S)Nr )�invocation_count)r r r r r f s zInvocationRecorder.__init__c O s" | j d 7_ | | f | _ d S)Nr, )r� �last_invocation)r rp rq r r r rr h s zInvocationRecorder.__call__N)r0 r1 r2 r rr r r r r r� e s r� c @ s� e Z d Z d d � Z d d � Z d d � Z e j e d � d d � � Z e j e
d
� d d � � Z d
d � Z d d � Z
d d � Z d d � Z e d d � � Z d S)�Test_hook_compressedc C s t � | _ d S)N)r� � fake_open)r r r r r� o s zTest_hook_compressed.setUpc C s | j d d � d S)Nr' r, )�do_test_use_builtin_open)r r r r �test_empty_stringr s z&Test_hook_compressed.test_empty_stringc C s | j d d � d S)NZabcdr9 )r� )r r r r � test_no_extu s z Test_hook_compressed.test_no_extzRequires gzip and zlibc C sn t j } | j t _ z t j d d � } Wd | t _ X| j | j j d � | j | j j d i f � d S)N�test.gzr; r, )r� r; )�gzipr r� r� �hook_compressedrY r� r� )r �
original_openr� r r r �test_gz_ext_fakex s
z%Test_hook_compressed.test_gz_ext_fakez Requires bz2c C sn t j } | j t _ z t j d d � } Wd | t _ X| j | j j d � | j | j j d i f � d S)N�test.bz2r= r, )r� r= )�bz2�BZ2Filer� r� r� rY r� r� )r r� r� r r r �test_bz2_ext_fake� s
z&Test_hook_compressed.test_bz2_ext_fakec C s | j d d � d S)Nz abcd.blahr< )r� )r r r r �
test_blah_ext� s z"Test_hook_compressed.test_blah_extc C s | j d d � d S)Nzabcd.GzrI )r� )r r r r �test_gz_ext_builtin� s z(Test_hook_compressed.test_gz_ext_builtinc C s | j d d � d S)Nzabcd.Bz2� )r� )r r r r �test_bz2_ext_builtin� s z)Test_hook_compressed.test_bz2_ext_builtinc C su | j | j � } z t j | | � } Wd | j | � X| j | j j d � | j | j j | | f i f � d S)Nr, )�replace_builtin_openr� r� r� rY r� r� )r r% r r� r� r r r r� � s z-Test_hook_compressed.do_test_use_builtin_openc C s t j } | t _ | S)N)�builtinsr )Z
new_open_funcr� r r r r� � s z)Test_hook_compressed.replace_builtin_openN)r0 r1 r2 r� r� r� r� Z
skipUnlessr� r� r� r� r� r� r� r� � staticmethodr� r r r r r� l s r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_hook_encodedc
C s� t � } t j | � } t � } t j } | t _ z% t � } t � } | | | � } Wd | t _ X| j | j d � | j \ } } | j | d | � | j | d | � | j | j
d � | � | j | � d S)Nr, r �encoding) r� r� r r� r� r rY r� r� r� �popr\ )
r r� r� r� r� r% r Z open_resultrp rq r r r �test� s
zTest_hook_encoded.testc " s� t t d � � } | j d � Wd QRX� j t t � � f d d � } | d d d d d g � � j t � � | d
d d d d g � Wd QRX� j t � � | d d d d d g � Wd QRX� j t � � | d d d
d d g � Wd QRXd S)Nrz s
A
B
C
D+IKw-c
sK t d t d | d t d � � � } t | � } Wd QRX� j | | � d S)NrJ r r& zutf-7)r r r rX rY )r Zexpected_linesrj r )r r r �check� s z+Test_hook_encoded.test_modes.<locals>.checkr� zA
zB
zC
u D€ZrUr{ r� zB
zC
) r r r r� r Z assertWarnsr r� r~ )r r r� r )r r �
test_modes� s z Test_hook_encoded.test_modesN)r0 r1 r2 r� r� r r r r r� � s r� �__main__).r� r` rU r� r� r� r� r� � ImportErrorr� �ior r r r Z test.supportr r r r r
r r r r r ZTestCaser4 rm rs r� r� r� r� r� r� r� r� r� r� r� r� r� r� r0 �mainr r r r �<module> sT
" $[� Y.A >.