Your IP : 216.73.216.196
o
0�i�� � @ sr d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZzd dl Z W n e
y6 dZ Y nw zd dl Z W n e
yH dZ Y nw d dl m Z m
Z
d dlmZmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lm Z G d d
� d
�Z G dd� d�Z G dd� de ej �Z G dd� d�Z!G dd� de ej �Z"G dd� d�Z#G dd� dej �Z$G dd� de$�Z%G d d � d e$�Z&G d d � d e$�Z'G d d!� d!e$�Z(G d"d#� d#e$�Z)G d$d%� d%e$�Z*G d&d'� d'e$�Z+G d(d)� d)e$�Z,G d*d+� d+e$�Z-G d,d-� d-�Z.G d.d/� d/ej �Z/G d0d1� d1ej �Z0G d2d3� d3ej �Z1e2d4k�r7e�3� dS dS )5� N)�BytesIO�StringIO)� FileInput� hook_encoded)�Path)�verbose)�TESTFN)�unlink)� os_helper)�warnings_helper)�support)�mockc @ s e Zd Zdd�dd�ZdS )� BaseTests�w��modec C sh t �� \}}| �tj|� d|v rd nd}t|||d��}|�|� W d � |S 1 s-w Y |S )N�b�utf-8��encoding)�tempfileZmkstemp�
addCleanupr
r �open�write)�selfZcontentr �fd�namer �f� r �:/opt/alt/python310/lib64/python3.10/test/test_fileinput.py�writeTmp, s
��zBaseTests.writeTmpN)�__name__�
__module__� __qualname__r r r r r r ) s r c @ sD e Zd Zdd� Zedd� �Zdd� Zddd �Zdd d �Zd
d� Z dS )�
LineReaderc C s
g | _ d S �N��
_linesread�r r r r �__init__6 �
zLineReader.__init__c C s z | j d d � W g | _ S g | _ w r% r& r( r r r � linesread9 s zLineReader.linesreadc C s t |�d��| _| S �NT)�iter�
splitlines�it)r �filenamer r r r �openhook@ s zLineReader.openhookNc C s t | jd�}| j�|� |S )N� )�nextr/ r' �append)r �size�liner r r �readlineD s zLineReader.readline���c C s> g }d} | � � }|s
|S |�|� |t|�7 }||kr |S q�Nr )r7 r4 �len)r Zhint�linesr5 r6 r r r � readlinesI s
�zLineReader.readlinesc C � d S r% r r( r r r �closeU s zLineReader.closer% )r8 )
r! r" r# r) �propertyr+ r1 r7 r< r> r r r r r$ 4 s
r$ c @ � e Zd Zdd� ZdS )�BufferSizesTestsc
C s� | � d�dd� td�D ���}| � d�dd� td�D ���}| � d�dd� td�D ���}| � d�d d� td
�D ���}t�d �}trGtd � t||||fd
d�}t|�}|� � | �
t |�d� | �
|d d� | �
|d d� | �
|� � d� | �
|�
� |� tr�td� t||||fd
d�}d}|r�|dkr�|�� }|r�|dks�| �
|�
� |� | �
|� � d� | �
|�� d� | �|�� � | �|�� � tr�td� |�� | �
|�� d� | �
|� � d� |� � tr�td � t||||d fd
d�}tj} z*td �t_t|�}| �
t |�d � | �
|d d!� | �
|�
� d"� |�� W | t_n| t_w t�r1td#� t||||fd
d�}| �
|� � d$� | �
|�
� d � |�� | �
|� � d$� | �
|�
� d � t�rftd%� tj}
z&t||||fd
d
d&�}|D ]} | d d'� �� } t| � �qw|� � W |
t_n|
t_w t||||fd
d�}|D ]'} | �
| d' d(� |�| d d'� �} | �| d � | �
t| � d
��|�� � �q�|� � d S ))Nr2 c s � � | ] }d |d V qdS )zLine %s of file 1
� Nr ��.0�ir r r � <genexpr>[ � � z5BufferSizesTests.test_buffer_sizes.<locals>.<genexpr>� c s rB )zLine %s of file 2
rC Nr rD r r r rG \ rH �
c s rB )zLine %s of file 3
rC Nr rD r r r rG ] rH � c s rB )zLine %s of file 4
rC Nr rD r r r rG ^ rH rC zLINE (\d+) OF FILE (\d+)z1. Simple iterationr ��filesr � � zLine 5 of file 1
� zLine 1 of file 4
z2. Status variables�xzLine 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
z<stdin>z5. Boundary conditionsr z
6. Inplace�rM �inplacer r8 �
) r �join�range�re�compiler �printr �listr> � assertEqualr: �linenor0 r7 �
filelineno� assertFalse� isfirstline�isstdin�nextfile�sys�stdinr �stdout�upper�match�assertNotEqual�int�group)
r �t1�t2�t3�t4Zpat�fir; �sZ savestdinZ
savestdoutr6 �mr r r �test_buffer_sizesY s�
�
z"BufferSizesTests.test_buffer_sizesN)r! r" r# rw r r r r rA X s rA c @ � e Zd Zdd� Zdd� ZdS )�UnconditionallyRaisec C s || _ d| _d S �NF)�exception_type�invoked)r r{ r r r r) � s
z UnconditionallyRaise.__init__c O s d| _ | �� �r, )r| r{ �r �args�kwargsr r r �__call__� s z UnconditionallyRaise.__call__N�r! r" r# r) r� r r r r ry � s ry c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zejed �d d!� �Zd"d#� Zejed �d$d%� �Zejed �d&d'� �Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8S )9�FileInputTestsc C s� | � d�}| � d�}| � d�}| � d�}t||||fdd�}|�� }| �|d� | �|�� d� | �|�� d� | �|�� |� |�� }| �|� | �|�� d� | �|�� d� | �|�� |� |�� d S )Nr2 zThe only line there is.
r rL rC r ) r r r7 ra rb rc r0 rd r> )r rp rq rr rs rt r6 r r r �test_zero_byte_files� s
z#FileInputTests.test_zero_byte_filesc C s` | � d�}| � d�}t||fdd�}t|�}| �|g d�� | �|�� d� | �|�� d� d S )N�A
B
C�D
E
Fr rL ��A
�B
�CzD
zE
�F� rS )r r r` ra rc rb �r rp rq rt r; r r r �%test_files_that_dont_end_with_newline� s
z4FileInputTests.test_files_that_dont_end_with_newlinec C s� | � d�}| � d�}t||fdd�}| �|�� d� t|�}| �|�� d� |�� | �|�� d� t|�}| �|�� d� d S )NzA
BzC
Dr rL r8 )r r ra �filenor3 rm rg r` )r rp rq rt r6 r r r � test_fileno� s
zFileInputTests.test_filenoc C s� z
t ddd�}| �d� W n ty Y nw | jddd�}t�dtf�� t |d dd
�}W d � n1 s7w Y t�dtf�� t|�}W d � n1 sRw Y | �|g d �� d S ) Nr r )r r z-FileInput should reject invalid mode argument� A
B
C
D�wbr r2 �U)rM r r )r� r� �C
�D) r �fail�
ValueErrorr r Zcheck_warnings�DeprecationWarningr` ra )r rt rp r; r r r �test_opening_mode� s � �
�z FileInputTests.test_opening_modec C s^ t �d�� }td�|_tdgdd�}t|�}| �|dg� W d � d S 1 s(w Y d S )Nz sys.stdin� spam, bacon, sausage, and spamrU �rb�rM r )r
Zpatchr �bufferr r` ra )r Zm_stdinrt r; r r r �test_stdin_binary_mode� s
"�z%FileInputTests.test_stdin_binary_modec C s\ t j}z&td�t _| �tt jd�� tdgdd�}t|�}| �|dg� W |t _d S |t _w )Nr� r� rU r� r� )rh ri r rd �hasattrr r` ra )r Z
orig_stdinrt r; r r r � test_detached_stdin_binary_mode� s
z.FileInputTests.test_detached_stdin_binary_modec C s� zt ddd� d�}| �d� W n ty Y nw z t dd�}| �d� W n ty. Y nw G dd � d �}| �d
�}|� }t |g|d�� }|�� W d � n1 sTw Y | �|jd � d S ) NrC c S r= r% r )r rv r r r �<lambda>
� z7FileInputTests.test_file_opening_hook.<locals>.<lambda>)rY r1 zGFileInput should raise if both inplace and openhook arguments are given�r1 z2FileInput should check openhook for being callablec @ rx )z=FileInputTests.test_file_opening_hook.<locals>.CustomOpenHookc S �
d| _ d S rz )r| r( r r r r) r* zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__init__c _ s d| _ t|ddi�S )NTr r )r| r )r r~ Zkargsr r r r� s zFFileInputTests.test_file_opening_hook.<locals>.CustomOpenHook.__call__Nr� r r r r �CustomOpenHook � r� rZ zopenhook not invoked)r r� r� r r7 �
assertTruer| )r rt r� �tZcustom_open_hookr r r �test_file_opening_hook
s& �
�
�z%FileInputTests.test_file_opening_hookc C s2 t td��}|�d� |�dd � |�d� W d � n1 s!w Y | �tt� tttd�d��[}z| �|�� d� | �|�� d � | �|�� d
� W n t y] | �
d � Y nw | � t �� t |� W d � n1 srw Y | �|�� d � | �|�� d � W d � d S 1 s�w Y d S )
Nr� s A
B
C
s
123456789
i� � ��ascii�rM r1 r� r� r� zRead to end of filer2 )
r r r r � safe_unlinkr r ra r7 �UnicodeDecodeErrorr� � assertRaisesr` �r r rt r r r �
test_readline% s.
� � �
�"�z FileInputTests.test_readlinec C s� t td��
}|�d� W d � n1 sw Y | �tt� ttdd��1}| �|�� d� | �|�� d� | �|�� d� | �|�� d� | �|�� d� W d � d S 1 sZw Y d S ) Nr� r� r� r� s A
s B
s C
D� )r r r r r� r ra r7 r� r r r �test_readline_binary_mode< s � "�z(FileInputTests.test_readline_binary_modec C s� | j ddd�}t|ddd��}|�� }| �|d� tj�d� W d � n1 s)w Y t|d��}| �|�� d� W d � d S 1 sGw Y d S )Ns
Initial text.r� r r� T)r rY s New line.) r r r7 ra rh rj r r �read�r Z temp_fileZfobjr6 r r r r � test_inplace_binary_write_modeI s � "�z-FileInputTests.test_inplace_binary_write_modec C s� | j ddd�}t|dddd��}|�� }| �|d� td � W d � n1 s(w Y t|d
��}| �|�� �d �d � W d � d S 1 sIw Y d S )
Ns Initial text �r� r Tr� �replace)rY r �errorsu Initial text �u New line r� s
s
New line ?)r r r7 ra r_ r r� �rstripr� r r r � test_inplace_encoding_errorsS s �
� "�z+FileInputTests.test_inplace_encoding_errorsc C sX dd� }| � d�}t|g|d�� }|�� }W d � n1 s w Y | �|d� d S )Nc S �
t �d�S )N�(I used to receive only filename and mode)�ior �r0 r r r r �old_hook^ r* zFFileInputTests.test_file_hook_backward_compatibility.<locals>.old_hookrZ r� r� )r r r7 ra )r r� r� rt �resultr r r �%test_file_hook_backward_compatibility] s
�z4FileInputTests.test_file_hook_backward_compatibilityc C s� | � d�}| � d�}t||fdd�� }t|�}W d � n1 s!w Y | �|g d�� | �|�� d� | �|�� d� | �|jd� d S ) Nr� r� r rL r� r� rS r )r r r` ra rc rb �_filesr� r r r �test_context_managere s
�z#FileInputTests.test_context_managerc C sV | � d�}zt|dd��}t�1 sw Y W d S ty* | �|jd� Y d S w )Nr2 r rL r )r r �OSErrorra r� )r rp rt r r r �test_close_on_exceptiono s
� �z&FileInputTests.test_close_on_exceptionc C s@ t g dd��}| �|jd� W d � d S 1 sw Y d S )Nr rL )rU )r ra r� )r rt r r r �.test_empty_files_list_specified_to_constructorw s "�z=FileInputTests.test_empty_files_list_specified_to_constructor)�categoryc C sh | � d�}t|gdd�� }|d }| �|d� |d }| �|d� W d � d S 1 s-w Y d S )N� line1
line2
r rL r �line1
rC zline2
)r r ra )r r� rt Zretval1Zretval2r r r �test__getitem__{ s
"�z FileInputTests.test__getitem__c C s� | � d�}| �td��/ t|gd��}| �|d d� W d � n1 s%w Y W d � d S W d � d S 1 s=w Y d S )Nr� z Use iterator protocol instead)rM r r� )r ZassertWarnsRegexr� r ra )r r� rt r r r �test__getitem___deprecation� s
� ��"�z*FileInputTests.test__getitem___deprecationc C � | � d�}t|gdd��!}| �t�� }|d W d � n1 s"w Y W d � n1 s1w Y | �|jjd� d S )Nr� r rL rC )z accessing lines out of order)r r r� � RuntimeErrorra � exceptionr~ �r r� rt �cmr r r �test__getitem__invalid_key� �
�� �z)FileInputTests.test__getitem__invalid_keyc C r� )Nr2 r rL r )zend of input reached)r r r� �
IndexErrorra r� r~ r� r r r �test__getitem__eof� r� z!FileInputTests.test__getitem__eofc C s� t j}tt�}z5| �d�}| �t|d � t|gddd��}t|� |t _|� � W d � n1 s4w Y W |t _n|t _w | �
|j d� d S )NrZ z.bakTr rX zos.unlink() was not invoked) �osr ry r� r r r� r r3 rg r� r| )r Zos_unlink_origZos_unlink_replacementr� rt r r r �%test_nextfile_oserror_deleting_backup� s
���z4FileInputTests.test_nextfile_oserror_deleting_backupc C �~ t j}tt�}z)| �d�}t|gddd��}|t _|�� W d � n1 s(w Y W |t _n|t _w | �|jd� d S �NrZ Tr rX zos.fstat() was not invoked) r� �fstatry r� r r r7 r� r| )r Z
os_fstat_origZos_fstat_replacementr� rt r r r �%test_readline_os_fstat_raises_OSError� �
���z4FileInputTests.test_readline_os_fstat_raises_OSErrorc C r� r� ) r� �chmodry r� r r r7 r� r| )r Z
os_chmod_origZos_chmod_replacementr� rt r r r �%test_readline_os_chmod_raises_OSError� r� z4FileInputTests.test_readline_os_chmod_raises_OSErrorc C s� G dd� dt �}|� }| �d�}t|gdd�� }|j}z ||_|�� }W ||_n||_w W d � n1 s6w Y | �|jd� | �|dd� d S ) Nc @ rx )zQFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueErrorc S � t �| t� d S r% )ry r) r� r( r r r r) � � zZFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.__init__c S s | � � d S r% )r� r( r r r r� � � zXFileInputTests.test_fileno_when_ValueError_raised.<locals>.FilenoRaisesValueError.filenoN)r! r" r# r) r� r r r r �FilenoRaisesValueError� r� r� rZ r rL z _file.fileno() was not invokedr8 zfileno() should return -1)ry r r �_filer� r� r| ra )r r� Z unconditionally_raise_ValueErrorr� rt Z file_backupr� r r r �"test_fileno_when_ValueError_raised� s
� �z1FileInputTests.test_fileno_when_ValueError_raisedc C s� t � }tddg|jd��`}| �|jg � | �|�� d� | �|jdg� | �|�� d� | �|jdg� | �|�� d� | �|jddg� | �|�� d� | �|jdg� | �|�� d� | �|jg � W d � d S 1 sow Y d S �Nz line1
line2zline3
r� r� Zline2r2 )r$ r r1 ra r+ r7 �r �srcrt r r r �test_readline_buffering� s" �"�z&FileInputTests.test_readline_bufferingc C s� t � }tddg|jd��^}| �|jg � | �t|�d� | �|jdg� | �t|�d� | �|jdg� | �t|�d� | �|jddg� | �tt|� | �|jdg� | �tt|� | �|jg � W d � d S 1 smw Y d S r� )r$ r r1 ra r+ r3 r� �
StopIterationr� r r r �test_iteration_buffering s" �"�z'FileInputTests.test_iteration_bufferingc C s� t | �d��}t|dd��.}|�� }| �|d� | �|�� d� | �|�� d� | �|�� t� |�� W d � d S 1 s>w Y d S )N�
Pathlib file.r r rC )
r r r r7 ra rb rc r0 r� �fspath)r rp rt r6 r r r �test_pathlib_file s "�z FileInputTests.test_pathlib_filec C s� t | �d��}t|ddd��}|�� }| �|d� td| � W d � n1 s)w Y t|dd��}| �|�� d� W d � d S 1 sHw Y d S )Nr� Tr )rY r z Modified %sr zModified Pathlib file.
)r r r r7 ra r_ r r� )r rp rt r6 r r r r �test_pathlib_file_inplace s �"�z(FileInputTests.test_pathlib_file_inplaceN)!r! r" r# r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Zignore_warningsr� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s>
r� c @ sd e Zd Z dddddd�dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Z dd� Z
dd� Z dS )�
MockFileInputNFr2 �r)r r1 r r� c C sJ || _ || _|| _|| _|| _|| _|| _d | _t� dd� �| _
i | _ d S )Nc S s dS r9 r r r r r r� 1 r� z(MockFileInput.__init__.<locals>.<lambda>) rM rY �backupr r1 r r� r� � collections� defaultdict�invocation_counts�
return_values)r rM rY r� r r1 r r� r r r r) ' s
zMockFileInput.__init__c C s | j d d7 < d S )Nr> rC )r� r( r r r r> 4 s zMockFileInput.closec C � | j d d7 < | jd S )Nrg rC �r� r� r( r r r rg 7 �
zMockFileInput.nextfilec C r� )Nr0 rC r� r( r r r r0 ; r� zMockFileInput.filenamec C r� )Nrb rC r� r( r r r rb ? r� zMockFileInput.linenoc C r� )Nrc rC r� r( r r r rc C r� zMockFileInput.filelinenoc C r� )Nr� rC r� r( r r r r� G r� zMockFileInput.filenoc C r� )Nre rC r� r( r r r re K r� zMockFileInput.isfirstlinec C r� )Nrf rC r� r( r r r rf O r� zMockFileInput.isstdin)NFr2 ) r! r" r# r) r> rg r0 rb rc r� re rf r r r r r� $ s �
r� c @ �& e Zd Z dd� Zdd� Zdd� ZdS )� BaseFileInputGlobalMethodsTestc C s t j| _t j| _tt _d S r% )� fileinput�_state� _orig_stater �_orig_FileInputr� r( r r r �setUpW s
z$BaseFileInputGlobalMethodsTest.setUpc C s | j t_| jt_d S r% )r� r� r r� r� r( r r r �tearDown\ s z'BaseFileInputGlobalMethodsTest.tearDownc C s2 |j | }| �|d|� t|j �}| �|d� d S )NrC )r� ra r: )r Zmock_file_inputZ method_nameZ actual_countZactual_total_countr r r �assertExactlyOneInvocation` s
z9BaseFileInputGlobalMethodsTest.assertExactlyOneInvocationN)r! r" r# r� r� r� r r r r r� S s
r� c @ s. e Zd Z dd� Zdd� Zdd� Zdd� Zd S )
�Test_fileinput_inputc C sl t � }t� |_|t_| �t�� }t�� W d � n1 s w Y | �d|j j
� | � |tjd� d S )N)zinput() already active�fileinput._state) r� �objectr� r� r� r� r� �inputra r� r~ �assertIs)r �instancer� r r r �1test_state_is_not_None_and_state_file_is_not_Nonek s
�zFTest_fileinput_input.test_state_is_not_None_and_state_file_is_not_Nonec C s t � }d |_|t_| �� d S r% )r� r� r� r� �do_test_call_input�r r r r r �-test_state_is_not_None_and_state_file_is_Nonex s
zBTest_fileinput_input.test_state_is_not_None_and_state_file_is_Nonec C s d t _| �� d S r% )r� r� r r( r r r �test_state_is_None� s z'Test_fileinput_input.test_state_is_Nonec C s� t � }t � }t � }t � }t � }t � }tj||||||d�}| �|tjd� | �||jd� | �||jd� | �||jd� | �||jd� | �||j d� d S )N)rM rY r� r r1 r r rM rY r� r r1 )
r r� r r r� rM rY r� r r1 )r rM rY r� r r1 r r� r r r r � s
�z'Test_fileinput_input.do_test_call_inputN)r! r" r# r r r
r r r r r r h s
r c @ � e Zd Z dd� Zdd� ZdS )�Test_fileinput_closec C s d t _t �� | �t j� d S r% )r� r� r> � assertIsNoner( r r r r
� s z'Test_fileinput_close.test_state_is_Nonec C s2 t � }|t_t�� | �|d� | �tj� d S )Nr> )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�r! r" r# r
r r r r r r � s r c @ r )�Test_fileinput_nextfilec C �Z d t _| �t�� }t �� W d � n1 sw Y | �d|jj� | �t j� d S �N)zno active input()) r� r� r� r� rg ra r� r~ r
�r r� r r r r
� �
�z*Test_fileinput_nextfile.test_state_is_Nonec C �P t � }t� }||jd< |t_t�� }| �|d� | �||� | �tj|� d S )Nrg )r r� r� r� r� rg r� r )r Znextfile_retvalr �retvalr r r r � �
z.Test_fileinput_nextfile.test_state_is_not_NoneNr r r r r r � �
r c @ r )�Test_fileinput_filenamec C r r ) r� r� r� r� r0 ra r� r~ r
r r r r r
� r z*Test_fileinput_filename.test_state_is_Nonec C r )Nr0 )r r� r� r� r� r0 r� r )r Zfilename_retvalr r r r r r � r z.Test_fileinput_filename.test_state_is_not_NoneNr r r r r r � r r c @ r )�Test_fileinput_linenoc C r r ) r� r� r� r� rb ra r� r~ r
r r r r r
� r z(Test_fileinput_lineno.test_state_is_Nonec C r )Nrb )r r� r� r� r� rb r� r )r Z
lineno_retvalr r r r r r � r z,Test_fileinput_lineno.test_state_is_not_NoneNr r r r r r � r r c @ r )�Test_fileinput_filelinenoc C r r ) r� r� r� r� rc ra r� r~ r
r r r r r
r z,Test_fileinput_filelineno.test_state_is_Nonec C r )Nrc )r r� r� r� r� rc r� r )r Zfilelineno_retvalr r r r r r r z0Test_fileinput_filelineno.test_state_is_not_NoneNr r r r r r r r c @ r )�Test_fileinput_filenoc C r r ) r� r� r� r� r� ra r� r~ r
r r r r r
) r z(Test_fileinput_fileno.test_state_is_Nonec C sV t � }t� }||jd< ||_|t_t�� }| �|d� | �||� | �tj|� d S )Nr� ) r r� r� �
fileno_retvalr� r� r� r� r )r r r r r r r r 3 s
z,Test_fileinput_fileno.test_state_is_not_NoneNr r r r r r & r r c @ r )�Test_fileinput_isfirstlinec C r r ) r� r� r� r� re ra r� r~ r
r r r r r
E r z-Test_fileinput_isfirstline.test_state_is_Nonec C r )Nre )r r� r� r� r� re r� r )r Zisfirstline_retvalr r r r r r O r z1Test_fileinput_isfirstline.test_state_is_not_NoneNr r r r r r B r r c @ r )�Test_fileinput_isstdinc C r r ) r� r� r� r� rf ra r� r~ r
r r r r r
` r z)Test_fileinput_isstdin.test_state_is_Nonec C r )Nrf )r r� r� r� r� rf r� r )r Zisstdin_retvalr r r r r r j r z-Test_fileinput_isstdin.test_state_is_not_NoneNr r r r r r ] r r c @ rx )�InvocationRecorderc C r� r9 )�invocation_countr( r r r r) z r* zInvocationRecorder.__init__c O s" | j d7 _ ||f| _t�d�S )NrC s
some bytes)r! �last_invocationr� r r} r r r r� } s
zInvocationRecorder.__call__Nr� r r r r r x s r c @ s� e Zd Z dd� Zdd� Zdd� Ze�ed�dd � �Z e�ed�d
d � �Z
e�e d �d
d� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zed d � �Zd S ) �Test_hook_compressedc C s t � | _d S r% )r � fake_openr( r r r r� � r� zTest_hook_compressed.setUpc C � | � dd� d S )Nr2 r� �� do_test_use_builtin_open_textr( r r r �test_empty_string� r� z&Test_hook_compressed.test_empty_stringc C r% �N�abcdr� r& r( r r r � test_no_ext� r� z Test_hook_compressed.test_no_extzRequires gzip and zlibc C �V t j}| jt _z t�dd�}W |t _n|t _w | �| jjd� | �| jjdi f� d S )N�test.gzr� rC )r- r� )�gzipr r$ r� �hook_compressedra r! r" �r �
original_openr� r r r �test_gz_ext_fake� � z%Test_hook_compressed.test_gz_ext_fakec C sJ t j}dd� t _z
tjdddd�}W |t _n|t _w | �t|�dg� d S )Nc S r� )Ns Ex-binary string)r� r r� r r r r� � s
zATest_hook_compressed.test_gz_with_encoding_fake.<locals>.<lambda>r- r� r r zEx-binary string)r. r r� r/ ra r` r0 r r r �test_gz_with_encoding_fake� s
z/Test_hook_compressed.test_gz_with_encoding_fakez Requires bz2c C r, )N�test.bz2r� rC )r5 r� )�bz2ZBZ2Filer$ r� r/ ra r! r" r0 r r r �test_bz2_ext_fake� r3 z&Test_hook_compressed.test_bz2_ext_fakec C r% )Nz abcd.blahr� �� do_test_use_builtin_open_binaryr( r r r �
test_blah_ext� r� z"Test_hook_compressed.test_blah_extc C r% )Nzabcd.Gzr� r8 r( r r r �test_gz_ext_builtin� r� z(Test_hook_compressed.test_gz_ext_builtinc C r% )Nzabcd.Bz2r� r8 r( r r r �test_bz2_ext_builtin� r� z)Test_hook_compressed.test_bz2_ext_builtinc C r% )Nr* r� r8 r( r r r �test_binary_mode_encoding� r� z.Test_hook_compressed.test_binary_mode_encodingc C r% r) r& r( r r r �test_text_mode_encoding� r� z,Test_hook_compressed.test_text_mode_encodingc C sf | � | j�}z
t�||�}W | � |� n| � |� w | �| jjd� | �| jj||fd d d�f� d S )NrC �r r� ��replace_builtin_openr$ r� r/ ra r! r" �r r0 r r1 r� r r r r9 � �
�z4Test_hook_compressed.do_test_use_builtin_open_binaryc C sf | � | j�}z
t�||�}W | � |� n| � |� w | �| jjd� | �| jj||fdd d�f� d S )NrC �localer? r@ rB r r r r' � rC z2Test_hook_compressed.do_test_use_builtin_open_textc C s t j}| t _|S r% )�builtinsr )Z
new_open_funcr1 r r r rA � s z)Test_hook_compressed.replace_builtin_openN)r! r" r# r� r( r+ �unittestZ
skipUnlessr. r2 r4 r6 r7 r: r; r< r= r> r9 r' � staticmethodrA r r r r r# � s(
r# c @ r� )�Test_hook_encodedc C s� t � }t � }tj||d�}t� }tj}|t_zt � }t � }|||�}W |t_n|t_w | �|jd� |j\} }
| � | d |� | � | d |� | � |
�
d�|� | � |
�
d�|� | � |
� d S )N�r� rC r r r� ) r r� r r rE r ra r! r" r �poprd ) r r r� r� r$ r1 r0 r Z open_resultr~ r r r r �test� s$
zTest_hook_encoded.testc s� t td��
}|�d� W d � n1 sw Y � �tt� � fdd�}|ddg� � �t�� |ddg� W d � n1 sBw Y |dd g� |d
d g� d S ) Nr� s �abcc sN t tdtd| d�d�� }t|�}W d � n1 sw Y � �||� d S )Nr� r rI �rM r r1 �r r r r` ra )r� �expected_linesrt r; r( r r �check� s
�
�z,Test_hook_encoded.test_errors.<locals>.check�ignore�abc�strictr� u �abc�backslashreplacez\x80abc)r r r r r� r� r� �r r rO r r( r � test_errors� s � � z Test_hook_encoded.test_errorsc s t td��
}|�d� W d � n1 sw Y � �tt� � fdd�}|dg d�� � �t�� |dg d�� W d � n1 sDw Y � �t�� |dg d�� W d � n1 s`w Y � �t�� |d g d
�� W d � d S 1 s}w Y d S ) Nr� s
A
B
C
D+IKw-c sJ t t| td�d�� }t|�}W d � n1 sw Y � �||� d S )Nzutf-7rL rM )r rN rt r; r( r r rO s �
�z+Test_hook_encoded.test_modes.<locals>.checkr� )r� r� r� � D€ZrUr� r� )r� zB
zC
rV ) r r r r r� Z assertWarnsr� r� r� rT r r( r �
test_modes s � � � "�z Test_hook_encoded.test_modesN)r! r" r# rK rU rW r r r r rH � s
rH c @ r@ )�MiscTestc C r� r% )r Z check__all__r� r( r r r �test_all r� zMiscTest.test_allN)r! r" r# rY r r r r rX s rX �__main__)4r� r� rh r] r� r� rE r rF r6 � ImportErrorr. r r r r Zpathlibr Z test.supportr Ztest.support.os_helperr r r� r
r rK r r
r r$ ZTestCaserA ry r� r� r� r r r r r r r r r r r# rH rX r! �mainr r r r �<module> sn � � $S s/? YA
�