o HXh@stdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z zddl Z Wn e y7dZ Ynwzddl Z Wn e yIdZ Ynwddlm Z mZddlmZmZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd l mZGd ddZGdddZGdddee j Z!GdddZ"Gdddee j Z#GdddZ$Gddde j Z%Gddde%Z&Gddde%Z'Gdd 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/d0d0e j Z0Gd1d2d2e j Z1Gd3d4d4e j Z2e3d5kr8e 4dSdS)6z, Tests for fileinput module. Nick Mathewson N)BytesIOStringIO) FileInput hook_encoded)Path)verbose)TESTFN)unlink) os_helper)warnings_helper)support)mockc@seZdZddddZdS) BaseTestswmodecCsht\}}|tj|d|vrdnd}t|||d}||Wd|S1s-wY|S)Nbutf-8encoding)tempfilemkstemp addCleanupr r openwrite)selfZcontentrfdnamerfr=/opt/python-3.10.19/usr/lib/python3.10/test/test_fileinput.pywriteTmp,s   zBaseTests.writeTmpN)__name__ __module__ __qualname__r!rrrr r)src@sDeZdZddZeddZddZddd Zdd d Zd dZ dS) LineReadercCs g|_dSN _linesreadrrrr __init__6 zLineReader.__init__cCs z |jddWg|_Sg|_wr&r'r)rrr linesread9szLineReader.linesreadcCst|d|_|SNT)iter splitlinesit)rfilenamerrrr openhook@szLineReader.openhookNcCst|jd}|j||S)N)nextr0r(append)rsizelinerrr readlineDs  zLineReader.readlinecCs>g}d} |}|s |S|||t|7}||kr|SqNr)r8r5len)rZhintlinesr6r7rrr readlinesIs  zLineReader.readlinescCdSr&rr)rrr closeUszLineReader.closer&)r9) r"r#r$r*propertyr,r2r8r=r?rrrr r%4s    r%c@eZdZddZdS)BufferSizesTestsc Cs|dddtdD}|dddtdD}|dddtdD}|dd dtd D}td }trGtd t||||fd d}t|}| | t |d| |dd| |dd| | d| | |trtdt||||fd d}d}|r|dkr|}|r|dks| | || | d| |d||||trtd|| |d| | d| trtdt||||dfd d}tj} z*tdt_t|}| t |d| |d d!| | d"|W| t_n| t_wtr1td#t||||fd d}| | d$| | d|| | d$| | dtrftd%tj} z&t||||fd d d&}|D]} | dd'} t| qw| W| t_n| t_wt||||fd d}|D]'} | | d'd(|| dd'} || d| t| d |q| dS))Nr3cs|] }d|dVqdS)zLine %s of file 1 Nr.0irrr [z5BufferSizesTests.test_buffer_sizes..csrC)zLine %s of file 2 rDNrrErrr rH\rI csrC)zLine %s of file 3 rDNrrErrr rH]rIcsrC)zLine %s of file 4 rDNrrErrr rH^rIrDzLINE (\d+) OF FILE (\d+)z1. Simple iterationrfilesrzLine 5 of file 1 zLine 1 of file 4 z2. Status variablesxzLine 6 of file 2 z 3. NextfilezLine 1 of file 3 z4. Stdin-z Line 1 of stdin Line 2 of stdin ! zLine 2 of stdin zz5. Boundary conditionsrz 6. InplacerNinplacerr9 )r!joinrangerecompilerprintrlistr? assertEqualr;linenor1r8 filelineno assertFalse isfirstlineisstdinnextfilesysstdinrstdoutuppermatchassertNotEqualintgroup) rt1t2t3t4patfir<sZ savestdinZ savestdoutr7mrrr test_buffer_sizesYs         z"BufferSizesTests.test_buffer_sizesN)r"r#r$ryrrrr rBXs rBc@eZdZddZddZdS)UnconditionallyRaisecCs||_d|_dSNF)exception_typeinvoked)rr}rrr r*s zUnconditionallyRaise.__init__cOsd|_|r-)r~r}rargskwargsrrr __call__szUnconditionallyRaise.__call__Nr"r#r$r*rrrrr r{s r{c@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZejedd d!Zd"d#Zejedd$d%Zejedd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Z d8S)9FileInputTestscCs|d}|d}|d}|d}t||||fdd}|}||d||d||d||||}||||d||d||||dS)Nr3zThe only line there is. rrMrDr) r!rr8rbrcrdr1rer?)rrqrrrsrtrvr7rrr test_zero_byte_filess       z#FileInputTests.test_zero_byte_filescCs`|d}|d}t||fdd}t|}||gd||d||ddS)NA B CD E FrrMA B CzD zE FrT)r!rrarbrdrcrrqrrrvr<rrr %test_files_that_dont_end_with_newlines  z4FileInputTests.test_files_that_dont_end_with_newlinecCs|d}|d}t||fdd}||dt|}||d|||dt|}||ddS)NzA BzC DrrMr9)r!rrbfilenor4rnrhra)rrqrrrvr7rrr test_filenos  zFileInputTests.test_filenocCsz tddd}|dWn tyYnw|jddd}tdtft|d dd }Wdn1s7wYtdtf t|}Wdn1sRwY||gd dS) Nrr)rrz-FileInput should reject invalid mode argumentA B C Dwbrr3U)rNrr)rrC D) rfail ValueErrorr!r Zcheck_warningsDeprecationWarningrarb)rrvrqr<rrr test_opening_modes   z FileInputTests.test_opening_modecCs^td }td|_tdgdd}t|}||dgWddS1s(wYdS)Nz sys.stdinspam, bacon, sausage, and spamrVrbrNr)r Zpatchrbufferrrarb)rZm_stdinrvr<rrr test_stdin_binary_modes  "z%FileInputTests.test_stdin_binary_modecCs\tj}z&tdt_|ttjdtdgdd}t|}||dgW|t_dS|t_w)NrrrVrr)rirjrrehasattrrrarb)rZ orig_stdinrvr<rrr test_detached_stdin_binary_modes z.FileInputTests.test_detached_stdin_binary_modecCsztdddd}|dWn tyYnwz tdd}|dWn ty.YnwGdd d }|d }|}t|g|d }|Wdn1sTwY||jd dS) NrDcSr>r&r)rrxrrr  z7FileInputTests.test_file_opening_hook..)rZr2zGFileInput should raise if both inplace and openhook arguments are givenr2z2FileInput should check openhook for being callablec@rz)z=FileInputTests.test_file_opening_hook..CustomOpenHookcS d|_dSr|)r~r)rrr r*r+zFFileInputTests.test_file_opening_hook..CustomOpenHook.__init__c_sd|_t|ddiS)NTrr)r~r)rrZkargsrrr rszFFileInputTests.test_file_opening_hook..CustomOpenHook.__call__Nrrrrr CustomOpenHook rr[zopenhook not invoked)rrrr!r8 assertTruer~)rrvrtZcustom_open_hookrrr test_file_opening_hook s&     z%FileInputTests.test_file_opening_hookc Cs2ttd}|d|dd|dWdn1s!wY|tttttdd[}z||d||d ||d Wnt y]| d Ynw| t  t |Wdn1srwY||d ||d WddS1swYdS) NrsA B C s 123456789 iasciirNr2rrrzRead to end of filer3) rrrr safe_unlinkrrrbr8UnicodeDecodeErrorr assertRaisesrarrrvrrr test_readline%s.       "zFileInputTests.test_readlinecCsttd }|dWdn1swY|ttttdd1}||d||d||d||d||dWddS1sZwYdS) NrrrrsA sB sC D)rrrrrrrbr8rrrr test_readline_binary_mode<s   "z(FileInputTests.test_readline_binary_modecCs|jddd}t|ddd}|}||dtjdWdn1s)wYt|d}||dWddS1sGwYdS)Ns Initial text.rrrT)rrZs New line.) r!rr8rbrirkrrreadrZ temp_fileZfobjr7rrrr test_inplace_binary_write_modeIs  "z-FileInputTests.test_inplace_binary_write_modecCs|jddd}t|dddd}|}||dtd Wdn1s(wYt|d }||d d WddS1sIwYdS) NsInitial text rrTrreplace)rZrerrorsuInitial text �u New line ˆrs s New line ?)r!rr8rbr`rrrstriprrrr test_inplace_encoding_errorsSs   "z+FileInputTests.test_inplace_encoding_errorscCsXdd}|d}t|g|d }|}Wdn1swY||ddS)NcS tdS)N(I used to receive only filename and mode)iorr1rrrr old_hook^r+zFFileInputTests.test_file_hook_backward_compatibility..old_hookr[rr)r!rr8rb)rrrrvresultrrr %test_file_hook_backward_compatibility]s   z4FileInputTests.test_file_hook_backward_compatibilitycCs|d}|d}t||fdd }t|}Wdn1s!wY||gd||d||d||jddS) NrrrrMrrrTr)r!rrarbrdrc_filesrrrr test_context_manageres   z#FileInputTests.test_context_managercCsV|d}zt|dd}t1swYWdSty*||jdYdSw)Nr3rrMr)r!rOSErrorrbr)rrqrvrrr test_close_on_exceptionos  z&FileInputTests.test_close_on_exceptioncCs@tgdd}||jdWddS1swYdS)NrrM)rV)rrbr)rrvrrr .test_empty_files_list_specified_to_constructorws"z=FileInputTests.test_empty_files_list_specified_to_constructor)categorycCsf|d}t|gdd}|d}||d|d}||dWddS1s,wYdS) zNTests invoking FileInput.__getitem__() with the current line number line1 line2 rrMrline1 rDzline2 N)r!rrb)rrrvZretval1Zretval2rrr test__getitem__{s  "zFileInputTests.test__getitem__c Cs|d}|td/t|gd}||ddWdn1s%wYWddSWddS1s=wYdS)NrzUse iterator protocol instead)rNrr)r!ZassertWarnsRegexrrrb)rrrvrrr test__getitem___deprecations "z*FileInputTests.test__getitem___deprecationc C~|d}t|gdd!}|t }|dWdn1s!wYWdn1s0wY||jjddS)zZTests invoking FileInput.__getitem__() with an index unequal to the line numberrrrMrDN)zaccessing lines out of order)r!rr RuntimeErrorrb exceptionrrrrvcmrrr test__getitem__invalid_key   z)FileInputTests.test__getitem__invalid_keyc Cr)zZTests invoking FileInput.__getitem__() with the line number but at end-of-inputr3rrMrN)zend of input reached)r!rr IndexErrorrbrrrrrr test__getitem__eofrz!FileInputTests.test__getitem__eofcCstj}tt}z5|d}|t|dt|gddd}t||t_| Wdn1s3wYW|t_n|t_w| |j ddS)zTests invoking FileInput.nextfile() when the attempt to delete the backup file would raise OSError. This error is expected to be silently ignoredr[z.bakTrrYNzos.unlink() was not invoked) osr r{rr!rrrr4rhrr~)rZos_unlink_origZos_unlink_replacementrrvrrr %test_nextfile_oserror_deleting_backups  z4FileInputTests.test_nextfile_oserror_deleting_backupcC|tj}tt}z)|d}t|gddd}|t_|Wdn1s'wYW|t_n|t_w||jddS)z{Tests invoking FileInput.readline() when os.fstat() raises OSError. This exception should be silently discarded.r[TrrYNos.fstat() was not invoked) rfstatr{rr!rr8rr~)rZ os_fstat_origZos_fstat_replacementrrvrrr %test_readline_os_fstat_raises_OSError  z4FileInputTests.test_readline_os_fstat_raises_OSErrorcCr)z{Tests invoking FileInput.readline() when os.chmod() raises OSError. This exception should be silently discarded.r[TrrYNr) rchmodr{rr!rr8rr~)rZ os_chmod_origZos_chmod_replacementrrvrrr %test_readline_os_chmod_raises_OSErrorrz4FileInputTests.test_readline_os_chmod_raises_OSErrorc CsGdddt}|}|d}t|gdd}|j}z ||_|}W||_n||_wWdn1s6wY||jd||dddS) Nc@rz)zQFileInputTests.test_fileno_when_ValueError_raised..FilenoRaisesValueErrorcSt|tdSr&)r{r*rr)rrr r*zZFileInputTests.test_fileno_when_ValueError_raised..FilenoRaisesValueError.__init__cSs |dSr&)rr)rrr r zXFileInputTests.test_fileno_when_ValueError_raised..FilenoRaisesValueError.filenoN)r"r#r$r*rrrrr FilenoRaisesValueErrorrrr[rrMz_file.fileno() was not invokedr9zfileno() should return -1)r{r!r_filerrr~rb)rrZ unconditionally_raise_ValueErrorrrvZ file_backuprrrr "test_fileno_when_ValueError_raiseds   z1FileInputTests.test_fileno_when_ValueError_raisedcCst}tddg|jd`}||jg||d||jdg||d||jdg||d||jddg||d||jdg||d||jgWddS1sowYdSNz line1 line2zline3 rrZline2r3)r%rr2rbr,r8rsrcrvrrr test_readline_bufferings""z&FileInputTests.test_readline_bufferingcCst}tddg|jd^}||jg|t|d||jdg|t|d||jdg|t|d||jddg|tt|||jdg|tt|||jgWddS1smwYdSr)r%rr2rbr,r4r StopIterationrrrr test_iteration_bufferings""z'FileInputTests.test_iteration_bufferingcCst|d}t|dd.}|}||d||d||d||t |WddS1s>wYdS)N Pathlib file.rrrD) rr!rr8rbrcrdr1rfspath)rrqrvr7rrr test_pathlib_files "z FileInputTests.test_pathlib_filecCst|d}t|ddd}|}||dtd|Wdn1s)wYt|dd}||dWddS1sHwYdS)NrTr)rZrz Modified %srzModified Pathlib file. )rr!rr8rbr`rr)rrqrvr7rrrr test_pathlib_file_inplaces "z(FileInputTests.test_pathlib_file_inplaceN)!r"r#r$rrrrrrrrrrrrrrrr Zignore_warningsrrrrrrrrrrrrrrrrr rs>             rc@sfeZdZdZddddddddZd d Zd d Zd dZddZddZ ddZ ddZ ddZ dS) MockFileInputzDA class that mocks out fileinput.FileInput for use during unit testsNFr3r)rr2rrcCsJ||_||_||_||_||_||_||_d|_t dd|_ i|_ dS)NcSsdSr:rrrrr r1rz(MockFileInput.__init__..) rNrZbackuprr2rrr collections defaultdictinvocation_counts return_values)rrNrZrrr2rrrrr r*'s zMockFileInput.__init__cCs|jdd7<dS)Nr?rD)rr)rrr r?4szMockFileInput.closecC|jdd7<|jdS)NrhrDrrr)rrr rh7 zMockFileInput.nextfilecCr)Nr1rDrr)rrr r1;rzMockFileInput.filenamecCr)NrcrDrr)rrr rc?rzMockFileInput.linenocCr)NrdrDrr)rrr rdCrzMockFileInput.filelinenocCr)NrrDrr)rrr rGrzMockFileInput.filenocCr)NrfrDrr)rrr rfKrzMockFileInput.isfirstlinecCr)NrgrDrr)rrr rgOrzMockFileInput.isstdin)NFr3) r"r#r$__doc__r*r?rhr1rcrdrrfrgrrrr r$s   rc@(eZdZdZddZddZddZdS) BaseFileInputGlobalMethodsTestzQBase class for unit tests for the global function of the fileinput module.cCstj|_tj|_tt_dSr&) fileinput_state _orig_stater_orig_FileInputrr)rrr setUpWs z$BaseFileInputGlobalMethodsTest.setUpcCs|jt_|jt_dSr&)rrrrrr)rrr tearDown\s z'BaseFileInputGlobalMethodsTest.tearDowncCs2|j|}||d|t|j}||ddS)NrD)rrbr;)rZmock_file_inputZ method_nameZ actual_countZactual_total_countrrr assertExactlyOneInvocation`s  z9BaseFileInputGlobalMethodsTest.assertExactlyOneInvocationN)r"r#r$rrrrrrrr rSs  rc@s0eZdZdZddZddZddZdd Zd S) Test_fileinput_inputz Unit tests for fileinput.input()cCsjt}t|_|t_|t }tWdn1swY|d|j j | |tjddS)zTests 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)zinput() already activefileinput._state) robjectrrrrrinputrbrrassertIs)rinstancerrrr 1test_state_is_not_None_and_state_file_is_not_Noneks  zFTest_fileinput_input.test_state_is_not_None_and_state_file_is_not_NonecCst}d|_|t_|dS)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)rrrrdo_test_call_inputrrrrr -test_state_is_not_None_and_state_file_is_Nonexs zBTest_fileinput_input.test_state_is_not_None_and_state_file_is_NonecCsdt_|dS)a4Tests 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)rrr r)rrr test_state_is_Nones z'Test_fileinput_input.test_state_is_NonecCst}t}t}t}t}t}tj||||||d}||tjd|||jd|||jd|||jd|||jd|||j ddS) aTests 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().)rNrZrrr2rrrNrZrrr2N) rrrrrrNrZrrr2)rrNrZrrr2rrrrr r s z'Test_fileinput_input.do_test_call_inputN)r"r#r$rr r r r rrrr rhs   rc@ eZdZdZddZddZdS)Test_fileinput_closez Unit tests for fileinput.close()cCsdt_t|tjdS)zPTests that fileinput.close() does nothing if fileinput._state is NoneN)rrr? assertIsNoner)rrr r sz'Test_fileinput_close.test_state_is_NonecCs0t}|t_t||d|tjdS)z`Tests that fileinput.close() invokes close() on fileinput._state and sets _state=Noner?N)rrrr?rrr rrr test_state_is_not_Nones  z+Test_fileinput_close.test_state_is_not_NoneNr"r#r$rr rrrrr rs rc@r)Test_fileinput_nextfilez#Unit tests for fileinput.nextfile()cCXdt_|t }tWdn1swY|d|jj|tjdS)zTests fileinput.nextfile() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNzno active input()) rrrrrhrbrrrrrrrr r    z*Test_fileinput_nextfile.test_state_is_NonecCNt}t}||jd<|t_t}||d||||tj|dS)aTests 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.rhN)rrrrrrhrr)rZnextfile_retvalrretvalrrr r   z.Test_fileinput_nextfile.test_state_is_not_NoneNrrrrr r rc@r)Test_fileinput_filenamez#Unit tests for fileinput.filename()cCr)zTests fileinput.filename() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrr1rbrrrrrrr r rz*Test_fileinput_filename.test_state_is_NonecCr)aTests 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.r1N)rrrrrr1rr)rZfilename_retvalrrrrr rrz.Test_fileinput_filename.test_state_is_not_NoneNrrrrr rrrc@r)Test_fileinput_linenoz!Unit tests for fileinput.lineno()cCr)zTests fileinput.lineno() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrrcrbrrrrrrr r rz(Test_fileinput_lineno.test_state_is_NonecCr)zTests 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.rcN)rrrrrrcrr)rZ lineno_retvalrrrrr rrz,Test_fileinput_lineno.test_state_is_not_NoneNrrrrr rrrc@r)Test_fileinput_filelinenoz%Unit tests for fileinput.filelineno()cCr)zTests fileinput.filelineno() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrrdrbrrrrrrr r rz,Test_fileinput_filelineno.test_state_is_NonecCr)aTests 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.rdN)rrrrrrdrr)rZfilelineno_retvalrrrrr rrz0Test_fileinput_filelineno.test_state_is_not_NoneNrrrrr r rrc@r)Test_fileinput_filenoz!Unit tests for fileinput.fileno()cCr)zTests fileinput.fileno() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrrrbrrrrrrr r )rz(Test_fileinput_fileno.test_state_is_NonecCsTt}t}||jd<||_|t_t}||d||||tj|dS)zTests 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.rN) rrr fileno_retvalrrrrr)rr rrrrr r3s   z,Test_fileinput_fileno.test_state_is_not_NoneNrrrrr r&rrc@r)Test_fileinput_isfirstlinez&Unit tests for fileinput.isfirstline()cCr)zTests fileinput.isfirstline() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrrfrbrrrrrrr r Erz-Test_fileinput_isfirstline.test_state_is_NonecCr)aTests 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.rfN)rrrrrrfrr)rZisfirstline_retvalrrrrr rOrz1Test_fileinput_isfirstline.test_state_is_not_NoneNrrrrr r!Brr!c@r)Test_fileinput_isstdinz"Unit tests for fileinput.isstdin()cCr)zTests fileinput.isstdin() when fileinput._state is None. Ensure that it raises RuntimeError with a meaningful error message and does not modify fileinput._stateNr) rrrrrgrbrrrrrrr r `rz)Test_fileinput_isstdin.test_state_is_NonecCr)zTests 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.rgN)rrrrrrgrr)rZisstdin_retvalrrrrr rjrz-Test_fileinput_isstdin.test_state_is_not_NoneNrrrrr r"]rr"c@rz)InvocationRecordercCrr:)invocation_countr)rrr r*zr+zInvocationRecorder.__init__cOs"|jd7_||f|_tdS)NrDs some bytes)r$last_invocationrrrrrr r}s  zInvocationRecorder.__call__Nrrrrr r#xs r#c@seZdZdZddZddZddZee dd d Z ee dd d Z ee d ddZ ddZddZddZddZddZddZddZeddZd S)!Test_hook_compressedz*Unit tests for fileinput.hook_compressed()cCs t|_dSr&)r# fake_openr)rrr rrzTest_hook_compressed.setUpcC|dddS)Nr3rdo_test_use_builtin_open_textr)rrr test_empty_stringrz&Test_hook_compressed.test_empty_stringcCr(Nabcdrr)r)rrr test_no_extrz Test_hook_compressed.test_no_extzRequires gzip and zlibcCVtj}|jt_z tdd}W|t_n|t_w||jjd||jjdifdS)Ntest.gzrrD)r0r)gziprr'rhook_compressedrbr$r%r original_openrrrr test_gz_ext_fakez%Test_hook_compressed.test_gz_ext_fakecCsJtj}ddt_z tjdddd}W|t_n|t_w|t|dgdS)NcSr)NsEx-binary string)rrrrrr rs zATest_hook_compressed.test_gz_with_encoding_fake..r0rrrzEx-binary string)r1rrr2rbrar3rrr test_gz_with_encoding_fakes  z/Test_hook_compressed.test_gz_with_encoding_fakez Requires bz2cCr/)Ntest.bz2rrD)r8r)bz2BZ2Filer'rr2rbr$r%r3rrr test_bz2_ext_faker6z&Test_hook_compressed.test_bz2_ext_fakecCr()Nz abcd.blahrdo_test_use_builtin_open_binaryr)rrr test_blah_extrz"Test_hook_compressed.test_blah_extcCr()Nzabcd.Gzrr<r)rrr test_gz_ext_builtinrz(Test_hook_compressed.test_gz_ext_builtincCr()Nzabcd.Bz2rr<r)rrr test_bz2_ext_builtinrz)Test_hook_compressed.test_bz2_ext_builtincCr()Nr-rr<r)rrr test_binary_mode_encodingrz.Test_hook_compressed.test_binary_mode_encodingcCr(r,r)r)rrr test_text_mode_encodingrz,Test_hook_compressed.test_text_mode_encodingc Csf||j}z t||}W||n||w||jjd||jj||fdddfdS)NrDrrreplace_builtin_openr'rr2rbr$r%rr1rr4rrrr r=  z4Test_hook_compressed.do_test_use_builtin_open_binaryc Csf||j}z t||}W||n||w||jjd||jj||fdddfdS)NrDlocalerCrDrFrrr r*rGz2Test_hook_compressed.do_test_use_builtin_open_textcCstj}|t_|Sr&)builtinsr)Z new_open_funcr4rrr rEsz)Test_hook_compressed.replace_builtin_openN)r"r#r$rrr+r.unittestZ skipUnlessr1r5r7r9r;r>r?r@rArBr=r* staticmethodrErrrr r&s(       r&c@r) Test_hook_encodedz'Unit tests for fileinput.hook_encoded()c Cst}t}tj||d}t}tj}|t_zt}t}|||}W|t_n|t_w||jd|j\} } | | d|| | d|| | d|| | d|| | dS)NrrDrrr) rrrr#rIrrbr$r%rpopre) rrrrr'r4r1rZ open_resultrrrrr tests$  zTest_hook_encoded.testcsttd }|dWdn1swYttfdd}|ddgt|ddgWdn1sBwY|dd g|d d gdS) NrsabccsNttdtd|dd }t|}Wdn1swY||dS)NrrrMrNrr2rrrrarb)rexpected_linesrvr<r)rr checks  z,Test_hook_encoded.test_errors..checkignoreabcstrictru�abcbackslashreplacez\x80abc)rrrrrrrrrrSrr)r test_errorss       zTest_hook_encoded.test_errorscsttd }|dWdn1swYttfdd}|dgdt|dgdWdn1sDwYt|dgdWdn1s`wYt|d gd WddS1s}wYdS) Nrs A B C D+IKw-csJtt|tdd }t|}Wdn1swY||dS)Nzutf-7rPrQ)rrRrvr<r)rr rSs z+Test_hook_encoded.test_modes..checkr)rrrD€ZrUrr)rzB zC rZ) rrrrrZ assertWarnsrrrrXrr)r test_modess       "zTest_hook_encoded.test_modesN)r"r#r$rrOrYr[rrrr rLs  rLc@rA)MiscTestcCrr&)r Z check__all__rr)rrr test_allrzMiscTest.test_allN)r"r#r$r]rrrr r\s r\__main__)5rrrrir^rrrIrrJr9 ImportErrorr1rrrrpathlibrZ test.supportrZtest.support.os_helperrr rr r rOr r rr%ZTestCaserBr{rrrrrrrrrrr!r"r#r&rLr\r"mainrrrr sn             $Ss/? YA