Your IP : 216.73.216.213
a
XC?h�} � @ sd d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m Z d dl m Z d dl m!Z! z d dl"Z"W n e#�y( dZ"Y n0 z d dl$Z$W n e#�yN dZ$Y n0 z d dl%Z%W n e#�yt dZ%Y n0 z d dl&Z&dd� e&�'� D �Z(W n e#e)f�y� g Z(Y n0 zd dl*m+Z+m,Z, W n e#�y� ej- Z+Z,Y n0 d dl.m/Z/ d d l m0Z0m1Z1 d
Z2e3e d ��r e �4� d kZ2e3ed ��rDej5j6�rDej5j6�7d
�Z8nd
Z8ej �7d��o`e �9� d kZ:dd� Z;d�dd�Z<G dd� dej=�Z>G dd� dej=�Z?G dd� dej=�Z@G dd� dej=�ZAd d l mBZB G d d � d eBjC�ZDG d d � d ej=�ZEe�Fe3e d!�d"�G d#d$� d$eE��ZGG d%d&� d&eE�ZHe�Fe3e d!�d"�G d'd(� d(eG��ZIG d)d*� d*ej=�ZJe�Fe3e d+�d,�G d-d.� d.ej=��ZKG d/d0� d0ej=�ZLG d1d2� d2ej=�ZMG d3d4� d4ej=�ZNe�Fe3e d5�d6�G d7d8� d8ej=��ZOe�Pd9�d:k�p�e�Pd;�d:k�p�e�Pd<�d:kZQe�ReQd=�e�Rej d>kd?�G d@dA� dAej=���ZSejTd�dBdC��ZUe�Fe3e dD�dE�G dFdG� dGej=��ZVe�Fej dHkdI�G dJdK� dKej=��ZWG dLdM� dMej=�ZXG dNdO� dOej=�ZYe�Rej dHkdP�G dQdR� dRej=��ZZe�Rej dHkdP�G dSdT� dTej=��Z[e�Fej dHkdI�G dUdV� dVej=��Z\e�Fej dHkdI�G dWdX� dXej=��Z]e�Fe3e dY�dZ�G d[d\� d\ej=��Z^e�Fej dHkdI�e j_G d]d^� d^ej=���Z`e�Fej dHkdI�G d_d`� d`ej=��Zae�Fej dHkdI�G dadb� dbej=��Zbe j_G dcdd� ddej=��ZcG dedf� dfej=�ZdG dgdh� dhej=�ZeG didj� djej=�ZfG dkdl� dlej=�Zge�hdm�e�Fe3e dn�do�G dpdq� dqej=���Zie�Fe3e dr��ove3e ds�dt�G dudv� dvej=��ZjG dwdx� dxejkejl�Zme�Fe3e dy�dz�G d{d|� d|ej=��Znd}d~� Zoe�Feo� d�e �pd�d�d��G d�d�� d�ej=���Zqe�Fe3e d��d��G d�d�� d�ej=��Zre�Fe3e d��d��e �pd�d��G d�d�� d�ej=���ZsG d�d�� d�ej=�ZtG d�d�� d�ej=�ZuG d�d�� d�ej=�ZvG d�d�� d�ej=�Zwe�Fe3e d��d��G d�d�� d�ej=��ZxG d�d�� d�ej=�ZyG d�d�� d�ej=�ZzG d�d�� d�ej=�Z{G d�d�� d�ej=�Z|G d�d�� d�ej=�Z}e;d��G d�d�� d�ej=��Z~e3e d���rNG d�d�� d�e|�Ze�d�k�r`e��� dS )�� N)�support)�
socket_helper)� win32_is_iotc C s g | ]
}|j �qS � )Zpw_uid)�.0�ur r �1/opt/alt/python39/lib64/python3.9/test/test_os.py�
<listcomp>3 � r )�INT_MAX�PY_SSIZE_T_MAX��assert_python_ok)�
unix_shell�FakePathF�geteuid� thread_infoZ linuxthreadsZfreebsdc C s t �tt| �d| �S )Nzrequires os.%s)�unittest�
skipUnless�hasattr�os��namer r r �requires_os_funcP s r � contentc C s: t | dd��}|�|� W d � n1 s,0 Y d S )N�xbr )�open�write)�filenameZcontent�fpr r r � create_fileT s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� MiscTestsc C s t �� }| �|t� d S �N)r �getcwd�assertIsInstance�str��self�cwdr r r � test_getcwdZ s zMiscTests.test_getcwdc
C s~ d}d}d}|d|t |� }t�� ��>}t�|���}|}t�� }| �||� |t |�t tjj � }|dkrv�q t |�|kr�|dkr�|d |� }tj�
||�}zt� |� t� |� W n\ t
y� Y �q Y nF t�y } z,| jtjk�rW Y d } ~ �q n� W Y d } ~ n
d } ~ 0 0 |}q>tj�r:tdt |�� �� W d � n1 �sP0 Y W d � n1 �sp0 Y d S )Ni� �� Zpython_test_dir_�ar z!Tested current directory length: )�len�tempfileZTemporaryDirectoryr Z
change_cwdr r# � assertEqual�path�sep�join�mkdir�chdir�FileNotFoundError�OSError�errnoZ ENAMETOOLONG�verbose�print)
r' Zmin_lenZdirlen�dirnameZtmpdirr/ �expectedr( Zneed�excr r r �test_getcwd_long_path^ s6
z MiscTests.test_getcwd_long_pathc C s. t �� }| �|t� | �t �|�t �� � d S r" )r �getcwdbr$ �bytesr. �fsdecoder# r& r r r � test_getcwdb� s zMiscTests.test_getcwdbN)�__name__�
__module__� __qualname__r) r< r@ r r r r r! Y s 5r! c @ s e Zd Zdd� ZeZdd� Zdd� Zejdd� �Z d d
� Z
eje � e
ek d �eje
d d
dd�dd� ���Zdd� Zdd� Ze � ejdkd�dd� �Zdd� Zd d � Zd d � Zd d!� Zd"d#� Ze � ee d$�d%�d&d'� �Z e � ee d$�d%�d(d)� �Z e � ee d$�d%�d*d+� �Z d,S )-� FileTestsc C s t j�tj�rt �tj� d S r" )r r/ �lexistsr �TESTFN�unlink�r' r r r �setUp� s zFileTests.setUpc C s: t �tjt jt jB �}t �|� | �t �tjt j �� d S r" )
r r r rF �O_CREAT�O_RDWR�close�
assertTrue�access�W_OK�r' �fr r r � test_access� s
zFileTests.test_accessc C s� t �tjt jt jB �}t �|�}zTd}||d krht �|� |d7 }|dkrV| �d� |t �|� }}q&W t �|� n t �|� 0 t � ||d � | �
t t j |d� d S )Nr � �
z%couldn't allocate two consecutive fds� � a)
r r r rF rJ rK �duprL �skipTest�
closerange� assertRaisesr5 r )r' �first�secondZretriesr r r �test_closerange� s
zFileTests.test_closerangec C s< t j}t�|�}| �ttj|d� t�|�}| �||� d S �Nr ) r rF �sys� getrefcountrZ � TypeErrorr �renamer. )r' r/ �old�newr r r � test_rename� s
zFileTests.test_renamec C s� t tjd��`}|�d� |�� |�� }t�|dd� t�|d�}| � t
|�t � | � |d� W d � n1 sr0 Y d S )N�w+b� spamr � ) r r rF r �flush�filenor �lseek�readr. �typer> )r' �fobj�fd�sr r r � test_read� s
zFileTests.test_readz needs INT_MAX < PY_SSIZE_T_MAXrT rS F)�sizeZmemuseZdry_runc C sh | � tjtj� ttjd� ttjd�� }t�|�� |�}W d � n1 sN0 Y | � |d� d S )Ns test�rb)
�
addCleanupr rG rF r r r rl rj r. )r' rr r �datar r r �test_large_read� s
.zFileTests.test_large_readc C s� t �tjt jt jB �}| �tt j|d� t �|d� t �|t d�� t �|t
d�� t � |� ttjd��(}| � |�
� �� g d�� W d � n1 s�0 Y d S )N�beanss bacon
s eggs
s spam
rs )s bacons eggsrg )r r r rF rJ �O_WRONLYrZ ra r � bytearray�
memoryviewrL r. rl �
splitlines)r' ro rn r r r �
test_write� s
�zFileTests.test_writec G s" t j|t jdd�}| �|d� d S )NT)�
creationflags�shellr )�
subprocessZcallZCREATE_NEW_CONSOLEr. )r' �argsZretcoder r r �write_windows_console� s
�z FileTests.write_windows_console�win32z$test specific to the Windows consolec C s* d}| � tjd|� | � tjdd|� d S )Nzprint('x' * 100000)�-cz-u)r� r_ �
executable�r' �coder r r �test_write_windows_console� s z$FileTests.test_write_windows_consolec G s. t �tjt j�}t j|g|�R � }|�� d S r" )r r r rF �O_RDONLY�fdopenrL )r' r� ro rQ r r r �
fdopen_helper� s zFileTests.fdopen_helperc C sB t �tjt jt jB �}t �|� | �� | �d� | �dd� d S )N�r�d )r r r rF rJ rK rL r� �r' ro r r r � test_fdopen s
zFileTests.test_fdopenc C s� t jd }| �t jt j� | �t j|� tt jd� t|d� t�t j|� | �ttj t j� t
|d�� }| � |� � d� W d � n1 s�0 Y d S )Nz.2� 1� 2r� �1)
r rF rt rG r r �replacerZ r4 �statr r. rl )r' �TESTFN2rQ r r r � test_replace s
zFileTests.test_replacec C s" t jtt jdd d�}t �|� d S )N� )r/ �flags�mode�dir_fd)r r �__file__r� rL rP r r r �test_open_keywords s �z FileTests.test_open_keywordsc C s> t �td�}z|dt jdd d� W n ttfy8 Y n0 d S )N�symlink�targetF)�src�dst�target_is_directoryr� )r Z
get_attributer rF �NotImplementedErrorr5 )r' r� r r r �test_symlink_keywords s �
z FileTests.test_symlink_keywords�copy_file_rangez test needs os.copy_file_range()c C s< | � t�� t�ddd� W d � n1 s.0 Y d S )Nr rS i����)rZ �
ValueErrorr r� rH r r r �#test_copy_file_range_invalid_values% s z-FileTests.test_copy_file_range_invalid_valuesc
C s t jd }d}tt j|� | �t jt j� tt jd�}| �|j� |�� }t|d�}| �t j|� | �|j� |�� }zt� ||d�}W n< t
y� } z$|j t j kr�� | �
|� W Y d }~n`d }~0 0 | �|tdd�� t|d��(}| �|�� |d |� � W d � n1 �s0 Y d S )Nz.3�
0123456789rs rf � r � )r rF r rt rG r rL rj r r� r5 r6 �ENOSYSrX �assertIn�ranger. rl ) r' r� ru �in_file�in_fd�out_file�out_fd�i�er r r �test_copy_file_range* s(
z FileTests.test_copy_file_rangec
C s^ t jd }d}d}d}d}tt j|� | �t jt j� tt jd�}| �|j� |�� }t|d�}| �t j|� | �|j� |�� } ztj || |||d�}
W n< t
y� } z$| j t j kr�� | �
| � W Y d } ~ n�d } ~ 0 0 | �|
td |d
�� t|d��}|�� } W d � n1 �s0 Y | �| d |� d | � | �| |d � ||||
� � d S ) Nz.4r� r� � r� rs rf )Z
offset_srcZ
offset_dstr rS � )r rF r rt rG r rL rj r r� r5 r6 r� rX r� r� rl r. )
r' ZTESTFN4ru Z
bytes_to_copyZin_skipZout_seekr� r� r� r� r� r� rl r r r �test_copy_file_range_offsetL s<
�
(�z%FileTests.test_copy_file_range_offsetN) rA rB rC rI �tearDownrR r] r � cpython_onlyre rq r r r r Z
bigmemtestrv r| r� r_ �platformr� r� r� r� r� r� r r r� r� r� r r r r rD � s<
�
�
!rD c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Ze� e
e d �d �d
d� �Z e� e
e d �d�dd� �Z
e� ejdkd�dd� �Ze� ejdkd�e� e
e d�d�dd� ��Zdd� Ze� ejdkd �d d � �Ze� ejdkd�d d � �Ze� ejdkd�d!d"� �Zd#S )$�StatAttributeTestsc C s( t j| _| �t j| j� t| jd� d S )N� ABC)r rF �fnamert rG r rH r r r rI { s zStatAttributeTests.setUpc
C s� t �|�}| �|tj d� | �|jd� t|�}tt�D ]d}|d d� dkr:|�� }|�d�rjdd� }ndd� }| �|t||��|tt|� � | � ||� q:d�
� D ]8}t t||�d �}t||d �d
}| j ||d d � q�z|d
| �
d� W n t�y Y n0 zd|_| �
d� W n t�y6 Y n0 zd|_| �
d� W n ttf�yd Y n0 zd|_| �
d� W n t�y� Y n0 zt �d�} | �
d� W n t�y� Y n0 zt �d�} W n t�y� Y n0 d S )Nr� ZST_ZTIMEc S s t | �S r" )�int��xr r r �trunc� r
z7StatAttributeTests.check_stat_attributes.<locals>.truncc S s | S r" r r� r r r r� � r
zst_atime st_mtime st_ctimei�� Z_nsi' rU ��deltar* �No exception raisedrS �rT �r rS rU r� rh r� r� � � � rT � � �
� )r r� r. �ST_SIZE�st_size�dir�lower�endswith�getattrr� �splitr� �assertAlmostEqual�fail�
IndexError�st_mode�AttributeError�st_rdevra �parrot� stat_result)
r' r� �result�membersr �attrr� ZfloatyZ nanosecondy�result2r r r �check_stat_attributes� s^
�
z(StatAttributeTests.check_stat_attributesc C s | � | j� d S r" )r� r� rH r r r �test_stat_attributes� s z'StatAttributeTests.test_stat_attributesc C sF z| j �t�� �}W n" ty6 | �d| j � Y n0 | �|� d S )Nz#cannot encode %a for the filesystem)r� �encoder_ �getfilesystemencoding�UnicodeEncodeErrorrX r� )r' r� r r r �test_stat_attributes_bytes� s
z-StatAttributeTests.test_stat_attributes_bytesc C sf t �| j�}ttjd �D ]F}t�||�}| �d|� |dk rJ| �d|� t�|�}| � ||� qd S )NrS s stat_resultrh s cos
stat_result
)
r r� r� r� �pickle�HIGHEST_PROTOCOL�dumpsr� �loadsr. �r' r� �proto�pZ unpickledr r r �test_stat_result_pickle� s
z*StatAttributeTests.test_stat_result_pickle�statvfsztest needs os.statvfs()c C s t �| j�}| �|j|d � d}t|�D ]"\}}| �t|d| �|| � q*| �t|j t
�� | �t |�d� zd|_| � d� W n t
y� Y n0 zd|_| � d� W n t
y� Y n0 zt �d�}| � d� W n ty� Y n0 zt �d�}W n t�y Y n0 d S ) Nr� )
ZbsizeZfrsizeZblocksZbfreeZbavail�filesZffreeZfavail�flagZnamemaxZf_rT rS r� r� r� )r r� r� r. �f_bfree� enumerater� rM �
isinstance�f_fsidr� r, r� r� r� �statvfs_resultra )r' r� r� �value�memberr� r r r �test_statvfs_attributes� s4
z*StatAttributeTests.test_statvfs_attributes�need os.statvfs()c C sf t �| j�}ttjd �D ]F}t�||�}| �d|� |dk rJ| �d|� t�|�}| � ||� qd S )NrS s statvfs_resultrh s cos
statvfs_result
)
r r� r� r� r� r� r� r� r� r. r� r r r �test_statvfs_result_pickle s
z-StatAttributeTests.test_statvfs_result_pickler� �Win32 specific testsc
C s\ zt �d� W nH ty* | �d� Y n. tyV } z| �d� W Y d }~n
d }~0 0 d S )Nzc:\pagefile.sysz c:\pagefile.sys does not existzCould not stat pagefile.sys)r r� r4 rX r5 r� �r' r� r r r � test_1686475 s z StatAttributeTests.test_1686475�pipezrequires os.pipe()c C s� t �� \}}z"t �|� W t �|� t �|� nt �|� t �|� 0 | �t��}t �|� W d � n1 sp0 Y | �|jjtj � d S r" )
r r� r� rL rZ r5 r. � exceptionr6 �EBADF)r' r� �w�ctxr r r �
test_15261 s
�
(z StatAttributeTests.test_15261c C sF | � t|d�� | � t|jt�� | � d|j ko:dkn � d S )N�st_file_attributesr � �� )rM r r� r r� )r' r� r r r �check_file_attributes& s z(StatAttributeTests.check_file_attributesz$st_file_attributes is Win32 specificc C sz t �| j�}| �|� | �|jtj@ d� tjd }t � |� | �
t j |� t �|�}| �|� | �|jtj@ tj� d S )Nr r� ) r r� r� r r. r �FILE_ATTRIBUTE_DIRECTORYr rF r2 rt �rmdir)r' r� r9 r r r �test_file_attributes+ s
�
�z'StatAttributeTests.test_file_attributesc C sd t j�t jd | j�}| �tj|� t|d� d}t j
d|ddg|d� t � |�}| � |j
d� d S ) NZTEMPr� r� z
icacls.exez/denyz*S-1-5-32-545:(S)�r} r )r r/ r1 �environr� rt r rG r r �
check_callr� �assertNotEqualr� )r' r� ZDETACHED_PROCESSr� r r r �test_access_denied@ s
�
z%StatAttributeTests.test_access_deniedc C s6 dt j�t �� �d }t �|�}| �|jtj� d S )Nz//./r )r r/ �
splitdriver# r� r. r� �S_IFBLK)r' r� r� r r r �test_stat_block_deviceU s
z)StatAttributeTests.test_stat_block_deviceN)rA rB rC rI r� r� r� r� r r r r r� r� r_ r� r� r r r r
r r r r r r� z s2 A
+ �
�
r� c @ s� e Zd Zdd� Zdd� Zd)dd�Zdd � Zed
d � �Zd d
� Z dd� Z
e � e
je
jv d�dd� �Ze � e
je
jv d�dd� �Ze � e
je
jv d�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Ze j d'd(� �Z dS )*�
UtimeTestsc C sD t j| _tj�| jd�| _| �t j| j� t� | j� t
| j� d S )N�f1) r rF r9 r r/ r1 r� rt �rmtreer2 r rH r r r rI _ s
zUtimeTests.setUpc C s4 t �|�}|j|d kp2|j|d kp2|j|d kS )Nr� r� r� )r r� �st_atime�st_mtime�st_ctime)r' r �str r r �support_subsecondg s
� �z UtimeTests.support_subsecondNc C s� |s
| j }| �|�}|r"d}d}nd}d}||||f� t�|�}|rt| j|j|d dd� | j|j|d dd� n$| �|j|d � | �|j|d � | �|j|� | �|j |� d S )Ni8Z�;l �
o] � r T l P�9 ��&� .>g���ư>r� )
r� r r r� r� r r r. � st_atime_ns� st_mtime_ns)r' �set_timer r �atime_ns�mtime_nsr r r r � _test_utimeo s"
zUtimeTests._test_utimec C s dd� }| � |� d S )Nc S s t j| |d� d S �N��ns�r �utime�r r# r r r r � s z'UtimeTests.test_utime.<locals>.set_time�r �r' r r r r �
test_utime� s zUtimeTests.test_utimec C s | d d S )Nr g��&� .>r r" r r r � ns_to_sec� s zUtimeTests.ns_to_secc s � fdd�}� � |� d S )Nc s0 |\}}� � |�}� � |�}t�| ||f� d S r" �r* r r% �r r# r r Zatime�mtimerH r r r � s
z2UtimeTests.test_utime_by_indexed.<locals>.set_timer' r( r rH r �test_utime_by_indexed� s z UtimeTests.test_utime_by_indexedc s � fdd�}� � |� d S )Nc s2 |\}}� � |�}� � |�}tj| ||fd� d S )N)�timesr+ r, rH r r r � s
z0UtimeTests.test_utime_by_times.<locals>.set_timer' r( r rH r �test_utime_by_times� s z UtimeTests.test_utime_by_timesz9follow_symlinks support for utime required for this test.c C s dd� }| � |� d S )Nc S s t j| |dd� d S )NF)r# �follow_symlinksr$ r& r r r r � s z9UtimeTests.test_utime_nofollow_symlinks.<locals>.set_timer' r( r r r � test_utime_nofollow_symlinks� s z'UtimeTests.test_utime_nofollow_symlinksz,fd support for utime required for this test.c C s dd� }| � |� d S )Nc S sB t | dd��"}tj|�� |d� W d � n1 s40 Y d S )N�wbr r" )r r r% rj )r r# r r r r r � s z*UtimeTests.test_utime_fd.<locals>.set_timer' r( r r r �
test_utime_fd� s zUtimeTests.test_utime_fdz0dir_fd support for utime required for this test.c C s dd� }| � |� d S )Nc S sN t j�| �\}}t�|�� }t j|||d� W d � n1 s@0 Y d S )N)r� r# )r r/ r� r � open_dir_fdr% )r r# r9 r �dirfdr r r r � s z.UtimeTests.test_utime_dir_fd.<locals>.set_timer' r( r r r �test_utime_dir_fd� s z UtimeTests.test_utime_dir_fdc C s dd� }| j || jd� d S )Nc S s t j| |d� d S r! r$ r& r r r r � s z1UtimeTests.test_utime_directory.<locals>.set_time�r )r r9 r( r r r �test_utime_directory� s z UtimeTests.test_utime_directoryc C sb t � � }|| j� | �| j�s$d}nd}t�| j�}d|j||j| f }| j|j|||d� d S )Ng �?g�������?z st_time=%r, current=%r, dt=%r)r� �msg)�timer� r r r� r r� )r' r Zcurrentr� r r: r r r �_test_utime_current� s
�
�z UtimeTests._test_utime_currentc s � fdd�}� � |� d S )Nc s t �� j� d S r" �r r% r� r8 rH r r r � s z/UtimeTests.test_utime_current.<locals>.set_time�r< r( r rH r �test_utime_current� s z UtimeTests.test_utime_currentc s � fdd�}� � |� d S )Nc s t �� jd � d S r" r= r8 rH r r r � s z3UtimeTests.test_utime_current_old.<locals>.set_timer> r( r rH r �test_utime_current_old� s z!UtimeTests.test_utime_current_oldc C sl t jdkrhtj�tj�|��d d }dd l}|jj}|� dd�}|�
|d dd d d |t |��}|rh|j S d S )Nr� r �\� r� )
r_ r� r r/ r �abspath�ctypes�windll�kernel32Zcreate_unicode_bufferZGetVolumeInformationWr, r� )r' r/ �rootrD rF �buf�okr r r �get_file_system� s
�zUtimeTests.get_file_systemc C sJ | � | j�dkr| �d� d}t�| j||f� | �t�| j�j|� d S )NZNTFSz
requires NTFSr ) rJ r9 rX r r% r� r. r� r )r' Zlarger r r �test_large_time s
zUtimeTests.test_large_timec C s� | � t��" tj| jddd� W d � n1 s20 Y | � t��" t�| jddg� W d � n1 sn0 Y | � t�� t�| jd� W d � n1 s�0 Y | � t�� t�| jd� W d � n1 s�0 Y | � t��$ tj| jddgd� W d � n1 �s 0 Y | � t�� tj| jdd� W d � n1 �sZ0 Y | � t�� tj| jdd� W d � n1 �s�0 Y tjtjv�r�| � t��" tj| jddd� W d � n1 �s�0 Y tjtjv�rft | jdd ��L}| � t�� t�|�
� d� W d � n1 �s<0 Y W d � n1 �s\0 Y tjtj v�r�| � t��" tj| jdd d
� W d � n1 �s�0 Y d S ) N)r� r� r" r� )r� )r� r� r� F�r1 r3 r �r� ) rZ r� r r% r� ra �supports_follow_symlinksr� � supports_fdr rj �supports_dir_fd)r' r r r r � test_utime_invalid_arguments s0 0 0 , , 4 0 0 2 P z'UtimeTests.test_utime_invalid_argumentsc C s� dd� }| � t��( tj| j|d�dfd� W d � n1 s@0 Y | � t��( tj| j|d�dfd� W d � n1 s�0 Y | � t��( tj| j|d�dfd� W d � n1 s�0 Y d S )Nc s G � fdd�d�}|� S )Nc s e Zd Z� fdd�ZdS )z?UtimeTests.test_issue31577.<locals>.get_bad_int.<locals>.BadIntc s � S r" r )r� ��divmod_ret_valr r �
__divmod__. s zJUtimeTests.test_issue31577.<locals>.get_bad_int.<locals>.BadInt.__divmod__N)rA rB rC rT r rR r r �BadInt- s rU r )rS rU r rR r � get_bad_int, s z/UtimeTests.test_issue31577.<locals>.get_bad_int�* rS r" r )rS rU r� )rZ ra r r% r� )r' rV r r r �test_issue31577( s 6 6 zUtimeTests.test_issue31577)N) rA rB rC rI r r r) � staticmethodr* r. r0 r r r r% rN r2 rO r4 rP r7 r9 r< r? r@ rJ rK rQ r r� rX r r r r r ^ s:
�
�
�
r )�
mapping_testsc @ s e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Ze �
e oBe j
�e �d �d d
� �Ze �
e obe j
�e �d �dd� �Zdd� Zdd� Zdd� Zdd� Ze �
e jd�dd� �Zdd � Ze�d d �d d � �Zd!d"� Zd#d$� Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z!d/d0� Z"d1d2� Z#d3d4� Z$d5d6� Z%dS )7� EnvironTestsz8check that os.environ object conform to mapping protocolNc C sB t tj�| _tjr t tj�| _| �� �� D ]\}}|tj|< q*d S r" ) �dictr r
�_EnvironTests__save�supports_bytes_environ�environb�_EnvironTests__saveb�
_reference�items�r' �keyr� r r r rI ? s
zEnvironTests.setUpc C s: t j�� t j�| j� t jr6t j�� t j�| j� d S r" )r r
�clear�updater] r^ r_ r` rH r r r r� F s
zEnvironTests.tearDownc C s dddd�S )NZVALUE1ZVALUE2ZVALUE3)ZKEY1ZKEY2ZKEY3r rH r r r ra M s zEnvironTests._referencec C s t j�� t jS r" )r r
re rH r r r �_empty_mappingP s
zEnvironTests._empty_mappingzrequires a shellc C sb t j�� t jjdd� t �dt ��(}|�� �� }| �|d� W d � n1 sT0 Y d S )NZWorld)ZHELLOz%s -c 'echo $HELLO') r r
re rf �popenr rl �stripr. )r' rh r� r r r � test_update2U s
zEnvironTests.test_update2c C sx t �dt ��V}t|�}| �t|�d� | �t|�d� | �t|�d� | �tt|� W d � n1 sj0 Y d S )Nz %s -c 'echo "line1
line2
line3"'zline1
zline2
zline3
)r rh r �iterr. �nextrZ �
StopIteration)r' rh �itr r r �test_os_popen_iter^ s �z EnvironTests.test_os_popen_iterc C s8 t j�� D ](\}}| �t|�t� | �t|�t� q
d S r" )r r
rb r. rm r% )r' rd �valr r r �test_keyvalue_typesk s z EnvironTests.test_keyvalue_typesc C s. | � � �� D ] \}}| �tj�|�|� q d S r" )ra rb r. r r
�getrc r r r �
test_itemsp s zEnvironTests.test_itemsc
C s4 t j}| �t|�d�d�dd� |�� D ���� dS )z>Check that the repr() of os.environ looks like environ({...}).zenviron({{{}}})z, c s s | ]\}}d � ||�V qdS )z
{!r}: {!r}N��format)r rd r� r r r � <genexpr>x s �z-EnvironTests.test___repr__.<locals>.<genexpr>N)r r
r. �reprru r1 rb )r' �envr r r �
test___repr__u s �zEnvironTests.test___repr__c C sZ t j�t j�}g d�}dt j�|�i}t j}z8t|�t _| �|t �� � | �|t jd d�� W |t _n|t _0 | �|t �i �� | �dt �ddi�� | �|t �|�� t j �rVz:t
j dd�� dd d
�}W d � n1 s�0 Y W n t y� Y n0 | �
tt j|� | �t �d d i�d
g� | �t �d d
i�d
g� | �t �dd i�d
g� d S )N)z/montyz/pythonrB z/flying/circus�PATH�rx �rB rB T)�recordr� r� )rz � PATHr~ � abc�abc)r �defpathr� �pathsepr1 r
r\ �assertSequenceEqual�
get_exec_pathr^ �warnings�catch_warnings� BytesWarningrZ r� )r' Z defpath_listZ test_pathZtest_envZ
saved_environZ mixed_envr r r �test_get_exec_path| s8
, ���z EnvironTests.test_get_exec_pathz#os.environb required for this test.c C s� d}z|� t�� d�}W n* tyB dt�� f }| �|� Y n0 |tjd< | �tjd |� | �tjd |� d}|tjd< | �tjd |� |� t�� d�}| �tjd |� d S ) Nu euro€�surrogateescapez'U+20AC character is not encodable to %sZunicodes unicode� �s bytesr> )
r� r_ r� r� rX r r
r. r_ �decode)r' r� Z value_bytesr: Z value_strr r r �
test_environb� s$
� �
zEnvironTests.test_environbc C s� d}d}d|�d�}t �� ��}|�|d � t�||� tjtjd|gdtj dd�}| �
|j � � t
|�� t�|� tjtjd|gdtj dd�}| �
|j � � t
d �� W d � n1 s�0 Y d S )NZ
PYTHONTESTVARZ testvaluez%import os; print(repr(os.environ.get(z)))r� T)�check�stdout�text)r ZEnvironmentVarGuard�popr �putenvr �runr_ r� �PIPEr. r� �rstriprw �unsetenv)r' r r� r� rx �procr r r �test_putenv_unsetenv� s
�
�z!EnvironTests.test_putenv_unsetenvrT r� c C s~ dD ].}| � ttftj|d� | � ttftj|� qtjdkrzdd }| � ttj|d� | � ttjd|� | � ttj|� d S )N)rB z=namezna=mezname=zname zna mer� r� r� � � r� �X)rZ r5 r� r r� r� r_ r� )r' r Zlongstrr r r �test_putenv_unsetenv_error� s
z'EnvironTests.test_putenv_unsetenv_errorc C s� d}| � |tj� | �t��}tj| W d � n1 s<0 Y | �|jjd |� | �|jj � | �t��}tj|= W d � n1 s�0 Y | �|jjd |� | �|jj � d S )NZ
missingkeyr )
� assertNotInr r
rZ �KeyError�assertIsr� r� rM �__suppress_context__)r' Zmissing�cmr r r �
test_key_type� s ( &zEnvironTests.test_key_typec C sT t |�}d}t|� dtj|< z&t|� | �tj| d� W tj|= n
tj|= 0 d S )NZ __new_key__Ztest_environ_iteration)rk rl r r
r. )r' Z
collection�iteratorZnew_keyr r r �_test_environ_iteration� s
z$EnvironTests._test_environ_iterationc C s | � tj� d S r" )r� r r
rH r r r �(test_iter_error_when_changing_os_environ� s z5EnvironTests.test_iter_error_when_changing_os_environc C s | � tj�� � d S r" )r� r r
rb rH r r r �.test_iter_error_when_changing_os_environ_items� s z;EnvironTests.test_iter_error_when_changing_os_environ_itemsc C s | � tj�� � d S r" )r� r r
�valuesrH r r r �/test_iter_error_when_changing_os_environ_values s z<EnvironTests.test_iter_error_when_changing_os_environ_valuesc C sf t rtj�t �sd S t�t � d|� d��� }|�� �� }W d � n1 sL0 Y | �||� d S )Nz -c 'echo $�')r r r/ �existsrh rl ri r. )r' �varr: rh r� r r r � _test_underlying_process_env s
*z)EnvironTests._test_underlying_process_envc C s� d}d}|t j|< dddd|di}tt j�}|�|� t j|B }| �||� | �d|| � |�� }| �tt j� |�� | �
dd� | �
||� d S � N�
_TEST_VAR_�original_value�_A_r� �_B_�2�3rB ) r r
r\ rf �assertDictEqualr. rb r� �NotImplemented�__or__r� �r' �overridden_keyr� �
new_vars_dictr: �actual�new_vars_itemsr r r �test_or_operator s
z EnvironTests.test_or_operatorc C s| d}dt j|< dddd|di}tt j�}|�|� t j|O _| �|t j� | �dt j| � | �dd� | �|d� d S )Nr� r� r� r� r� r� r� �r r
r\ rf r. r� )r' r� r� r: r r r �test_ior_operator! s
z EnvironTests.test_ior_operatorc C s� t j�� }| �t��& ddi}t j|O _W d � n1 s@0 Y | �t��& ddi}t j|O _W d � n1 s�0 Y | �|t j� d S )NrS r� )r r
�copyrZ ra r. )r' Zos_environ_copyZdict_with_bad_keyZdict_with_bad_valr r r � test_ior_operator_invalid_dicts0 s
, ,z,EnvironTests.test_ior_operator_invalid_dictsc C sz d}dt j|< dd|dff}tt j�}|�|� t j|O _| �|t j� | �dt j| � | �dd� | �|d� d S )Nr� r� )r� r� )r� r� r� r� r� r� )r' r� r� r: r r r �$test_ior_operator_key_value_iterable= s
z1EnvironTests.test_ior_operator_key_value_iterablec C s� d}d}|t j|< dddd|di}t|�}|�t j� |t jB }| �||� | �||| � |�� }| �tt j� |�� | �
dd� | �
||� d S r� ) r r
r\ rf r� r. rb r� r� �__ror__r� r� r r r �test_ror_operatorL s
z EnvironTests.test_ror_operator)&rA rB rC �__doc__Z type2testrI r� ra rg r r r r r/ r� rj ro rq rs ry r� r^ r� r� r Zrequires_mac_verr� r� r� r� r� r� r� r� r� r� r� r� r r r r r[ ; sH �
�
)�
r[ c @ sZ e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )� WalkTestszTests for os.walk().c K s( d|v r|� d�|d< tj|fi |��S �Nr1 � followlinks)r� r �walk)r' �top�kwargsr r r r� e s zWalkTests.walkc C s> t jj}| �tjtj� |tjd�| _|| jd�| _|| jd�| _ || jd�}||d�}|| jd�}|| jd�}||d�}||d�}||d �| _
|tjd
�}|tjd
d �} ||d �}
||d
�} ||d�} t � | j � t � |� t � |� t � |� |||| |fD ]H}
t |
ddd��"}|�
d|
d � W d � n1 �s<0 Y �q t�� �r�t �t j�|�| j
� t �d|
d� t �|dd�| d� t �|dd�| d� |dd gg d�f| _n|dgdgf| _t �|d� zt �|� W n& t�y | �t j|tj� Y n40 t �|tj� t �|� t �|� | jd d d�= d S )NZTEST1�SUB1�SUB11�SUB2ZSUB21�tmp1�tmp2�tmp3�linkZTEST2�tmp4� broken_link� broken_link2� broken_link3r� �utf-8��encodingzI'm z" and proud of it. Blame test_os.
ZbrokenTZtmp5)r� r� r� r� r rS )r r/ r1 rt r r rF � walk_path� sub1_path�
sub11_path� link_path�makedirsr r � can_symlinkr� rC � sub2_tree�chmod�listdir�PermissionErrorr� �S_IRWXUrG r )r' r1 Z sub2_pathZ
sub21_pathZ tmp1_pathZ tmp2_pathZ tmp3_pathZ tmp5_pathZt2_pathZ tmp4_pathZbroken_link_pathZbroken_link2_pathZbroken_link3_pathr/ rQ r r r rI j sT
6
�
zWalkTests.setUpc C s� t | �| j��}| �t|�d� |d d d dk}|d d �� |dd| d �� |dd| d �� | �|d | jddgd gf� | �|d| | jd
gd gf� | �|d| | jg g f� | �|dd| | j� d S ) Nrh r rS r� r� rU ���r� r� r� r� ) �listr� r� r. r, �sortr� r� r� �r' �allZflippedr r r �test_walk_topdown� s zWalkTests.test_walk_topdownNc C s� |d u r| j }g }| �|�D ],\}}}|�|||f� d|v r |�d� q | �t|�d� | �|d | j dgdgf� |d d �� |d d �� | �|d | j� d S )Nr� rU r r� r� rS r� )r� r� �append�remover. r, r� r� )r' r� r� rG �dirsr� r r r �test_walk_prune� s zWalkTests.test_walk_prunec C s | � t| j�� d S r" )r� r r� rH r r r �test_file_like_path� s z WalkTests.test_file_like_pathc C s� t | j| jdd��}| �t|�d|� |d d d dk}|d d �� |dd| d �� |dd| d �� | �|d | jdd
gd gf� | �|| | jg g f� | �||d | jd gd
gf� | �|dd| | j� d S )NF��topdownrh r� rS r r� rU r� r� r� r� r� ) r� r� r� r. r, r� r� r� r� r� r r r �test_walk_bottom_up� s$
�
���z WalkTests.test_walk_bottom_upc C sh t �� s| �d� | j| jdd�}|D ]2\}}}|| jkr&| �|g � | �|dg� qdq&| �d� d S )Nzneed symlink supportTrL r� z+Didn't follow symlink with followlinks=True)r r� rX r� r� r� r. r� )r' �walk_itrG r� r� r r r �test_walk_symlink� s
zWalkTests.test_walk_symlinkc
C s� g }| j | j|jd�}t|�\}}}| �|g � d}tj�||�}tj�||d �}t�||� zhdd� |D �} | � |� | �
|| � | �
|| � |D ]"}
|
|kr�| � tj�||
�| � q�W t�||� nt�||� 0 d S )N)�onerrorr� z.newc S s g | ]\}}}|�qS r r )r r� �drQ r r r r r
z/WalkTests.test_walk_bad_dir.<locals>.<listcomp>) r� r� r� rl r. r r/ r1 rb rM r� r� ) r' �errorsr� rG r� r� �dir1Zpath1Zpath1new�roots�dir2r r r �test_walk_bad_dir� s"
zWalkTests.test_walk_bad_dirc s d}t j�tjd�� t jj� gdg| �R � }t �|� � �fdd�td�D �}t|d �D ]>}||rldgng g f}|D ]}��t|�|� qxt j� |�}q\� �fdd�td�D �}� }t|d �D ]D}|||k r�dgng g f}|D ]}��t|�|� q�t j�|d�}q�d S ) N� Zdeepr� c s g | ]}�j � d d��qS )Fr� �r� �r �j��baser' r r r r
z7WalkTests.test_walk_many_open_files.<locals>.<listcomp>r� rS c s g | ]}�j � d d��qS )Tr� r� r� r� r r r r
)
r r/ r1 r rF r� r� r. rl r9 )r' Zdepthr� Zitersr� r: rn r r� r �test_walk_many_open_files s"
z#WalkTests.test_walk_many_open_files)N)
rA rB rC r� r� rI r� r� r� r� r� r� r� r r r r r� ` s E
r� �fwalkzTest needs os.fwalk()c @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dZ dS )�
FwalkTestszTests for os.fwalk().c k s0 | j |fi |��D ]\}}}}|||fV qd S r" )r )r' r� r� rG r� r� Zroot_fdr r r r� ( s zFwalkTests.walkc O s t j|i |��S r" )r r �r' r� r� r r r r , s zFwalkTests.fwalkc
C s� |� � }|� � }tjddd�D ]�\}}|j||d� |j||d� i }tjf i |��D ] \}}}t|�t|�f||< qV| jf i |��D ]4\}}}} | �||� | � || t|�t|�f� q�q dS )z.
compare with walk() results.
�TFrU ��repeat)r� r� )r� r1 N)
r� � itertools�productrf r r� �setr r� r. )
r' � walk_kwargs� fwalk_kwargsr� r1 r: rG r� r� �rootfdr r r �_compare_to_walk/ s zFwalkTests._compare_to_walkc C s dt ji}| �||� d S )Nr� )r rF r )r' r� r r r �test_compare_to_walkA s
z FwalkTests.test_compare_to_walkc C sT zBt �dt j�}dtji}|�� }||d< | �||� W t �|� n t �|� 0 d S )N�.r� r� )r r r� r rF r� r rL )r' ro r r
r r r � test_dir_fdE s
zFwalkTests.test_dir_fdc C s| t jddd�D ]h\}}tj|d f}| j|d|i�D ]B\}}}}t�|� t�|� | �t t�
|��t |�t |�B � q2qd S )Nr rU r r1 ) r r r rF r r �fstatr� r. r r� )r' r� r1 r� rG r� r� r r r r �test_yields_correct_dir_fdO s
z%FwalkTests.test_yields_correct_dir_fdc C s\ t �d�}t �|� td�D ]}| �tj�D ]}q,q t �d�}| �t j|� | �||� d S )NrS � ) r rW rL r� r r rF rt r. )r' Zminfdr� r� Znewfdr r r � test_fd_leak[ s
zFwalkTests.test_fd_leakN) rA rB rC r� r� r r r
r r r r� r r r r r $ s
r c @ s e Zd ZdZdd� ZdS )�BytesWalkTests� Tests for os.walk() with bytes.c k s� d|v r|� d�|d< tjt�|�fi |��D ]p\}}}t�|�}tttj|��}tttj|��}|||fV tttj|��|d d �< tttj|��|d d �< q.d S r� )r� r r� �fsencoder? r� �map) r' r� r� �broot�bdirs�bfilesrG r� r� r r r r� n s "
zBytesWalkTests.walkN)rA rB rC r� r� r r r r r l s r c @ s e Zd ZdZddd�ZdS )�BytesFwalkTestsr r c o s� t jt �|�g|�R i |��D ]t\}}}}t �|�}ttt j|��} ttt j|��}
|| |
|fV ttt j| ��|d d �< ttt j|
��|d d �< q d S r" )r r r r? r� r ) r' r� r� r� r r r �topfdrG r� r� r r r r | s *
zBytesFwalkTests.fwalkN)r )rA rB rC r� r r r r r r y s r c @ s^ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze �
e j d
kd�dd� �Z
dd� ZdS )� MakedirTestsc C s t �tj� d S r" �r r2 r rF rH r r r rI � s zMakedirTests.setUpc
C s� t j}tj�|ddd�}t�|� tj�|dddd�}t�|� | �ttjtj� tj�|dddddtj�}t�|� tj�|dtjddddd�}t�|� d S �Nr� r� Zdir3Zdir4Zdir5Zdir6) r rF r r/ r1 r� rZ r5 �curdir)r' r� r/ r r r � test_makedir� s
�zMakedirTests.test_makedirc C s� t �d��� t j}tj�|d�}tj�|d�}t�|d� | �tj�|�� | �tj� |�� tj
dkr�| � t� |�j
d@ d� | � t� |�j
d@ d� W d � n1 s�0 Y d S )NrU r� r� im �ntr� i� )r Z
temp_umaskrF r r/ r1 r� rM r� �isdirr r. r� r� )r' r� �parentr/ r r r � test_mode� s
zMakedirTests.test_modec C s� t j�tjd�}d}t �d�}t �||� | �tt j||� | jtt j||dd� t j|ddd� t j||dd� t �|� t jt j� d �dd� d S )
Nr� r� � F��exist_oki� T)r� r( �/)
r r/ r1 r rF �umaskr� rZ r5 rC )r' r/ r� �old_maskr r r � test_exist_ok_existing_directory� s
z-MakedirTests.test_exist_ok_existing_directoryc C s t j�tjd�}tj}d}t �d�}z�t�t � tj�j
�}zt � tj||B � W n t yl t
�d��Y n0 t � tj�j
|@ |kr�t
�d��t �|||B � t j||dd� t � |t�t � |�j
�| @ � t j|||B dd� W t �|� n t �|� 0 d S )Nr� r� r& zCannot set S_ISGID for dir.z No support for S_ISGID dir mode.Tr' )r r/ r1 r rF r� �S_ISGIDr* �S_IMODE�lstatr� r� r� r �SkipTestr� )r' r/ r- r� r+ Zexisting_testfn_moder r r � test_exist_ok_s_isgid_directory� s&
�
z,MakedirTests.test_exist_ok_s_isgid_directoryc C s� t j}tj�t jd�}t|d��}|�d� W d � n1 s@0 Y | �ttj |� | jttj |dd� | jttj |dd� t�
|� d S )Nr� r r� Fr' T) r rF r r/ r1 r r rZ r5 r� r� )r' r� r/ rQ r r r �#test_exist_ok_existing_regular_file� s (z0MakedirTests.test_exist_ok_existing_regular_filer" zrequires Windowsc C sd t j}tj�tj�t jd��}tj|dd� tjd|dgdd�}t� |� | �
|� � |� d�� d S ) Nr� i� )r� z cacls.exez/sZoemr� z6 "D:P(A;OICI;FA;;;SY)(A;OICI;FA;;;BA)(A;OICI;FA;;;OW)") r rF r r/ rC r1 r2 r � check_outputr r. ri )r' r� r/ �outr r r �test_win32_mkdir_700� s
�z!MakedirTests.test_win32_mkdir_700c C sL t j�tjdddddd�}t j�|�s>|tjkr>t j�|�}qt �|� d S r )r r/ r1 r rF r� r9 �
removedirs�r' r/ r r r r� � s �zMakedirTests.tearDownN)rA rB rC rI r! r% r, r1 r2 r r r r r5 r� r r r r r � s
r �chownzTest needs chownc @ s� e Zd Zedd� �Zdd� Ze�ee d�d�dd� �Z
e�e oHe e
�d kd
�d d � �Ze�e oje e
�d kd
�dd� �Zedd� �ZdS )�ChownFileTestsc C s t �tj� d S r" r ��clsr r r �
setUpClass� s zChownFileTests.setUpClassc C s� t �tj�}|j}|j}ddt�d�t� dd�fD ]0}| �
t t j tj||� | �
t t j tj||� q2| �
t � tj||�� | �
t � tjdd�� d S )Ng �y � �r� ���rU )r r� r rF �st_uid�st_gid�decimalZDecimal� fractionsZFractionrZ ra r8 � assertIsNone)r' r� �uid�gidr� r r r �*test_chown_uid_gid_arguments_must_be_index� s z9ChownFileTests.test_chown_uid_gid_arguments_must_be_index� getgroupszneed os.getgroupsc C s� t �� }t|�dk r | �d� |d d� \}}t �tj�j}t �tj||� t �tj�j }| �
||� t �tj||� t �tj�j }| �
||� d S )NrU z test needs at least 2 groups) r rF r, rX r� r rF r> r8 r? r. )r' �groupsZgid_1Zgid_2rC rD r r r �test_chown_gid s
z ChownFileTests.test_chown_gidrS z0test needs root privilege and more than one userc C sv t d d� \}}t�tj�j}t�tj||� t�tj�j}| �||� t�tj||� t�tj�j}| �||� d S �NrU ) � all_usersr r� r rF r? r8 r> r. )r' �uid_1�uid_2rD rC r r r �test_chown_with_root s z#ChownFileTests.test_chown_with_rootz2test needs non-root account and more than one userc C sl t d d� \}}t�tj�j}| �t��0 t�tj||� t�tj||� W d � n1 s^0 Y d S rI ) rJ r r� r rF r? rZ r� r8 )r' rK rL rD r r r � test_chown_without_permission s
z,ChownFileTests.test_chown_without_permissionc C s t �tj� d S r" )r r r rF r: r r r �
tearDownClass' s z ChownFileTests.tearDownClassN)rA rB rC � classmethodr< rE r r r r rH �
root_in_posixr, rJ rM rN rO r r r r r9 � s
�
�
r9 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �RemoveDirsTestsc C s t �tj� d S r" )r r� r rF rH r r r rI - s zRemoveDirsTests.setUpc C s t �t j� d S r" )r r rF rH r r r r� 0 s zRemoveDirsTests.tearDownc C sx t j�tjd�}t �|� t j�|d�}t �|� t �|� | �t j�|�� | �t j�|�� | �t j�tj�� d S )N�dira�dirb) r r/ r1 r rF r2 r6 � assertFalser� �r' rS rT r r r �test_remove_all3 s
z RemoveDirsTests.test_remove_allc C s� t j�tjd�}t �|� t j�|d�}t �|� tt j�|d�� t �|� | �t j� |�� | �
t j� |�� | �
t j� tj�� d S �NrS rT �file.txt) r r/ r1 r rF r2 r r6 rU r� rM rV r r r �test_remove_partial= s
z#RemoveDirsTests.test_remove_partialc C s� t j�tjd�}t �|� t j�|d�}t �|� tt j�|d�� | �t�� t � |� W d � n1 sn0 Y | �
t j� |�� | �
t j� |�� | �
t j� tj�� d S rX ) r r/ r1 r rF r2 r rZ r5 r6 rM r� rV r r r �test_remove_nothingH s
(z#RemoveDirsTests.test_remove_nothingN)rA rB rC rI r� rW rZ r[ r r r r rR , s
rR c @ s e Zd Zdd� ZdS )� DevNullTestsc C s� t tjdd��"}|�d� |�� W d � n1 s60 Y t tjd�� }| �|�� d� W d � n1 sr0 Y d S )Nr3 r � hellors r
)r r �devnullr rL r. rl rP r r r � test_devnullV s
&zDevNullTests.test_devnullN)rA rB rC r_ r r r r r\ U s r\ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� URandomTestsc C sr | � tt�d��d� | � tt�d��d� | � tt�d��d� | � tt�d��d� | � tt�d��d� d S )Nr rS rT r� i� )r. r, r �urandomrH r r r �test_urandom_length_ s
z URandomTests.test_urandom_lengthc C s0 t �d�}| �|t� t �d�}| �||� d S �N� )r ra r$ r> r �r' Zdata1Zdata2r r r �test_urandom_valuef s
z URandomTests.test_urandom_valuec C s< d� dd| ddf�}td|�}|d }| �t|�|� |S )N�
zimport os, syszdata = os.urandom(%s)z sys.stdout.buffer.write(data)zsys.stdout.buffer.flush()r� rS )r1 r r. r, )r' �countr� r4 r� r r r �get_urandom_subprocessl s �
z#URandomTests.get_urandom_subprocessc C s$ | � d�}| � d�}| �||� d S rc )ri r re r r r �test_urandom_subprocessw s
z$URandomTests.test_urandom_subprocessN)rA rB rC rb rf ri rj r r r r r` ^ s r` � getrandomzneed os.getrandom()c @ s@ e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�GetRandomTestsc
C sR zt �d� W n> tyL } z&|jtjkr6t�d��n� W Y d }~n
d }~0 0 d S )NrS z%getrandom() syscall fails with ENOSYS)r rk r5 r6 r� r r0 )r; r; r r r r< s zGetRandomTests.setUpClassc C s* t �d�}| �|t� | �t|�d� d S rc )r rk r$ r> r. r, �r' ru r r r �test_getrandom_type� s
z"GetRandomTests.test_getrandom_typec C s t �d�}| �|d� d S )Nr r
)r rk r. )r' �emptyr r r �test_getrandom0� s
z GetRandomTests.test_getrandom0c C s | � ttd�� d S )N� GRND_RANDOM)rM r r rH r r r �test_getrandom_random� s z$GetRandomTests.test_getrandom_randomc C s* zt �dt j� W n ty$ Y n0 d S �NrS )r rk �
GRND_NONBLOCK�BlockingIOErrorrH r r r �test_getrandom_nonblock� s z&GetRandomTests.test_getrandom_nonblockc C s$ t �d�}t �d�}| �||� d S rc )r rk r re r r r �test_getrandom_value� s
z#GetRandomTests.test_getrandom_valueN)
rA rB rC rP r< rn rp rr rv rw r r r r rl } s
rl ZHAVE_GETENTROPYrS ZHAVE_GETRANDOMZHAVE_GETRANDOM_SYSCALLz*os.random() does not use a file descriptorZvxworksz$VxWorks can't set RLIMIT_NOFILE to 1c @ s0 e Zd Ze�ed�dd� �Zdd� Zdd� ZdS ) �URandomFDTestsz!test requires the resource modulec C s d}t d|� d S )Na� if 1:
import errno
import os
import resource
soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_NOFILE)
resource.setrlimit(resource.RLIMIT_NOFILE, (1, hard_limit))
try:
os.urandom(16)
except OSError as e:
assert e.errno == errno.EMFILE, e.errno
else:
raise AssertionError("OSError not raised")
r� r
r� r r r �test_urandom_failure� s z#URandomFDTests.test_urandom_failurec C s d}t d|�\}}}d S )Na if 1:
import os
import sys
import test.support
os.urandom(4)
with test.support.SuppressCrashReport():
os.closerange(3, 256)
sys.stdout.buffer.write(os.urandom(4))
�-Scr
)r' r� �rcr4 �errr r r �test_urandom_fd_closed� s z%URandomFDTests.test_urandom_fd_closedc C s� | � tjtj� ttjd� djtjd�}td|�\}}}| �t|�d� | � |dd� |dd� � td|�\}}}| �t|�d� | � ||� d S )Ns xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxa� if 1:
import os
import sys
import test.support
os.urandom(4)
with test.support.SuppressCrashReport():
for fd in range(3, 256):
try:
os.close(fd)
except OSError:
pass
else:
# Found the urandom fd (XXX hopefully)
break
os.closerange(3, 256)
with open({TESTFN!r}, 'rb') as f:
new_fd = f.fileno()
# Issue #26935: posix allows new_fd and fd to be equal but
# some libc implementations have dup2 return an error in this
# case.
if new_fd != fd:
os.dup2(new_fd, fd)
sys.stdout.buffer.write(os.urandom(4))
sys.stdout.buffer.write(os.urandom(4))
)rF rz r� r rh )
rt r rG rF r ru r r. r, r )r' r� r{ r4 r| Zout2Zerr2r r r �test_urandom_fd_reopened� s � z'URandomFDTests.test_urandom_fd_reopenedN) rA rB rC r r �resourcery r} r~ r r r r rx � s
rx c # s~ g � � fdd�}� fdd�}zHt j}t j}t j}|t _|t _| durJ| t _� V W |t _|t _|t _n|t _|t _|t _0 dS )z�
Stubs out execv and execve functions when used as context manager.
Records exec calls. The mock execv and execve functions always raise an
exception as they would normally never return.
c s � � d| |f� td��d S )N�execvz execv called)r� � RuntimeError�r r� ��callsr r �
mock_execv s z#_execvpe_mockup.<locals>.mock_execvc s � � d| |f� ttjd��d S )N�execvez
execve called)r� r5 r6 ZENOTDIRr� r� r r � mock_execve s z$_execvpe_mockup.<locals>.mock_execveN)r r� r� r� )r� r� r� Z
orig_execvZ orig_execveZ orig_defpathr r� r �_execvpe_mockup s$ �r� r� zneed os.execv()c @ st e Zd Ze�ed�dd� �Zdd� Zdd� Ze� e
e d�d �d
d � �Z d d
� Z
dd� Ze� ejdkd�dd� �ZdS )� ExecTestsz4avoid triggering a linuxthreads bug: see issue #4970c C s | � ttjddgd � d S )Nz no such app-)rZ r5 r �execvperH r r r � test_execvpe_with_bad_program$ s �z'ExecTests.test_execvpe_with_bad_programc C sN | � ttjdd� | � ttjdg � | � ttjdd� | � ttjddg� d S )N�notepadr r| rB )rZ r� r r� rH r r r �test_execv_with_bad_arglist* s z%ExecTests.test_execv_with_bad_arglistc C sB | � ttjdg d � | � ttjdg i � | � ttjddgi � d S )Nr� rB )rZ r� r r� rH r r r � test_execvpe_with_bad_arglist0 s z'ExecTests.test_execvpe_with_bad_arglist�_execvpez)No internal os._execvpe function to test.c
C s� t jd }|tu r8d}t j�t �|�|�}|}g d�}n4d}g d�}t j�||�}t jdkrht �|�}n|}ddi}t� �J}| �t t j
||� | � t |�d � | � |d
d ||ff� W d � n1 s�0 Y t|d ��P}| jt
t j
|||d
� | � t |�d � | �|d
d|||ff� W d � n1 �s60 Y t� �t}|�� } |tu �rd|| d< n|| d< | jt
t j
||| d
� | � t |�d � | �|d
d||| ff� W d � n1 �s�0 Y d S )NZ absolutepaths
executable)s progname�arg1�arg2r� )Zprognamer� r� r" Zspamrw rS r r� )r� r{ r� r~ rz )r r0 r> r/ r1 r r r� rZ r� r� r. r, r5 r� r� )
r' Z test_typeZ program_pathZprogram�fullpathZnative_fullpathZ argumentsrx r� Zenv_pathr r r �_test_internal_execvpe5 sN
�6
�
�$
�
�z ExecTests._test_internal_execvpec C s" | � t� tjdkr | � t� d S )Nr" )r� r% r r r> rH r r r �test_internal_execvpe_strf s
z#ExecTests.test_internal_execvpe_strc C s� t jddg}tj�� }d|d< | �t��" t�|d ||� W d � n1 sP0 Y tj�� }d|d< | �t��" t�|d ||� W d � n1 s�0 Y tj�� }d|d < | �t��" t�|d ||� W d � n1 s�0 Y d S )
Nr� �pass�cabbage�FRUIT VEGETABLEr �orange VEGETABLE=cabbage�FRUIT�lemon� FRUIT=ORANGE)r_ r� r r
r� rZ r� r� )r' r� �newenvr r r �test_execve_invalid_envk s
0
0
z!ExecTests.test_execve_invalid_envr� zWin32-specific testc
C sb zt �ddgi � W n> tyR } z&| �|jd u p:|jdk� W Y d }~nd }~0 0 | �d� d S )NrB �argr zNo OSError raised)r r� r5 rM Zwinerrorr� r� r r r �test_execve_with_empty_path� s
0z%ExecTests.test_execve_with_empty_pathN)rA rB rC r �skipIf�USING_LINUXTHREADSr� r� r� r r r r� r� r� r_ r� r� r r r r r� ! s �
�
/r� r� r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�Win32ErrorTestsc
C sv zt �tj� W nP ty& d}Y nL ty` } z$d}| �dtj|f � W Y d }~nd }~0 0 | �dtj � d S )NFTz.file %s must not exist; os.stat failed with %szfile %s must not exist)r r� r rF r4 r5 r� )r' r� r; r r r rI � s � zWin32ErrorTests.setUpc C s | � ttjtjtjd � d S )Nz.bak)rZ r5 r rb r rF rH r r r re � s zWin32ErrorTests.test_renamec C s | � ttjtj� d S r" )rZ r5 r r� r rF rH r r r � test_remove� s zWin32ErrorTests.test_removec C s | � ttjtj� d S r" )rZ r5 r r3 r rF rH r r r �
test_chdir� s zWin32ErrorTests.test_chdirc C sR | � tjtj� ttjd��"}| �ttjtj� W d � n1 sD0 Y d S )Nr� ) rt r rG rF r rZ r5 r r2 rP r r r �
test_mkdir� s zWin32ErrorTests.test_mkdirc C s | � ttjtjd � d S r" )rZ r5 r r% r rF rH r r r r) � s zWin32ErrorTests.test_utimec C s | � ttjtjd� d S r^ )rZ r5 r r� r rF rH r r r �
test_chmod� s zWin32ErrorTests.test_chmodN)
rA rB rC rI re r� r� r� r) r� r r r r r� � s r� c @ s� e Zd Zg d�Zdd� ZeD ]Zee�e� de < q dd� Ze� e
e d�d�d d
� �Z e� e
e d �d �d
d� �Z
e� e
e d�d�dd� �Ze� e
e d�d�dd� �Ze� e
e d�d�dd� �Ze� e
e d�d �d d � �Ze� e
e d �d �d!d"� �Ze� e
e d#�d$�d%d&� �Ze� e
e d'�d(�d)d*� �Ze� e
e d+�d,�d-d.� �Ze� e
e d/�d0�d1d2� �Ze� e
e d3�d4�d5d6� �Ze� e
e d7�d8�d9d:� �Zd;d<� Ze� e
e d=�d>�d?d@� �ZdAS )B�
TestInvalidFD) �fchdirrW r� � fdatasyncr �fstatvfs�fsync� tcgetpgrp�ttynamec s � fdd�}|S )Nc s t t� �r| �tt� �� d S r" )r r r� r� rH �rQ r r �helper� s
z(TestInvalidFD.get_single.<locals>.helperr )rQ r� r r� r �
get_single� s zTestInvalidFD.get_singleZtest_c
G s` z|t �� g|�R � W n4 tyL } z | �|jtj� W Y d }~nd }~0 0 | �d| � d S )Nz5%r didn't raise an OSError with a bad file descriptor)r � make_bad_fdr5 r. r6 r� r� )r' rQ r� r� r r r r� � s &�zTestInvalidFD.check�isattyztest needs os.isatty()c C s | � t�t�� �d� d S �NF)r. r r� r r� rH r r r � test_isatty� s zTestInvalidFD.test_isattyrY ztest needs os.closerange()c C sr t �� }td�D ].}zt�|| � W n ty8 Y q0 q@q|dk rRt�d��| �t� ||| d �d � d S )NrT rU z5Unable to acquire a range of invalid file descriptorsrS )
r r� r� r r r5 r r0 r. rY )r' ro r� r r r r] � s �z TestInvalidFD.test_closerange�dup2ztest needs os.dup2()c C s | � tjd� d S )N� )r� r r� rH r r r � test_dup2� s zTestInvalidFD.test_dup2�fchmodztest needs os.fchmod()c C s | � tjd� d S r^ )r� r r� rH r r r � test_fchmod� s zTestInvalidFD.test_fchmod�fchownztest needs os.fchown()c C s | � tjdd� d S )Nr� )r� r r� rH r r r � test_fchown� s zTestInvalidFD.test_fchown� fpathconfztest needs os.fpathconf()c C s | � tjd� | � tjd� d S )N� PC_NAME_MAX)r� r �pathconfr� rH r r r �test_fpathconf� s z TestInvalidFD.test_fpathconf� ftruncateztest needs os.ftruncate()c C s | � tjd� | � tjd� d S r^ )r� r �truncater� rH r r r �test_ftruncate� s z TestInvalidFD.test_ftruncaterk ztest needs os.lseek()c C s | � tjdd� d S r^ )r� r rk rH r r r �
test_lseek� s zTestInvalidFD.test_lseekrl ztest needs os.read()c C s | � tjd� d S rs )r� r rl rH r r r rq � s zTestInvalidFD.test_read�readvztest needs os.readv()c C s t d�}| �tj|g� d S )NrT )ry r� r r� )r' rH r r r �
test_readv� s zTestInvalidFD.test_readv� tcsetpgrpztest needs os.tcsetpgrp()c C s | � tjd� d S r^ )r� r r� rH r r r �test_tcsetpgrpt� s z TestInvalidFD.test_tcsetpgrptr ztest needs os.write()c C s | � tjd� d S )N� )r� r r rH r r r r| s zTestInvalidFD.test_write�writevztest needs os.writev()c C s | � tjdg� d S )Nr )r� r r� rH r r r � test_writev s zTestInvalidFD.test_writevc C s | � tj� | � tjd� d S �NT)r� r �get_inheritable�set_inheritablerH r r r �test_inheritable s z TestInvalidFD.test_inheritable� get_blocking�-needs os.get_blocking() and os.set_blocking()c C s | � tj� | � tjd� d S r� )r� r r� � set_blockingrH r r r �
test_blocking s zTestInvalidFD.test_blockingN)rA rB rC Zsinglesr� rQ �localsr� r r r r r� r] r� r� r� r� r� r� rq r� r� r| r� r� r� r r r r r� � sH
�r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� LinkTestsc C s t j| _tj�t jd �| _d S )Nr� )r rF �file1r r/ r1 �file2rH r r r rI s zLinkTests.setUpc C s, | j | jfD ]}tj�|�r t�|� q d S r" )r� r� r r/ r� rG )r' �filer r r r� s zLinkTests.tearDownc
C s� t |� zt�||� W n2 tyJ } z| �d| � W Y d }~n
d }~0 0 t|d��V}t|d��,}| �tj�|� � |� � �� W d � n1 s�0 Y W d � n1 s�0 Y d S )N�
os.link(): %sr� )
r r r� r� rX r rM r/ � sameopenfilerj )r' r� r� r� r �f2r r r �
_test_link s $zLinkTests._test_linkc C s | � | j| j� d S r" )r� r� r� rH r r r � test_link' s zLinkTests.test_linkc C s( | � t| jt�� �t| jt�� �� d S r" )r� r> r� r_ r� r� rH r r r �test_link_bytes* s �zLinkTests.test_link_bytesc C sZ zt �d� W n ty* t�d��Y n0 | jd7 _| jd | _| �| j| j� d S )N� ñz#Unable to encode for this platform.r� )r r � UnicodeErrorr r0 r� r� r� rH r r r �test_unicode_name. s zLinkTests.test_unicode_nameN) rA rB rC rI r� r� r� r� r� r r r r r� s
r� zPosix specific testsc @ s� e Zd ZdZdZe�eed�d�dd� �Z e�eed�d�dd � �Z
e�eed
�d �d d
� �Z e�eed�d�dd� �Z e�eed�d�dd� �Z
e�eed�d�dd� �Ze�eed�d�dd� �Ze�eed�d�d d � �Zd S ) �PosixUidGidTests� �setuidztest needs os.setuid()c C sB t �� dkr | �tt jd� | �tt jd� | �tt j| j� d S �Nr z
not an int)r �getuidrZ r5 r� ra �
OverflowError� UID_OVERFLOWrH r r r � test_setuid> s z PosixUidGidTests.test_setuid�setgidztest needs os.setgid()c C sF t �� dkr ts | �tt jd� | �tt jd� | �tt j| j� d S r� ) r r� �HAVE_WHEEL_GROUPrZ r5 r� ra r� � GID_OVERFLOWrH r r r � test_setgidE s z PosixUidGidTests.test_setgid�seteuidztest needs os.seteuid()c C sB t �� dkr | �tt jd� | �tt jd� | �tt j| j� d S r� ) r r� rZ r5 r� ra �setegidr� r� rH r r r � test_seteuidL s z PosixUidGidTests.test_seteuidr� ztest needs os.setegid()c C sF t �� dkr ts | �tt jd� | �tt jd� | �tt j| j� d S r� ) r r� r� rZ r5 r� ra r� r� rH r r r � test_setegidS s z PosixUidGidTests.test_setegid�setreuidztest needs os.setreuid()c C sn t �� dkr | �tt jdd� | �tt jdd� | �tt jdd� | �tt j| jd� | �tt jd| j� d S r� )r r� rZ r5 r� ra r� r� rH r r r �
test_setreuidZ s z PosixUidGidTests.test_setreuidc C s t �tjddg� d S )Nr� z,import os,sys;os.setreuid(-1,-1);sys.exit(0)�r r r_ r� rH r r r �test_setreuid_neg1c s �z#PosixUidGidTests.test_setreuid_neg1�setregidztest needs os.setregid()c C sr t �� dkr"ts"| �tt jdd� | �tt jdd� | �tt jdd� | �tt j| jd� | �tt jd| j� d S r� ) r r� r� rZ r5 r� ra r� r� rH r r r �
test_setregidk s z PosixUidGidTests.test_setregidc C s t �tjddg� d S )Nr� z,import os,sys;os.setregid(-1,-1);sys.exit(0)r� rH r r r �test_setregid_neg1t s �z#PosixUidGidTests.test_setregid_neg1N)rA rB rC r� r� r r r r r� r� r� r� r� r� r� r� r r r r r� 8 s$
r� c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Ze�e e
d �d
�d d � �Z d
d� Z dS )� Pep383Testsc s� t jrt j| _nt jr t j| _nt j| _t�| j�| _g � � fdd�}|t j� t jr`|t j� t jrp|t j� � s~| � d� t
� | _ t� | j� zL� D ]B}t �
tj�| j|�� t�|�}|| j v r�td��| j �|� q�W n t�| j� � Y n0 d S )Nc s2 zt �| �} W n ty" Y d S 0 � �| � d S r" )r r r� r� )�fn��bytesfnr r � add_filename� s
z'Pep383Tests.setUp.<locals>.add_filenamez&couldn't create any non-ascii filenamezduplicate filename)r �TESTFN_UNENCODABLEr� ZTESTFN_NONASCIIrF r r ZbdirZTESTFN_UNICODErX r � unicodefnr2 Zcreate_empty_filer/ r1 r? r� �add�shutilr )r' r� r� r r� r rI ~ s6
zPep383Tests.setUpc C s t �| j� d S r" )r r r� rH r r r r� � s zPep383Tests.tearDownc C sv | j }tt�| j��}| �||� t�� }z:t�tj� | �tt�� �tt�tj��� W t�|� n t�|� 0 d S r" ) r� r r r� r� r. r# r3 r0 )r' r: �foundZcurrent_directoryr r r � test_listdir� s "zPep383Tests.test_listdirc C s. | j D ]"}ttj�| j|�d�}|�� qd S �Nrs )r� r r r/ r1 r� rL )r' r� rQ r r r � test_open� s
zPep383Tests.test_openr� r� c C s* | j D ] }tj�| j|�}t�|� qd S r" )r� r r/ r1 r� r� )r' r� �fullnamer r r � test_statvfs� s
zPep383Tests.test_statvfsc C s& | j D ]}t�tj�| j|�� qd S r" )r� r r� r/ r1 r� )r' r� r r r � test_stat� s
zPep383Tests.test_statN)
rA rB rC rI r� r r r r r r r r r r r r r� | s % �
r� c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Ze�d �d
d � �Z d d
� Z
dS )�Win32KillTestsc C sv dd l }ddl m} dd l}|jjj}|j|_|j|� |j
�|j |� |j �|� |j �|� |j �f|_ d}t
jtjdd�|�gt
jt
jt
jd�}| �|jj� | �|jj� | �|jj� d\}} || k �rJ|�� d u �rJ|�t|��}
||�|j� � �|
|� |
�d d d �} | � | dd� |
j �r6| � ||
j �!� � �qTt"�#d � |d
7 }q�| �$d � t%�&|j'|� | � |�(� |� d S ) Nr ��wintypes�runningr� z<import sys;sys.stdout.write('{}');sys.stdout.flush();input())r� �stderr�stdin�r r� zPeekNamedPipe failed皙�����?rS z1Did not receive communication from the subprocess))rD r
�msvcrtrE rF �
PeekNamedPipe�BOOL�restype�HANDLE�POINTERZc_char�DWORD�argtypesr �Popenr_ r� ru r� rt r� rL r r
�pollZcreate_string_bufferr, Z
get_osfhandlerj Zsizeofr r� r. r� r; �sleepr� r �kill�pid�wait) r' ZsigrD r
r r r: r� rh �maxrH Zrsltr r r �_kill� sP
�
����
zWin32KillTests._killc C s | � tj� d S r" )r �signal�SIGTERMrH r r r �test_kill_sigterm� s z Win32KillTests.test_kill_sigtermc C s | � d� d S )Nr� )r rH r r r �
test_kill_int s z Win32KillTests.test_kill_intc C s� dt �� }t�dd|�}d|d< tjtjtj� tj�
t �d�|gtj d�}d\}}||k r�|�
� d u r�|d dkrvq�t�d� |d7 }qTt�|jtj� | �d � t�|j|� t�d
� |�
� s�t�|jtj� | �d �|�� d S ) Nz
test_os_%sr� rS r zwin_console_handler.pyr r r z'Subprocess didn't finish initializationg �?z subprocess did not stop on {})�uuidZuuid1�mmapr r r_ r� r r/ r1 r9 r� ZCREATE_NEW_PROCESS_GROUPr r; r r r r �SIGINTr� ru )r' Zeventr Ztagname�mr� rh r r r r �_kill_with_event s0 ���
z Win32KillTests._kill_with_eventz.subprocesses aren't inheriting Ctrl+C propertyc C sd ddl m} dd l }|�|j�� }|jjj}|�|j�|jf|_|j|_ ||d� | �
t j d� d S )Nr r � CTRL_C_EVENT)
rD r
r Zc_intrE rF �SetConsoleCtrlHandlerr r r r( r r) )r' r
rD ZNULLr* r r r �test_CTRL_C_EVENT! s
�
z Win32KillTests.test_CTRL_C_EVENTc C s | � tjd� d S )N�CTRL_BREAK_EVENT)r( r r, rH r r r �test_CTRL_BREAK_EVENT4 s z$Win32KillTests.test_CTRL_BREAK_EVENTN) rA rB rC r r" r# r( r �skipr+ r- r r r r r � s 6
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Win32ListdirTestszTest listdir on Windows.c C s� g | _ td�D ]�}d| }tj�tj|�}d| }tj�tj|�}t�|� t|ddd�� }|� d| � W d � n1 s~0 Y | j �
||g� q| j � � d S )NrU zSUB%dzFILE%dr r� r� z'I'm %s and proud of it. Blame test_os.
) �
created_pathsr� r r/ r1 r rF r� r r �extendr� )r' r� Zdir_nameZdir_path� file_nameZ file_pathrQ r r r rI < s
,zWin32ListdirTests.setUpc C s t �tj� d S r" )r r r rF rH r r r r� I s zWin32ListdirTests.tearDownc C sH | � tt�tj��| j� | � tt�t�tj���dd� | jD �� dS )z-Test when the path is not an "extended" path.c S s g | ]}t �|��qS r �r r �r r/ r r r r V r
zCWin32ListdirTests.test_listdir_no_extended_path.<locals>.<listcomp>N)r. �sortedr r� r rF r0 r rH r r r � test_listdir_no_extended_pathL s ��z/Win32ListdirTests.test_listdir_no_extended_pathc C sh dt j�tj� }| �tt �|��| j� dt � t j�tj�� }| �tt �|��dd� | jD �� dS )z&Test when the path starts with '\\?\'.�\\?\� \\?\c S s g | ]}t �|��qS r r3 r4 r r r r e r
z@Win32ListdirTests.test_listdir_extended_path.<locals>.<listcomp>N)
r r/ rC r rF r. r5 r� r0 r r7 r r r �test_listdir_extended_pathX s � �z,Win32ListdirTests.test_listdir_extended_pathN)rA rB rC r� rI r� r6 r9 r r r r r/ 8 s
r/ �readlinkzneeds os.readlink()c @ sz e Zd ZdZej�e�Ze� e�Z
e� e�Z dd� Z dd� Z
dd� Zdd � Zejd
d � �Zejd d
� �Zejdd� �ZdS )�
ReadlinkTestsZ readlinktestc C sb t j�|�}t j�|�}tjdkrRdd� }||�r>|dd � }||�rR|dd � }| �||� d S )Nr� c S s | � t| t�rdnd�S )Nr8 r7 )�
startswithr� r> )r� r r r �<lambda>u s �z/ReadlinkTests.assertPathEqual.<locals>.<lambda>rh )r r/ �normcaser_ r� r. )r' �left�rightZ
has_prefixr r r �assertPathEqualo s
z ReadlinkTests.assertPathEqualc C sT | � tj�| j�� | � tj�| j�� | �tj�| j�� | �tj�| j�� d S r" ) rM r r/ r� �filelink_target�filelinkb_targetrU �filelink� filelinkbrH r r r rI } s zReadlinkTests.setUpc C s0 t | j�}| �ttj| j� | �ttj|� d S r" )r rB rZ r5 r r: )r' rB r r r �test_not_symlink� s
z ReadlinkTests.test_not_symlinkc C s( | � ttjd� | � ttjtd�� d S )Nz missing-link)rZ r4 r r: r rH r r r �test_missing_link� s
�z ReadlinkTests.test_missing_linkc C sB t �| j| j� | �tj| j� t| j�}| �t � |�| j� d S r" )
r r� rB rD rt r rG r rA r: )r' rD r r r �
test_pathlike� s
zReadlinkTests.test_pathlikec C sN t �| j| j� | �tj| j� t �t| j��}| � || j� | �
|t � d S r" ) r r� rC rE rt r rG r: r rA r$ r> r7 r r r �test_pathlike_bytes� s
z!ReadlinkTests.test_pathlike_bytesc C sJ t �| j| j� | �tj| j� t �| j�}| �|| j� | � |t
� d S r" ) r r� rC rE rt r rG r: rA r$ r> r7 r r r �
test_bytes� s
zReadlinkTests.test_bytesN)rA rB rC rD r r/ rC r� rB r rE rC rA rI rF rG r �skip_unless_symlinkrH rI rJ r r r r r; h s
r; c @ s� e Zd ZdZej�e�ZdZ ej�
e�Z dZ dd� Z
dd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Ze�ej�d�o�ej�d�d�dd � �Zd d � Z d d � Z d!S )"�Win32SymlinkTestsZ filelinktestZ dirlinktestz missing linkc C s d S r" r rH r r r rI � s zWin32SymlinkTests.setUpc C sR t j�| j�rt �| j� t j�| j�r4t �| j� t j�| j�rNt �| j� d S r" ) r r/ r� rD r� �dirlinkr rE � missing_linkrH r r r r� � s zWin32SymlinkTests.tearDownc C s` t �| j| j� | �t j�| j�� | �t j�| j�� | �t j�| j�� | � | j| j� d S r" )
r r� �dirlink_targetrM rM r/ r� r# �islink�
check_statrH r r r �test_directory_link� s
z%Win32SymlinkTests.test_directory_linkc C s` t �| j| j� | �t j�| j�� | �t j�| j�� | �t j�| j�� | � | j| j� d S r" )
r r� rB rD rM r/ r� �isfilerP rQ rH r r r �test_file_link� s
z Win32SymlinkTests.test_file_linkc C s6 | j }tj�|�r t�|� d}d}t�|||� dS )z2Create a "directory" link to a non-existent targetz"c:\\target does not exist.29r3c740TN)rN r r/ rE r� r� )r' Zlinknamer� Z
target_is_dirr r r �_create_missing_dir_link� s
z*Win32SymlinkTests._create_missing_dir_linkc C s | � � t�| j� d S r" )rU r r� rN rH r r r �,test_remove_directory_link_to_missing_target� s z>Win32SymlinkTests.test_remove_directory_link_to_missing_targetc C s | � � | �tj�| j�� d S r" )rU rU r r/ r# rN rH r r r �.test_isdir_on_directory_link_to_missing_target� s z@Win32SymlinkTests.test_isdir_on_directory_link_to_missing_targetc C s | � � t�| j� d S r" )rU r r rN rH r r r �.test_rmdir_on_directory_link_to_missing_target� s z@Win32SymlinkTests.test_rmdir_on_directory_link_to_missing_targetc C sn | � t�|�t�|�� | �t�|�t�|�� t�|�}| � t�|�t�|�� | �t�|�t�|�� d S r" )r. r r� r r/ r )r' r� r� Z
bytes_linkr r r rQ � s
z Win32SymlinkTests.check_statc C sJ t j�tj�}t j�|d�}t j�|d�}| �tj|� t �|� t �|� t �|� t j�t j�|d��}t |� t �
� }z�t � |� t j�|d�}t � t j�
|�d� | �dt �t �
� �� | �t �|�t �d�� t � |� | �t �|�t �t j�
|��� t � |� | �t �|�t �t j�
|��� W t � |� n t � |� 0 d S )N�level2�level3r� r� )r r/ rC r rF r1 rt r r2 r r# r3 r� �relpathr� r� r. r� )r' Zlevel1rY rZ r� Zorig_dirr� r r r �
test_12084� s2
�
�z Win32SymlinkTests.test_12084�C:\Users\All Users�C:\ProgramDatazTest directories not foundc C s" t �d�}| �t j�|d�� d S )Nr] r^ )r r: rM r/ �samefile)r' r� r r r �
test_29248
s
z Win32SymlinkTests.test_29248c C s� d}t jj|gd � }d| |f||f|d d� |d d� fg}|D ]�\}}zt �||� W n tyn Y n$0 zt �|� W n ty� Y n0 z t �t �|�t �|�� W n ty� Y qD0 zt �|� W qD ty� Y qD0 qDd S )NZXXXXXXXXXXXXXXXXXXXXXXXXXXXrT rA � )r r/ r1 r� r4 r� r5 r )r' Zsegmentr/ Z
test_casesr� �destr r r �test_buffer_overflow
s.
� z&Win32SymlinkTests.test_buffer_overflowc s� t j�d�� t j�� �s"| �d� � fdd�t�t �� �d�D �}|D ]\}tj r`t
� t
d|� t � |�}| � |t �
|�� | �t
�|j�� | � |jt
j� q�qF| �d� d S )Nz$%LOCALAPPDATA%\Microsoft\WindowsAppsz%test requires a WindowsApps directoryc s g | ]}t j�� |��qS r )r r/ r1 �r r+ �rG r r r F
s �z6Win32SymlinkTests.test_appexeclink.<locals>.<listcomp>z*.exez Testing withz$test requires an app execution alias)r r/ �
expandvarsr# rX �fnmatch�filterr� r r7 r8 r/ r. r� rU �S_ISLNKr� Zst_reparse_tagZIO_REPARSE_TAG_APPEXECLINK)r' �aliases�aliasr r re r �test_appexeclinkA
s
�
z"Win32SymlinkTests.test_appexeclinkN) rA rB rC rD r r/ rC r� rB rM r9 rO rN rI r� rR rT rU rV rW rX rQ r\ r r rE r� r` rc rl r r r r rL � s,
#
�
#rL c @ sD e Zd ZdZej�ej�e��Z dd� Z
dd� Z dd� Z dd � Z
d
S ) �Win32JunctionTestsZ junctiontestc C s d S r" r rH r r r rI [
s zWin32JunctionTests.setUpc C s t j�| j�rt �| j� d S r" )r r/ rE �junctionrG rH r r r r� _
s zWin32JunctionTests.tearDownc C s� t �| j| j� | �tj�| j�� | �tj�| j�� | �tj� | j�� | �
t� | j�t� | j�� | �
t� | j�t� | j�� | �tj�| j�� | �
tj�d| j �tj�t�| j��� d S )Nr7 )�_winapi�CreateJunction�junction_targetrn rM r r/ rE r� r# r r� r/ r. rU rP r> r: rH r r r �test_create_junctionc
s �z'Win32JunctionTests.test_create_junctionc C s\ t �| j| j� | �tj�| j�� | �tj�| j�� t� | j� | �
tj�| j�� d S r" ) ro rp rq rn rM r r/ r� rE rG rU rH r r r � test_unlink_removes_junctionp
s
z/Win32JunctionTests.test_unlink_removes_junctionN)rA rB rC rn r r/ r9 rC r� rq rI r� rr rs r r r r rm V
s
rm c @ s e Zd Zdd� Zdd� ZdS )� Win32NtTestsc
C s, t �d�}t �d�}dd l}|jddd�}|jj|j_|jj|j _|jj|jj
f|j _ |�� }|j� � }|� ||�
|��}| �d|� |j}dd d
tg}td �D ]R} |D ]H}
z|�|
� W n ty� Y n0 zt�|
� W q� ty� Y q�0 q�q�|� ||�
|��}| �d|� |j| } | �d| � d S ) Nr" rD r z Kernel32.dllT)Zuse_last_errorrS z\\?\C:z\\?\NULz \\?\CONINrT )r �
import_moduleZctypes.wintypesZWinDLLr
r ZGetCurrentProcessr r ZGetProcessHandleCountZLPDWORDr r Zbyrefr. r� r� r� Z_getfinalpathname� Exceptionr r� ) r' r" rD ZkernelZhprocZ handle_countrI Z before_count� filenames�_r Z handle_deltar r r � test_getfinalpathname_handlesz
sB
�
�
z*Win32NtTests.test_getfinalpathname_handlesc C s� t j}| �t j|� t�� d }t�d�}t�t j
d||t |�g��~}t�� |k r�zt �
|� W qF ty� } z W Y d }~qFd }~0 0 qFz|�d� W n tjy� |�� Y n0 W d � n1 s�0 Y d S )Nr� a� import os
import sys
import time
filename = sys.argv[1]
deadline = float(sys.argv[2])
while time.time() < deadline:
try:
with open(filename, "w") as f:
pass
except OSError:
pass
try:
os.remove(filename)
except OSError:
pass
r� rS )r rF rt rG r; �textwrap�dedentr r r_ r� r% r r� r4 r ZTimeoutExpiredZ terminate)r' r ZdeadlineZcommandr� r� r r r �test_stat_unlink_race�
s
z"Win32NtTests.test_stat_unlink_raceN)rA rB rC ry r| r r r r rt x
s 0rt c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�NonLocalSymlinkTestsc C s t �d� dS )zM
Create this structure:
base
\___ some_dir
z
base/some_dirN)r r� rH r r r rI �
s zNonLocalSymlinkTests.setUpc C s t �d� d S )Nr� )r r rH r r r r� �
s z NonLocalSymlinkTests.tearDownc C s t j�dd�}t �d|� dS )a�
The symlink target should resolve relative to the link, not relative
to the current directory.
Then, link base/some_link -> base/some_dir and ensure that some_link
is resolved as a directory.
In issue13772, it was discovered that directory detection failed if
the symlink target was not specified relative to the current
directory, which was a defect in the implementation.
r� Z some_linkZsome_dirN)r r/ r1 r� )r' r� r r r � test_directory_link_nonlocal�
s z1NonLocalSymlinkTests.test_directory_link_nonlocalN)rA rB rC rI r� r~ r r r r r} �
s r} c @ s e Zd Zdd� Zdd� ZdS )�FSEncodingTestsc C s( | � t�d�d� | � t�d�d� d S )Ns abc�u abcŁ)r. r r r? rH r r r �test_nop�
s zFSEncodingTests.test_nopc C sF dD ]<}zt �|�}W n ty, Y qY n0 | �t �|�|� qd S )N)u unicodeŁu latiné�ascii)r r r� r. r? )r' r� r� r r r �
test_identity�
s
z FSEncodingTests.test_identityN)rA rB rC r� r� r r r r r �
s r c @ sV e Zd Zdd� Ze�e�d�oDe� oDe j
� d�pDe e
d�oDe e
d�d�dd � �Zd
S ) �DeviceEncodingTestsc C s | � t�d�� d S )Ni@� )rB r �device_encodingrH r r r � test_bad_fd s z DeviceEncodingTests.test_bad_fdr �win� nl_langinfo�CODESETz>test requires a tty and either Windows or nl_langinfo(CODESET)c C s( t �d�}| �|� | �t�|�� d S r^ )r r� ZassertIsNotNonerM �codecs�lookup)r' r� r r r �test_device_encoding
s
z(DeviceEncodingTests.test_device_encodingN)rA rB rC r� r r r r� r r_ r� r< r �localer� r r r r r� s "�r� c @ s� e Zd Ze�eed�d�dd� �Zddd�Zdd � Z d
d � Z
e�e j d kd
�dd� �Z
e�e j d kd
�dd� �Ze�eed�d�dd� �ZdS )�PidTests�getppidztest needs os.getppidc C s< t jtjddgt jd�}|�� \}}| �t|�t� � � d S )Nr� z import os; print(os.getppid()))r� )
r r r_ r� r� Z communicater. r� r �getpid)r' r� r� rx r r r � test_getppid s
�� zPidTests.test_getppidNc C s� t jdkr(dt j� d�dd|� d�g}n t jd|g}t�tjt j|�}|d urV||� t�|d�\}}| �t�|�|� | �||� d S )Nr� �"r� r ) r_ r� r� r �spawnv�P_NOWAIT�waitpidr. �waitstatus_to_exitcode)r' r� �exitcode�callbackr� r Zpid2Zstatusr r r �
check_waitpid s
zPidTests.check_waitpidc C s | j ddd� d S )Nr� r )r� r� �r� rH r r r � test_waitpid/ s zPidTests.test_waitpidc C sV d}d|� d�}| j ||d� | �t�� t�d� W d � n1 sH0 Y d S )N� zimport sys; sys.exit(�)�r� g )r� rZ ra r r� )r' r� r� r r r �test_waitstatus_to_exitcode2 s
z$PidTests.test_waitstatus_to_exitcoder� �win32-specific testc C s" d}d|� d�}| j ||d� d S )Nl : z$import _winapi; _winapi.ExitProcess(r� r� r� )r' ZSTATUS_CONTROL_C_EXITr� r r r �test_waitpid_windows: s z PidTests.test_waitpid_windowsc C s� d}ddd|fD ]}| � t�|d> �|� q| �t��" t�|d d> � W d � n1 s^0 Y | �t�� t�d� W d � n1 s�0 Y d S )Nr r rS r� r� r� )r. r r� rZ r� r� )r' Z max_exitcoder� r r r �#test_waitstatus_to_exitcode_windowsB s � 0 z,PidTests.test_waitstatus_to_exitcode_windows�SIGKILLzneed signal.SIGKILLc s6 dt j� d�}tj� � fdd�}| j|� |d� d S )Nzimport time; time.sleep(r� c s t �| � � d S r" )r r )r �Zsignumr r � kill_processU s z?PidTests.test_waitstatus_to_exitcode_kill.<locals>.kill_process)r� r� )r Z LONG_TIMEOUTr r� r� )r' r� r� r r� r � test_waitstatus_to_exitcode_killP s z)PidTests.test_waitstatus_to_exitcode_kill)N)rA rB rC r r r r r� r� r� r� r_ r� r� r� r r� r r r r r� s
r� c @ s$ e Zd Zddd�dd�Zed�dd� �Zed�d d
� �Zed �d d
� �Zed�dd� �Zed�dd� �Z ed�dd� �Z
ed�dd� �Z ed�dd � �Z ed�d d � �Z
ed�d d � �Zed�d!d"� �Zed�d#d$� �Zed�d%d&� �Zed�d'd(� �Zd)d*� Zed�d+d,� �Zed�d-d.� �Zd/S )0�
SpawnTestsF��with_env� use_bytesc C s� d| _ tj}| �tj|� |s*d| j }n8ttj�| _t t
� � �| _ | j | j| j < d| j | j f }t
|d��}|�|� W d � n1 s�0 Y tj|g}|r�dd� |D �}dd� | j�� D �| _|S ) N� zimport sys; sys.exit(%s)z4import sys, os; magic = os.environ[%r]; sys.exit(%s)r c S s g | ]}t �|��qS r r3 rd r r r r r r
z*SpawnTests.create_args.<locals>.<listcomp>c S s" i | ]\}}t �|�t �|��qS r r3 )r �k�vr r r �
<dictcomp>s s �z*SpawnTests.create_args.<locals>.<dictcomp>)r� r rF rt rG r\ r r
rx r% r$ Zuuid4rd r r r_ r� rb )r' r� r� r r� r r� r r r � create_args\ s(
� (
�zSpawnTests.create_args�spawnlc C s4 | � � }tjtj|d g|�R � }| �|| j� d S r^ )r� r r� �P_WAITr. r� �r' r� r� r r r � test_spawnlx s zSpawnTests.test_spawnl�spawnlec C s> | j dd�}tjtj|d g|�| j�R � }| �|| j� d S �NT)r� r )r� r r� r� rx r. r� r� r r r � test_spawnle~ s zSpawnTests.test_spawnle�spawnlpc C s4 | � � }tjtj|d g|�R � }| �|| j� d S r^ )r� r r� r� r. r� r� r r r � test_spawnlp� s zSpawnTests.test_spawnlp�spawnlpec C s> | j dd�}tjtj|d g|�| j�R � }| �|| j� d S r� )r� r r� r� rx r. r� r� r r r �
test_spawnlpe� s zSpawnTests.test_spawnlper� c C sT | � � }t�tj|d |�}| �|| j� t�tjt|d �|�}| �|| j� d S r^ )r� r r� r� r. r� r r� r r r � test_spawnv� s
zSpawnTests.test_spawnv�spawnvec C s6 | j dd�}t�tj|d || j�}| �|| j� d S r� �r� r r� r� rx r. r� r� r r r � test_spawnve� s zSpawnTests.test_spawnve�spawnvpc C s. | � � }t�tj|d |�}| �|| j� d S r^ )r� r r� r� r. r� r� r r r � test_spawnvp� s zSpawnTests.test_spawnvp�spawnvpec C s6 | j dd�}t�tj|d || j�}| �|| j� d S r� )r� r r� r� rx r. r� r� r r r �
test_spawnvpe� s zSpawnTests.test_spawnvpec C s0 | � � }t�tj|d |�}tj|| jd� d S )Nr r� )r� r r� r� r Z wait_processr� )r' r� r r r r � test_nowait� s zSpawnTests.test_nowaitc C s8 | j ddd�}t�tj|d || j�}| �|| j� d S )NTr� r r� r� r r r �test_spawnve_bytes� s z SpawnTests.test_spawnve_bytesc C s> | � � }| �ttjtj|d � | �ttjtj|d d� d S �Nr rB )r� rZ r� r r� r� �r' r� r r r �test_spawnl_noargs� s z SpawnTests.test_spawnl_noargsc C sB | � � }| �ttjtj|d i � | �ttjtj|d di � d S r� )r� rZ r� r r� r� r� r r r �test_spawnle_noargs� s z SpawnTests.test_spawnle_noargsc C sv | � � }| �ttjtj|d d� | �ttjtj|d g � | �ttjtj|d d� | �ttjtj|d dg� d S �Nr r r| rB )r� rZ r� r r� r� r� r r r �test_spawnv_noargs� s
z SpawnTests.test_spawnv_noargsc C s~ | � � }| �ttjtj|d di � | �ttjtj|d g i � | �ttjtj|d di � | �ttjtj|d dgi � d S r� )r� rZ r� r r� r� r� r r r �test_spawnve_noargs� s
z SpawnTests.test_spawnve_noargsc C st t jddg}tj�� }d|d< z|tj|d ||�}W n tyH Y n0 | �|d� tj�� }d|d< z|tj|d ||�}W n ty� Y n0 | �|d� tj�� }d |d
< z|tj|d ||�}W n ty� Y n0 | �|d� tj }| �
tj |� t |d ��}|�
d � W d � n1 �s*0 Y t j|g}tj�� }d
|d< |tj|d ||�}| �|d� d S )Nr� r� r� r� r � r� r� r� r� r zPimport sys, os
if os.getenv("FRUIT") != "orange=lemon":
raise AssertionErrorz orange=lemon)r_ r� r r
r� r� r� r. r rF rt rG r r )r' Zspawnr� r� r� r r r r r �_test_invalid_env� s>
*
z SpawnTests._test_invalid_envc C s | � tj� d S r" )r� r r� rH r r r �test_spawnve_invalid_env s z#SpawnTests.test_spawnve_invalid_envc C s | � tj� d S r" )r� r r� rH r r r �test_spawnvpe_invalid_env s z$SpawnTests.test_spawnvpe_invalid_envN)rA rB rC r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� [ sD
.
r� z>Skip due to platform/environment differences on *NIX buildbots�getloginztest needs os.getloginc @ s e Zd Zdd� ZdS )�
LoginTestsc C s t �� }| �t|�d� d S r^ )r r� r r, )r' Z user_namer r r �
test_getlogin s zLoginTests.test_getloginN)rA rB rC r� r r r r r� s r� � getpriority� setpriorityz'needs os.getpriority and os.setpriorityc @ s e Zd ZdZdd� ZdS )�ProgramPriorityTestsz0Tests for os.getpriority() and os.setpriority().c C s t �t jt �� �}t �t jt �� |d � z�t �t jt �� �}|dkr^|dkr^t�d| ��n| �||d � W zt �t jt �� |� W n2 ty� } z|j t j
kr�� W Y d }~n
d }~0 0 nPzt �t jt �� |� W n4 t�y
} z|j t j
kr�� W Y d }~n
d }~0 0 0 d S )NrS � z?unable to reliably test setpriority at current nice level of %s) r r� � PRIO_PROCESSr� r� r r0 r. r5 r6 ZEACCES)r' r� Znew_prior| r r r �test_set_get_priority s& � � z*ProgramPriorityTests.test_set_get_priorityN)rA rB rC r� r� r r r r r� s r� c @ sv e Zd ZG dd� dej�Zdd� Zedd� �Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZeZdd� Zdd� ZdS )�SendfileTestServerc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) zSendfileTestServer.Handlerc C s. t j�| |� g | _d| _d| _| �d� d S )NTFs 220 ready
)�asynchat�
async_chat�__init__� in_buffer�
accumulate�closed�push)r' �connr r r r� 2 s
z#SendfileTestServer.Handler.__init__c C s | � d�}| jr | j�|� d S )N� )�recvr� r� r� rm r r r � handle_read9 s
z&SendfileTestServer.Handler.handle_readc C s d� | j�S )Nr
)r1 r� rH r r r �get_data> s z#SendfileTestServer.Handler.get_datac C s | � � d| _d S r� )rL r� rH r r r � handle_closeA s z'SendfileTestServer.Handler.handle_closec C s � d S r" r rH r r r � handle_errorE s z'SendfileTestServer.Handler.handle_errorN)rA rB rC r� r� r� r� r� r r r r �Handler0 s
r� c C sp t j�| � tj�| � | �tjtj� | � |� | �
d� | j� � d d� \| _ | _
d | _d| _t �� | _d S )Nr� rU F)� threading�Threadr� �asyncore�
dispatcherZ
create_socket�socketZAF_INETZ SOCK_STREAMZbindZlistenZ getsockname�host�port�handler_instance�_activeZLock� _active_lock)r' Zaddressr r r r� H s
zSendfileTestServer.__init__c C s | j S r" )r� rH r r r r U s zSendfileTestServer.runningc C s$ t �� | _t j�| � | j�� d S r" )r� ZEvent�_SendfileTestServer__flagr� �startr rH r r r r� Y s
zSendfileTestServer.startc C s d| _ | �� d S r� )r� r1 rH r r r �stop_ s zSendfileTestServer.stopc C s& t | jdd�st�d� q | �� d S )Nr� F���MbP?)r� r� r; r r� rH r r r r d s zSendfileTestServer.waitc C sL d| _ | j�� | j r@tjr@| j�� tjddd� | j�� qt� � d S )NTr� rS )Ztimeoutrh )
r� r� r r� Z
socket_mapr� �acquireZloop�releaseZ close_allrH r r r r� l s
zSendfileTestServer.runc C s | � � \}}| �|�| _d S r" )Zacceptr� r� )r' r� Zaddrr r r �
handle_acceptu s z SendfileTestServer.handle_acceptc C s | � � d S r" )rL rH r r r �handle_connecty s z!SendfileTestServer.handle_connectc C s dS r^ r rH r r r �writable} s zSendfileTestServer.writablec C s � d S r" r rH r r r r� � s z SendfileTestServer.handle_errorN)rA rB rC r� r� r� r� �propertyr r� r� r r� r� r� r� r� r� r r r r r� . s
r� �sendfileztest needs os.sendfile()c @ s
e Zd Zdd Zej�d� o8ej�d� o8ej�d� Ze� ed�Z
e� ej dk d�Z e
d d
� �Ze
d d � �Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Ze
d d � �Ze
d d � �Ze
e d!d"� ��Ze
e d#d$� ��Ze
e� e e d%�d&�d'd(� ��Z d)S )*� TestSendfiles� 12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde� ZlinuxZsolarisZsunosz%requires headers and trailers supportr� z(test is only meaningful on 32-bit buildsc C s t �� | _tt j| j� d S r" )r Zthreading_setuprd r rF �DATAr: r r r r< � s
zTestSendfile.setUpClassc C s t j| j� t �t j� d S r" )r Zthreading_cleanuprd rG rF r: r r r rO � s zTestSendfile.tearDownClassc C s~ t tjdf�| _| j�� t�� | _| j�| jj| jj f� | j�
d� | j� d� | j� � | _
ttjd�| _| j� � | _ d S )Nr rS r� rs )r� r ZHOST�serverr� r� �clientZconnectr� r� Z
settimeoutr� rj �socknor r rF r� rH r r r rI � s
zTestSendfile.setUpc C s0 | j �� | j�� | jjr&| j�� d | _d S r" )r� rL r r r r� rH r r r r� � s
zTestSendfile.tearDownc
O sp zt j|i |��W S tyh } z>|jtjkr2� n"|jtjtjfv rRW Y d}~q n� W Y d}~q d}~0 0 q dS )zfA higher level wrapper representing how an application is
supposed to use sendfile().
N)r r� r5 r6 Z
ECONNRESET�EAGAIN�EBUSY)r' r� r� r| r r r �sendfile_wrapper� s z TestSendfile.sendfile_wrapperc C s� d}d}d}|t | j�k rd| �| j| j||�}|dkr8qd||7 }||7 }| �||k� | �||� q | �|t | j�� | j�t j
� | j� � | j �
� | j j�� }| �t |�t | j�� | �|| j� d S )Nr r� )r, r� r r rj rM r. r �shutdownr� � SHUT_RDWRrL r r r� r� )r' �
total_sent�offset�nbytes�sentru r r r �test_send_whole_file� s$
z!TestSendfile.test_send_whole_filec C s� d}t | j�d }t | j�| }d}||k rj| �| j| j||�}|dkrJqj||7 }||7 }| �||k� q$| j�tj � | j�
� | j � � | j j
�� }| jt | j�d d � }| �|t |�� | �t |�t |�� | �||� d S )Nr rU r� )r, r� r r rj rM r r r� r rL r r r� r� r. )r' r r Z must_sendr
r ru r: r r r �test_send_at_certain_offset� s&
z(TestSendfile.test_send_at_certain_offsetc
C s� t | j�d }zt�| j| j|d�}W n2 tyX } z|jtjkrD� W Y d }~nd }~0 0 | � |d� | j
� t j
� | j
�� | j�� | jj�� }| � |d� d S )Nr� r r
)r, r� r r� r rj r5 r6 �EINVALr. r r r� r rL r r r� r� )r' r r r� ru r r r �test_offset_overflow� s
z!TestSendfile.test_offset_overflowc C sT | � t��$}t�| j| jdd� W d � n1 s40 Y | �|jjtj � d S )Nr� r� )
rZ r5 r r� r rj r. r� r6 r �r' r� r r r �test_invalid_offset� s 2z TestSendfile.test_invalid_offsetc C s< t j| j| jddd� | jr8t j| j| jdddddd� d S )Nr r� )r� r� r rh r )r� r� r rh �headers�trailersr� )r r� r rj �SUPPORT_HEADERS_TRAILERSrH r r r �
test_keywords
s � �zTestSendfile.test_keywordsc C s� d}d| j d d� }tj| j| jddddgd�}| �|d� ||7 }d}|t|�k r�tt|�| d�}| �| j| j||�}|dkr�q�| �||� ||7 }||7 }qJ| � |t|�� | j
� � | j �
� | j j�� }| � t|�t|�� d S ) Nr s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyr� r� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy�r i )r� r r� r rj ZassertLessEqualr, �minr r. r rL r r r� r� �hash)r' r Z
expected_datar r r
ru r r r � test_headers
s. � �
zTestSendfile.test_headersc C s� t jd }d}| �t j|� t||� t|d��Z}tj| j|� � ddddgd� | j
� � | j �
� | j j�� }| �|d � W d � n1 s�0 Y d S )
Nr� s abcdefrs r r� s 123456s 789�r s abcde123456789)r rF rt rG r r r r� r rj r rL r r r� r� r. )r' r� Z file_datarQ ru r r r �
test_trailers%
s
�
zTestSendfile.test_trailersc C sl d| j j_| �t��2}tj| j| jdddd gd d� W d � n1 sL0 Y | � |j
j t j � d S )NFr � x� r� r �
r r� r� rZ r5 r r� r rj r. r� r6 r r r r r � test_headers_overflow_32bits5
s
�$z)TestSendfile.test_headers_overflow_32bitsc C sl d| j j_| �t��2}tj| j| jdddd gd d� W d � n1 sL0 Y | � |j
j t j � d S )NFr r r r� r r r r r r � test_trailers_overflow_32bits>
s
�$z*TestSendfile.test_trailers_overflow_32bits� SF_NODISKIOztest needs os.SF_NODISKIOc
C s\ z t j| j| jddt jd� W n8 tyV } z |jtjtjfvrB� W Y d }~n
d }~0 0 d S )Nr r� )r� ) r r� r rj r! r5 r6 r r )r' r| r r r �
test_flagsG
s �
zTestSendfile.test_flagsN) rA rB rC r� r_ r� r< r r r Zrequires_headers_trailers�maxsizeZ requires_32brP r< rO rI r� r r r
r r r r r r r r r r" r r r r r� � sN � �� �
�r� c C s� t td�sdS z�ttjdd��V} zt�| �� dd� W n. tyd Y W d � W t�tj� dS 0 W d � n1 sz0 Y W t�tj� nt�tj� 0 dS )N�setxattrFr r � user.testr
T) r r r r rF r$ rj r5 rG )r r r r � supports_extended_attributesS
s
�& r& z(no non-broken extended attribute supportrU r� �' c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ExtendedAttributeTestsc
K sJ t j}| �t j|� t|� | �t��&}|||d�fi |�� W d � n1 sR0 Y | �|jj t j
� ||�} | � | t � |||d�dfi |�� t
| �}
|
�d� | �t
||��|
� | �||dfi |��d� |||d�dtjfi |�� | �||dfi |��d� | �t��,}|||d�dtjfi |�� W d � n1 �sN0 Y | �|jj t j� | �t��,}|||d�dtjfi |�� W d � n1 �s�0 Y | �|jj t j
� |||d�dtjfi |�� |
�d� | �t
||��|
� |||d�fi |�� | �t��&}|||d�fi |�� W d � n1 �sL0 Y | �|jj t j
� |
�d� | �t
||��|
� | �|||d�fi |��d� |||d�dfi |�� | �|||d�fi |��d� |||d�fi |�� td d
� td �D ��} | D ]} ||| d fi |�� �q
| �t
||��t
| �t
| �B � d S )
N� user.testr
r% r] s byez
user.test2s foos aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaac s s | ]}d � |�V qdS )z user.test{}Nrt )r r� r r r rv �
r
z;ExtendedAttributeTests._check_xattrs_str.<locals>.<genexpr>r� r )r rF rt rG r rZ r5 r. r� r6 ZENODATAr$ r� r r� r �
XATTR_REPLACE� XATTR_CREATEZEEXISTr� r5 r� )
r' rp �getxattrr$ � removexattr� listxattrr� r� r� Z
init_xattrZxattr�many�thingr r r �_check_xattrs_stri
sL 4
< <
6
z(ExtendedAttributeTests._check_xattrs_strc O sN | j tg|�R i |�� t�tj� | j tjg|�R i |�� t�tj� d S r" )r1 r% r rG rF r r r r r r �
_check_xattrs�
s z$ExtendedAttributeTests._check_xattrsc C s | � tjtjtjtj� d S r" �r2 r r, r$ r- r. rH r r r � test_simple�
s �z"ExtendedAttributeTests.test_simplec C s | j tjtjtjtjdd� d S )NFrL r3 rH r r r �
test_lpath�
s �z!ExtendedAttributeTests.test_lpathc C s4 dd� }dd� }dd� }dd� }| � ||||� d S ) Nc W sD t | d��&}tj|�� g|�R � W d � S 1 s60 Y d S r )r r r, rj �r/ r� r r r r r, �
s z1ExtendedAttributeTests.test_fds.<locals>.getxattrc W sF t | dd��&}tj|�� g|�R � W d � n1 s80 Y d S �Nr3 r )r r r$ rj r6 r r r r$ �
s z1ExtendedAttributeTests.test_fds.<locals>.setxattrc W sF t | dd��&}tj|�� g|�R � W d � n1 s80 Y d S r7 )r r r- rj r6 r r r r- �
s z4ExtendedAttributeTests.test_fds.<locals>.removexattrc W sD t | d��&}tj|�� g|�R � W d � S 1 s60 Y d S r )r r r. rj r6 r r r r. �
s z2ExtendedAttributeTests.test_fds.<locals>.listxattr)r2 )r' r, r$ r- r. r r r �test_fds�
s
z ExtendedAttributeTests.test_fdsN)rA rB rC r1 r2 r4 r5 r8 r r r r r( c
s
0r( �get_terminal_sizez requires os.get_terminal_sizec @ s e Zd Zdd� Zdd� ZdS )�
TermsizeTestsc
C sz z t �� }W nL tyX } z4tjdks8|jtjtjfv rB| �d� � W Y d}~n
d}~0 0 | � |j
d� | � |j d� dS )z�Check if get_terminal_size() returns a meaningful value.
There's no easy portable way to actually check the size of the
terminal, so let's check if it returns something sensible instead.
r� � failed to query terminal sizeNr ) r r9 r5 r_ r� r6 r �ENOTTYrX ZassertGreaterEqual�columns�lines)r' rr r� r r r �test_does_not_crash�
s
z!TermsizeTests.test_does_not_crashc
C s� z t jddgt jdd��� }W n$ tt jtfyB | �d� Y n0 t|d �t|d �f}zt �
t j �
� �}W nL ty� } z4t jdks�|jtjtjfv r�| �d � � W Y d
}~n
d
}~0 0 | �||� d
S ) z�Check if stty returns the same results
stty actually tests stdin, so get_terminal_size is invoked on
stdin explicitly. If stty succeeded, then get_terminal_size()
should work too.
Zsttyrr T)r r� zstty invocation failedrS r r� r; N)r r3 ZDEVNULLr� r4 ZCalledProcessErrorr� rX r� r r9 r_ � __stdin__rj r5 r� r6 r r< r. )r' rr r: r� r� r r r �test_stty_match�
s" ���
z TermsizeTests.test_stty_matchN)rA rB rC r? rA r r r r r: �
s r: � memfd_createzrequires os.memfd_creater� r� c @ s e Zd Zdd� ZdS )�MemfdCreateTestsc C s� t �dt j�}| �|d� | �t j|� | �t �|�� t|ddd��*}|� d� | �
|� � d� W d � n1 sv0 Y t �d�}| �t j|� | �t �|�� d S )NZHir� r3 F)�closefds memfd_creater� ) r rB � MFD_CLOEXECr rt rL rU r� r r r. �tell)r' ro rQ �fd2r r r �test_memfd_create�
s
.
z"MemfdCreateTests.test_memfd_createN)rA rB rC rH r r r r rC �
s rC c @ s e Zd Zdd� Zdd� ZdS )� OSErrorTestsc C s� G dd� dt �}g | _g | _tjd ur.tj}ntj}| j�|� | j�||�� tjd urbtj}n t� tj�}| j�|� | j�t
|�� | j�t |�� | j| j | _ d S )Nc @ s e Zd ZdS )z OSErrorTests.setUp.<locals>.StrN�rA rB rC r r r r �Str�
s rK )
r% �bytes_filenames�unicode_filenamesr r� rF r� ZTESTFN_UNDECODABLEr r ry rz rw )r' rK ZdecodedZencodedr r r rI �
s
zOSErrorTests.setUpc C sJ | j tjf| j tjdf| j tjf| j tjtjf| j tjf| j tjf| j tj fg}t
j dkr�|� | j
tjdf| j
tjdf| jtjdf| jtjdf| jtjff� n,|� | j tjf| j tjdf| j tjdff� ttd�r�|�| j tjddf� ttd��r|�| j tjddf� ttd��r.|�| j tjdf� ttd ��rN|�| j tjdf� ttd
��rn|�| j tjdf� ttd ��r�|�| j tjf� ttd ��r�t
j dk�r�|�| j
tjdf� |�| jtjdf� n|�| j tjdf� ttd
��r(|� | j tjf| j tj df| j tj ddf| j tj dff� ttd��rH|�| j tj df� ttd��rf|�| j tj f� |D ]�^}}}|D ]�}zdt!|t"t#f��r�||g|�R � n>| �$t%d�� ||g|�R � W d � n1 �s�0 Y W nL t&�y } z | �'|j(|t"|�� W Y d }~n,d }~0 t)�y, Y n0 | �*d�+|�� �qx�qjd S )Nr� r� s dstr� r8 r �lchownr� �chflags�lchflags�chrootr� r. r) s user�lchmodr: z should bezNo exception thrown by {}),rw r r3 r� r/ r r� r r� rG r_ r� r1 rL rb r� rM r� r r� r8 rN r� rO rP rQ r� r. r, r$ r- rR r: r� r% r> ZassertWarnsRegex�DeprecationWarningr5 r� r �UnicodeDecodeErrorr� ru )r' Zfuncsrw �funcZ func_argsr r| r r r �test_oserror_filename sz
�
�
�
� 4(z"OSErrorTests.test_oserror_filenameN)rA rB rC rI rV r r r r rI �
s rI c @ s e Zd Zdd� ZdS )�
CPUCountTestsc C s8 t �� }|d ur*| �|t� | �|d� n
| �d� d S )Nr z&Could not determine the number of CPUs)r � cpu_countr$ r� �
assertGreaterrX )r' Zcpusr r r �test_cpu_countX s
z CPUCountTests.test_cpu_countN)rA rB rC rZ r r r r rW W s rW c @ s� e Zd Zdd� Ze�edu d�dd� �Ze�edu d�dd� �Ze� e
e d �d
�d d � �Z d
d� Z
dd� Ze� e
e d�d�dd� �Zdd� Zdd� Ze� ejdkd�dd � �Ze� e
e d �d �d d � �Ze� e
e d!�d"�d#d$� �ZdS )%�FDInheritanceTestsc C sP t �tt j�}| �t j|� | �t �|�d� t �|d� | �t �|�d� d S �NFT) r r r� r� rt rL r. r� r� r� r r r �test_get_set_inheritableb s
z+FDInheritanceTests.test_get_set_inheritableNz
need fcntlc C sn t �tt j�}| �t j|� | �t �|�d� t�|tj �}|tj
M }t�|tj |� | �t �|�d� d S r\ ) r r r� r� rt rL r. r� �fcntl�F_GETFD�
FD_CLOEXECZF_SETFD)r' ro r� r r r � test_get_inheritable_cloexecj s z/FDInheritanceTests.test_get_inheritable_cloexecc C sf t �tt j�}| �t j|� | �t�|tj�tj @ tj � t �
|d� | �t�|tj�tj @ d� d S )NTr ) r r r� r� rt rL r. r^ r_ r` r� r� r r r � test_set_inheritable_cloexecw s � �z/FDInheritanceTests.test_set_inheritable_cloexec�O_PATHzneed os.O_PATHc C sn t �tt j�}| �t j|� | �t �|�d� t �|d� | �t �|�d� t �|d� | �t �|�d� d S r\ ) r r r� rc rt rL r. r� r� r� r r r � test_get_set_inheritable_o_path� s z2FDInheritanceTests.test_get_set_inheritable_o_pathc C s� t �� }| �t��}t�|� W d � n1 s20 Y | �|jjtj � | �t�� }t�
|d� W d � n1 sz0 Y | �|jjtj � | �t�� }t�
|d� W d � n1 s�0 Y | �|jjtj � d S �NTF) r r� rZ r5 r r� r. r� r6 r� r� )r' ro r r r r � test_get_set_inheritable_badf� s ( * *z0FDInheritanceTests.test_get_set_inheritable_badfc C s2 t �tt j�}| �t j|� | �t �|�d� d S r� )r r r� r� rt rL r. r� r� r r r r � s z FDInheritanceTests.test_openr� zneed os.pipe()c C sP t �� \}}| �t j|� | �t j|� | �t �|�d� | �t �|�d� d S r� )r r� rt rL r. r� )r' ZrfdZwfdr r r � test_pipe� s
z FDInheritanceTests.test_pipec C sJ t �tt j�}| �t j|� t �|�}| �t j|� | �t �|�d� d S r� ) r r r� r� rt rL rW r. r� �r' Zfd1rG r r r �test_dup� s
zFDInheritanceTests.test_dupc C s( t �d�}| �t j|� | �|d� d S �NrS r )r rW rt rL rY r� r r r �test_dup_standard_stream� s
z+FDInheritanceTests.test_dup_standard_streamr� r� c C sH t �dt j�}| �t j|� t �|�}| �t j|� | �t �|�� d S )NZNUL)r r r� rt rL rW rU r� rh r r r � test_dup_nul� s
z FDInheritanceTests.test_dup_nulr� zneed os.dup2()c C s� t �tt j�}| �t j|� t �tt j�}| �t j|� | �t �||�|� | �t � |�� t �tt j�}| �t j|� | �t j||dd�|� | �
t � |�� d S )NF)Z inheritable) r r r� r� rt rL r. r� rM r� rU )r' ro rG Zfd3r r r r� � s z FDInheritanceTests.test_dup2�openptyzneed os.openpty()c C sP t �� \}}| �t j|� | �t j|� | �t �|�d� | �t �|�d� d S r� )r rm rt rL r. r� )r' Z master_fdZslave_fdr r r � test_openpty� s
z FDInheritanceTests.test_openpty)rA rB rC r] r r� r^ ra rb r r r rd rf r rg ri rk r_ r� rl r� rn r r r r r[ a s&
r[ c @ sH e Zd Zddddejfdfddddd eed
d�fgZd d � Zd
d� ZdS )�PathTConverterTests)r� Tr N)r/ Fr NrN FN)rO F�r N)rP Frp Nr rp rL c C s t j}tjdkrd }}nt�t j�}t|�}t�t|�tjtjB �}| � t j
t j� | � tj |� t|�}t|�}| j D �]�\}}} }
| j
|d���V ztt|�} W n" ty� Y W d � qzY n0 ||||fD ]`} | d u r�q�| j
|| d��2 | | g| �R � }
|
d u�r|
|
� W d � q�1 �s,0 Y q�| �td�� | |g| �R � W d � n1 �sl0 Y |�r�| |g| �R � }
|
d u�r�|
|
� n>| �td�� | |g| �R � W d � n1 �s�0 Y W d � qz1 �s�0 Y qzd S )Nr" r )r r/ zto return str or bytesz os.PathLike)r rF r r r r r rx rJ rt rG rL � functionsZsubTestr� r� �assertRaisesRegexra )r' Z str_filenameZ bytes_fspath�bytes_filenamero Z
int_fspathZ
str_fspathr Zallow_fdZ
extra_argsZ
cleanup_fnr� r/ r� r r r �test_path_t_converter� sL
�
*�0
�z)PathTConverterTests.test_path_t_converterc C s� d}| � t|d �� t�td�� W d � n1 s80 Y | � t|d �� t�td�� W d � n1 sv0 Y | � t|d �� t�tt� �� W d � n1 s�0 Y d S )Nz-__fspath__\(\) to return str or bytes, not %sr� rU �floatg� ��Q�@�object)rr ra r r� r rv )r' r: r r r �&test_path_t_converter_and_custom_class s ,,z:PathTConverterTests.test_path_t_converter_and_custom_class) rA rB rC r �F_OKr� rq rt rw r r r r ro � s � ,ro r� r� c @ s e Zd Zdd� ZdS )�
BlockingTestsc C sn t �tt j�}| �t j|� | �t �|�d� t �|d� | �t �|�d� t �|d� | �t �|�d� d S re ) r r r� r� rt rL r. r� r� r� r r r r� s zBlockingTests.test_blockingN)rA rB rC r� r r r r ry s ry c @ s e Zd Zdd� ZdS )� ExportsTestsc C s | � dtj� | � dtj� d S )Nr r� )r� r �__all__rH r r r � test_os_all- s zExportsTests.test_os_allN)rA rB rC r| r r r r rz , s rz c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestDirEntryc C s0 t j�tj�| _| �tj| j� t �| j� d S r" )r r/ �realpathr rF rt r r2 rH r r r rI 3 s zTestDirEntry.setUpc C s | � ttj� d S r" )rZ ra r �DirEntryrH r r r �test_uninstantiable8 s z TestDirEntry.test_uninstantiablec C sj t tj�| jd�d�}dd� t�| j�D ��� }| �|tj� | �|j d� dd l
}| � t |j
||� d S )NrY � pythonc S s g | ]}|�qS r r �r �entryr r r r = r
z0TestDirEntry.test_unpickable.<locals>.<listcomp>r )r r r/ r1 �scandirr� r$ r r. r r� rZ ra r� )r' r r� r� r r r �test_unpickable; s z TestDirEntry.test_unpickableN)rA rB rC rI r� r� r r r r r} 2 s r} c @ s
e Zd ZejZdd� Zd<dd�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
d=dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Ze�ejej v d(�d)d*� �Z d+d,� Z d-d.� Z d/d0� Z d1d2� Z!d3d4� Z"d5d6� Z#d7d8� Z$d9d:� Z%d;S )>� TestScandirc C s> t j�tj�| _t �| j�| _| �tj| j� t � | j� d S r" )
r r/ r~ r rF r �
bytes_pathrt r r2 rH r r r rI G s zTestScandir.setUprY c C s2 t |t�r| jn| j}tj�||�}t|d� |S )Nr� )r� r> r� r/ r r1 r )r' r r/ r r r r r M s
zTestScandir.create_filec C s2 t dd� t�| j�D ��}| �t|�� �|� |S )Nc s s | ]}|j |fV qd S r" r r� r r r rv T s �z*TestScandir.get_entries.<locals>.<genexpr>)r\ r r� r/ r. r5 �keys)r' �names�entriesr r r � get_entriesS s
�zTestScandir.get_entriesc C sZ |rJt |�D ]:}|�d�s q |dv r&q | �t||�t||�|||f� q n | �||� d S )NZst_)�st_dev�st_ino�st_nlink)r� r< r. r� )r' Zstat1Zstat2Z skip_fieldsr� r r r �assert_stat_equalY s
�z TestScandir.assert_stat_equalc C s( t �| j�}| �tt|�� |�� d S r" )r r� r/ rZ ra rm rL )r' � scandir_iterr r r r� f s z TestScandir.test_uninstantiablec C s< | � d�}t�| j�}dd l}| �t|j||� |�� d S )NrY r ) r r r� r/ r� rZ ra r� rL )r' r r� r� r r r r� k s
zTestScandir.test_unpickablec C s2 | � |tj� | �|j|� | �|jtj�| j|�� | �|�� tj|jdd�j � t�|j�}| �|�
� t� |j �� | �|�
� t�|j �� | �|�� tj�|j�� tj|jdd�}| �|j
dd�t� |j �� | �|j
dd�t�|j �� | �|�� |tjdk�o| � | �|jdd�|tjdk� d S )NFrL r" )r$ r r r. r r/ r1 �inoder� r� �is_dir�S_ISDIRr� �is_file�S_ISREG�
is_symlinkrP r� )r' r� r r� r� r� Z
entry_statZ entry_lstatr r r � check_entryr s>
�
�
�
�
�
�
��zTestScandir.check_entryc
C s� t td�}t�� }tj�| jd�}t�|� | �d�}|r�z t�|tj�| jd�� W n2 t y� } z| �
d| � W Y d }~n
d }~0 0 |r�tj |tj�| jd�dd� t� |tj�| jd �� ddg}|r�|� d� |r�|�
d
� | �|�}|d }| �|ddd d � |d }| �|dd dd � |�rD|d }| �|dd dd � |�r~|d }| �|ddd d� |d }| �|d d dd� d S ) Nr� r� rY z
link_file.txtr� � symlink_dirT)r� �symlink_file.txt)r� r� F)r r r r� r/ r1 r2 r r� r� rX r� r� r1 r� r� ) r' r� r� r9 r r� r� r� r� r r r �test_attributes� sB
$�
zTestScandir.test_attributesc C sN t |t�r| jn| j}tt�|��}| �t|�d� |d }| �|j |� |S rj )
r� r> r� r/ r� r r� r. r, r )r' r r/ r� r� r r r � get_entry� s zTestScandir.get_entryc C s | j |d�}| �tj�|��S )Nr )r r� r r/ �basename)r' r r r r r �create_file_entry� s z TestScandir.create_file_entryc C sp | � � }t�� }zNt�| j� tdd� t�� D ��}| �t|� � �tj�
|�g� W t�|� n t�|� 0 d S )Nc s s | ]}|j |fV qd S r" r r� r r r rv � r
z5TestScandir.test_current_directory.<locals>.<genexpr>) r r r# r3 r/ r\ r� r. r5 r� r� )r' r Zold_dirr� r r r �test_current_directory� s �z"TestScandir.test_current_directoryc C s | � � }| �t|�d� d S )Nz<DirEntry 'file.txt'>)r� r. rw �r' r� r r r � test_repr� s zTestScandir.test_reprc C s* | � � }| �t�|�tj�| jd�� d S )NrY )r� r. r �fspathr/ r1 r� r r r �test_fspath_protocol� s z TestScandir.test_fspath_protocolc C sN t �d�}| j|d�}t �|�}| �|t� | �|t j�t �| j�|�� d S )Nz
bytesfile.txtr ) r r r� r� r$ r> r. r/ r1 )r' rs Z bytes_entryr� r r r �test_fspath_protocol_bytes� s
�z&TestScandir.test_fspath_protocol_bytesc C s� t j�| jd�}t �|� | �d�}t �|� t jdkrF| �|�� � | � |�
� � | � |� � � t jdkr�| � t
|j� |�� |jdd� n0| �|�� d� | � t
|j� | j t
|jdd� d S )Nr� r" FrL r )r r/ r1 r2 r� r r rM r� rU r� r� rZ r4 r� r� rY )r' r/ r� r r r �test_removed_dir� s
z TestScandir.test_removed_dirc C s� | � � }t�|j� | �|�� � tjdkr:| �|�� � | �|� � � tjdkrv| �
t |j � |�
� |j
dd� n0| �|� � d� | �
t |j
� | j
t |j
dd� d S )Nr" FrL r )r� r rG r/ rU r� r rM r� r� rZ r4 r� r� rY r� r r r �test_removed_file� s
z TestScandir.test_removed_filec C s� t �� s| �d�S | �d�}t�|tj�| jd�� | �ddg�}|d }t� |� | �
|� � d� | � |�
� � | � |�� � | � |j
dd�� | � |jdd�� | �|�� � | �t|j� |jdd� d S )Nzcannot create symbolic linkrY z symlink.txtr FrL )r r� rX r r r� r/ r1 r� rG rY r� rU r� r� rM r� rZ r4 r� )r' r r� r� r r r �test_broken_symlink s"
�
z TestScandir.test_broken_symlinkc C sp | � d� t�| j�}tt�|��}| �t|�d|� |d }| �|jd� | �|jt�tj� | jd��� d S �NrY rS r s file.txt)
r r r r/ r� r� r. r, r r1 )r' �
path_bytesr� r� r r r rJ s
�zTestScandir.test_bytesc
C s� | � d� ttfD ]�}|t�| j��}| �t�� tt� |��}W d � n1 sT0 Y | �
t |�d|� |d }| �
|j d� | �
|jt�tj�
| jd��� | �t|j �t� | �t|j�t� qd S r� )r ry rz r r r/ � assertWarnsrS r� r� r. r, r r1 r� rm r> )r' r; r� r� r� r r r �test_bytes_like, s
,�zTestScandir.test_bytes_likez.fd support for listdir required for this test.c C sv | � tjtj� | �d� dg}t�� rJt�dtj� | jd�� |�
d� t� | j���}t�|��}t |�}W d � n1 s�0 Y dd� |D �}| �
t|�|� | �
|t�|�� |D ]�}| �
|j|j� | �
t�|�|j� | �
|�� |jdk� tjtjv r�tj|j|d�}| �
|�� |� tj|j|dd�}| �
|jdd�|� q�W d � n1 �sh0 Y d S ) NrY r� c S s g | ]
}|j �qS r r r� r r r r I r
z'TestScandir.test_fd.<locals>.<listcomp>rM F)r� r1 rL )r� r r� rO r r r� r� r/ r1 r� r5 r� r. r5 r� r r� r� r� rP )r' Zexpected_namesro rn r� r� r� r r r r �test_fd< s*
& zTestScandir.test_fdc C s | � ttjd� d S )NrB )rZ r4 r r� rH r r r �test_empty_pathV s zTestScandir.test_empty_pathc C sN | � d� t�| j�}t|�}| �t|�d|� t|�}| �t|�d|� d S )NrY rS r )r r r� r/ r� r. r, )r' r� r� Zentries2r r r �test_consume_iterator_twiceY s
z'TestScandir.test_consume_iterator_twicec C s$ di g fD ]}| � ttj|� q
d S )NgX9��v��?)rZ ra r r� )r' �objr r r �test_bad_path_typed s z TestScandir.test_bad_path_typec C sf | � d� | � d� t�| j�}t|� |�� |�� | �� � ~W d � n1 sX0 Y d S �NrY � file2.txt)r r r� r/ rl rL �check_no_resource_warning�r' r� r r r �
test_closeh s
zTestScandir.test_closec C sv | � d� | � d� t�| j��}t|� W d � n1 s>0 Y | �� � ~W d � n1 sh0 Y d S r� )r r r� r/ rl r� r� r r r �test_context_managers s
&
z TestScandir.test_context_managerc C sT | � d� | � d� t�| j�� }t|� |�� W d � n1 sF0 Y d S r� )r r r� r/ rl rL r� r r r �test_context_manager_close{ s
z&TestScandir.test_context_manager_closec C s� | � d� | � d� | �t��L t�| j�� }t|� dd W d � n1 sR0 Y W d � n1 sp0 Y | �� � ~W d � n1 s�0 Y d S )NrY r� rS r )r rZ �ZeroDivisionErrorr r� r/ rl r� r� r r r � test_context_manager_exception� s
D
z*TestScandir.test_context_manager_exceptionc C s� | � d� | � d� t�| j�}t|� | �t�� ~t�� W d � n1 sR0 Y t�| j�}t |� | �
� � ~W d � n1 s�0 Y d S r� ) r r r� r/ rl r� �ResourceWarningr Z
gc_collectr� r� r� r r r �test_resource_warning� s
&
z!TestScandir.test_resource_warningN)rY )rY )&rA rB rC r r� rI r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rJ r� r r r r� rO r� r� r� r� r� r� r� r� r� r r r r r� D s@
)
�
r� c @ s^ e Zd Zeej�Zdd� Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� ZdS )�
TestPEP519c C s dD ]}| � || �|�� qd S )N)r] s goodbyes some/path/and/file�r. r� )r' �br r r �test_return_bytes� s z TestPEP519.test_return_bytesc C s dD ]}| � || �|�� qd S )N)ZhelloZgoodbyezsome/path/and/filer� )r' rp r r r �test_return_string� s z TestPEP519.test_return_stringc C sL dD ]B}t |�}| �|| �|�� | �dt�|�� | �dt�|�� qd S )N)�path/like/object� path/like/objectr� r� )r r. r� r r r? )r' r� Zpathliker r r �test_fsencode_fsdecode� s
z!TestPEP519.test_fsencode_fsdecodec C sB | � d| �td��� | �tttj�� | �ttd�tj�� d S )Nz #feelthegilr� )r. r� r rM �
issubclassr �PathLiker� rH r r r rH � s zTestPEP519.test_pathlikec C s4 t ddi �}tt t|� fD ]}| �t| j|� qd S )NZblahr )rm r� r rZ ra r� )r' Zvapor�or r r � test_garbage_in_exception_out� s z(TestPEP519.test_garbage_in_exception_outc C s | � t| j� d S r" )rZ ra r� rH r r r �test_argument_required� s z!TestPEP519.test_argument_requiredc C sR | � t| jtd�� tddi �}d|_| � t| j|� � | � t| jtt� �� d S )NrW Zfoor rS )rZ ra r� r rm �
__fspath__r� )r' �cr r r �test_bad_pathlike� s
�z TestPEP519.test_bad_pathlikec C s8 G dd� dt j�}| �tt|�� | �ttt j�� d S )Nc @ s e Zd ZdS )z0TestPEP519.test_pathlike_subclasshook.<locals>.ANrJ r r r r �A� s r� )r r� rU r� r rM )r' r� r r r �test_pathlike_subclasshook� s z%TestPEP519.test_pathlike_subclasshookc C s | � tjt tj� d S r" )r$ r r� r> �types� GenericAliasrH r r r �test_pathlike_class_getitem� s z&TestPEP519.test_pathlike_class_getitemN)rA rB rC rY r r� r� r� r� rH r� r� r� r� r� r r r r r� � s
r� c @ s e Zd Zdd� ZdS )�
TimesTestsc C sn t �� }| �|t j� dD ]}t||�}| �|t� qt jdkrj| �|jd� | �|j d� | �|j
d� d S )N)�user�system�
children_user�children_system�elapsedr" r ) r r/ r$ � times_resultr� ru r r. r� r� r� )r' r/ Zfieldr� r r r �
test_times� s
zTimesTests.test_timesN)rA rB rC r� r r r r r� � s r� �forkc @ s e Zd Zdd� ZdS )� ForkTestsc C s d}t d|� t d|dd� d S )Nz�if 1:
import os
from test import support
pid = os.fork()
if pid != 0:
support.wait_process(pid, exitcode=0)
r� Z malloc_debug)Z PYTHONMALLOCr
r� r r r � test_fork� s
zForkTests.test_forkN)rA rB rC r� r r r r r� � s r� �_fspathc @ s e Zd ZdZeej�ZdS )�TestPEP519PurePythonz>Explicitly test the pure Python implementation of os.fspath().N)rA rB rC r� rY r r� r� r r r r r� � s r� �__main__)r )N)�r� r� r� �
contextlibr@ r6 rg rA r r� r% r r� r r r� r� r r_ Z sysconfigr- rz r� r; r� r r$ r� �testr Z test.supportr r� r r � ImportErrorr^ ro �pwdZgetpwallrJ r� Z _testcapir r r# Ztest.support.script_helperr r r rQ r r r �versionr< r� �getgidr� r r ZTestCaser! rD r� r rZ ZBasicTestMappingProtocolr[ r� r r r r r r9 rR r\ r` rl Zget_config_varZOS_URANDOM_DONT_USE_FDr� rx �contextmanagerr� r� r� r� r� r� r� r r/ r; rK rL rm rt r} r r� r� r� r. r� r� r� r� r� r� r& Zrequires_linux_versionr( r: rC rI rW r[ ro ry rz r} r� r� r� r� r� rA �mainr r r r �<module> sd
A a e \ ' EG
j;) - � �� �M �h#d%CGs/< 0![ H 4�V O� Q0
]
yB � Y: