shell bypass 403
o
a b� � @ s 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 dlm Z d dl m Z d dl m Z d dl m Z d dl m!Z! d dl m"Z" d dl#m$Z$ e�%� � e�&d e'� d dl(Z(d dl)Z)W d � n1 s�w Y zd dl*Z*W n e+y� dZ*Y nw zd dl,Z,W n e+y� dZ,Y nw zd dl-Z-W n e+y� dZ-Y nw zd dl.Z.d
d � e.�/� D �Z0W n e+e1f�y g Z0Y nw z
d d l2m3Z3m4Z4 W n e+�y% ej5 Z3Z4Y nw d d
l6m7Z7 d dl m8Z8 d dl9m:Z: dZ;e<e d��rFe �=� d kZ;e<ed��rYej>j?�rYej>j?�@d�ZAndZAej#�@d��oge �B� d kZCdd� ZDd�dd�ZEe�Fej#�@d�d�ZGG d d � d ejH�ZIG d d � d ejH�ZJG d d!� d!ejH�ZKG d"d#� d#ejH�ZLd d$lmMZM G d%d&� d&eMjN�ZOG d'd(� d(ejH�ZPe�Qe<e d)�d*�G d+d,� d,eP��ZRG d-d.� d.eP�ZSe�Qe<e d)�d*�G d/d0� d0eR��ZTG d1d2� d2ejH�ZUe�Qe<e d3�d4�G d5d6� d6ejH��ZVG d7d8� d8ejH�ZWG d9d:� d:ejH�ZXG d;d<� d<ejH�ZYe�Qe<e d=�d>�G d?d@� d@ejH��ZZe�[dA�dBk�p?e�[dC�dBk�p?e�[dD�dBkZ\e�Fe\dE�e�Fej#dFkdG�G dHdI� dIejH���Z]ej^d�dJdK��Z_e�Qe<e dL�dM�G dNdO� dOejH��Z`e�Qej#dPkdQ�G dRdS� dSejH��ZaG dTdU� dUejH�ZbG dVdW� dWejH�Zce�Fej#dPkdX�G dYdZ� dZejH��Zde�Fej#dPkdX�G d[d\� d\ejH��Zee�Qej#dPkdQ�G d]d^� d^ejH��Zfe�Qej#dPkdQ�G d_d`� d`ejH��Zge�Qe<e da�db�G dcdd� ddejH��Zhe�Qej#dPkdQ�e jiG dedf� dfejH���Zje�Qej#dPkdQ�G dgdh� dhejH��Zke�Qej#dPkdQ�G didj� djejH��Zle jiG dkdl� dlejH��ZmG dmdn� dnejH�ZnG dodp� dpejH�ZoG dqdr� drejH�ZpG dsdt� dtejH�Zqe�rdu�e�Qe<e dv�dw�G dxdy� dyejH���Zse�Qe<e dz��o|e<e d{�d|�G d}d~� d~ejH��ZtG dd�� d�e)juejv�Zwe�Qe<e d��d��G d�d�� d�ejH��Zxd�d�� Zye�Qey� d��e �zd�d�d��G d�d�� d�ejH���Z{e�Qe<e d��d��G d�d�� d�ejH��Z|e�Qe<e d��d��e �zd�d��G d�d�� d�ejH���Z}e�Qe<e d��d��e �zd�d�d��G d�d�� d�ejH���Z~G d�d�� d�ejH�ZG d�d�� d�ejH�Z�G d�d�� d�ejH�Z�G d�d�� d�ejH�Z�e�Qe<e d��d��G d�d�� d�ejH��Z�G d�d�� d�ejH�Z�G d�d�� d�ejH�Z�G d�d�� d�ejH�Z�G d�d�� d�ejH�Z�G d�d�� d�ejH�Z�eDd��G d�d�� d�ejH��Z�e<e d���r�G d�d�� d�e��Z�e�d�k�r�e��� dS dS )�� N)�support)�
import_helper)� os_helper)�
socket_helper)�threading_helper)�warnings_helper)� win32_is_iot�ignorec C � g | ]}|j �qS � )�pw_uid)�.0�ur r �5/usr/local/python-3.10/lib/python3.10/test/test_os.py�
<listcomp>; � 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_funcZ � r � contentc C s< t | dd��}|�|� W d � d S 1 sw Y d S )N�xbr )�open�write)�filenameZcontent�fpr r r � create_file^ s "�r( Zaixz%on AIX, splice() only accepts socketsc @ �$ e Zd Zdd� Zdd� Zdd� ZdS )� MiscTestsc C s t �� }| �|t� d S �N)r �getcwd�assertIsInstance�str��self�cwdr r r � test_getcwdi � zMiscTests.test_getcwdc
C sz d}t jdkr d}d}d}|d|t|� }t�� ��}t�|��z}|} t�� }| � ||� |t|�ttj
j � }|dkr@q�t|�|krP|dkrP|d |� }tj
� ||�}z t�
|� t�|� W n! tyl Y q� ty� } z
| jtjkrW Y d } ~ q�� d } ~ ww |}q%tjr�td t|�� �� W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )
Ni� �vxworks�� �� Zpython_test_dir_�aTr z!Tested current directory length: )�sys�platform�len�tempfile�TemporaryDirectoryr Z
change_cwdr r, � assertEqual�path�sep�join�mkdir�chdir�FileNotFoundError�OSError�errnoZ ENAMETOOLONGr �verbose�print)
r0 Zmin_lenZdirlen�dirnameZtmpdirr> �expectedr1 Zneed�excr r r �test_getcwd_long_pathm sL
��� � ��"�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__r2 rK rO r r r r r* h s 9r* 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 e � ee d,�d-�d.d/� �Z e � ee d,�d-�e!d0d1� ��Z"e � ee d,�d-�e!d2d3� ��Z#e � ee d,�d-�e!d4d5� ��Z$d6S )7� FileTestsc C s" t j�tj�rt �tj� d S d S r+ )r r> �lexistsr �TESTFN�unlink�r0 r r r �setUp� � �zFileTests.setUpc C s: t �tjt jt jB �}t �|� | �t �tjt j �� d S r+ )
r r$ r rU �O_CREAT�O_RDWR�close�
assertTrue�access�W_OK�r0 �fr r r � test_access� s
zFileTests.test_accessc C s� t �tjt jt jB �}t �|�}z/d}||d kr9t �|� |d7 }|dkr+| �d� |t �|�}}||d ksW t �|� nt �|� w t � ||d � | �
t t j |d� d S )Nr � �
z%couldn't allocate two consecutive fds� � a)
r r$ r rU rZ r[ �dupr\ �skipTest�
closerange� assertRaisesrD r% )r0 �first�secondZretriesr r r �test_closerange� s
��zFileTests.test_closerangec C s< t j}t�|�}| �ttj|d� t�|�}| �||� d S �Nr ) r rU r8 � getrefcountrj � TypeErrorr �renamer= )r0 r> �old�newr r r � test_rename� s
zFileTests.test_renamec C s� t tjd��1}|�d� |�� |�� }t�|dd� t�|d�}| � t
|�t � | � |d� W d � d S 1 s:w Y d S )N�w+b� spamr � ) r$ r rU r% �flush�filenor �lseek�readr= �typerM )r0 �fobj�fd�sr r r � test_read� s
"�zFileTests.test_readz needs INT_MAX < PY_SSIZE_T_MAXrd rc F)�sizeZmemuse�dry_runc C sh | � tjtj� ttjd� ttjd��}t�|�� |�}W d � n1 s'w Y | � |d� d S )Ns test�rb)
�
addCleanupr rV rU r( r$ r r{ ry r= )r0 r� 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 � d S 1 sMw Y d S )N�beanss bacon
s eggs
s spam
r� )s bacons eggsrv )r r$ r rU rZ �O_WRONLYrj rp r% � bytearray�
memoryviewr\ r= r{ �
splitlines)r0 r~ r} r r r �
test_write� s
�"�zFileTests.test_writec G s" t j|t jdd�}| �|d� d S )NT)�
creationflags�shellr )�
subprocess�callZCREATE_NEW_CONSOLEr= )r0 �args�retcoder 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� r8 �
executable�r0 �coder r r �test_write_windows_console s z$FileTests.test_write_windows_consolec G s4 t �tjt j�}t j|g|�R ddi�}|�� d S )N�encoding�utf-8)r r$ r rU �O_RDONLY�fdopenr\ )r0 r� r~ ra 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 rU rZ r[ r\ r� �r0 r~ 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
|ddd��}| � |� � d� W d � d S 1 sJw Y d S )Nz.2� 1� 2r� r� �r� �1)
r rU r� rV r( r �replacerj rC �statr$ r= r{ )r0 �TESTFN2ra 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� r\ r` r r r �test_open_keywords+ s �z FileTests.test_open_keywordsc C s> t �td�}z |dtjdd d� W d S ttfy Y d S w )N�symlink�targetF)�src�dst�target_is_directoryr� )r Z
get_attributer r rU �NotImplementedErrorrD )r0 r� r r r �test_symlink_keywords0 s ��z FileTests.test_symlink_keywords�copy_file_rangez test needs os.copy_file_range()c C �> | � t�� t�ddd� W d � d S 1 sw Y d S �Nr rc i����)rj �
ValueErrorr r� rW r r r �#test_copy_file_range_invalid_values8 � "�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� |�� }z t� ||d�}W n t
yc } z|j t j krS� | �
|� W Y d }~d S d }~ww | �|tdd�� t|d��}| �|�� |d |� � W d � d S 1 s�w Y d S )N�.3�
0123456789r� ru � r � )r rU r( r� rV r$ r\ ry r r� rD rE �ENOSYSrh �assertIn�ranger= r{ ) r0 r� r� �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� |�� } z tj || |||d�}
W n t
yl } z| j t j kr\� | �
| � W Y d } ~ d S d } ~ ww | �|
td |d
�� t|d�� }|�� } W d � n1 s�w Y | �| d |� d | � | �| |d � ||||
� � d S ) N�.4r� r� � r� r� ru )�
offset_src�
offset_dstr rc � )r rU r( r� rV r$ r\ ry r r� rD rE r� rh r� r� r{ r= )
r0 �TESTFN4r� �
bytes_to_copy�in_skip�out_seekr� r� r� r� r� r� r{ r r r �test_copy_file_range_offset_ sB
� ��
��z%FileTests.test_copy_file_range_offset�spliceztest needs os.splice()c C r� r� )rj r� r r� rW r r r �test_splice_invalid_values� r� z$FileTests.test_splice_invalid_valuesc
s� t jd }d}tt j|� | �t jt j� tt jd�}| �|j� |�� }t� � \� �| �� fdd�� | ��fdd�� z t�
|�d�}W n t ye } z|j t j
krU� | �|� W Y d }~d S d }~ww | �|tdd �� | �t�� d
�|d |� � d S ) Nr� r� r� c �
t �� �S r+ �r r\ r ��read_fdr r �<lambda>� �
z'FileTests.test_splice.<locals>.<lambda>c r� r+ r� r ��write_fdr r r� � r� r� r r� r� )r rU r( r� rV r$ r\ ry r �piper� rD rE r� rh r� r� r= r{ )r0 r� r� r� r� r� r� r �r� r� r � test_splice� s(
��
zFileTests.test_splicec
s t jd }d}d}d}tt j|� | �t jt j� tt jd�}| �|j� |�� }t� � \� �| �� fdd�� | ��fdd�� z tj
|�||d �}W n t yk } z|j t j
kr[� | �|� W Y d }~d S d }~ww | �|td
|d �� t�� d �} | �| |||| � � d S )
Nr� r� r� r� r� c r� r+ r� r r� r r r� � r� z1FileTests.test_splice_offset_in.<locals>.<lambda>c r� r+ r� r r� r r r� � r� )r� r rc r� )r rU r( r� rV r$ r\ ry r r� r� rD rE r� rh r� r� r{ r= )
r0 r� r� r� r� r� r� r� r� r{ r r� r �test_splice_offset_in� s.
��
z FileTests.test_splice_offset_inc
sp t jd }d}d}d}tt j|� | �t jt j� t�� \� �| �� fdd�� | ��fdd�� t��|� t|d�}| �t j|� | �|j � |�
� }z tj � |||d �}W n t yw } z|j
t
jkrg� | �|� W Y d }~d S d }~ww | �|td
|d �� t|d �� } | �� }
W d � n1 s�w Y | �|
d |� d
| � | �|
|d � |d |� � d S )Nr� r� r� r� c r� r+ r� r r� r r r� � r� z2FileTests.test_splice_offset_out.<locals>.<lambda>c r� r+ r� r r� r r r� � r� ru )r� r rc r� r� )r rU r( r� rV r r� r% r$ r\ ry r� rD rE r� rh r� r� r{ r= ) r0 r� r� r� r� r� r� r� r� r� r{ r r� r �test_splice_offset_out� s8
��
� z FileTests.test_splice_offset_outN)%rP rQ rR rX �tearDownrb rm r � cpython_onlyrt r� r r r r Z
bigmemtestr� r� r� r8 r9 r� r� r� r� r� r� r r r� r� r� r� �requires_splice_piper� r� r� r r r r rS � sT
�
�
!
,
$rS 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 rU �fnamer� rV r( rW r r r rX s zStatAttributeTests.setUpc
C s� t �|�}| �|tj d� | �|jd� t|�}tt�D ]2}|d d� dkrO|�� }|�d�r5dd� }ndd� }| �|t||��|tt|� � | � ||� q d�
� D ] }t t||�d �}t||d �d
}| j ||d d � qTz |d
| �
d� W n ty� Y nw z
d|_| �
d� W n ty� Y nw z
d|_| �
d� W n ttfy� Y nw z
d|_| �
d� W n ty� Y nw z t �d�} | �
d� W n ty� Y nw zt �d�} W d S ty� Y d S w )Nr� ZST_ZTIMEc S s t | �S r+ )�int��xr r r �trunc s z7StatAttributeTests.check_stat_attributes.<locals>.truncc S s | S r+ r r� r r r r� s zst_atime st_mtime st_ctimei�� Z_nsi' re ��deltar6 �No exception raisedrc �rd �r rc re r� rw r� r� � � � rd � � �
� )r r� r= �ST_SIZE�st_size�dir�lower�endswith�getattrr� �splitr� �assertAlmostEqual�fail�
IndexError�st_mode�AttributeError�st_rdevrp �parrot� stat_result)
r0 r� �result�membersr �attrr� ZfloatyZ nanosecondy�result2r r r �check_stat_attributes sl
� � � �� �
� �z(StatAttributeTests.check_stat_attributesc C s | � | j� d S r+ )r r� rW r r r �test_stat_attributesL � z'StatAttributeTests.test_stat_attributesc C sF z
| j �t�� �}W n ty | �d| j � Y nw | �|� d S )Nz#cannot encode %a for the filesystem)r� �encoder8 �getfilesystemencoding�UnicodeEncodeErrorrh r )r0 r� r r r �test_stat_attributes_bytesO s �z-StatAttributeTests.test_stat_attributes_bytesc C �f t �| j�}ttjd �D ]#}t�||�}| �d|� |dk r%| �d|� t�|�}| � ||� q
d S )Nrc s stat_resultrw s cos
stat_result
)
r r� r� r� �pickle�HIGHEST_PROTOCOL�dumpsr� �loadsr= �r0 r
�proto�pZ unpickledr r r �test_stat_result_pickleV 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� z
d|_| � d� W n t
yK Y nw z
d|_| � d� W n t
y_ Y nw z t �d�}| � d� W n tyu Y nw zt �d�}W d S ty� Y d S w ) Nr� )
ZbsizeZfrsizeZblocksZbfreeZbavail�filesZffreeZfavail�flagZnamemaxZf_rd rc r� r� r� )r r r� r= �f_bfree� enumerater r] �
isinstance�f_fsidr� r: r r r �statvfs_resultrp )r0 r
r �value�memberr
r r r �test_statvfs_attributes` s< � �
� �z*StatAttributeTests.test_statvfs_attributes�need os.statvfs()c C r )Nrc s statvfs_resultrw 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 d S ty | �d� Y d S ty. } z | �d� W Y d }~d S d }~ww )Nzc:\pagefile.sysz c:\pagefile.sys does not existzCould not stat pagefile.sys)r r� rC rh rD r �r0 r� r r r � test_1686475� s ��z StatAttributeTests.test_1686475r� zrequires os.pipe()c C s� t �� \}}zt �|� W t �|� t �|� n t �|� t �|� w | �t��
}t �|� W d � n1 s8w Y | �|jjtj � d S r+ )
r r� r� r\ rj rD r= � exceptionrE �EBADF)r0 r� �w�ctxr r r �
test_15261� s
� �z StatAttributeTests.test_15261c C sL | � t|d�� | � t|jt�� | � d|j ko dk� d S � d S )N�st_file_attributesr � �� )r] r r# r3 r� )r0 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� r5 r= r3 �FILE_ATTRIBUTE_DIRECTORYr rU rA r� �rmdir)r0 r
rH 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 ) N�TEMPr� r� z
icacls.exez/denyz*S-1-5-32-545:(S)�r� r )r r> r@ �environr� r� r rV r( r� �
check_callr� �assertNotEqualr� )r0 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)r0 r� r
r r r �test_stat_block_device� s
z)StatAttributeTests.test_stat_block_deviceN)rP rQ rR rX r r r r r r r r r( r* r8 r9 r- r2 r5 r8 r> rA r r r r r� s4 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 rU rH r r> r@ r� r� �rmtreerA r( rW r r r rX � s
zUtimeTests.setUpc C s4 t �|�}|j|d kp|j|d kp|j|d kS )Nr� r� r� )r r� �st_atime�st_mtime�st_ctime)r0 r& �str r r �support_subsecond� s
� �z UtimeTests.support_subsecondNc C s� |s| j }| �|�}|rd}d}nd}d}||||f� t�|�}|r:| 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� rI r r� r rE rF r= � st_atime_ns� st_mtime_ns)r0 �set_timer& rI �atime_ns�mtime_nsrH r r r � _test_utime� s"
zUtimeTests._test_utimec C � dd� }| � |� d S )Nc S � t j| |d� d S �N��ns�r �utime�r& rV r r r rN � z'UtimeTests.test_utime.<locals>.set_time�rQ �r0 rN r r r �
test_utime s zUtimeTests.test_utimec C s | d d S )NrK g��&� .>r rU r r r � ns_to_sec s zUtimeTests.ns_to_secc � � fdd�}� � |� d S )Nc s0 |\}}� � |�}� � |�}t�| ||f� d S r+ �r^ r rX �r& rV rO rP Zatime�mtimerW r r rN ( s
z2UtimeTests.test_utime_by_indexed.<locals>.set_timer[ r\ r rW r �test_utime_by_indexed& s z UtimeTests.test_utime_by_indexedc r_ )Nc s2 |\}}� � |�}� � |�}tj| ||fd� d S )N)�timesr` ra rW r r rN 2 s
z0UtimeTests.test_utime_by_times.<locals>.set_timer[ r\ r rW r �test_utime_by_times1 s z UtimeTests.test_utime_by_timesz9follow_symlinks support for utime required for this test.c C rR )Nc S s t j| |dd� d S )NF)rV �follow_symlinksrW rY r r r rN > 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 rR )Nc S sD t | dd��}tj|�� |d� W d � d S 1 sw Y d S )N�wbr rU )r$ r rX ry )r& rV r' r r r rN G s "�z*UtimeTests.test_utime_fd.<locals>.set_timer[ r\ r r r �
test_utime_fdD s zUtimeTests.test_utime_fdz0dir_fd support for utime required for this test.c C rR )Nc S sL t j�| �\}}t �|t j�}zt j|||d� W t �|� d S t �|� w )N)r� rV )r r> r r$ r� rX r\ )r& rV rH r �dirfdr r r rN Q s
z.UtimeTests.test_utime_dir_fd.<locals>.set_timer[ r\ r r r �test_utime_dir_fdN s z UtimeTests.test_utime_dir_fdc C s dd� }| j || jd� d S )Nc S rS rT rW rY r r r rN \ rZ z1UtimeTests.test_utime_directory.<locals>.set_time�r& )rQ rH r\ r r r �test_utime_directory[ s z UtimeTests.test_utime_directoryc C sb t � � }|| j� | �| j�sd}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� rI r r� rF r )r0 rN �currentr� rH rn r r r �_test_utime_currenta s
�
�z UtimeTests._test_utime_currentc r_ )Nc s t �� j� d S r+ �r rX r� rl rW r r rN x � z/UtimeTests.test_utime_current.<locals>.set_time�rq r\ r rW r �test_utime_currentw � z UtimeTests.test_utime_currentc r_ )Nc s t �� jd � d S r+ rr rl rW r r rN ~ rZ z3UtimeTests.test_utime_current_old.<locals>.set_timert r\ r rW r �test_utime_current_old} rv z!UtimeTests.test_utime_current_oldc C sp t jdkr4tj�tj�|��d d }dd l}|jj}|� dd�}|�
|d dd d d |t |��}|r6|j S d S d S )Nr� r �\� r� )
r8 r9 r r> r? �abspath�ctypes�windll�kernel32Zcreate_unicode_bufferZGetVolumeInformationWr: r& )r0 r> �rootr{ r} �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 NTFSrJ ) r� rH rh r rX r� r= r� rF )r0 Zlarger r r �test_large_time� s
zUtimeTests.test_large_timec C s� | � t�� tj| jddd� W d � n1 sw Y | � t�� t�| jddg� W d � n1 s7w Y | � t�� t�| jd� W d � n1 sSw Y | � t�� t�| jd� W d � n1 sow Y | � t�� tj| jddgd� W d � n1 s�w Y | � t�� tj| jdd� W d � n1 s�w Y | � t�� tj| jdd� W d � n1 s�w Y tjtjvr�| � t�� tj| jddd� W d � n1 s�w Y tjtjv�r.t | jdd ��&}| � t�� t�|�
� d� W d � n 1 �sw Y W d � n 1 �s)w Y tjtj v�rW| � t�� tj| jdd d
� W d � d S 1 �sPw Y d S d S ) N)r� r� rU r� )r� )r� r� r� F�rf rh r �r� ) rj r� r rX r� rp �supports_follow_symlinksr� � supports_fdr$ ry �supports_dir_fd)r0 r' r r r � test_utime_invalid_arguments� sJ � � � � � � � � �� � $��z'UtimeTests.test_utime_invalid_argumentsc C s� dd� }| � t�� tj| j|d�dfd� W d � n1 s w Y | � t�� tj| j|d�dfd� W d � n1 sAw Y | � t�� tj| j|d�dfd� W d � d S 1 scw 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__� � zJUtimeTests.test_issue31577.<locals>.get_bad_int.<locals>.BadInt.__divmod__N)rP rQ rR r� r r� r r �BadInt� s r� r )r� r� r r� r � get_bad_int� s z/UtimeTests.test_issue31577.<locals>.get_bad_int�* rc rU r )rc re r� )rj rp r rX r� )r0 r� r r r �test_issue31577� s � � "�zUtimeTests.test_issue31577r+ ) rP rQ rR rX rI rQ r] � staticmethodr^ rc re r r r rX r� rg r� ri r� rk rm rq ru rw r� r� r� r r� r� r r r r rB � s<
�
�
�
rB )�
mapping_testsc @ sB e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Ze �
e o!e j
�e �d �e �
ee d �d
�dd� ��Ze �
e o:e j
�e �d �e �
ee d �d
�dd� ��Zdd� Zdd� Zdd� Zdd� Ze �
e jd�dd � �Zd d � Ze�d d �d!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%d7d8� Z&dS )9� EnvironTestsz8check that os.environ object conform to mapping protocolNc C sB t tj�| _tjrt tj�| _| �� �� D ] \}}|tj|< qd S r+ ) �dictr r; �_EnvironTests__save�supports_bytes_environ�environb�_EnvironTests__saveb�
_reference�items�r0 �keyr& r r r rX � s �zEnvironTests.setUpc C s> t j�� t j�| j� t jr t j�� t j�| j� d S d S r+ )r r; �clear�updater� r� r� r� rW r r r r� � s
�zEnvironTests.tearDownc C s dddd�S )NZVALUE1ZVALUE2ZVALUE3)ZKEY1ZKEY2ZKEY3r rW r r r r� � � zEnvironTests._referencec C s t j�� t jS r+ )r r; r� rW r r r �_empty_mapping� s
zEnvironTests._empty_mappingzrequires a shell�popenzneeds os.popen()c C sd t j�� t jjdd� t �dt ��}|�� �� }| �|d� W d � d S 1 s+w Y d S )NZWorld)ZHELLOz%s -c 'echo $HELLO') r r; r� r� r� r r{ �stripr= )r0 r� r& r r r � test_update2� s
"�zEnvironTests.test_update2c C sz t �dt ��,}t|�}| �t|�d� | �t|�d� | �t|�d� | �tt|� W d � d S 1 s6w Y d S )Nz %s -c 'echo "line1
line2
line3"'zline1
zline2
zline3
)r r� r �iterr= �nextrj �
StopIteration)r0 r� �itr r r �test_os_popen_iter� s �"�z EnvironTests.test_os_popen_iterc C s8 t j�� D ]\}}| �t|�t� | �t|�t� qd S r+ )r r; r� r= r| r. )r0 r� �valr r r �test_keyvalue_types� s �z EnvironTests.test_keyvalue_typesc C s. | � � �� D ]\}}| �tj�|�|� qd S r+ )r� r� r= r r; �getr� r r r �
test_items 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
r� r& r r r � <genexpr> s
�
�
�z-EnvironTests.test___repr__.<locals>.<genexpr>N)r r; r= �reprr� r@ r� )r0 �envr r r �
test___repr__ s �zEnvironTests.test___repr__c C s\ t j�t j�}g d�}dt j�|�i}t j}z t|�t _| �|t �� � | �|t jd d�� W |t _n|t _w | �|t �i �� | �dt �ddi�� | �|t �|�� t j r�z t
j dd��
dd d
�}W d � n1 snw Y W n t y} Y n w | �
tt j|� | �t �d d i�d
g� | �t �d d
i�d
g� | �t �dd i�d
g� d S d S )N)z/montyz/pythonry z/flying/circus�PATH�r� �ry ry T)�recordr� r� )r� � PATHr� � abc�abc)r �defpathr �pathsepr@ r; r� �assertSequenceEqual�
get_exec_pathr� �warnings�catch_warnings� BytesWarningrj r� )r0 Z defpath_listZ test_pathZtest_envZ
saved_environZ mixed_envr r r �test_get_exec_path s@
�� �����z EnvironTests.test_get_exec_pathz#os.environb required for this test.c C s� d}z
|� t�� d�}W n ty! dt�� f }| �|� Y nw |tjd< | �tjd |� | �tjd |� d}|tjd< | �tjd |� |� t�� d�}| �tjd |� d S ) Nu euro€�surrogateescapez'U+20AC character is not encodable to %s�unicodes unicode� �s bytesrM )
r r8 r r rh r r; r= r� �decode)r0 r& Z value_bytesrn Z value_strr r r �
test_environb5 s&
� ��
zEnvironTests.test_environbc C s� d}d}d|�d�}t �� �L}|�|d � t�||� tjtjd|gdtj dd�}| �
|j � � t
|�� t�|� tjtjd|gdtj dd�}| �
|j � � t
d �� W d � d S 1 s]w Y d S )NZ
PYTHONTESTVARZ testvaluez%import os; print(repr(os.environ.get(z)))r� T)�check�stdout�text)r ZEnvironmentVarGuard�popr �putenvr� �runr8 r� �PIPEr= r� �rstripr� �unsetenv)r0 r r& r� r� �procr r r �test_putenv_unsetenvL s
�
�"�z!EnvironTests.test_putenv_unsetenvrd r� c C s� dD ]}| � ttftj|d� | � ttftj|� qtjdkr?dd }| � ttj|d� | � ttjd|� | � ttj|� d S d S )N)ry z=namezna=mezname=zname zna mer& r� r� � � r� �X)rj rD r� r r� r� r8 r9 )r0 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 w Y | �|jjd |� | �|jj � | �t�� }tj|= W d � n1 sHw Y | �|jjd |� | �|jj � d S )NZ
missingkeyr )
� assertNotInr r; rj �KeyError�assertIsr. r� r] �__suppress_context__)r0 �missing�cmr r r �
test_key_typeo s �
�zEnvironTests.test_key_typec C sR t |�}d}t|� dtj|< zt|� | �tj| d� W tj|= d S tj|= w )NZ __new_key__Ztest_environ_iteration)r� r� r r; r= )r0 Z
collection�iteratorZnew_keyr r r �_test_environ_iteration} s
z$EnvironTests._test_environ_iterationc C � | � tj� d S r+ )r� r r; rW r r r �(test_iter_error_when_changing_os_environ� r z5EnvironTests.test_iter_error_when_changing_os_environc C � | � tj�� � d S r+ )r� r r; r� rW r r r �.test_iter_error_when_changing_os_environ_items� � z;EnvironTests.test_iter_error_when_changing_os_environ_itemsc C r� r+ )r� r r; �valuesrW r r r �/test_iter_error_when_changing_os_environ_values� r� z<EnvironTests.test_iter_error_when_changing_os_environ_valuesc C sf t rtj�t �s
d S t�t � d|� d���}|�� �� }W d � n1 s&w Y | �||� d S )Nz -c 'echo $�')r r r> �existsr� r{ r� r= )r0 �varrI r� 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�3ry ) r r; r� r� �assertDictEqualr= r� r� �NotImplemented�__or__r� �r0 �overridden_keyr� �
new_vars_dictrI �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� r� r= r� )r0 r� r� rI 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 w Y | �t�� ddi}t j|O _W d � n1 s@w Y | �|t j� d S )Nrc r� )r r; �copyrj rp r= )r0 Zos_environ_copyZdict_with_bad_keyZdict_with_bad_valr r r � test_ior_operator_invalid_dicts� 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� )r0 r� r� rI 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� r� r� r= r� r� r� �__ror__r� r� r r r �test_ror_operator� s
z EnvironTests.test_ror_operator)'rP rQ rR �__doc__Z type2testrX r� r� r� r r r r r> r� r r� r� r� r� r� r� r� r� r� r Zrequires_mac_verr� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � sN � � )�
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 �Nrf � followlinks)r� r �walk)r0 �top�kwargsr r r r � 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 ]"}
t |
ddd��}|�
d|
d � W d � n1 s�w 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 d S w t �|tj� t �|� t �|� | jd d d�= d S )NZTEST1�SUB1�SUB11�SUB2ZSUB21�tmp1�tmp2�tmp3�linkZTEST2�tmp4� broken_link� broken_link2� broken_link3r� r� r� zI'm z" and proud of it. Blame test_os.
�brokenTZtmp5)r r r r r rc )r r> r@ r� r rD rU � walk_path� sub1_path�
sub11_path� link_path�makedirsr$ r% � can_symlinkr� rz � sub2_tree�chmod�listdir�PermissionErrorr� �S_IRWXUrV r7 )r0 r@ Z sub2_pathZ
sub21_pathZ tmp1_pathZ tmp2_pathZ tmp3_pathZ tmp5_pathZt2_pathZ tmp4_pathZbroken_link_pathZbroken_link2_pathZbroken_link3_pathr> ra r r r rX � sZ
��� �
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 ) Nrw r rc r r� re ���r r
r r ) �listr r r= r: �sortr r r �r0 �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 re r r r
rc r )r r �append�remover= r: r r )r0 r r" r~ �dirsr r r r �test_walk_pruneP s
� zWalkTests.test_walk_prunec C s | � t| j�� d S r+ )r' r r rW r r r �test_file_like_pathc r� 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��topdownrw r� rc r r re r r r
r r ) r r r r= r: r r r r r! r r r �test_walk_bottom_upf s$
�
���z WalkTests.test_walk_bottom_upc C sj t �� s | �d� | j| jdd�}|D ]\}}}|| jkr-| �|g � | �|dg� d S q| �d� d S )Nzneed symlink supportTr� r z+Didn't follow symlink with followlinks=True)r r rh r r r r= r )r0 �walk_itr~ 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�||� z5dd� |D �} | � |� | �
|| � | �
|| � |D ]}
|
|kr\| � tj�||
�| � qKW t�||� d S t�||� w )N)�onerrorr z.newc S s g | ]\}}}|�qS r r )r
r� �dra r r r r � � z/WalkTests.test_walk_bad_dir.<locals>.<listcomp>) r r r$ r� r= r r> r@ rq r] r� r� ) r0 �errorsr, r~ 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 ] }||r6dgng g f}|D ]
}��t|�|� q<t j� |�}q.� �fdd�td�D �}� }t|d �D ]"}|||k rldgng g f}|D ]
}��t|�|� qrt j�|d�}qbd S ) N� Zdeepr/ c � g | ] }�j � d d��qS )Fr) �r �r
�j��baser0 r r r � � z7WalkTests.test_walk_many_open_files.<locals>.<listcomp>r� rc c r7 )Tr) r8 r9 r; r r r � r= )
r r> r@ r rU r r� r= r� rH )r0 �depthr Zitersr� rI r� r r; r �test_walk_many_open_files� s$
�z#WalkTests.test_walk_many_open_filesr+ )
rP rQ rR r r rX r# r' r( r+ r- r5 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 s2 � | j |fi |��D ] \}}}}|||fV q
d S r+ )r@ )r0 r r r~ r& r Zroot_fdr r r r � s � �zFwalkTests.walkc O s t j|i |��S r+ )r r@ �r0 r� r r r r r@ � r zFwalkTests.fwalkc
C s� |� � }|� � }tjddd�D ]O\}}|j||d� |j||d� i }tjdi |��D ]\}}}t|�t|�f||< q+| jdi |��D ]\}}}} | �||� | � || t|�t|�f� qCqdS )z.
compare with walk() results.
�TFre ��repeat)r* r )r* rf Nr )
r� � itertools�productr� r r �setr@ r� r= )
r0 � walk_kwargs� fwalk_kwargsr* rf rI r~ r& r �rootfdr r r �_compare_to_walk� s ��zFwalkTests._compare_to_walkc C s dt ji}| �||� d S )Nr )r rU rL )r0 r r r r �test_compare_to_walk� �
z FwalkTests.test_compare_to_walkc C sR z"t �dt j�}dtji}|�� }||d< | �||� W t �|� d S t �|� w )N�.r r� )r r$ r� r rU r� rL r\ )r0 r~ rI rJ r r r � test_dir_fd� s
zFwalkTests.test_dir_fdc C s| t jddd�D ]4\}}tj|d f}| j|d|i�D ]!\}}}}t�|� t�|� | �t t�
|��t |�t |�B � qqd S )NrC re rD rf ) rF rG r rU r@ r �fstatr� r= rH r )r0 r* rf r� r~ r& r rK r r r �test_yields_correct_dir_fd� s
$��z%FwalkTests.test_yields_correct_dir_fdc C s\ t �d�}t �|� td�D ] }| �tj�D ]}qqt �d�}| �t j|� | �||� d S )Nrc � ) r rg r\ r� r@ r rU r� r= )r0 Zminfdr� r� Znewfdr r r � test_fd_leak� s
�
zFwalkTests.test_fd_leakN) rP rQ rR r r r@ rL rM rP rR rT r? r r r r rA � s
rA c @ � e Zd ZdZdd� ZdS )�BytesWalkTests� Tests for os.walk() with bytes.c k s� � d|v r |� d�|d< tjt�|�fi |��D ]8\}}}t�|�}tttj|��}tttj|��}|||fV tttj|��|d d �< tttj|��|d d �< qd S r )r� r r �fsencoderN r �map) r0 r r �broot�bdirs�bfilesr~ r& r r r r r � s �"
�zBytesWalkTests.walkN)rP rQ rR r r r r r r rV � s rV c @ s e Zd ZdZddd�ZdS )�BytesFwalkTestsrW rO c o s� � t jt �|�g|�R i |��D ]:\}}}}t �|�}ttt j|��} ttt j|��}
|| |
|fV ttt j| ��|d d �< ttt j|
��|d d �< qd S r+ )r r@ rX rN r rY ) r0 r r� r rZ r[ r\ �topfdr~ r& r r r r r@ s �*
�zBytesFwalkTests.fwalkN)rO )rP rQ rR r r@ r r r r r] s r] c @ �D e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� MakedirTestsc C � t �tj� d S r+ �r rA r rU rW r r r rX r zMakedirTests.setUpc
C s� t j}tj�|ddd�}t�|� tj�|dddd�}t�|� | �ttjtj� tj�|dddddtj�}t�|� tj�|dtjddddd�}t�|� d S �Nr2 r4 Zdir3Zdir4Zdir5Zdir6) r rU r r> r@ r rj rD �curdir)r0 r< r> r r r � test_makedir s
�zMakedirTests.test_makedirc C s� t �d��W t j}tj�|d�}tj�|d�}t�|d� | �tj�|�� | �tj� |�� tj
dkrT| � t� |�j
d@ d� | � t� |�j
d@ d� W d � d S W d � d S 1 s_w Y d S )Nre r2 r4 im �ntr� i� )r Z
temp_umaskrU r r> r@ r r] r� �isdirr r= r� r )r0 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 )
Nr2 r� � F��exist_oki� T)r� rl �/)
r r> r@ r rU �umaskr rj rD rz )r0 r> r� �old_maskr r r � test_exist_ok_existing_directory5 s
z-MakedirTests.test_exist_ok_existing_directoryc C s� t j�tjd�}tj}d}t �d�}zdt�t � tj�j
�}z t � tj||B � W n t y4 t
�d��w t � tj�j
|@ |krEt
�d��t �|||B � t j||dd� t � |t�t � |�j
�| @ � t j|||B dd� W t �|� d S t �|� w )Nr2 r� rj zCannot set S_ISGID for dir.z No support for S_ISGID dir mode.Trk )r r> r@ r rU r� �S_ISGIDrn �S_IMODE�lstatr r r r �SkipTestr )r0 r> rq r� ro Zexisting_testfn_moder r r � test_exist_ok_s_isgid_directoryC s(
�
�
z,MakedirTests.test_exist_ok_s_isgid_directoryc C s� t j}tj�t jd�}t|ddd��
}|�d� W d � n1 s"w Y | �ttj |� | jttj |dd� | jttj |dd� t�
|� d S ) Nr2 r0 r� r� r� Frk T) r rU r r> r@ r$ r% rj rD r r% )r0 r< r> ra r r r �#test_exist_ok_existing_regular_file^ s �z0MakedirTests.test_exist_ok_existing_regular_filec C s` t j�tjdddddd�}t j�|�s)|tjkr)t j�|�}t j�|�s)|tjkst �|� d S rc )r r> r@ r rU r� rH �
removedirs�r0 r> r r r r� h s � �zMakedirTests.tearDownN)
rP rQ rR rX re ri rp ru rv r� r r r r r` s
r` �chownzTest needs chownc @ s� e Zd Zedd� �Zdd� Ze�ee d�d�dd� �Z
e�e o$e e
�d kd
�d d � �Ze�e o5e e
�d kd
�dd� �Zedd� �ZdS )�ChownFileTestsc C ra r+ rb ��clsr r r �
setUpClassw rs zChownFileTests.setUpClassc C s� t �tj�}|j}|j}ddt�d�t� dd�fD ]}| �
t t j tj||� | �
t t j tj||� q| �
t � tj||�� | �
t � tjdd�� d S )Ng �y � �r ���re )r r� r rU �st_uid�st_gid�decimalZDecimal� fractionsZFractionrj rp ry � assertIsNone)r0 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 )Nre z test needs at least 2 groups) r r� r: rh r� r rU r ry r� r= )r0 �groupsZgid_1Zgid_2r� r� r r r �test_chown_gid� s
z ChownFileTests.test_chown_gidrc 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 �Nre ) � all_usersr r� r rU r� ry r r= )r0 �uid_1�uid_2r� r� r r r �test_chown_with_root� s z#ChownFileTests.test_chown_with_rootz2test needs non-root account and more than one userc C sn t d d� \}}t�tj�j}| �t�� t�tj||� t�tj||� W d � d S 1 s0w Y d S r� ) r� r r� r rU r� rj r ry )r0 r� r� r� r r r � test_chown_without_permission� s "�z,ChownFileTests.test_chown_without_permissionc C ra r+ )r r7 r rU r{ r r r �
tearDownClass� rs z ChownFileTests.tearDownClassN)rP rQ rR � classmethodr} r� r r r r r� �
root_in_posixr: r� r� r� r� r r r r rz t s
�
�
rz c @ �4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �RemoveDirsTestsc C ra r+ )r r r rU rW r r r rX � r zRemoveDirsTests.setUpc C s t �t j� d S r+ )r rD rU rW r r r r� � r 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> r@ r rU rA rw � assertFalser� �r0 r� r� r r r �test_remove_all� 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 �Nr� r� �file.txt) r r> r@ r rU rA r( rw r� r� r] r� 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 s7w Y | �
t j� |�� | �
t j� |�� | �
t j� tj�� d S r� ) r r> r@ r rU rA r( rj rD rw r] r� r� r r r �test_remove_nothing� s
�z#RemoveDirsTests.test_remove_nothingN)rP rQ rR rX r� r� r� r� r r r r r� � s
r� c @ � e Zd Zdd� ZdS )� DevNullTestsc C s� t tjdd��}|�d� |�� W d � n1 sw Y t tjd��}| �|�� d� W d � d S 1 s:w Y d S )Nrh r � hellor� � )r$ r �devnullr% r\ r= r{ r` r r r � test_devnull� s
�"�zDevNullTests.test_devnullN)rP rQ rR r� r r r r r� � � 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 rc rd r� r5 )r= r: r �urandomrW r r r �test_urandom_length� s
z URandomTests.test_urandom_lengthc C s0 t �d�}| �|t� t �d�}| �||� d S �N� )r r� r- rM r= �r0 Zdata1Zdata2r r r �test_urandom_value� 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� rc )r@ r r= r: )r0 �countr� �outr� r r r �get_urandom_subprocess� s �
z#URandomTests.get_urandom_subprocessc C s$ | � d�}| � d�}| �||� d S r� )r� r= r� r r r �test_urandom_subprocess� �
z$URandomTests.test_urandom_subprocessN)rP rQ rR r� r� r� r� 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 sB zt �d� W d S ty } z |jtjkrt�d��� d }~ww )Nrc z%getrandom() syscall fails with ENOSYS)r r� rD rE r� r rt )r| rJ r r r r} s
��zGetRandomTests.setUpClassc C s* t �d�}| �|t� | �t|�d� d S r� )r r� r- rM r= r: �r0 r� r r r �test_getrandom_type s
z"GetRandomTests.test_getrandom_typec C s t �d�}| �|d� d S )Nr r� )r r� r= )r0 �emptyr r r �test_getrandom0 rN z GetRandomTests.test_getrandom0c C s | � ttd�� d S )N� GRND_RANDOM)r] r r rW r r r �test_getrandom_random r� z$GetRandomTests.test_getrandom_randomc C s* z
t �dt j� W d S ty Y d S w �Nrc )r r� �
GRND_NONBLOCK�BlockingIOErrorrW r r r �test_getrandom_nonblock s
�z&GetRandomTests.test_getrandom_nonblockc C s$ t �d�}t �d�}| �||� d S r� )r r� r= r� r r r �test_getrandom_value& r� z#GetRandomTests.test_getrandom_valueN)
rP rQ rR r� r} r� r� r� r� r� r r r r r� s
r� ZHAVE_GETENTROPYrc ZHAVE_GETRANDOMZHAVE_GETRANDOM_SYSCALLz*os.random() does not use a file descriptorr4 z$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_failure8 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 )r0 r� �rcr� �errr r r �test_urandom_fd_closedO 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))
)rU r� r� r rw )
r� r rV rU r( r� r r= r: r= )r0 r� r� r� r� Zout2Zerr2r r r �test_urandom_fd_reopened] s � z'URandomFDTests.test_urandom_fd_reopenedN) rP rQ rR r r �resourcer� r� r� r r r r r� 3 s
r� c # s~ � g � � fdd�}� fdd�}z%t j}t j}t j}|t _|t _| dur&| t _� V W |t _|t _|t _dS |t _|t _|t _w )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��)N�execvz execv called)r$ � RuntimeError�r r� ��callsr r �
mock_execv� s z#_execvpe_mockup.<locals>.mock_execvc s � � d| |f� ttjd��)N�execvez
execve called)r$ rD rE �ENOTDIRr� 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-)rj rD r �execvperW 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� ry )rj r� r r� rW 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� ry )rj r� r r� rW r r r � test_execvpe_with_bad_arglist� s z'ExecTests.test_execvpe_with_bad_arglist�_execvpez)No internal os._execvpe function to test.c
C s� t jd }|tu r d}t j�t �|�|�}|}g d�}nd}g d�}t j�||�}t jdkr4t �|�}n|}ddi}t� �%}| �t t j
||� | � t |�d � | � |d
d ||ff� W d � n1 sew Y t|d ��(}| jt
t j
|||d
� | � t |�d � | �|d
d|||ff� W d � n1 s�w Y t� �:}|�� } |tu r�|| d< n|| d< | jt
t j
||| d
� | � t |�d � | �|d
d||| ff� W d � d S 1 s�w Y d S )NZ absolutepaths
executable)s progname�arg1�arg2r� )Zprognamer� r� rf Zspamr� rc r r� )r� r� r� r� r� )r r? rM r> r@ rX r r� rj r� r� r= r: rD r� r� )
r0 Z test_typeZ program_pathZprogram�fullpathZnative_fullpathZ argumentsr� r� Zenv_pathr r r �_test_internal_execvpe� sT
� �
�
� �
�
�"�z ExecTests._test_internal_execvpec C s& | � t� tjdkr| � t� d S d S )Nrf )r� r. r r rM rW r r r �test_internal_execvpe_str� s
�z#ExecTests.test_internal_execvpe_strc C s� t jddg}tj�� }d|d< | �t�� t�|d ||� W d � n1 s(w Y tj�� }d|d< | �t�� t�|d ||� W d � n1 sOw Y tj�� }d|d < | �t�� t�|d ||� W d � d S 1 sww Y d S )
Nr� �pass�cabbage�FRUIT VEGETABLEr �orange VEGETABLE=cabbage�FRUIT�lemon� FRUIT=ORANGE)r8 r� r r; r� rj r� r� )r0 r� �newenvr r r �test_execve_invalid_env� s
�
�
"�z!ExecTests.test_execve_invalid_envr� zWin32-specific testc
C sd z
t �ddgi � W n ty* } z| �|jd u p |jdk� W Y d }~d S d }~ww | �d� d S )Nry �argr zNo OSError raised)r r� rD r] �winerrorr r, r r r �test_execve_with_empty_path s (��z%ExecTests.test_execve_with_empty_pathN)rP rQ rR r �skipIf�USING_LINUXTHREADSr� r� r� r r r r� r� r� r8 r9 r� r r r r r� � s �
�
/r� r� r+ c @ r_ )�Win32ErrorTestsc
C sz zt �tj� W n* ty d}Y d S ty2 } zd}| �dtj|f � W Y d }~d S d }~ww | �dtj � d S )NFTz.file %s must not exist; os.stat failed with %szfile %s must not exist)r r� r rU rC rD r )r0 r� rJ r r r rX s
���zWin32ErrorTests.setUpc C s | � ttjtjtjd � d S )Nz.bak)rj rD r rq r rU rW r r r rt s zWin32ErrorTests.test_renamec C � | � ttjtj� d S r+ )rj rD r r% r rU rW r r r � test_remove r! zWin32ErrorTests.test_removec C r� r+ )rj rD r rB r rU rW r r r �
test_chdir# r! zWin32ErrorTests.test_chdirc C sT | � tjtj� ttjd��}| �ttjtj� W d � d S 1 s#w Y d S )Nr� ) r� r rV rU r$ rj rD r rA r` r r r �
test_mkdir& s "�zWin32ErrorTests.test_mkdirc C s | � ttjtjd � d S r+ )rj rD r rX r rU rW r r r r] , � zWin32ErrorTests.test_utimec C s | � ttjtjd� d S rn )rj rD r r r rU rW r r r �
test_chmod/ r� zWin32ErrorTests.test_chmodN)
rP rQ rR rX rt r� r� r� r] r� r r r r r� s r� c @ s� e Zd Zg d�Zdd� ZeD ] Zee�e� de < qdd� Zdd� Ze �
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�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/d0� �Ze �
e e d1�d2�d3d4� �Ze �
e e d5�d6�d7d8� �Ze �
e e d9�d:�d;d<� �Zd=d>� Ze �
e e d?�d@�dAdB� �ZdCS )D�
TestInvalidFD)�fchdirrg � fdatasyncrQ �fstatvfs�fsync� tcgetpgrp�ttynamec s � fdd�}|S )Nc s" t t� �r| �tt� �� d S d S r+ )r r r� r rW �ra r r �helper9 s
�z(TestInvalidFD.get_single.<locals>.helperr )ra r� r r� r �
get_single8 s zTestInvalidFD.get_singleZtest_c
O sh z|t �� g|�R i |�� W n ty* } z| �|jtj� W Y d }~d S d }~ww | �d| � d S )Nz5%r didn't raise an OSError with a bad file descriptor)r � make_bad_fdrD r= rE r/ r )r0 ra r� r r� r r r r� @ s ��
�zTestInvalidFD.checkc C s | j tjdd� d S )Nr� r� )r� r r� rW r r r r� I r� zTestInvalidFD.test_fdopen�isattyztest needs os.isatty()c C s | � t�t�� �d� d S �NF)r= r r� r r� rW r r r � test_isattyL s zTestInvalidFD.test_isattyri ztest needs os.closerange()c C sn t �� }td�D ]}z t�|| � W n ty Y qw |dk r't�d��| �t� ||| d �d � d S )Nrd re z5Unable to acquire a range of invalid file descriptorsrc )
r r� r� r rQ rD r rt r= ri )r0 r~ r� r r r rm P s �� z TestInvalidFD.test_closerange�dup2ztest needs os.dup2()c C � | � tjd� d S )N� )r� r r rW r r r � test_dup2` rZ zTestInvalidFD.test_dup2�fchmodztest needs os.fchmod()c C r rn )r� r r rW r r r � test_fchmodd rZ zTestInvalidFD.test_fchmod�fchownztest needs os.fchown()c C � | � tjdd� d S )Nr )r� r r rW r r r � test_fchownh � zTestInvalidFD.test_fchown� fpathconfztest needs os.fpathconf()c C � | � tjd� | � tjd� d S )N� PC_NAME_MAX)r� r �pathconfr
rW r r r �test_fpathconfl � z TestInvalidFD.test_fpathconf� ftruncateztest needs os.ftruncate()c C r rn )r� r �truncater rW r r r �test_ftruncateq r z TestInvalidFD.test_ftruncaterz ztest needs os.lseek()c C r rn )r� r rz rW r r r �
test_lseekv r zTestInvalidFD.test_lseekr{ ztest needs os.read()c C r r� )r� r r{ rW r r r r� z rZ zTestInvalidFD.test_read�readvztest needs os.readv()c C s t d�}| �tj|g� d S )Nrd )r� r� r r )r0 r r r r �
test_readv~ s zTestInvalidFD.test_readv� tcsetpgrpztest needs os.tcsetpgrp()c C r rn )r� r r rW r r r �test_tcsetpgrpt� rZ z TestInvalidFD.test_tcsetpgrptr% ztest needs os.write()c C r )N� )r� r r% rW r r r r� � rZ zTestInvalidFD.test_write�writevztest needs os.writev()c C s | � tjdg� d S )Nr� )r� r r rW r r r � test_writev� r zTestInvalidFD.test_writevc C � | � tj� | � tjd� d S �NT)r� r �get_inheritable�set_inheritablerW r r r �test_inheritable� � z TestInvalidFD.test_inheritable� get_blocking�-needs os.get_blocking() and os.set_blocking()c C r r )r� r r! � set_blockingrW r r r �
test_blocking� s zTestInvalidFD.test_blockingN) rP rQ rR Zsinglesr� ra �localsr� r� r r r r r� rm r r r r r r r� r r r� r r r$ r r r r r� 3 sL
�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 rU �file1r r> r@ �file2rW r r r rX � � zLinkTests.setUpc C s, | j | jfD ]
}tj�|�rt�|� qd S r+ )r' r( r r> r� rV )r0 �filer r r r� � s
��zLinkTests.tearDownc
C s� t |� zt�||� W n ty% } z
| �d| � W Y d }~nd }~ww t|d��4}t|d��}| �tj�|� � |� � �� W d � n1 sJw Y W d � d S W d � d S 1 sbw Y d S )N�
os.link(): %sr� )
r( r r
r rh r$ r] r> � sameopenfilery )r0 r' r( r� rC �f2r r r �
_test_link� s �� P�zLinkTests._test_linkc C s | � | j| j� d S r+ )r. r' r( rW r r r � test_link� r� zLinkTests.test_linkc C s( | � t| jt�� �t| jt�� �� d S r+ )r. rM r' r8 r r( rW r r r �test_link_bytes� s �zLinkTests.test_link_bytesc C sV zt �d� W n ty t�d��w | jd7 _| jd | _| �| j| j� d S )N� ñz#Unable to encode for this platform.r� )r rX � UnicodeErrorr rt r' r( r. rW r r r �test_unicode_name� s
� zLinkTests.test_unicode_nameN) rP rQ rR rX r� r. r/ r0 r3 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 �getuidrj rD r6 rp �
OverflowError� UID_OVERFLOWrW r r r � test_setuid� � z PosixUidGidTests.test_setuid�setgidztest needs os.setgid()c C �F t �� dkrts| �tt jd� | �tt jd� | �tt j| j� d S r7 ) r r8 �HAVE_WHEEL_GROUPrj rD r= rp r9 � GID_OVERFLOWrW r r r � test_setgid� � z PosixUidGidTests.test_setgid�seteuidztest needs os.seteuid()c C sB t �� dkr| �tt jd� | �tt jd� | �tt j| j� d S r7 ) r r8 rj rD rC rp �setegidr9 r: rW r r r � test_seteuid� r< z PosixUidGidTests.test_seteuidrD ztest needs os.setegid()c C r> r7 ) r r8 r? rj rD rD rp r9 r@ rW r r r � test_setegid� rB 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 r7 )r r8 rj rD rG rp r9 r: rW r r r �
test_setreuid� s z PosixUidGidTests.test_setreuidc C � t �tjddg� d S )Nr� z,import os,sys;os.setreuid(-1,-1);sys.exit(0)�r� r< r8 r� rW r r r �test_setreuid_neg1� �
�z#PosixUidGidTests.test_setreuid_neg1�setregidztest needs os.setregid()c C sr t �� dkrts| �tt jdd� | �tt jdd� | �tt jdd� | �tt j| jd� | �tt jd| j� d S r7 ) r r8 r? rj rD rM rp r9 r@ rW r r r �
test_setregid� s z PosixUidGidTests.test_setregidc C rI )Nr� z,import os,sys;os.setregid(-1,-1);sys.exit(0)rJ rW r r r �test_setregid_neg1� rL z#PosixUidGidTests.test_setregid_neg1N)rP rQ rR r: r@ r r r r r; rA rE rF rH rK rN rO r r r r r4 � s&
r4 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| _n t jrt j| _nt j| _t�| j�| _g � � fdd�}|t j� t jr0|t j� t jr8|t j� � s?| � d� t
� | _ t� | j� z'� D ]!}t �
tj�| j|�� t�|�}|| j v rgtd��| j �|� qLW d S t�| j� � )Nc s2 zt �| �} W n
ty Y d S w � �| � d S r+ )r rX 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_NONASCIIrU r rX ZbdirZTESTFN_UNICODErh rH � unicodefnrA Zcreate_empty_filer> r@ rN r� �add�shutilrD )r0 rT rQ r rR r rX s8
� zPep383Tests.setUpc C s t �| j� d S r+ )rX rD r� rW r r r r� * r zPep383Tests.tearDownc C st | j }tt�| j��}| �||� t�� }z t�tj� | �tt�� �tt�tj��� W t�|� d S t�|� w r+ ) rV rH r r r� r= r, rB r? )r0 rI �foundZcurrent_directoryr r r � test_listdir- s "zPep383Tests.test_listdirc C s. | j D ]}ttj�| j|�d�}|�� qd S �Nr� )rV r$ r r> r@ r� r\ )r0 rQ ra r r r � test_open9 s
�zPep383Tests.test_openr r) c C s* | j D ]}tj�| j|�}t�|� qd S r+ )rV r r> r@ r� r )r0 rQ �fullnamer r r � test_statvfs> s
�zPep383Tests.test_statvfsc C s& | j D ]
}t�tj�| j|�� qd S r+ )rV r r� r> r@ r� )r0 rQ r r r � test_statG s
�zPep383Tests.test_statN)
rP rQ rR rX r� rZ r\ r r r r r^ r_ r r r r rP s % �
rP c @ sF e Zd Zdd� Zdd� Zdd� Zdd� Ze�d �d
d � �Z d d
� Z
dS )�Win32KillTestsc C s� 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 r�|�� d u r�|�t|��}
||�|j� � �|
|� |
�d d d �} | � | dd� |
j r�| � ||
j �!� � q�t"�#d � |d
7 }|| k r�|�� d u sj| �$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皙�����?rc z1Did not receive communication from the subprocess))r{ rb �msvcrtr| r} �
PeekNamedPipe�BOOL�restype�HANDLE�POINTERZc_char�DWORD�argtypesr� �Popenr8 r� r� r� r� r� r\ rd re �pollZcreate_string_bufferr: Z
get_osfhandlery Zsizeofr= r& r= r� ro �sleepr r �kill�pid�wait) r0 �sigr{ rb rh ri rn r� r� �maxr Zrsltr r r �_killM sP
�
���
�
zWin32KillTests._killc C r� r+ )rx �signal�SIGTERMrW r r r �test_kill_sigterm� rs z Win32KillTests.test_kill_sigtermc C s | � d� d S )Nr� )rx rW 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 rN|�
� d u rN|d dkr;q[t�d� |d7 }||k rN|�
� d u s4t�|jtj� | �d � t�|j|� t�d
� |�
� s}t�|jtj� | �d �|�� d S d S ) Nz
test_os_%sr rc r zwin_console_handler.pyr: rf rg z'Subprocess didn't finish initializationg �?z subprocess did not stop on {})�uuidZuuid1�mmapr� rp r8 r� r r> r@ rH r� ZCREATE_NEW_PROCESS_GROUPrq ro rr rs rt ry �SIGINTr r� )r0 �eventr Ztagname�mr� r� rw r r r �_kill_with_event� s4 ���
�
�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 ra � CTRL_C_EVENT)
r{ rb rm Zc_intr| r} �SetConsoleCtrlHandlerrj ro rk r� ry r� )r0 rb r{ ZNULLr� r r r �test_CTRL_C_EVENT� s
�
z Win32KillTests.test_CTRL_C_EVENTc C r )N�CTRL_BREAK_EVENT)r� ry r� rW r r r �test_CTRL_BREAK_EVENT� � z$Win32KillTests.test_CTRL_BREAK_EVENTN) rP rQ rR rx r{ r| r� r �skipr� r� r r r r r` K s 6
r` c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Win32ListdirTestszTest listdir on Windows.c C s� g | _ td�D ]E}d| }tj�tj|�}d| }tj�tj|�}t�|� t|ddd��}|� d| � W d � n1 s?w Y | j �
||g� q| j � � d S )Nre zSUB%dzFILE%dr0 r� r� z'I'm %s and proud of it. Blame test_os.
) �
created_pathsr� r r> r@ r rU r r$ r% �extendr )r0 r� Zdir_nameZdir_path� file_nameZ file_pathra r r r rX � s
�zWin32ListdirTests.setUpc C ra r+ )rX rD r rU rW r r r r� � r 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 � g | ]}t �|��qS r �r rX �r
r> r r r r � r0 zCWin32ListdirTests.test_listdir_no_extended_path.<locals>.<listcomp>N)r= �sortedr r r rU r� rX rW r r r � test_listdir_no_extended_path� 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 r� r r� r� r r r r � r0 z@Win32ListdirTests.test_listdir_extended_path.<locals>.<listcomp>N)
r r> rz r rU r= r� r r� rX rx r r r �test_listdir_extended_path� s � �z,Win32ListdirTests.test_listdir_extended_pathN)rP rQ rR r rX r� r� r� r r r r r� � 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�|�}tjdkr)dd� }||�r |dd � }||�r)|dd � }| �||� d S )Nr� c S s | � t| t�r
d�S d�S )Nr� r� )�
startswithr# rM )r r r r r� � s
��z/ReadlinkTests.assertPathEqual.<locals>.<lambda>rw )r r> �normcaser8 r9 r= )r0 �left�rightZ
has_prefixr r r �assertPathEqual� s
z ReadlinkTests.assertPathEqualc C sT | � tj�| j�� | � tj�| j�� | �tj�| j�� | �tj�| j�� d S r+ ) r] r r> r� �filelink_target�filelinkb_targetr� �filelink� filelinkbrW r r r rX
s zReadlinkTests.setUpc C s0 t | j�}| �ttj| j� | �ttj|� d S r+ )r r� rj rD r r� )r0 r� r r r �test_not_symlink
s
z ReadlinkTests.test_not_symlinkc C s( | � ttjd� | � ttjtd�� d S )Nz missing-link)rj rC r r� r rW 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� r� r� r� r rV r r� r� )r0 r� 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� r� r� r� r rV r� r r� r- rM rx 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� r� r� r� r rV r� r� r- rM rx r r r �
test_bytes#
s
zReadlinkTests.test_bytesN)rP rQ rR r� r r> rz r� r� rX r� r� r� rX r� r� r �skip_unless_symlinkr� r� r� r r r r r� � 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�oKej�d�d�dd � �Zd d � Z d d � Z d!S )"�Win32SymlinkTestsZ filelinktestZ dirlinktestz missing linkc C s^ t j�| j�s J �t j�| j�sJ �t j�| j�rJ �t j�| j�r$J �t j�| j�r-J �d S r+ )r r> r� �dirlink_targetr� �dirlinkr� � missing_linkrW r r r rX 5
s
zWin32SymlinkTests.setUpc C sV t j�| j�r
t �| j� t j�| j�rt �| j� t j�| j�r)t �| j� d S d S r+ ) r r> r� r� r% r� r7 rT r� rW r r r r� <
s �zWin32SymlinkTests.tearDownc C �` t �| j| j� | �t j�| j�� | �t j�| j�� | �t j�| j�� | � | j| j� d S r+ )
r r� r� r� r] r> r� rg �islink�
check_statrW r r r �test_directory_linkD
�
z%Win32SymlinkTests.test_directory_linkc C r� r+ )
r r� r� r� r] r> r� �isfiler� r� rW r r r �test_file_linkK
r� z Win32SymlinkTests.test_file_linkc C sF | j }tj�|�rt�|� d}tj�|�rJ �d}t�|||� dS )z2Create a "directory" link to a non-existent targetz"c:\\target does not exist.29r3c740TN)r� r r> rT r% r� r� )r0 Zlinknamer� Z
target_is_dirr r r �_create_missing_dir_linkR
s
z*Win32SymlinkTests._create_missing_dir_linkc C � | � � t�| j� d S r+ )r� r r% r� rW 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+ )r� r� r r> rg r� rW r r r �.test_isdir_on_directory_link_to_missing_targetc
r) z@Win32SymlinkTests.test_isdir_on_directory_link_to_missing_targetc C r� r+ )r� r r7 r� rW r r r �.test_rmdir_on_directory_link_to_missing_targetg
r3 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= rs rX )r0 r
r� Z
bytes_linkr r r r� k
s
z Win32SymlinkTests.check_statc C sH 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 � |� d S t � |� w )N�level2�level3r' r
)r r> rz r rU r@ r� rD rA r( r, rB r� �relpathr� r r= r� )r0 Zlevel1r� r� r' Zorig_dirr
r r r �
test_12084s
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� r] r> �samefile)r0 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 ]O\}}zt �||� W n ty7 Y nw zt �|� W n tyH Y nw zt �t �|�t �|�� W n ty` Y q"w zt �|� W q" tyq Y q"w d S )NZXXXXXXXXXXXXXXXXXXXXXXXXXXXrd rx � )r r> r@ r� rC r% rD rX )r0 Zsegmentr> Z
test_casesr� �destr r r �test_buffer_overflow�
s8
� � � � ��z&Win32SymlinkTests.test_buffer_overflowc s� t j�d�� t j�� �s| �d� � fdd�t�t �� �d�D �}|D ].}tj r0t
� t
d|� t � |�}| � |t �
|�� | �t
�|j�� | � |jt
j� d S | �d� d S )Nz$%LOCALAPPDATA%\Microsoft\WindowsAppsz%test requires a WindowsApps directoryc s g | ] }t j�� |��qS r )r r> r@ �r
r7 �r~ r r r �
s �z6Win32SymlinkTests.test_appexeclink.<locals>.<listcomp>z*.exez Testing withz$test requires an app execution alias)r r> �
expandvarsrg rh �fnmatch�filterr r rF rG rs r= r� r� �S_ISLNKr �st_reparse_tagZIO_REPARSE_TAG_APPEXECLINK)r0 �aliases�aliasrH r r� r �test_appexeclink�
s
�
z"Win32SymlinkTests.test_appexeclinkN) rP rQ rR r� r r> rz r� r� r� rH r� r� rX r� r� r� r� r� r� r� r� r� r r rT r� r� r� r� r r r r r� ,
s.
#
�
#r� c @ sD e Zd ZdZej�ej�e��Z dd� Z
dd� Z dd� Z dd � Z
d
S ) �Win32JunctionTestsZ junctiontestc C s( t j�| j�s J �t j�| j�rJ �d S r+ )r r> r� �junction_targetrT �junctionrW r r r rX �
� zWin32JunctionTests.setUpc C s" t j�| j�rt �| j� d S d S r+ )r r> rT r� rV rW r r r r� �
rY 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 )Nr� )�_winapi�CreateJunctionr� r� r] r r> rT r� rg r= r� rs r= r� r� r� r� rW r r r �test_create_junction�
s �z'Win32JunctionTests.test_create_junctionc C s\ t �| j| j� | �tj�| j�� | �tj�| j�� t� | j� | �
tj�| j�� d S r+ ) r� r� r� r� r] r r> r� rT rV r� rW r r r � test_unlink_removes_junction�
s
z/Win32JunctionTests.test_unlink_removes_junctionN)rP rQ rR r� r r> rH rz r� r� rX r� r� r� r r r r r� �
s
r� c @ r� )� 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 ](} |D ]#}
z|�|
� W n tyf Y nw zt�|
� W qT tyw Y qTw qP|� ||�
|��}| �d|� |j| } | �d| � d S ) Nrf r{ r z Kernel32.dllT)Zuse_last_errorrc z\\?\C:z\\?\NULz \\?\CONINrd )r �
import_moduleZctypes.wintypesZWinDLLrb rl ZGetCurrentProcessrk rj ZGetProcessHandleCountZLPDWORDro rn Zbyrefr= r& r� r� Z_getfinalpathname� Exceptionr r� ) r0 rf r{ ZkernelZhprocZ handle_countr� Z before_count� filenames�_r Z handle_deltar r r � test_getfinalpathname_handles sH
�
� � ��
z*Win32NtTests.test_getfinalpathname_handlesN)rP rQ rR r� r r r r r� �
� r� c @ r) )�NonLocalSymlinkTestsc C s t �d� dS )zM
Create this structure:
base
\___ some_dir
z
base/some_dirN)r r rW r r r rX 4 s zNonLocalSymlinkTests.setUpc C s t �d� d S )Nr< )rX rD rW r r r r� = s z NonLocalSymlinkTests.tearDownc C s. t j�dd�}t �d|� t j�|�sJ �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> r@ r� rg )r0 r� r r r � test_directory_link_nonlocal@ s z1NonLocalSymlinkTests.test_directory_link_nonlocalN)rP rQ rR rX r� r� r r r r r� 1 s r� c @ � e Zd Zdd� Zdd� ZdS )�FSEncodingTestsc C s( | � t�d�d� | � t�d�d� d S )Ns abc�u abcŁ)r= r rX rN rW r r r �test_nopR r� zFSEncodingTests.test_nopc C sB dD ] }zt �|�}W n ty Y qw | �t �|�|� qd S )N)u unicodeŁu latiné�ascii)r rX r r= rN )r0 rQ rS r r r �
test_identityV s ��z FSEncodingTests.test_identityN)rP rQ rR r� r� r r r r r� Q s r� c @ sV e Zd Zdd� Ze�e�d�o"e� o"e j
� d�p"e e
d�o"e e
d�d�dd � �Zd
S ) �DeviceEncodingTestsc C s | � t�d�� d S )Ni@� )r� r �device_encodingrW r r r � test_bad_fdc r 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 rn )r r� ZassertIsNotNoner] �codecs�lookup)r0 r� r r r �test_device_encodingg s
z(DeviceEncodingTests.test_device_encodingN)rP rQ rR r� r r r r� r r8 r9 r� r �localer� r r r r r� a 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� rp r8 r� r� � communicater= r� r �getpid)r0 r r� r� r r r � test_getppidq s
�� zPidTests.test_getppidNc C s� t jdkrdt j� d�dd|� d�g}nt jd|g}t�tjt j|�}|d ur+||� t�|d�\}}| �t�|�|� | �||� d S )Nr� �"r� r ) r8 r9 r� r �spawnv�P_NOWAIT�waitpidr= �waitstatus_to_exitcode)r0 r� �exitcode�callbackr� rt Zpid2�statusr r r �
check_waitpidz s
zPidTests.check_waitpidc C s | j ddd� d S )Nr� r )r� r� �r� rW r r r � test_waitpid� r� zPidTests.test_waitpidc C sX d}d|� d�}| j ||d� | �t�� t�d� W d � d S 1 s%w Y d S )N� zimport sys; sys.exit(�)�r� g )r� rj rp r r� )r0 r� r� r r r �test_waitstatus_to_exitcode� 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� )r0 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/w Y | �t�� t�d� W d � d S 1 sJw Y d S )Nr4 r rc r� r� r )r= r r� rj r� r9 )r0 Z max_exitcoder� r r r �#test_waitstatus_to_exitcode_windows� s � � "�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 rs )rt ��signumr r � kill_process� r z?PidTests.test_waitstatus_to_exitcode_kill.<locals>.kill_process)r� r� )r Z LONG_TIMEOUTry r r� )r0 r� r r r r � test_waitstatus_to_exitcode_kill� s z)PidTests.test_waitstatus_to_exitcode_killr+ )rP rQ rR r r r r r� r� r� r� r8 r9 r r ry r r r r r r� p 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|� |sd| j }n ttj�| _t t
� � �| _ | j | j| j < d| j | j f }t
|ddd��
}|�|� W d � n1 sHw Y tj|g}|rfdd� |D �}d d
� | j�� D �| _|S ) N� zimport sys; sys.exit(%s)z4import sys, os; magic = os.environ[%r]; sys.exit(%s)r0 r� r� c S r� r r� r� r r r r � r0 z*SpawnTests.create_args.<locals>.<listcomp>c S s" i | ]
\}}t �|�t �|��qS r r� )r
�k�vr r r �
<dictcomp>� s �z*SpawnTests.create_args.<locals>.<dictcomp>)r� r rU r� rV r� r r; r� r. r} Zuuid4r� r$ r% r8 r� r� )r0 r
r r& r� r' r� r r r � create_args� s*
� �
�zSpawnTests.create_args�spawnlc C �4 | � � }tjtj|d g|�R � }| �|| j� d S rn )r r r �P_WAITr= r� �r0 r� r� r r r � test_spawnl� � zSpawnTests.test_spawnl�spawnlec C �> | j dd�}tjtj|d g|�| j�R � }| �|| j� d S �NT)r
r )r r r r r� r= r� r r r r � test_spawnle� � zSpawnTests.test_spawnle�spawnlpc C r rn )r r r r r= r� r r r r � test_spawnlp� r zSpawnTests.test_spawnlp�spawnlpec C r r )r r r r r� r= r� r r r r �
test_spawnlpe� r zSpawnTests.test_spawnlper� c C sT | � � }t�tj|d |�}| �|| j� t�tjt|d �|�}| �|| j� d S rn )r r r� r r= r� r r r r r � test_spawnv� s
zSpawnTests.test_spawnv�spawnvec C �6 | j dd�}t�tj|d || j�}| �|| j� d S r �r r r! r r� r= r� r r r r � test_spawnve� � zSpawnTests.test_spawnve�spawnvpc C s. | � � }t�tj|d |�}| �|| j� d S rn )r r r& r r= r� r r r r � test_spawnvp� s zSpawnTests.test_spawnvp�spawnvpec C r" r )r r r( r r� r= r� r r r r �
test_spawnvpe r% zSpawnTests.test_spawnvpec C s0 | � � }t�tj|d |�}tj|| jd� d S )Nr r� )r r r� r� r Z wait_processr� )r0 r� rt 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 ry )r rj r� r r r� �r0 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 rj 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� ry )r rj 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 r0 )r rj r� r r! r� r- r r r �test_spawnve_noargs* s
"z SpawnTests.test_spawnve_noargsc C sv t jddg}tj�� }d|d< z |tj|d ||�}W n ty$ Y nw | �|d� tj�� }d|d< z |tj|d ||�}W n tyI Y nw | �|d� tj�� }d |d
< z |tj|d ||�}W n tyn Y nw | �|d� tj }| �
tj |� t |d d d
��
}|�
d� W d � n1 s�w Y t j|g}tj�� }d|d< |tj|d ||�}| �|d� d S )Nr� r� r� r� r � r� r� r� r� r0 r� r� zPimport sys, os
if os.getenv("FRUIT") != "orange=lemon":
raise AssertionErrorz orange=lemon)r8 r� r r; r� r r� r= r rU r� rV r$ r% )r0 �spawnr� r� r� r& r' r r r �_test_invalid_env2 sF
�
�
� �
z SpawnTests._test_invalid_envc C r� r+ )r5 r r! rW r r r �test_spawnve_invalid_env` rs z#SpawnTests.test_spawnve_invalid_envc C r� r+ )r5 r r( rW r r r �test_spawnvpe_invalid_envd rs z$SpawnTests.test_spawnvpe_invalid_envN)rP rQ rR r r r r r r r r$ r' r) r* r+ r. r/ r1 r2 r5 r6 r7 r r r r r � sF
.
r z>Skip due to platform/environment differences on *NIX buildbots�getloginztest needs os.getloginc @ r� )�
LoginTestsc C s t �� }| �t|�d� d S rn )r r8 r= r: )r0 Z user_namer r r �
test_getloginn � zLoginTests.test_getloginN)rP rQ rR r: r r r r r9 k � r9 � getpriority� setpriorityz'needs os.getpriority and os.setpriorityc @ rU )�ProgramPriorityTestsz0Tests for os.getpriority() and os.setpriority().c C s
t �t jt �� �}t �t jt �� |d � zIt �t jt �� �}|dkr.|dkr.t�d| ��| �||d � W z
t �t jt �� |� W d S ty^ } z|j t j
krS� W Y d }~d S d }~ww z t �t jt �� |� W w ty� } z
|j t j
krz� W Y d }~w d }~ww )Nrc � z?unable to reliably test setpriority at current nice level of %s) r r= � PRIO_PROCESSr� r> r rt r= rD rE ZEACCES)r0 r< Znew_prior� r r r �test_set_get_priorityx s2 � ���� ���z*ProgramPriorityTests.test_set_get_priorityN)rP rQ rR r rB r r r r r? s 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 @ r� ) zSendfileTestServer.Handlerc C s. t j�| |� g | _d| _d| _| �d� d S )NTFs 220 ready
)�asynchat�
async_chat�__init__� in_buffer�
accumulate�closed�push)r0 �connr r r rF � s
z#SendfileTestServer.Handler.__init__c C s$ | � d�}| jr| j�|� d S d S )N� )�recvrH rG r$ r� r r r � handle_read� s
�z&SendfileTestServer.Handler.handle_readc C s d� | j�S )Nr� )r@ rG rW r r r �get_data� r� z#SendfileTestServer.Handler.get_datac C s | � � d| _d S r )r\ rI rW r r r � handle_close� s
z'SendfileTestServer.Handler.handle_closec C � � r+ r rW r r r � handle_error� � z'SendfileTestServer.Handler.handle_errorN)rP rQ rR rF rN rO rP rR r r r r �Handler� s rT c C sp t j�| � tj�| � | �tjtj� | � |� | �
d� | j� � d d� \| _ | _
d | _d| _t �� | _d S )Nr� re F)� threading�ThreadrF �asyncore�
dispatcherZ
create_socket�socket�AF_INET� SOCK_STREAM�bind�listen� getsockname�host�port�handler_instance�_active�Lock� _active_lock)r0 �addressr r r rF � s
zSendfileTestServer.__init__c C s | j S r+ )rb rW r r r rc � s zSendfileTestServer.runningc C s. | j rJ �t�� | _tj�| � | j�� d S r+ )rc rU �Event�_SendfileTestServer__flagrV �startru rW r r r rh � s
zSendfileTestServer.startc C s | j sJ �d| _| �� d S r� )rc rb r@ rW r r r �stop� s
zSendfileTestServer.stopc C s2 t | jdd�st�d� t | jdd�r| �� d S )NrI F���MbP?)r ra ro rr ri rW r r r ru � s
� zSendfileTestServer.waitc C sV d| _ | j�� | j r%tjr%| j�� tjddd� | j�� | j r%tjst� � d S )NTrj rc )�timeoutr� )
rb rg rH rW Z
socket_maprd �acquireZloop�releaseZ close_allrW r r r r� � s
� zSendfileTestServer.runc C s | � � \}}| �|�| _d S r+ )�acceptrT ra )r0 rK �addrr r r �
handle_accept� s z SendfileTestServer.handle_acceptc C s | � � d S r+ )r\ rW r r r �handle_connect� r� z!SendfileTestServer.handle_connectc C s dS rn r rW r r r �writable� r� zSendfileTestServer.writablec C rQ r+ r rW r r r rR � rS z SendfileTestServer.handle_errorN)rP rQ rR rD rE rT rF �propertyrc rh ri ru r� rp rq rN rr rR r r r r rC � s
rC �sendfileztest needs os.sendfile()c @ s
e Zd Zdd Zej�d� o ej�d� o ej�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� �linuxZsolarisZsunosz%requires headers and trailers supportr5 z(test is only meaningful on 32-bit buildsc C s t �� | _ttj| j� d S r+ )r Zthreading_setupr� r( r rU �DATAr{ r r r r} � s
zTestSendfile.setUpClassc C s t j| j� t�tj� d S r+ )r Zthreading_cleanupr� r rV rU r{ r r r r� � s zTestSendfile.tearDownClassc C s~ t tjdf�| _| j�� t�� | _| j�| jj| jj f� | j�
d� | j� d� | j� � | _
ttjd�| _| j� � | _ d S )Nr rc rv r� )rC r ZHOST�serverrh rY �client�connectr_ r` �
settimeoutrM ry �socknor$ r rU r* rW r r r rX � s
zTestSendfile.setUpc C s0 | j �� | j�� | jjr| j�� d | _d S r+ )r* r\ rz ry rc ri rW r r r r�
s
zTestSendfile.tearDownc
O s\ z t j|i |��W S ty- } z|jtjkr� |jtjtjfv r(W Y d}~q � d}~ww )zfA higher level wrapper representing how an application is
supposed to use sendfile().
TN)r rt rD rE Z
ECONNRESET�EAGAIN�EBUSY)r0 r� r r� r r r �sendfile_wrapper
s ��z TestSendfile.sendfile_wrapperc C s� d}d}d}|t | j�k r8| �| j| j||�}|dkr q8||7 }||7 }| �||k� | �||� |t | j�k s
| �|t | j�� | j�t j
� | j� � | j �
� | j j�� }| �t |�t | j�� | �|| j� d S )Nr rL )r: rx r� r} ry r] r= rz �shutdownrY � SHUT_RDWRr\ ry ru ra rO )r0 �
total_sent�offset�nbytes�sentr� r r r �test_send_whole_file
s& �
z!TestSendfile.test_send_whole_filec C s� d}t | j�d }t | j�| }d}||k r8| �| j| j||�}|dkr%q8||7 }||7 }| �||k� ||k s| j�tj � | j�
� | j � � | j j
�� }| jt | j�d d � }| �|t |�� | �t |�t |�� | �||� d S )Nr re rL )r: rx r� r} ry r] rz r� rY r� r\ ry ru ra rO r= )r0 r� r� Z must_sendr� r� r� rI r r r �test_send_at_certain_offset1
s( �
z(TestSendfile.test_send_at_certain_offsetc
C s� t | j�d }z t�| j| j|d�}W n ty, } z
|jtjkr"� W Y d }~n d }~ww | � |d� | j
� t j
� | j
�� | j�� | jj�� }| � |d� d S )NrL r r� )r: rx r rt r} ry rD rE �EINVALr= rz r� rY r� r\ ry ru ra rO )r0 r� r� r� r� r r r �test_offset_overflowH
s ���
z!TestSendfile.test_offset_overflowc C sT | � t��}t�| j| jdd� W d � n1 sw Y | �|jjtj � d S )Nr rL )
rj rD r rt r} ry r= r. rE r� �r0 r� r r r �test_invalid_offsetY
s �z TestSendfile.test_invalid_offsetc C s@ t j| j| jddd� | jr t j| j| jdddddd� d S d S )Nr rL )r� r� r� r� r )r� r� r� r� �headers�trailersr� )r rt r} ry �SUPPORT_HEADERS_TRAILERSrW r r r �
test_keywords^
s �
��zTestSendfile.test_keywordsc C s� d}d| j d d� }tj| j| jddddgd�}| �|d� ||7 }d}|t|�k rWtt|�| d�}| �| j| j||�}|dkrCqW| �||� ||7 }||7 }|t|�k s+| � |t|�� | j
� � | j �
� | j j�� }| � t|�t|�� d S ) Nr s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyr rL s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxs yyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyyy�r� i )rx r rt r} ry ZassertLessEqualr: �minr� r= rz r\ ry ru ra rO �hash)r0 r� Z
expected_datar� r� r� r� r r r � test_headersi
s0 � � �
zTestSendfile.test_headersc C s� t jd }d}| �t j|� t||� t|d��.}tj| j|� � ddddgd� | j
� � | j �
� | j j�� }| �|d � W d � d S 1 sIw Y d S )
Nr� s abcdefr� r r� s 123456s 789�r� s abcde123456789)r rU r� rV r( r$ r rt r} ry rz r\ ry ru ra rO r= )r0 r� Z file_datara r� r r r �
test_trailers�
s
�
"�zTestSendfile.test_trailersc C �l d| j j_| �t��}tj| j| jdddd gd d� W d � n1 s&w Y | � |j
j t j � d S )NFr � x� r� r� �
ry ra rH rj rD r rt r} ry r= r. rE r� r� r r r � test_headers_overflow_32bits�
�
� �z)TestSendfile.test_headers_overflow_32bitsc C r� )NFr r� r� r� r� r� r� r r r � test_trailers_overflow_32bits�
r� z*TestSendfile.test_trailers_overflow_32bits� SF_NODISKIOztest needs os.SF_NODISKIOc
C s\ zt j| j| jddt jd� W d S ty- } z|jtjtjfvr"� W Y d }~d S d }~ww )Nr rL )r� ) r rt r} ry r� rD rE r r~ )r0 r� r r r �
test_flags�
s ����zTestSendfile.test_flagsN) rP rQ rR rx r8 r9 r� r� r r Zrequires_headers_trailers�maxsizeZ requires_32br� r} r� rX r� r� r� r� r� r� r� r� r� r� r� r r r� r r r r ru � sP � �� �
�ru c C s� t td�sdS zLttjdd��+} z t�| �� dd� W n ty2 Y W d � W t�tj� dS w W d � n1 s=w Y W t�tj� dS W t�tj� dS t�tj� w )N�setxattrFr# r � user.testr� T) r r r$ r rU r� ry rD rV )r' r r r � supports_extended_attributes�
s&
��� � � �r� z(no non-broken extended attribute supportre r� �' c @ r� ) �ExtendedAttributeTestsc
K sF t j}| �t j|� t|� | �t��}|||d�fi |�� W d � n1 s)w Y | �|jj t j
� ||�} | � | t � |||d�dfi |�� t
| �}
|
�d� | �t
||��|
� | �||dfi |��d� |||d�dtjfi |�� | �||dfi |��d� | �t��}|||d�dtjfi |�� W d � n1 s�w Y | �|jj t j� | �t��}|||d�dtjfi |�� W d � n1 s�w Y | �|jj t j
� |||d�dtjfi |�� |
�d� | �t
||��|
� |||d�fi |�� | �t��}|||d�fi |�� W d � n 1 �s$w 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{}Nr� )r
r� r r r r� �
� � z;ExtendedAttributeTests._check_xattrs_str.<locals>.<genexpr>r� r� )r rU r� rV r( rj rD r= r. rE �ENODATAr- r rH rW r �
XATTR_REPLACE� XATTR_CREATE�EEXISTr% r� r� )
r0 r �getxattrr� � removexattr� listxattrr rQ r� Z
init_xattrZxattr�many�thingr r r �_check_xattrs_str�
sT �
� �
�
$z(ExtendedAttributeTests._check_xattrs_strc O sN | j tg|�R i |�� t�tj� | j tjg|�R i |�� t�tj� d S r+ )r� r. r rV rU r rX rB r r r �
_check_xattrs�
s z$ExtendedAttributeTests._check_xattrsc C s | � tjtjtjtj� d S r+ �r� r r� r� r� r� rW r r r � test_simple�
s �z"ExtendedAttributeTests.test_simplec C s | j tjtjtjtjdd� d S )NFr� r� rW r r r �
test_lpath s
�z!ExtendedAttributeTests.test_lpathc C s4 dd� }dd� }dd� }dd� }| � ||||� d S ) Nc W �D t | d��}tj|�� g|�R � W d � S 1 sw Y d S r[ )r$ r r� ry �r> r� r' r r r r� � $�z1ExtendedAttributeTests.test_fds.<locals>.getxattrc W �H t | dd��}tj|�� g|�R � W d � d S 1 s w Y d S �Nrh r )r$ r r� ry r� r r r r� � "�z1ExtendedAttributeTests.test_fds.<locals>.setxattrc W r� r� )r$ r r� ry r� r r r r� r� z4ExtendedAttributeTests.test_fds.<locals>.removexattrc W r� r[ )r$ r r� ry r� r r r r� r� z2ExtendedAttributeTests.test_fds.<locals>.listxattr)r� )r0 r� r� r� r� r r r �test_fds s
z ExtendedAttributeTests.test_fdsN)rP rQ rR r� r� r� r� r� r r r r r� �
s 0 r� �get_terminal_sizez requires os.get_terminal_sizec @ r� )�
TermsizeTestsc
C sn zt �� }W n ty& } ztjdks |jtjtjfv r!| �d� � d}~ww | � |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 r� rD r8 r9 rE r� �ENOTTYrh ZassertGreaterEqual�columns�lines)r0 r� r� r r r �test_does_not_crash s
��z!TermsizeTests.test_does_not_crashc
C s� zt jddgt jdd��� }W n tt jtfy! | �d� Y nw t|d �t|d �f}z
t �
t j �
� �}W n tyX } zt jdksN|jtjtjfv rS| �d � � d
}~ww | �||� 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.
Zsttyr� T)rd r� zstty invocation failedrc r r� r� N)r� � check_output�DEVNULLr rC �CalledProcessErrorr rh r� r r� r8 � __stdin__ry rD r9 rE r� r� r= )r0 r� rI r� r� r r r �test_stty_match) s* ��
��
��z TermsizeTests.test_stty_matchN)rP rQ rR r� r� r r r r r� s r� � memfd_createzrequires os.memfd_creater� r c @ r� )�MemfdCreateTestsc C s� t �dt j�}| �|d� | �t j|� | �t �|�� t|ddd��}|� d� | �
|� � d� W d � n1 s;w Y t �d�}| �t j|� | �t �|�� d S )NZHir rh F)�closefds memfd_creater� ) r r� � MFD_CLOEXECr= r� r\ r� r r$ r% r= �tell)r0 r~ ra �fd2r r r �test_memfd_createI s
�
z"MemfdCreateTests.test_memfd_createN)rP rQ rR r� r r r r r� F r< r� �eventfdzrequires os.eventfdr6 c @ r) )� EventfdTestsc C s� dd� }d}d}t �|�}| �|d� | �t j|� | �t �|�� t �||�}| �|||�� t � ||d�� t �||�}| �||d�� t � ||d�� t � ||d�� t �||�}| �||d�� t �
|d � t �
|d
� t � |�}| �|d� d S ) Nc S s t �d| �S )z$Pack as native uint64_t
z@Q)�struct�pack)r& r r r r� [ s z/EventfdTests.test_eventfd_initval.<locals>.packr� r� r r� �( re r r� ) r r� r= r� r\ r� r r{ r= r% �
eventfd_write� eventfd_read)r0 r� r� �initvalr~ �resr r r �test_eventfd_initvalZ s(
z!EventfdTests.test_eventfd_initvalc C s. d}t jt jB t jB }t �||�}| �|d� | �t j|� t �|�}| � |d� t �|�}| � |d� | �
t ��
t �|� W d � n1 sIw Y | �
t �� t � |d� W d � n1 sdw Y t �
|d� t �|�}| � |d� | �
t �� t �|� W d � d S 1 s�w Y d S )Nre r rc r� )r � EFD_CLOEXEC�
EFD_SEMAPHORE� EFD_NONBLOCKr� r= r� r\ r� r= rj r� r{ r� )r0 r� r� r~ r� r r r �test_eventfd_semaphorey s*
� �
"�z#EventfdTests.test_eventfd_semaphorec C s t jt jB }t �d|�}| �|d� | �t j|� t�|g|g|gd�\}}}| �|||fg |gg f� t � |d� t�|g|g|gd�\}}}| �|||f|g|gg f� | �t �
|�d� t � |d� t�|g|g|gd�\}}}| �|||f|gg g f� t �
|� d S )Nr r r� l ���� ) r r� r� r� r= r� r\ �selectr= r� r� )r0 r� r~ �rfd�wfdZxfdr r r �test_eventfd_select� s z EventfdTests.test_eventfd_selectN)rP rQ rR r� r� r� r r r r r� W s r� c @ r� )� OSErrorTestsc C s� G dd� dt �}g | _g | _tjd urtj}ntj}| j�|� | j�||�� tjd ur1tj}nt� tj�}| j�|� | j�t
|�� | j�t |�� | j| j | _ d S )Nc @ � e Zd ZdS )z OSErrorTests.setUp.<locals>.StrN�rP rQ rR r r r r �Str� � r� )
r. �bytes_filenames�unicode_filenamesr rU rU r$ ZTESTFN_UNDECODABLEr rX r� r� r� )r0 r� Zdecoded�encodedr r r rX � s
zOSErrorTests.setUpc C s@ | j tjf| j tjdf| j tjf| j tjtjf| j tjf| j tjf| j tj fg}t
j dkrP|� | j
tjdf| j
tjdf| jtjdf| jtjdf| jtjff� n|� | j tjf| j tjdf| j tjdff� ttd�rv|�| j tjddf� ttd�r�|�| j tjddf� ttd�r�|�| j tjdf� ttd �r�|�| j tjdf� ttd
�r�|�| j tjdf� ttd �r�|�| j tjf� ttd �r�t
j dkr�|�| j
tjdf� |�| jtjdf� n
|�| j tjdf� ttd
��r
|� | j tjf| j tj df| j tj ddf| j tj dff� ttd��r |�| j tj df� ttd��r,|�| j tj f� |D ]o^}}}|D ]f}z2t!|t"t#f��rI||g|�R � n | �$t%d�� ||g|�R � W d � n 1 �scw Y W n( t&�y� } z| �'|j(|t"|�� W Y d }~�q5d }~w t)�y� Y �q5w | �*d�+|�� �q5�q.d S )Nr� r� s dstr� ry r �lchownr �chflags�lchflags�chrootr
r� r� s user�lchmodr� z should bezNo exception thrown by {}),r� r rB r rs r$ r� r7 r� rV r8 r9 r� r� rq r� r� r r r$ ry r� r r� r� r� r
r� r� r� r� r� r� r# r. rM ZassertWarnsRegex�DeprecationWarningrD r� r& �UnicodeDecodeErrorr r� )r0 Zfuncsr� �funcZ func_argsr r� r r r �test_oserror_filename� s�
�
�
�
� ��"����z"OSErrorTests.test_oserror_filenameN)rP rQ rR rX r� r r r r r� � s r� c @ r� )�
CPUCountTestsc C s: t �� }|d ur| �|t� | �|d� d S | �d� d S )Nr z&Could not determine the number of CPUs)r � cpu_countr- r� �
assertGreaterrh )r0 Zcpusr r r �test_cpu_count s
z CPUCountTests.test_cpu_countN)rP rQ rR r� r r r r r� r� r� 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� r� r\ r= r r r� r r r �test_get_set_inheritable 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� r� r\ r= r �fcntl�F_GETFD�
FD_CLOEXEC�F_SETFD)r0 r~ r� r r r � test_get_inheritable_cloexec 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� r� r\ r= r� r� r� r r� r r r � test_set_inheritable_cloexec' s � �z/FDInheritanceTests.test_set_inheritable_cloexec�O_PATHzneed os.O_PATHc C �n t �tt j�}| �t j|� | �t �|�d� t �|d� | �t �|�d� t �|d� | �t �|�d� d S r� ) r r$ r� r r� r\ r= r r r� r r r � test_get_set_inheritable_o_path2 s z2FDInheritanceTests.test_get_set_inheritable_o_pathc C s� t �� }| �t��
}t�|� W d � n1 sw Y | �|jjtj � | �t��}t�
|d� W d � n1 s=w Y | �|jjtj � | �t��}t�
|d� W d � n1 saw Y | �|jjtj � d S �NTF) r r� rj rD r r r= r. rE r/ r )r0 r~ r1 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� r� r\ r= r r� r r r r\ M s z FDInheritanceTests.test_openr� zneed os.pipe()c C �P t �� \}}| �t j|� | �t j|� | �t �|�d� | �t �|�d� d S r� )r r� r� r\ r= r )r0 r� r� r r r � test_pipeR �
z FDInheritanceTests.test_pipec C sJ t �tt j�}| �t j|� t �|�}| �t j|� | �t �|�d� d S r� ) r r$ r� r� r� r\ rg r= r �r0 �fd1r� r r r �test_dupZ s
zFDInheritanceTests.test_dupc C s( t �d�}| �t j|� | �|d� d S �Nrc r )r rg r� r\ r� r� r r r �test_dup_standard_streamb s
z+FDInheritanceTests.test_dup_standard_streamr� r c C sH t �dt j�}| �t j|� t �|�}| �t j|� | �t �|�� d S )N�NUL)r r$ r� r� r\ rg r� r r r r r � test_dup_nulg 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)� inheritable) r r$ r� r� r� r\ r= r r] r r� )r0 r~ r� Zfd3r r r r p s z FDInheritanceTests.test_dup2�openptyzneed os.openpty()c C r r� )r r r� r\ r= r )r0 Z master_fdZslave_fdr r r � test_openpty� r z FDInheritanceTests.test_openpty)rP rQ rR r� r r� r� r� r� r r r r r r\ r r
r r8 r9 r r r r r r r r� s(
r� c @ sH e Zd Zddddejfdfddddd eed
d�fgZd d � Zd
d� ZdS )�PathTConverterTests)r� Tr N)rs Fr Nr^ FN)r� F�r N)r� Fr Nr$ r r\ c C s� t j}tjdkr
d }}n
t�t j�}t|�}t�t|�tjtjB �}| � t j
t j� | � tj |� t|�}t|�}| j D ]�\}}} }
| j
|d��� ztt|�} W n ty` Y W d � q=w ||||fD ].} | d u rnqg| j
|| d�� | | g| �R � }
|
d ur�|
|
� W d � n1 s�w Y qg| �td�� | |g| �R � W d � n1 s�w Y |r�| |g| �R � }
|
d ur�|
|
� n | �td�� | |g| �R � W d � n1 s�w Y W d � n1 s�w Y q=d S )Nrf r )r r> zto return str or bytesz os.PathLike)r rU r r rX r r$ r� rZ r� rV r\ � functionsZsubTestr r �assertRaisesRegexrp )r0 Z str_filenameZ bytes_fspath�bytes_filenamer~ Z
int_fspathZ
str_fspathr Zallow_fd�
extra_argsZ
cleanup_fnrQ r> r
r r r �test_path_t_converter� sd
��� ��� ��� �� ���z)PathTConverterTests.test_path_t_converterc C s� d}| � t|d �� t�td�� W d � n1 s w Y | � t|d �� t�td�� W d � n1 s;w Y | � t|d �� t�tt� �� W d � d S 1 s\w Y d S )Nz-__fspath__\(\) to return str or bytes, not %sr� re �floatg� ��Q�@�object)r rp r r� r r )r0 rn r r r �&test_path_t_converter_and_custom_class� s � �"�z:PathTConverterTests.test_path_t_converter_and_custom_class) rP rQ rR r �F_OKr r r r r r r r r � s � ,r r! r" c @ r� )�
BlockingTestsc C r r ) r r$ r� r� r� r\ r= r! r# r� r r r r$ � s zBlockingTests.test_blockingN)rP rQ rR r$ r r r r r � r< r c @ r� )� ExportsTestsc C s | � dtj� | � dtj� d S )Nr$ r )r� r �__all__rW r r r � test_os_all� s zExportsTests.test_os_allN)rP rQ rR r r r r r r � r� r c @ r) )� TestDirEntryc C s0 t j�tj�| _| �tj| j� t �| j� d S r+ )r r> �realpathr rU r� rD rA rW r r r rX � s zTestDirEntry.setUpc C s | � ttj� d S r+ )rj rp r �DirEntryrW r r r �test_uninstantiable� r� z TestDirEntry.test_uninstantiablec C sj t tj�| jd�d�}dd� t�| j�D ��� }| �|tj� | �|j d� dd l
}| � t |j
||� d S )Nr� � pythonc S s g | ]}|�qS r r �r
�entryr r r r � s z0TestDirEntry.test_unpickable.<locals>.<listcomp>r )r( r r> r@ �scandirr� r- r# r= r r rj rp r )r0 r& r' r r r r �test_unpickable� s z TestDirEntry.test_unpickableN)rP rQ rR rX r$ r) r r r r r! � 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 rU rX �
bytes_pathr� rD rA rW r r r rX � s zTestScandir.setUpr� c C s2 t |t�r| jn| j}tj�||�}t|d� |S )Nr% )r# rM r+ r> r r@ r( )r0 r r> r& r r r r( � s
zTestScandir.create_filec C s2 t dd� t�| j�D ��}| �t|�� �|� |S )Nc s � � | ]}|j |fV qd S r+ r r& r r r r� s � �z*TestScandir.get_entries.<locals>.<genexpr>)r� r r( r> r= r� �keys)r0 �names�entriesr r r � get_entries s
�zTestScandir.get_entriesc C s\ |r&t |�D ] }|�d�sq|dv rq| �t||�t||�|||f� qd S | �||� d S )NZst_)�st_dev�st_ino�st_nlink)r� r� r= r )r0 Zstat1Zstat2Z skip_fieldsr r r r �assert_stat_equal s
�� z TestScandir.assert_stat_equalc C s( t �| j�}| �tt|�� |�� d S r+ )r r( r> rj rp r| r\ )r0 � scandir_iterr r r r$ s z TestScandir.test_uninstantiablec C s< | � d�}t�| j�}dd l}| �t|j||� |�� d S )Nr� r ) r( r r( r> r rj rp r r\ )r0 r& r5 r r r r r) s
zTestScandir.test_unpickablec C s0 | � |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 �� | �|�� |tjdko�| � | �|jdd�|tjdk� d S )NFr� rf )r- r r# r= r r> r@ �inoder� r2 �is_dir�S_ISDIRr �is_file�S_ISREG�
is_symlinkr� r4 )r0 r' r r7 r9 r; Z
entry_statZ entry_lstatr r r � check_entry" s>
�
�
�
�
�
�
��zTestScandir.check_entryc
C s� t td�}t�� }tj�| jd�}t�|� | �d�}|rEzt�|tj�| jd�� W n t yD } z
| �
d| � W Y d }~nd }~ww |ratj |tj�| jd�dd� t� |tj�| jd �� ddg}|rl|� d� |rs|�
d
� | �|�}|d }| �|ddd d � |d }| �|dd dd � |r�|d }| �|dd dd � |r�|d }| �|ddd d� |d }| �|d d dd� d S d S ) Nr
r� r� z
link_file.txtr+ � symlink_dirT)r� �symlink_file.txt)r= r> F)r r r r r> r@ rA r( r
r rh r� r$ r� r0 r< ) r0 r
r� rH r& r� r. r/ r' r r r �test_attributes> sH
���
�zTestScandir.test_attributesc C sN t |t�r| jn| j}tt�|��}| �t|�d� |d }| �|j |� |S r )
r# rM r+ r> r r r( r= r: r )r0 r r> r/ r' r r r � get_entryg s zTestScandir.get_entryc C s | j |d�}| �tj�|��S )Nr )r( r@ r r> �basename)r0 r r& r r r �create_file_entryp r z TestScandir.create_file_entryc C sn | � � }t�� }z(t�| j� tdd� t�� D ��}| �t|� � �tj�
|�g� W t�|� d S t�|� w )Nc s r, r+ r r& r r r r� | r� z5TestScandir.test_current_directory.<locals>.<genexpr>) r( r r, rB r> r� r( r= r� r- rA )r0 r& Zold_dirr/ r r r �test_current_directoryt s �z"TestScandir.test_current_directoryc C s | � � }| �t|�d� d S )Nz<DirEntry 'file.txt'>)rB r= r� �r0 r' r r r � test_repr� r; zTestScandir.test_reprc C s* | � � }| �t�|�tj�| jd�� d S )Nr� )rB r= r �fspathr> r@ rD 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 rX rB rF r- rM r= r> r@ )r0 r Z bytes_entryrF r r r �test_fspath_protocol_bytes� s
�z&TestScandir.test_fspath_protocol_bytesc C s� t j�| jd�}t �|� | �d�}t �|� t jdkr#| �|�� � | � |�
� � | � |� � � t jdkrI| � t
|j� |�� |jdd� d S | �|�� d� | � t
|j� | j t
|jdd� d S )Nr� rf Fr� r )r r> r@ rA r@ r7 r r] r7 r� r9 r; rj rC r6 r� r� )r0 r> r' r r r �test_removed_dir� s
z TestScandir.test_removed_dirc C s� | � � }t�|j� | �|�� � tjdkr | �|�� � | �|� � � tjdkr<| �
t |j � |�
� |j
dd� d S | �|� � d� | �
t |j
� | j
t |j
dd� d S )Nrf Fr� r )rB r rV r> r� r7 r r] r9 r; rj rC r6 r� r� rD 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 linkr� z symlink.txtr Fr� )r r rh r( r r� r> r@ r0 rV r� r6 r� r7 r9 r] r; rj rC r� )r0 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 �Nr� rc r s file.txt)
r( r rX r> r r( r= r: r r@ )r0 �
path_bytesr/ r' r r r r� � s
�zTestScandir.test_bytesc
C s� | � d� ttfD ]\}|t�| j��}| �t�� tt� |��}W d � n1 s*w Y | �
t |�d|� |d }| �
|j d� | �
|jt�tj�
| jd��� | �t|j �t� | �t|j�t� q d S rL )r( r� r� r rX r> � assertWarnsr� r r( r= r: r r@ r� r| rM )r0 r| rM r/ r' r r r �test_bytes_like� s
���zTestScandir.test_bytes_likez.fd support for listdir required for this test.c C sp | � tjtj� | �d� dg}t�� r%t�dtj� | jd�� |�
d� t� | jtj �}z�t�|�� }t
|�}W d � n1 sBw Y dd� |D �}| �t|�|� | �|t�|�� |D ]H}| �|j|j� | �t�|�|j� | �|�� |jdk� tjtjv r�tj|j|d�}| �|�� |� tj|j|dd�}| �|jdd�|� qaW t�|� d S t�|� w ) Nr� r
c S r
r r r& r r r r � r z'TestScandir.test_fd.<locals>.<listcomp>r� F)r� rf r� )r� r r( r� r( r r r� r> r@ r$ r$ r� r r= r� r r rF r; r� r� r\ )r0 Zexpected_namesr~ r� r/ r. r' rH r r r �test_fd� s4
� ��
zTestScandir.test_fdc C s | � ttjd� d S )Nry )rj rC r r( rW r r r �test_empty_path r� zTestScandir.test_empty_pathc C sN | � d� t�| j�}t|�}| �t|�d|� t|�}| �t|�d|� d S )Nr� rc r )r( r r( r> r r= r: )r0 r� r/ Zentries2r r r �test_consume_iterator_twice s
z'TestScandir.test_consume_iterator_twicec C s$ di g fD ]
}| � ttj|� qd S )NgX9��v��?)rj rp r r( )r0 �objr r r �test_bad_path_type s �z TestScandir.test_bad_path_typec C sh | � d� | � d� t�| j�}t|� |�� |�� | �� �
~W d � d S 1 s-w Y d S �Nr� � file2.txt)r( r r( r> r� r\ �check_no_resource_warning�r0 r� r r r �
test_close s
"�zTestScandir.test_closec C sx | � d� | � d� t�| j�� }t|� W d � n1 s w Y | �� �
~W d � d S 1 s5w Y d S rU )r( r r( r> r� rW rX r r r �test_context_manager& s
�
"�z TestScandir.test_context_managerc C sV | � d� | � d� t�| j��}t|� |�� W d � d S 1 s$w Y d S rU )r( r r( r> r� r\ rX r r r �test_context_manager_close. s
"�z&TestScandir.test_context_manager_closec C s� | � d� | � d� | �t��& t�| j��}t|� dd W d � n1 s)w Y W d � n1 s8w Y | �� �
~W d � d S 1 sNw Y d S )Nr� rV rc r )r( rj �ZeroDivisionErrorr r( r> r� rW rX r r r � test_context_manager_exception5 s
�� �
"�z*TestScandir.test_context_manager_exceptionc C s� | � d� | � d� t�| j�}t|� | �t��
~t�� W d � n1 s)w Y t�| j�}t |� | �
� �
~W d � d S 1 sIw Y d S rU ) r( r r( r> r� rN �ResourceWarningr �
gc_collectr rW rX r r r �test_resource_warning? s
�
"�z!TestScandir.test_resource_warningN)r� )&rP rQ rR r rW rX r( r0 r4 r$ r) r< r? r@ rB rC rE rG rH rI rJ rK r� rO r r r r r� rP rQ rR rT rY rZ r[ r] r` r r r r r* � sB
)
�
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 � dD ] }| � || �|�� qd S )N)r� s goodbyes some/path/and/file�r= rF )r0 �br r r �test_return_bytesT � �z TestPEP519.test_return_bytesc C rb )N)ZhelloZgoodbyezsome/path/and/filerc )r0 r r r r �test_return_stringX rf z TestPEP519.test_return_stringc C sL dD ]!}t |�}| �|| �|�� | �dt�|�� | �dt�|�� qd S )N)�path/like/object� path/like/objectri rh )r r= rF r rX rN )r0 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= rF r r] �
issubclassr �PathLiker# rW r r r r� d s zTestPEP519.test_pathlikec C s4 t ddi �}tt t|� fD ]
}| �t| j|� q
d S )NZblahr )r| r� r rj rp rF )r0 Zvapor�or r r � test_garbage_in_exception_outi s �z(TestPEP519.test_garbage_in_exception_outc C s | � t| j� d S r+ )rj rp rF rW r r r �test_argument_requiredn r� z!TestPEP519.test_argument_requiredc C sR | � t| jtd�� tddi �}d|_| � t| j|� � | � t| jtt� �� d S )Nr� Zfoor rc )rj rp rF r r| �
__fspath__r\ )r0 �cr r r �test_bad_pathlikeq s
�z TestPEP519.test_bad_pathlikec C s8 G dd� dt j�}| �tt|�� | �ttt j�� d S )Nc @ r� )z0TestPEP519.test_pathlike_subclasshook.<locals>.ANr� r r r r �A r� rs )r rl r� rk r r] )r0 rs r r r �test_pathlike_subclasshook| s z%TestPEP519.test_pathlike_subclasshookc C s | � tjt tj� d S r+ )r- r rl rM �types� GenericAliasrW r r r �test_pathlike_class_getitem� r� z&TestPEP519.test_pathlike_class_getitemN)rP rQ rR r� r rF re rg rj r� rn ro rr rt rw r r r r ra N s
ra c @ r� )�
TimesTestsc C sr t �� }| �|t j� dD ]
}t||�}| �|t� q
t jdkr7| �|jd� | �|j d� | �|j
d� d S d S )N)�user�system�
children_user�children_system�elapsedrf r ) r rd r- � times_resultr r r r= r{ r| r} )r0 rd Zfieldr& r r r �
test_times� s
�zTimesTests.test_timesN)rP rQ rR r r r r r rx � r� rx �forkc @ r� )� 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)rP rQ rR r� r r r r r� � r� r� �_fspathc @ s e Zd ZdZeej�ZdS )�TestPEP519PurePythonz>Explicitly test the pure Python implementation of os.fspath().N)rP rQ rR r r� r r� rF r r r r r� � s r� �__main__)r" r+ )�r� �
contextlibr� rE r� r� rF r� r~ r r r� rX ry rY r� r� r� r8 Z sysconfigr; rU ro ru r r} r� �testr Z test.supportr r r r r r9 r r� � simplefilterr� rD rW r� � ImportErrorr� r� �pwdZgetpwallr� r Z _testcapir r r� Ztest.support.script_helperr r Ztest.support.os_helperr r� r r r �versionr� r� �getgidr? r r( r� r� ZTestCaser* rS r� rB r� ZBasicTestMappingProtocolr� r r rA rV r] r` rz r� r� r� r� Zget_config_varZOS_URANDOM_DONT_USE_FDr� �contextmanagerr� r� r� r� r& r4 rP r` r� r� r� r� r� r� r� r� r� r� r r� r9 r? rX rV rC ru r� Zrequires_linux_versionr� r� r� r� r� r� r� r r r r! r* ra rx r� r� rP �mainr r r r �<module> s�
� � � ���
�E Z e _ ) EG
^;) - � �� �M �h#g%CGs/< 0!1 H 4�V O� Q0
Q]
y B� \:
�