Your IP : 216.73.216.213
��Yf� � @ s� d 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 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/ d0 � d0 e j � Z, e- d1 k r�e j. � d S)2z,
Tests for fileinput module.
Nick Mathewson
� 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 �8/opt/alt/python35/lib64/python3.5/test/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 Zpat�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 rm r r r r � s z UnconditionallyRaise.__init__c O s d | _ | j � � d S)NT)rn rm )r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N)r0 r1 r2 r rq r r r r rl � s rl 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 ri 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 ri 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 ri 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 ri 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_stdinri 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_stdinri 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 rk 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)rn )r r r r r 1 s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c W s d | _ t | � S)NT)rn r )r ro r r r rq 3 s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__N)r0 r1 r2 r rq r r r r �CustomOpenHook0 s r� rT zopenhook not invoked) r r| r} r �
addCleanupr r+ �
assertTruern )r ri 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)Nry 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 ri 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)
Nry s A
B
C
DrJ r r� s A
s B
s C
D� )r r r r� r r rY r+ )r r ri 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
rt zD
zE
ru r; rI )r r rX rY r[ rZ �_filesr )r rD rE ri 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 ri 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 ri 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)zNTests invoking FileInput.__getitem__() with the current
line numberr, zline1
zline2
rJ r N)r r� r r rY )r r� ri 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)zZTests invoking FileInput.__getitem__() with an index unequal to
the line numberr, zline1
zline2
rJ N� accessing lines out of order)r� ) r r� r r r� � RuntimeErrorrY � exceptionro )r r� ri �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)zZTests invoking FileInput.__getitem__() with the line number but at
end-of-inputr, rJ r N�end of input reached)r� ) r r� r r r� �
IndexErrorrY r� ro )r r� ri 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)z�Tests invoking FileInput.nextfile() when the attempt to delete
the backup file would raise OSError. This error is expected to be
silently ignoredr, rT rJ rS TNzos.unlink() was not invoked) �osr
rl r� r r� r r r( r_ r� rn )r Zos_unlink_origZos_unlink_replacementr� ri 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)z{Tests invoking FileInput.readline() when os.fstat() raises OSError.
This exception should be silently discarded.r, rT rJ rS TNzos.fstat() was not invoked) r� �fstatrl r� r r� r r r+ r� rn )r Z
os_fstat_origZos_fstat_replacementr� ri 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)z{Tests invoking FileInput.readline() when os.chmod() raises OSError.
This exception should be silently discarded.r, rT rJ rS TNzos.fstat() was not invoked) r� r� rl r� r r� r r r+ r� rn )r Z
os_chmod_origZos_chmod_replacementr� ri 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)rl r r} )r r r r r � s zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | j � d S)N)rq )r r r r rw � s zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r0 r1 r2 r rw r r r r �FilenoRaisesValueError� s r� r, rT rJ z _file.fileno() was not invokedzfileno() should return -1r/ )
rl r r� r r �_filerw r� rn rY )r r� Z unconditionally_raise_ValueErrorr� ri 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 �srcri 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� ri r r r �test_iteration_buffering s z'FileInputTests.test_iteration_bufferingN) r0 r1 r2 rs rv rx r r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr� r� r� r� r� r� r r r r rr � s*
( rr c @ s� e Z d 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)�
MockFileInputzDA class that mocks out fileinput.FileInput for use during unit testsNFr' 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)Nrw r, )r� r� )r r r r rw 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 �__doc__r r r_ r% rZ r[ rw r] r^ r r r r r� s r� c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) � BaseFileInputGlobalMethodsTestzQBase class for unit tests for the global function of
the fileinput module.c 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 r� > s r� c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) �Test_fileinput_inputz Unit tests for fileinput.input()c C sq t � } t � | _ | t _ | j t � � } t j � Wd QRX| j d | j j
� | j | t j d � d S)z�Tests invoking fileinput.input() when fileinput._state is not None
and its _file attribute is also not None. Expect RuntimeError to
be raised with a meaningful error message and for fileinput._state
to *not* be modified.N�input() already activezfileinput._state)r� ) r� �objectr� r� r� r� r� �inputrY r� ro �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)a\ Tests invoking fileinput.input() when fileinput._state is not None
but its _file attribute *is* None. Expect it to create and return
a new fileinput.FileInput object with all method parameters passed
explicitly to the __init__() method; also ensure that
fileinput._state is set to the returned instance.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)a4 Tests invoking fileinput.input() when fileinput._state is None
Expect it to create and return a new fileinput.FileInput object
with all method parameters passed explicitly to the __init__()
method; also ensure that fileinput._state is set to the returned
instance.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) a Tests that fileinput.input() creates a new fileinput.FileInput
object, passing the given parameters unmodified to
fileinput.FileInput.__init__(). Note that this test depends on the
monkey patching of fileinput.FileInput done by setUp().rJ rS r� rK r r& zfileinput._stateN) 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 r� S s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_closez Unit tests for fileinput.close()c C s' d t _ t j � | j t j � d S)zPTests that fileinput.close() does nothing if fileinput._state
is NoneN)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)z`Tests that fileinput.close() invokes close() on fileinput._state
and sets _state=Noner N)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 r� � s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_nextfilez#Unit tests for fileinput.nextfile()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.nextfile() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� r_ rY r� ro 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)a Tests fileinput.nextfile() when fileinput._state is not None.
Ensure that it invokes fileinput._state.nextfile() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r_ N)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 r� � s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_filenamez#Unit tests for fileinput.filename()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.filename() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� r% rY r� ro 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)a Tests fileinput.filename() when fileinput._state is not None.
Ensure that it invokes fileinput._state.filename() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r% N)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 r� � s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_linenoz!Unit tests for fileinput.lineno()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.lineno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� rZ rY r� ro 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)z�Tests fileinput.lineno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.lineno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rZ N)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 r� � s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_filelinenoz%Unit tests for fileinput.filelineno()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.filelineno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� r[ rY r� ro 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)a Tests fileinput.filelineno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.filelineno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r[ N)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 r� � s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_filenoz!Unit tests for fileinput.fileno()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.fileno() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� rw rY r� ro 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)z�Tests fileinput.fileno() when fileinput._state is not None.
Ensure that it invokes fileinput._state.fileno() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.rw N) r� r� r� �
fileno_retvalr� r� rw 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 r� s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_isfirstlinez&Unit tests for fileinput.isfirstline()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.isfirstline() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� r] rY r� ro 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)a Tests fileinput.isfirstline() when fileinput._state is not None.
Ensure that it invokes fileinput._state.isfirstline() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r] N)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 r� / s
r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�Test_fileinput_isstdinz"Unit tests for fileinput.isstdin()c
C sV d t _ | j t � � } t j � Wd QRX| j d | j j � | j t j � d S)z�Tests fileinput.isstdin() when fileinput._state is None.
Ensure that it raises RuntimeError with a meaningful error message
and does not modify fileinput._stateN�no active input())r� ) r� r� r� r� r^ rY r� ro 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)z�Tests fileinput.isstdin() when fileinput._state is not None.
Ensure that it invokes fileinput._state.isstdin() exactly once,
returns whatever it returns, and does not modify fileinput._state
to point to a different object.r^ N)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 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 ro rp r r r rq h s zInvocationRecorder.__call__N)r0 r1 r2 r rq r r r r r� e s r� c @ s� e Z d 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_compressedz*Unit tests for fileinput.hook_compressed()c 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= )�bz2ZBZ2Filer� 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� 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 Z d d � Z d d � Z d S)�Test_hook_encodedz'Unit tests for fileinput.hook_encoded()c
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_resultro rp 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)Nry 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_linesri r )r r r �check� s z+Test_hook_encoded.test_modes.<locals>.checkr� zA
zB
zC
u D€ZrUrz 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 r� � s r� �__main__)/r� 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 rl rr r� r� r� r� r� r� r� r� r� r� r� r� r� r� r0 �mainr r r r �<module> sV
" $[� Y.A >.