File: //usr/local/lib/python3.7/test/__pycache__/test_tempfile.cpython-37.pyc
B
��g � � @ s: d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl Z d dl
mZ d dlmZ eed�r�d dlZdZnd Ze je jkZeed�Zej�d�r�d Znd
ZG d d � d e j�Z G d
d� de j�Z G dd� de �Z G dd� de �Z G dd� de �Z G dd� de �Z!G dd� de �Z"e j#dd� �Z$dd � Z%G d d � d �Z&G d d � d e&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&e �Z+G d)d*� d*e �Z,G d+d,� d,e �Z-G d-d.� d.e �Z.e j/e j0k �rG d/d0� d0e �Z1G d1d2� d2�Z2G d3d4� d4e �Z3e4d5k�r6e �5� dS )6� N)�mock)�support)�
script_helper�stat� �spawnlZopenbsd�0 �d c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestLowLevelInternalsc C s: | � tt�d�� | � tt�d�� | � tt�d �� d S )N� � )�assertIs�str�tempfile�_infer_return_type�bytes)�self� r �./usr/local/lib/python3.7/test/test_tempfile.py� test_infer_return_type_singles) s z4TestLowLevelInternals.test_infer_return_type_singlesc C sp | � tt�dd�� | � tt�dd�� | �t�� t�dd� W d Q R X | �t�� t�dd� W d Q R X d S )Nr r )r
r r r r � assertRaises� TypeError)r r r r � test_infer_return_type_multiples. s z6TestLowLevelInternals.test_infer_return_type_multiplesc C s� | � tt�d d�� | � tt�dd �� | � tt�d d �� | � tt�dd �� | � tt�d d�� | �t�� t�dd d� W d Q R X | �t�� t�dd d� W d Q R X d S )Nr r )r
r r r r r r )r r r r �)test_infer_return_type_multiples_and_none6 s z?TestLowLevelInternals.test_infer_return_type_multiples_and_nonec C s | � tt�t�d��� d S )N�/)r
r r r �pathlib�Path)r r r r � test_infer_return_type_pathlibA s z4TestLowLevelInternals.test_infer_return_type_pathlibN)�__name__�
__module__� __qualname__r r r r r r r r r
( s r
c @ s8 e Zd Ze�d�Ze�d�Zdd� Zdd� Zdd� Z d S )
� BaseTestCasez^[a-z0-9_-]{8}$s ^[a-z0-9_-]{8}$c C s* t �� | _| j�� tjdtdtd� d S )N�ignore�mktemp)�category�message�module)r �check_warnings�_warnings_manager� __enter__�warnings�filterwarnings�RuntimeWarningr )r r r r �setUpL s
zBaseTestCase.setUpc C s | j �d d d � d S )N)r( �__exit__)r r r r �tearDownR s zBaseTestCase.tearDownc
C s� t j�|�\}}|d t|�� }|t|�t|� d � }|d k rn| �t|�t|�tksbt|t j�rftnt d� |d k r�| �t|�t|�tkr�tnt d� |d k r�| �t|�t|�tkr�tnt d� |||fdkr�| �t|�td� | �
t j� |�t j� |�d||f � | �
||d||f � | �
||d||f � |t|�t|�t|� � }t|t��rb| j n| j
} | �| �|�d|| jf � d S )Nzunexpected return type)NNNz default return type must be strzfile %r not in directory %rz file %r does not begin with %rz file %r does not end with %rz$random characters %r do not match %r)�os�path�split�lenr
�typer �
isinstance�PathLiker � assertEqual�abspath� str_check�b_check�
assertTrue�match�pattern)
r �name�dir�pre�suf�ndir�nbaseZnpreZnsufZcheckr r r � nameCheckU s8 zBaseTestCase.nameCheckN)
r r r �re�compiler9 r: r- r/ rD r r r r r! G s
r! c @ s e Zd Zdd� ZdS )� TestExportsc C st t j}ddddddddddddddd�}g }x*|D ]"}|d dkr2||kr2|�|� q2W | �t|�dkd| � d S )Nr )�NamedTemporaryFile�
TemporaryFile�mkstemp�mkdtempr# �TMP_MAX�
gettempprefix�gettempprefixb�
gettempdir� gettempdirb�tempdir�template�SpooledTemporaryFile�TemporaryDirectoryr �_zunexpected keys: %s)r �__dict__�appendr; r3 )r �dict�expectedZunexp�keyr r r � test_exports{ s*
zTestExports.test_exportsN)r r r r[ r r r r rG z s rG c sR e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Ze� e
e d
�d �d d
� �Z � Z
S )�TestRandomNameSequencez6Test the internal iterator object _RandomNameSequence.c s t �� | _t� �� d S )N)r �_RandomNameSequence�r�superr- )r )� __class__r r r- � s
z TestRandomNameSequence.setUpc C s t | j�}| �|ddd� d S )Nr )�nextr^ rD )r �sr r r �test_get_six_char_str� s
z,TestRandomNameSequence.test_get_six_char_strc C sL i }| j }x<tt�D ]0}t|�}| �|ddd� | �||� d||< qW d S )Nr r )r^ �range�
TEST_FILESra rD � assertNotIn)r rX r^ �irb r r r � test_many� s z TestRandomNameSequence.test_manyc C s. d}| j }x |D ]}|d7 }|dkrP qW d S )Nr r � )r^ )r rg r^ rb r r r �
supports_iter� s
z$TestRandomNameSequence.supports_iter�forkz!os.fork is required for this testc C s� t �� \}}d }zft �� }|sTt �|� t �|t| j��d�� t �|� t �d� t| j�}t � |t
|��� d�}W d |r�yt � |t
j� W n tk
r� Y nX t �|d� t �|� t �|� X | �||� d S )N�asciir )r0 �piperk �close�writera r^ �encode�_exit�readr3 �decode�kill�signal�SIGKILL�OSError�waitpid�assertNotEqual)r Zread_fdZwrite_fd�pidZ parent_valueZ child_valuer r r �test_process_awareness� s(
z-TestRandomNameSequence.test_process_awareness)r r r �__doc__r- rc rh rj �unittest�
skipUnless�hasattrr0 r{ �
__classcell__r r )r` r r\ � s
r\ c @ s e Zd ZdZdd� Zdd� ZdS )�TestCandidateTempdirListz3Test the internal function _candidate_tempdir_list.c C s8 t �� }| �t|�dk� x|D ]}| �|t� q W d S )Nr )r �_candidate_tempdir_list� assertFalser3 �assertIsInstancer )r �cand�cr r r �test_nonempty_list� s
z+TestCandidateTempdirList.test_nonempty_listc
C s� t �� ��}x*dD ]"}t�|�}|stj�|�||< qW t�� }x*dD ]"}t�|�}|sZt�| � ||� qDW y t�
� }W n t t fk
r� tj
}Y nX | � ||� W d Q R X d S )N)�TMPDIRZTEMPZTMP)r �EnvironmentVarGuardr0 �getenvr1 r8 r r� �
ValueError�assertIn�getcwd�AttributeErrorrw �curdir)r �envZenvname�dirnamer� r r r �test_wanted_dirs� s
z)TestCandidateTempdirList.test_wanted_dirsN)r r r r| r� r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�TestGetDefaultTempdirz Test _get_default_tempdir().c s t �� ���fdd�}t�t d|��� t �� | �t���g � dd� �t�td���6 | � t
�� t �� W d Q R X | �t���g � W d Q R X � �fdd�}t�td|��6� | � t
�� t �� W d Q R X | �t���g � W d Q R X W d Q R X W d Q R X d S ) Nc s � gS )Nr r )�our_temp_directoryr r �our_candidate_list s zKTestGetDefaultTempdir.test_no_files_left_behind.<locals>.our_candidate_listr� c _ s
t � �d S )N)rw )�args�kwargsr r r �
raise_OSError s zFTestGetDefaultTempdir.test_no_files_left_behind.<locals>.raise_OSError�openc s � | |�}�|_ |S )N)ro )r� r� �fp)� orig_openr� r r �
bad_writer# s
zCTestGetDefaultTempdir.test_no_files_left_behind.<locals>.bad_writer) r rT r � swap_attrZ_get_default_tempdirr7 r0 �listdir�ior �FileNotFoundError)r r� r� r )r� r� r� r �test_no_files_left_behind
s
z/TestGetDefaultTempdir.test_no_files_left_behindN)r r r r| r� r r r r r�
s r� c @ s e Zd ZdZdd� Zdd� ZdS )�TestGetCandidateNamesz0Test the internal function _get_candidate_names.c C s t �� }| �|t j� d S )N)r �_get_candidate_namesr� r] )r �objr r r � test_retval2 s z!TestGetCandidateNames.test_retvalc C s" t �� }t �� }| �||k� d S )N)r r� r; )r �a�br r r �test_same_thing7 s z%TestGetCandidateNames.test_same_thingN)r r r r| r� r� r r r r r� / s r� c c s> t �� } z$t�t d| �� d V W d Q R X W d t�| � X d S )NrQ )r rK r r� �rmtree)r? r r r �_inside_empty_temp_dir? s
r� c s t �td� fdd��S )Nr� c s t � �S )N)�iterr )�namesr r �<lambda>L r z'_mock_candidate_names.<locals>.<lambda>)r r� r )r� r )r� r �_mock_candidate_namesI s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestBadTempdirc C s� t � �� t�tj�j }}|tjtjB tjB M }t� tj|� zPt�
tjtj �r\| � d� | �
t�� | �� W d Q R X | �t�tj�g � W d t� tj|� X W d Q R X d S )Nz!can't set the directory read-only)r� r0 r r rQ �st_mode�S_IWUSR�S_IWGRP�S_IWOTH�chmod�access�W_OKZskipTestr �PermissionError� make_tempr7 r� )r Zoldmode�moder r r �test_read_only_directoryQ s
z'TestBadTempdir.test_read_only_directoryc C s^ t � �N tj�tjd�}t�td|��$ | �t �� | �
� W d Q R X W d Q R X W d Q R X d S )N� nonexistentrQ ) r� r0 r1 �joinr rQ r r� r r� r� )r rQ r r r �test_nonexisting_directory_ s
z)TestBadTempdir.test_nonexisting_directoryc C sp t � �` tj�tjd�}t|d��� t� td|��( | �
t t f�� | �
� W d Q R X W d Q R X W d Q R X d S )N�file�wbrQ )r� r0 r1 r� r rQ r� rn r r� r �NotADirectoryErrorr� r� )r rQ r r r �test_non_directoryf s z!TestBadTempdir.test_non_directoryN)r r r r� r� r� r r r r r� O s r� c @ s� e Zd ZdZG dd� d�Zd dd�Zdd � Zd
d � Zd d
� Zdd� Z e
� e d�dd� �Z
e
� ed�dd� �Ze
� ed�dd� �Zdd� Zdd � Zd d � ZdS ) �TestMkstempInnerz*Test the internal function _mkstemp_inner.c @ s< e Zd ZejZejZej Z
ej Z dd� Z
dd� Zdd� ZdS )zTestMkstempInner.mkstempedc C s> |r | j }n| j}t�|||�}t�|||||�\| _| _d S )N)�_bflags�_tflagsr r �_mkstemp_inner�fdr> )r r? r@ rA �bin�flags� output_typer r r �__init__x s
z#TestMkstempInner.mkstemped.__init__c C s t �| j|� d S )N)r0 ro r� )r r r r r ro s z TestMkstempInner.mkstemped.writec C s | � | j� | �| j� d S )N)�_closer� �_unlinkr> )r r r r �__del__� s z"TestMkstempInner.mkstemped.__del__N)r r r r �_bin_openflagsr� �_text_openflagsr� r0 rn r� �unlinkr� r� ro r� r r r r � mkstempedr s r� Nr c C sr t �|||�}|d kr0|tkr(t �� }nt �� }|d kr>|� }|d krL|� }| �||||�}| �|j|||� |S )N)r r r rO rP r� rD r> )r r? r@ rA r� r� r� r r r � do_create� s
zTestMkstempInner.do_createc C s^ | � � �d� | j dd��d� | j dd��d� | j ddd��d� | j ddd��d� d S ) Ns blatr� )r@ r� )rA )r@ rA �aaz.txt)r� ro )r r r r �
test_basic� s
zTestMkstempInner.test_basicc C s� t �� }| j|dd��d� | j|dd��d� | j|dd��d� | j|ddd��d� | j|dd d��d� | �t�� | jd
dd��d� W d Q R X | �t�� | j|d
d��d� W d Q R X | �t�� | j|dd
d��d� W d Q R X d S ) Nr )r? rA s blat� a)r? r@ � b)r? r@ rA s aas .txtr )r rP r� ro r r )r Zdir_br r r �test_basic_with_bytes_names� s z,TestMkstempInner.test_basic_with_bytes_namesc C s. t tt��}x |D ]}| jdd�||< qW d S )Nr� )r@ )�listrd re r� )r �extantrg r r r �test_basic_many� s
z TestMkstempInner.test_basic_manyc C sH t �� }z.| j|d��d� | jt�|�d��d� W d t�|� X d S )N)r? s blat)r rK r� ro r r r0 �rmdir)r r? r r r �test_choose_directory� s
z&TestMkstempInner.test_choose_directoryzos.stat not availablec C sJ | � � }t�t�|j�j�}d}tjdkr:|d? }|d }| �||� d S )Ni� �win32� �I ) r� r �S_IMODEr0 r> r� �sys�platformr7 )r r� r� rY �userr r r �test_file_mode� s
z TestMkstempInner.test_file_modezos.spawnl not availablec C s� t jr d}nd}| �� }| �t�|j�d� d|j }yt}W n tk
r\ t j
d }Y nX tj � tj �
tj �|��d�}t jdkr�dt j }d| }nt j}t�tjt j||||�}| �|dk d | � | �|dkd
| � d S ) N�v�qFz%dr ztf_inherit_check.pyr� z"%s"z$child process caught fatal signal %dz child process reports failure %d)r �verboser� r7 r0 �get_inheritabler� �__file__� NameErrorr� �argvr1 r� r� r8 r� �
executabler �P_WAITr� )r r� r� r� �meZtesterZ decoratedZretvalr r r �test_noinherit� s(
z TestMkstempInner.test_noinheritztext mode not availablec C sL | j dd�}|�d� |�d� t�|jdtj� | �t�|jd�d� d S )Nr )r� s blats extra
ri s blat)r� ro r0 �lseekr� �SEEK_SETr7 rr )r �fr r r �
test_textmode� s
z TestMkstempInner.test_textmodec C s t �t �� t �� dt jt�S )Nr )r r� rO rM r� r )r r r r r� � s
zTestMkstempInner.make_tempc
C sz t � �j tddd��R | �� \}}t�|� | �|�d�� | �� \}}t�|� | �|�d�� W d Q R X W d Q R X d S )N�aaa�bbb)r� r� r� r0 rn r; �endswith)r Zfd1Zname1Zfd2Zname2r r r �!test_collision_with_existing_file s
z2TestMkstempInner.test_collision_with_existing_filec
C sl t � �\ tddd��D t�� }| �|�d�� | �� \}}t�|� | �|�d�� W d Q R X W d Q R X d S )Nr� r� ) r� r� r rK r; r� r� r0 rn )r r? r� r> r r r �&test_collision_with_existing_directory s
z7TestMkstempInner.test_collision_with_existing_directory)NNNr )r r r r| r� r� r� r� r� r� r} r~ �has_statr� �
has_spawnlr� � has_textmoder� r� r� r� r r r r r� o s
&
r� c @ s e Zd ZdZdd� Zdd� ZdS )�TestGetTempPrefixzTest gettempprefix().c C sL t �� }| �|t� | �t|�d� t �� }| �|t� | �t|�d� d S )Nr )r rM r� r Z
assertGreaterr3 rN r )r �pZpbr r r �test_sane_template! s z$TestGetTempPrefix.test_sane_templatec C sd t �� d }t jdd�}z:tj�||�}t�|tjtjB �}t� |� t�
|� W d t� |� X d S )Nz
xxxxxx.xxxr )�prefix) r rM rK r0 r1 r� r� �O_RDWR�O_CREATrn r� r� )r r� �dr� r r r �test_usable_template- s
z&TestGetTempPrefix.test_usable_templateN)r r r r| r� r r r r r r� s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �TestGetTempDirzTest gettempdir().c C sX xRt �� t �� fD ]>}| �tj�|�p.|tjkd| � | �tj�|�d| � qW d S )Nz%r is not an absolute pathz%r is not a directory) r rO rP r; r0 r1 �isabsr� �isdir)r r r r r �test_directory_existsA s
z$TestGetTempDir.test_directory_existsc C s t �� }|�d� |�� d S )Ns blat)r rH ro rn )r r� r r r �test_directory_writableJ s
z&TestGetTempDir.test_directory_writablec C sP t �� }t �� }t �� }| �||k� | �t|�t|�� | �|t�|�� d S )N) r rO rP r; ry r4 r7 r0 �fsdecode)r r� r� r� r r r r� T s z TestGetTempDir.test_same_thingc C s` t �d�}t jd }t _z0t�� � }||d< | �t �� |� W d Q R X W d |t _t�|� X d S )Nz-Tempr� )r rK rQ r r� r7 rO r� )r Zcase_sensitive_tempdirZ_tempdirr� r r r �test_case_sensitive^ s
z"TestGetTempDir.test_case_sensitiveN)r r r r| r r r� r
r r r r r > s
r c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) � TestMkstempzTest mkstemp().Nc
C s� t �|||�}|d kr0|tkr(t �� }nt �� }|d kr>|� }|d krL|� }t j|||d�\}}tj�|�\}}tj� |�} | �
| |d| |f � z| � ||||� W d t� |� t�
|� X d S )N)r? r� �suffixz+Directory '%s' incorrectly returned as '%s')r r r rO rP rJ r0 r1 r2 r8 r7 rD rn r� )
r r? r@ rA r� r� r> rB rC Zadirr r r r� p s$
zTestMkstemp.do_createc C sL | � � | j dd� | j dd� | j ddd� | j ddd� | j dd � d S )
Nr� )r@ r� )rA )r@ rA r� z.txt�.)r? )r� )r r r r r� � s zTestMkstemp.test_basicc C s� t �� }| j|dd� | j|dd� | j|dd� | j|ddd� | j|ddd� | jd d
� | �t�� | jd ddd� W d Q R X | �t�� | jd d dd� W d Q R X | �t�� | jd dd
d� W d Q R X d S )Nr )r? rA r� )r? r@ r� )r? r@ rA s aas .txt� .)r? r
r� z.txt)r rP r� r r )r r r r r r� � s z'TestMkstemp.test_basic_with_bytes_namesc C s< t �� }z"| j|d� | jt�|�d� W d t�|� X d S )N)r? )r rK r� r r r0 r� )r r? r r r r� � s
z!TestMkstemp.test_choose_directory)NNN)r r r r| r� r� r� r� r r r r r m s
r c @ sf e Zd ZdZdd� Zddd�Zdd� Zd d
� Zd d � Zd
d� Z e
� e d�dd� �Z
dd� Zdd� ZdS )� TestMkdtempzTest mkdtemp().c C s t �� S )N)r rK )r r r r r� � s zTestMkdtemp.make_tempNc C s� t �|||�}|d kr0|tkr(t �� }nt �� }|d kr>|� }|d krL|� }t j|||d�}y| �||||� |S t�|� � Y nX d S )N)r? r� r ) r r r rO rP rK rD r0 r� )r r? r@ rA r� r> r r r r� � s
zTestMkdtemp.do_createc C s^ t �| �� � t �| jdd�� t �| jdd�� t �| jddd�� t �| jddd�� d S )Nr� )r@ r� )rA )r@ rA r� z.txt)r0 r� r� )r r r r r� � s
zTestMkdtemp.test_basicc C s� t �� }t�| j|d�� t�| j|dd�� t�| j|dd�� t�| j|ddd�� t�| j|ddd�� | �t�� t�| j|d dd�� W d Q R X | �t�� t�| j|dd
d�� W d Q R X | �t�� t�| jd ddd�� W d Q R X d S ) N)r? r� )r? r@ r� )r? rA )r? r@ rA s aas .txtr� z.txtr )r rP r0 r� r� r r )r r r r r r� � s z'TestMkdtemp.test_basic_with_bytes_namesc
C sX t tt��}z"x |D ]}| jdd�||< qW W d x |D ]}t|t�r6t�|� q6W X d S )Nr� )r@ )r� rd re r� r5 r r0 r� )r r� rg r r r r� � s
zTestMkdtemp.test_basic_manyc C sH t �� }z.t�| j|d�� t�| jt�|�d�� W d t�|� X d S )N)r? )r rK r0 r� r� r r )r r? r r r r� � s
z!TestMkdtemp.test_choose_directoryzos.stat not availablec C sb | � � }zHt�t�|�j�}|dM }d}tjdkrB|d? }|d }| �||� W d t�|� X d S )Ni� i� r� r� r� ) r� r r� r0 r� r� r� r7 r� )r r? r� rY r� r r r � test_mode� s
zTestMkdtemp.test_modec
C sl t � �\ tddd��D tjdd�}|�� | �|j�d�� t�� }| �|�d�� W d Q R X W d Q R X d S )Nr� r� F)�delete) r� r� r rH rn r; r> r� rK )r r� r? r r r r� s z-TestMkdtemp.test_collision_with_existing_filec
C s^ t � �N tddd��6 t�� }| �|�d�� t�� }| �|�d�� W d Q R X W d Q R X d S )Nr� r� )r� r� r rK r; r� )r Zdir1Zdir2r r r r� s z2TestMkdtemp.test_collision_with_existing_directory)NNN)r r r r| r� r� r� r� r� r� r} r~ r� r r� r� r r r r r � s
r c sT e Zd ZdZ� fdd�Z� fdd�ZG dd� d�Zdd d
�Zd d � Zd
d� Z � Z
S )�
TestMktempzTest mktemp().c s t �� | _t� �� d S )N)r rK r? r_ r- )r )r` r r r- s
zTestMktemp.setUpc s& | j rt�| j � d | _ t� �� d S )N)r? r0 r� r_ r/ )r )r` r r r/ " s zTestMktemp.tearDownc @ s( e Zd ZejZejZdd� Z dd� Z
dS )zTestMktemp.mktempedc C s. t j|||d�| _t�t�| j| jd�� d S )N)r? r� r i� )r r# r> r0 rn r� r� )r r? r@ rA r r r r� , s z TestMktemp.mktemped.__init__c C s | � | j� d S )N)r� r> )r r r r r� 2 s zTestMktemp.mktemped.__del__N) r r r r0 r� r� r r� r� r� r� r r r r �mktemped( s r r c C s( | � | j||�}| �|j| j||� |S )N)r r? rD r> )r r@ rA r� r r r r� 5 s zTestMktemp.do_createc C s@ | � � | j dd� | j dd� | j ddd� | j ddd� d S )Nr� )r@ r� )rA )r@ rA r� z.txt)r� )r r r r r� ; s
zTestMktemp.test_basicc C s. t tt��}x |D ]}| jdd�||< qW d S )Nr� )r@ )r� rd re r� )r r� rg r r r rh C s
zTestMktemp.test_many)r r ) r r r r| r- r/ r r� r� rh r� r r )r` r r s
r c @ sj e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS ) �TestNamedTemporaryFilezTest NamedTemporaryFile().Nr Tc C s8 |d krt �� }t j||||d�}| �|j|||� |S )N)r? r� r r )r rO rH rD r> )r r? r@ rA r r� r r r r� X s
z TestNamedTemporaryFile.do_createc C s@ | � � | j dd� | j dd� | j ddd� | j ddd� d S )Nr� )r@ r� )rA )r@ rA r� z.txt)r� )r r r r r� b s
z!TestNamedTemporaryFile.test_basicc C sP | � � }t�|�}|j}|j}~|d� ~|d� ~tjdd�rL| �|� � d S )Ns foos barT)Zcpython)r� �weakref�refro r Zcheck_impl_detail� assertIsNone)r r� �wrro Zwrite2r r r �test_method_lookupj s
z)TestNamedTemporaryFile.test_method_lookupc sV dddg� � fdd�}x&t |� �D ]\}}| �|� | � q"W | �|t� �d � d S )Ns spam
s eggs
s beans
c s* t jdd�} | �d�� �� | �d� | S )Nzw+b)r� r r )r rH ro r� �seek)r� )�linesr r � make_file~ s
z3TestNamedTemporaryFile.test_iter.<locals>.make_filer )� enumerater7 r3 )r r rg �lr )r r � test_iterz s
z TestNamedTemporaryFile.test_iterc C s( t �� }| �tj�|j�d|j � d S )Nz$NamedTemporaryFile %s does not exist)r rH r; r0 r1 �existsr> )r r� r r r �test_creates_named� s z)TestNamedTemporaryFile.test_creates_namedc C sX t �� }z>t j|d�}|�d� |�� | �tj�|j �d|j � W d t�
|� X d S )N)r? s blatz(NamedTemporaryFile %s exists after close) r rK rH ro rn r� r0 r1 r r> r� )r r? r� r r r �test_del_on_close� s
z(TestNamedTemporaryFile.test_del_on_closec C sv t �� }d }zFt j|dd�}|j}|�d� |�� | �tj� |j�d|j � W d |d k rft�
|� t� |� X d S )NF)r? r s blatz)NamedTemporaryFile %s missing after close) r rK rH r> ro rn r; r0 r1 r r� r� )r r? �tmpr� r r r �test_dis_del_on_close� s
z,TestNamedTemporaryFile.test_dis_del_on_closec C s. t �� }|�d� |�� |�� |�� d S )Ns abc
)r rH ro rn )r r� r r r �test_multiple_close� s
z*TestNamedTemporaryFile.test_multiple_closec sX t �� �� | �tj�� j�� W d Q R X | �tj�� j�� � fdd�}| �t |� d S )Nc s � � W d Q R X d S )Nr r )r� r r �
use_closed� s z?TestNamedTemporaryFile.test_context_manager.<locals>.use_closed)
r rH r; r0 r1 r r> r� r r� )r r& r )r� r �test_context_manager� s
z+TestNamedTemporaryFile.test_context_managerc
sn g � t j�� �fdd�}tjd|d��> tjdtd��$ | �ttj� | �t � �d� W d Q R X W d Q R X d S )Nc s � � | � �| � d S )N)rW )r� )�closed�os_closer r rn � s
z5TestNamedTemporaryFile.test_no_leak_fd.<locals>.closezos.close)� side_effectzio.openr )
r0 rn r �patchr� r r rH r7 r3 )r rn r )r( r) r �test_no_leak_fd� s z&TestNamedTemporaryFile.test_no_leak_fdc C st t �� }| �tj|� | �t�� t jd|d� W d Q R X | �t�� t jd|d� W d Q R X | � t
� |�g � d S )Nr )r� r? � ) r rK Z
addCleanupr r� r r� rH r r7 r0 r� )r r? r r r �
test_bad_mode� s z$TestNamedTemporaryFile.test_bad_mode)Nr r T)r r r r| r� r� r r r! r" r$ r% r' r, r. r r r r r U s
r c @ s� e Zd ZdZd+dd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� ZdS ),�TestSpooledTemporaryFilez Test SpooledTemporaryFile().r Nr c C s& |d krt �� }t j||||d�}|S )N)�max_sizer? r� r )r rO rS )r r0 r? r@ rA r� r r r r� � s z"TestSpooledTemporaryFile.do_createc C s4 | � � }| �|j� | j dddd�}| �|j� d S )Nr r� z.txt)r0 r@ rA )r� r� �_rolled)r r� r r r r� � s z#TestSpooledTemporaryFile.test_basicc C s~ t �� }zdt jd|d�}| �|j� |�d� | �|j� |j}|�� | �t |t
�o`t j �
|�d| � W d t �|� X d S )N�
)r0 r? s blat blat blat blat blat z*SpooledTemporaryFile %s exists after close)r rK rS r� r1 ro r; r> rn r5 r r0 r1 r r� )r r? r� �filenamer r r r" � s
z*TestSpooledTemporaryFile.test_del_on_closec C sP | j dd�}| �|j� x&td�D ]}|�dd� |�d� q"W | �|j� d S )N� )r0 � r s xxxxxxxxxxxxxxxxxxxx)r� r� r1 rd r ro )r r� rg r r r �test_rewrite_small� s z+TestSpooledTemporaryFile.test_rewrite_smallc C s^ | j dd�}| �|j� |�d� | �|j� |�d� | �|j� |�d� | �|j� d S )Nr4 )r0 s xxxxxxxxxxxxxxxxxxxxs
xxxxxxxxxx� x)r� r� r1 ro r; )r r� r r r �test_write_sequential� s
z.TestSpooledTemporaryFile.test_write_sequentialc C s@ | � � }|�d� |�d�}| �|d� |�� }| �|d� d S )N)r7 � y� zr s xyz)r� �
writelinesr r7 rr )r r� �pos�bufr r r �test_writelines s
z(TestSpooledTemporaryFile.test_writelinesc C s< | j dd�}|�d� | �|j� |�d� | �|j� d S )N�# )r0 )s xxxxxxxxxxxxxxxxxxxxs
xxxxxxxxxxs xxxxxr7 )r� r; r� r1 ro r; )r r� r r r �test_writelines_sequential s
z3TestSpooledTemporaryFile.test_writelines_sequentialc C sV | j dd�}| �|j� |�dd�}| �|d� | �|j� |�d� | �|j� d S )Nr4 )r0 r r r7 )r� r� r1 r r7 ro r; )r r� r< r r r � test_sparse s
z$TestSpooledTemporaryFile.test_sparsec C s: | j dd�}| �|j� | �|�� dk� | �|j� d S )Nr4 )r0 r )r� r� r1 r; �fileno)r r� r r r � test_fileno& s z$TestSpooledTemporaryFile.test_filenoc C s: t �� }|�d� | �|j� |�� |�� |�� d S )Ns abc
)r rS ro r� r1 rn )r r� r r r �#test_multiple_close_before_rollover- s
z<TestSpooledTemporaryFile.test_multiple_close_before_rolloverc C s> t jdd�}|�d� | �|j� |�� |�� |�� d S )Nr )r0 s abc
)r rS ro r; r1 rn )r r� r r r �"test_multiple_close_after_rollover6 s
z;TestSpooledTemporaryFile.test_multiple_close_after_rolloverc C sL | j dd�}|j}|j}|j}|d� |d� |dd� | �|d�d� d S )Nr4 )r0 s# aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaas# bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbr �F sF aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb)r� rr ro r r7 )r r� rr ro r r r r �test_bound_methods? s
z+TestSpooledTemporaryFile.test_bound_methodsc C s� t jdd�}|�d� | �|j� | �|jd� | �|j� | � t
�� |j W d Q R X | � t
�� |j W d Q R X |�d� | �
|j� | �|jd� | �|j� | � t
�� |j W d Q R X | � t
�� |j W d Q R X d S )Nr2 )r0 s
xxxxxxxxxxzw+br7 zrb+)r rS ro r� r1 r7 r� r r> r r� �newlines�encodingr; �assertIsNotNone)r r� r r r �test_propertiesM s"
z(TestSpooledTemporaryFile.test_propertiesc C s, t jdddd�}|�d� |�d� | �|�� d� |�d� |�d� | �|�� d� | �|j� | �|jd� | � |j
� | �|j t j
� | �|jd� |�d � |�d� | �|�� d
� |�d � |�d� | �|�� d � | �|j� | �|jd� | �|j
� | �|j t j
� | �|jd� d S )
Nzw+r2 zutf-8)r� r0 rI zabc
r zdef
zabc
def
zxyzzy
zabc
def
xyzzy
zfoobar
zabc
def
xyzzy
foobar
)r rS ro r r7 rr r� r1 r� r r> rH r0 �lineseprI r; rJ )r r� r r r �test_text_modea s0
z'TestSpooledTemporaryFile.test_text_modec C s� t jddddd�}|�d� |�d� | �|�� d� | �|j� | �|jd� | � |j
� | � |j � | �|j
d� |�d� |�d � |�d� | �|�� d
� | �|j� | �|jd� | � |j
� | � |j � | �|j
d� d S ) Nzw+r2 r zutf-8)r� r0 �newlinerI u Λ
r u ΜΜΜΜΜΜΜΜΜΜ
u( ΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝuB Λ
ΜΜΜΜΜΜΜΜΜΜ
ΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝΝ)r rS ro r r7 rr r� r1 r� r r> rJ rH rI r; )r r� r r r � test_text_newline_and_encoding s(
z7TestSpooledTemporaryFile.test_text_newline_and_encodingc sX t jdd�� � | �� j� | �� j� W d Q R X | �� j� � fdd�}| �t|� d S )Nr )r0 c s � � W d Q R X d S )Nr r )r� r r r& � s zQTestSpooledTemporaryFile.test_context_manager_before_rollover.<locals>.use_closed)r rS r� r1 r( r; r r� )r r& r )r� r �$test_context_manager_before_rollover� s z=TestSpooledTemporaryFile.test_context_manager_before_rolloverc sv t jdd��<� | �� j� � �d� � �� | �� j� | �� j� W d Q R X | �� j� � fdd�}| �t |� d S )Nr )r0 s abc
c s � � W d Q R X d S )Nr r )r� r r r& � s zQTestSpooledTemporaryFile.test_context_manager_during_rollover.<locals>.use_closed)
r rS r� r1 ro �flushr; r( r r� )r r& r )r� r �$test_context_manager_during_rollover� s
z=TestSpooledTemporaryFile.test_context_manager_during_rolloverc sn t jdd�� � �d� � �� | �� j� � � | �� j� W d Q R X | �� j� � fdd�}| �t |� d S )Nr )r0 s abc
c s � � W d Q R X d S )Nr r )r� r r r& � s zPTestSpooledTemporaryFile.test_context_manager_after_rollover.<locals>.use_closed)
r rS ro rQ r; r1 r� r( r r� )r r& r )r� r �#test_context_manager_after_rollover� s
z<TestSpooledTemporaryFile.test_context_manager_after_rolloverc C s� t jdd�}|�d� |�d� |�� | �|j� | �|j� � d� t jdd�}|�d� |�d� | �|j� | �|j� � d� t jdd�}|�d� |�d� | �
|j� t r�| �t �
|�� �jd� d S ) Nr2 )r0 s abcdefg
r r � s abcdri )r rS ro r �truncater� r1 r7 Z_file�getvaluer; r� r0 �fstatrB �st_size)r r� r r r �!test_truncate_with_size_parameter� s"
z:TestSpooledTemporaryFile.test_truncate_with_size_parameter)r Nr r )r r r r| r� r� r" r6 r8 r> r@ rA rC rD rE rG rK rM rO rP rR rS rY r r r r r/ � s(
r/ c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TestTemporaryFilezTest TemporaryFile().c C s t �� d S )N)r rI )r r r r r� � s z TestTemporaryFile.test_basicc C sR t �� }t j|d�}|�d� yt�|� W n |�� t�|� � Y nX d S )N)r? s blat)r rK rI ro r0 r� rn )r r? r� r r r �test_has_no_name� s
z"TestTemporaryFile.test_has_no_namec C s. t �� }|�d� |�� |�� |�� d S )Ns abc
)r rI ro rn )r r� r r r r% � s
z%TestTemporaryFile.test_multiple_closec s@ � fdd�}|dd� |dd� |dddd � |d
dd d � d S )
Nc s@ t j||��*}|�| � |�d� � �| |�� � W d Q R X d S )Nr )r rI ro r r7 rr )�inputr� r� Zfileobj)r r r � roundtrip� s
z;TestTemporaryFile.test_mode_and_encoding.<locals>.roundtrips 1234zw+bzabdc
zw+u Λzutf-16)rI zfoo
r )rN r )r r] r )r r �test_mode_and_encoding� s
z(TestTemporaryFile.test_mode_and_encodingc
sn g � t j�� �fdd�}tjd|d��> tjdtd��$ | �ttj� | �t � �d� W d Q R X W d Q R X d S )Nc s � � | � �| � d S )N)rW )r� )r( r) r r rn s
z0TestTemporaryFile.test_no_leak_fd.<locals>.closezos.close)r* zio.openr )
r0 rn r r+ r� r r rI r7 r3 )r rn r )r( r) r r, s z!TestTemporaryFile.test_no_leak_fdN) r r r r| r� r[ r% r^ r, r r r r rZ � s
rZ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
NulledModulesc G s& dd� |D �| _ dd� | j D �| _d S )Nc S s g | ]
}|j �qS r )rV )�.0�modr r r �
<listcomp> s z*NulledModules.__init__.<locals>.<listcomp>c S s g | ] }|� � �qS r )�copy)r` r r r r rb s )�refs�contents)r �modulesr r r r� s zNulledModules.__init__c C s* x$| j D ]}x|D ] }d ||< qW qW d S )N)rd )r r rZ r r r r) s
zNulledModules.__enter__c G s2 x,t | j| j�D ]\}}|�� |�|� qW d S )N)�ziprd re �clear�update)r �exc_infor r� r r r r. s zNulledModules.__exit__N)r r r r� r) r. r r r r r_ s r_ c @ sn e Zd ZdZddd�Zdd� Zd d
� Zejd d � �Z ej
d
d� �Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS )�TestTemporaryDirectoryzTest TemporaryDirectory().Nr r c C s� |d krt �� }t j|||d�}| �|j|||� |rR| �|j|||d �}d |_ttj� |jd�d��}|�
d� W d Q R X |S )N)r? r� r r ztest.txtr� s Hello world!) r rO rT rD r> r� r� r0 r1 r� ro )r r? r@ rA �recurser# �d1r� r r r r� & s z TestTemporaryDirectory.do_createc C sL t �� �}W d Q R X | �t��}t j|d� W d Q R X | �|jjtj� d S )N)r? )r rT r r� r7 Z exception�errno�ENOENT)r r� �cmr r r �test_mkdtemp_failure3 s
z+TestTemporaryDirectory.test_mkdtemp_failurec C sj t �� }zP| j|d�}| �tj�|j�d|j � |�� | � tj�|j�d|j � W d t�
|� X d S )N)r? z$TemporaryDirectory %s does not existz*TemporaryDirectory %s exists after cleanup) r rK r� r; r0 r1 r r> �cleanupr� r� )r r? r r r r �test_explicit_cleanup= s z,TestTemporaryDirectory.test_explicit_cleanupc C s� | � � }| j dd�}t�|jtj�|jd�� |�� | �tj�|j�d|j � | � tj�|j�d� | �
t� |j�dgd� |�� d S )Nr )rl Zfooz*TemporaryDirectory %s exists after cleanupz-Directory pointed to by a symlink was deletedztest.txtz>Contents of the directory pointed to by a symlink were deleted) r� r0 �symlinkr> r1 r� rr r� r r; r7 r� )r rm Zd2r r r �(test_cleanup_with_symlink_to_a_directoryJ s z?TestTemporaryDirectory.test_cleanup_with_symlink_to_a_directoryc C sJ t �� }z0| j|d�}|j}~| �tj�|�d| � W d t�|� X d S )N)r? z*TemporaryDirectory %s exists after __del__) r rK r� r> r� r0 r1 r r� )r r? r r> r r r �test_del_on_collection_ s z-TestTemporaryDirectory.test_del_on_collectionc C s� | � � �|}xtdD ]l}dj||d�}t�d|�\}}}|�� �� }| �tj� |�d| � |�dd�}| �
d|� | � d |� qW W d Q R X d S )
N)�builtinsr0 Zshutilr� r r* a� if True:
import builtins
import os
import shutil
import sys
import tempfile
import warnings
tmp = tempfile.TemporaryDirectory(dir={dir!r})
sys.stdout.buffer.write(tmp.name.encode())
tmp2 = os.path.join(tmp.name, 'test_dir')
os.mkdir(tmp2)
with open(os.path.join(tmp2, "test.txt"), "w") as f:
f.write("Hello world!")
{mod}.tmp = tmp
warnings.filterwarnings("always", category=ResourceWarning)
)r? ra z-cz*TemporaryDirectory %s exists after cleanupzutf-8�backslashreplacez
Exception z'ResourceWarning: Implicitly cleaning up) r� �formatr �assert_python_okrs �stripr� r0 r1 r rf r� )r r? ra �code�rc�out�err�tmp_namer r r �test_del_on_shutdownl s
z+TestTemporaryDirectory.test_del_on_shutdownc C s~ | � � �l}dj|d�}t�d|�\}}}|�� �� }| �tj� |�d| � |�dd�}| �
d|� | � d|� W d Q R X d S ) Na� if True:
import sys
import tempfile
import warnings
def generator():
with tempfile.TemporaryDirectory(dir={dir!r}) as tmp:
yield tmp
g = generator()
sys.stdout.buffer.write(next(g).encode())
warnings.filterwarnings("always", category=ResourceWarning)
)r? z-cz*TemporaryDirectory %s exists after cleanupzutf-8rx z
Exception z'ResourceWarning: Implicitly cleaning up) r� ry r rz rs r{ r� r0 r1 r rf r� )r r? r| r} r~ r r� r r r �test_exit_on_shutdown� s
z,TestTemporaryDirectory.test_exit_on_shutdownc
C sz | � � �h}| j |dd�}|j}tjdtfdd�� tjdtd� ~t�� W d Q R X | �t j
� |�d| � W d Q R X d S ) N� )r? rl Z
ImplicitlyF)�quiet�always)r$ z*TemporaryDirectory %s exists after __del__) r� r> r r' �ResourceWarningr* r+ Z
gc_collectr� r0 r1 r )r r? r r> r r r �test_warnings_on_cleanup� s
z/TestTemporaryDirectory.test_warnings_on_cleanupc C s$ | � � }|�� |�� |�� d S )N)r� rr )r r r r r r% � s z*TestTemporaryDirectory.test_multiple_closec C sN | � � }|�&}| �tj�|�� | �||j� W d Q R X | �tj�|�� d S )N)r� r; r0 r1 r r7 r> r� )r r r> r r r r' � s
z+TestTemporaryDirectory.test_context_manager)Nr r r )r r r r| r� rq rs r Zskip_unless_symlinkru Z cpython_onlyrv r� r� r� r% r' r r r r rk # s
rk �__main__)6r rn r� r0 r ru r� rE r* �
contextlibr r} r Ztestr Z test.supportr r r r� r� r� r� r� r� �
startswithre ZTestCaser
r! rG r\ r� r� r� �contextmanagerr� r� r� r� r� r r r r r r/ rH rI rZ r_ rk r �mainr r r r �<module> sf
3 H*%
0 /?m<~ >