File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_posix.cpython-39.pyc
a
��h D � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z e
�rNddl Z ddl Z ddl m Z d,dd�Z!dd� Z"d d � Z#d d � Z$d d!� Z%d"d#� Z&ej'j(e
d$d%�G d&d'� d'e��Z)ej'j(e
d$d%�G d(d)� d)e��Z*ej'j(e
d$d%�G d*d+� d+e��Z+dS )-zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�AARCH64)�HAS_NET_IO_COUNTERS)�
PYTHON_EXE)� QEMU_USER)�PsutilTestCase)�mock)�pytest)�retry_on_failure)�sh)�skip_on_access_denied)�spawn_testproc)� terminate)�which)� getpagesizec C s� dg}t r|�d� |dur0|�dt|�g� n ts8trD|�d� n
|�d� tr^| �dd�} |�d | g� t|�}t r�|�� n|�� d
d� }g }|D ]6}|� � }z t
|�}W n t y� Y n0 |�|� q�|du r�|S |d S dS ) zwWrapper for calling the ps command with a little bit of cross-platform
support for a narrow range of features.
�psz --no-headersNz-pz-A�ax�startZstimez-o� r ) r �append�extend�strr r �replacer �
splitlines�strip�int�
ValueError)�fmt�pid�cmd�output�
all_output�line� r) �C/usr/local/lib64/python3.9/site-packages/psutil/tests/test_posix.pyr / s0
r c C s> d}t r d}t|| ��� }tr6d|d v s.J �|d S |d S )N�commandZcommz
/bin/qemu-r r )r r �splitr )r$ �fieldr+ r) r) r* �ps_nameg s r. c C s0 d}t s trd}t|| �}t�dd|�}|�� S )Nr+ �argsz\(python.*?\)$� )r r r �re�subr )r$ r- �outr) r) r* �ps_argsr s
r4 c C s d}t r d}t|| �S )NZrssZrssize�r r �r$ r- r) r) r* �ps_rss| s r7 c C s d}t r d}t|| �S )NZvszZvsizer5 r6 r) r) r* �ps_vsz� s r8 c
C s� zt d| ��� }W n@ tyT } z(dt|��� v r>t�d��� W Y d }~n
d }~0 0 |�d�d }|�� }t|d �d }t|d �d }t|d �d }t |d �
d
d ��}||||fS ) Nzdf -k %sz device busyzdf returned EBUSY�
r � � � � �%r0 ) r r � RuntimeErrorr �lowerr �skipr, r! �floatr ) �devicer3 �errr( �fields� sys_total�sys_used�sys_free� sys_percentr) r) r* �df� s
rJ z
POSIX only��reasonc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z e � e
� dd� ��Ze � e
� dd� ��Zdd� Zdd� Zdd� Zdd� Zejjep�ed d �d d � �Zd d!� Ze
� d"d#� �Zejje d$d �ejje d%d �d&d'� ��Z d(S ))� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s t tddgtjd�j| _d S )Nz-Ez-O)�stdin)r r �
subprocess�PIPEr$ ��clsr) r) r* �
setUpClass� s �zTestProcess.setUpClassc C s t | j� d S �N)r r$ rQ r) r) r* �
tearDownClass� s zTestProcess.tearDownClassc C s, t d| j�}t�| j��� }||ks(J �d S )N�ppid)r r$ �psutil�ProcessrV )�selfZppid_psZ ppid_psutilr) r) r* � test_ppid� s zTestProcess.test_ppidc C s. t d| j�}t�| j��� j}||ks*J �d S )N�uid)r r$ rW rX �uids�real)rY Zuid_psZ
uid_psutilr) r) r* �test_uid� s zTestProcess.test_uidc C s. t d| j�}t�| j��� j}||ks*J �d S )NZrgid)r r$ rW rX �gidsr] )rY Zgid_psZ
gid_psutilr) r) r* �test_gid� s zTestProcess.test_gidc C s, t d| j�}t�| j��� }||ks(J �d S )N�user)r r$ rW rX �username)rY Z username_psZusername_psutilr) r) r* �
test_username� s zTestProcess.test_usernamec C s^ t �� }tjdtd��4}|�� t|�� j�ks2J �|j s<J �W d � n1 sP0 Y d S )Nzpsutil.pwd.getpwuid�Z side_effect)
rW rX r �patch�KeyErrorrb r r\ r] �called)rY �pZfunr) r) r* �test_username_no_resolution� s z'TestProcess.test_username_no_resolutionc C s< t �d� t| j�}t�| j��� d d }||ks8J �d S )N皙�����?r r: )�time�sleepr7 r$ rW rX � memory_info)rY Zrss_psZ
rss_psutilr) r) r* �test_rss_memory� s
zTestProcess.test_rss_memoryc C s< t �d� t| j�}t�| j��� d d }||ks8J �d S )Nrj r r: )rk rl r8 r$ rW rX rm )rY Zvsz_psZ
vsz_psutilr) r) r* �test_vsz_memory� s
zTestProcess.test_vsz_memoryc C sv t | j�}tj�|��� }t�| j��� �� }t �
dd|�}t �
dd|�}t �
dd|�}t �
dd|�}||ksrJ �d S )Nz\d.\dr0 z\d) r. r$ �os�path�basenamer@ rW rX �namer1 r2 )rY Zname_psZ name_psutilr) r) r* � test_name� s
zTestProcess.test_namec C s� d}g d�}t jd|d��V t jd|d��( t�� }|�� dksDJ �W d � n1 sX0 Y W d � n1 sv0 Y d S )N�long-program-name)�long-program-name-extendedZfoo�bar� psutil._psplatform.Process.name�Z return_value�"psutil._psplatform.Process.cmdlinerv )r re rW rX rs )rY rs �cmdlinerh r) r) r* �test_name_long� s �zTestProcess.test_name_longc C s� d}t jd|d��^ t jdt�dd�d��( t�� }|�� dksDJ �W d � n1 sX0 Y W d � n1 sv0 Y d S �Nru rx ry rz r r0 rd )r re rW Z AccessDeniedrX rs �rY rs rh r) r) r* � test_name_long_cmdline_ad_exc� s
�z)TestProcess.test_name_long_cmdline_ad_excc
C s� d}t jd|d��� t jdt�dd�d��L t�� }t�tj�� |�� W d � n1 s^0 Y W d � n1 s|0 Y W d � n1 s�0 Y d S r} )r re rW Z
NoSuchProcessrX r �raisesrs r~ r) r) r* � test_name_long_cmdline_nsp_exc s
�z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablerK c C s\ t d| j�}t�| j��� }tj�|��d�}t|�}tj�|��d�}|||fv sXJ �d S )Nr z%H:%M:%S) r r$ rW rX Z create_time�datetime�
fromtimestamp�strftime�round)rY Ztime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr) r) r* �test_create_time s ����z TestProcess.test_create_timec C s^ t | j�}t�| j��� }z||ks(J �W n. tyX |d t|�� }||ksTJ �Y n0 d S rT )r. r$ rW rX Zexe�AssertionError�len)rY Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer) r) r* �test_exe s
zTestProcess.test_exec C sT t | j�}d�t�| j��� �}trDt|�t|�k rD|�|�sPJ �n ||ksPJ �d S )N� ) r4 r$ �joinrW rX r{ r r� �
startswith)rY Z
ps_cmdlineZpsutil_cmdliner) r) r* � test_cmdline1 s
zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}t�� �� }||ks$J �d S )N�nice)r r$ rW rX r� )rY Zps_niceZ psutil_nicer) r) r* � test_nice? s zTestProcess.test_niceN) �__name__�
__module__� __qualname__�__doc__� classmethodrS rU rZ r^ r` rc ri r r rn ro rt r| r r� r �mark�skipifr r r� r� r� r r r� r) r) r) r* rM � s8
rM c @ s� e Zd ZdZe� dd� �Zejje dd�ejje
d� dd�ejje dd�d d
� ���Z e� d d � �Z
e� d
d� �Zdd� Zdd� Zdd� Zdd� Zejjedd�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� t�� �ts$tr0d� vr0� �dd� t� �t�� dkrz� fdd��D ��fdd�� D � }| �dt |� ��d S )Nr$ r r c s g | ]}|� vr|�qS r) r) ��.0�x)�pids_psr) r* �
<listcomp>X � z,TestSystemAPIs.test_pids.<locals>.<listcomp>c s g | ]}|� vr|�qS r) r) r� )� pids_psutilr) r* r� X s z difference: )
�sortedr rW Zpidsr r �insertr� �failr )rY �
differencer) )r� r� r* � test_pidsK s �zTestSystemAPIs.test_pidszunreliable on SUNOSrK �ifconfigzno ifconfig cmdz
not supportedc C sL t d�}tjdd�D ]2}|�� D ]}|�|�r qq | �d||f ��qd S )Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output
%s)r rW Znet_io_countersr, r� r� )rY r& Znicr( r) r) r* �test_nic_names_ s
��z TestSystemAPIs.test_nic_namesc C s� t d�}|�� st�d��|�d�}dd� |D �}dd� |D �}t|�tt�� �ksXJ �| jt�� |d��` t t�� �D ]B\}}|j
|| ks�J �|j || ks�J �|j d urxt�
|j � qxW d � n1 s�0 Y d S )N�who -u�no users on this systemr9 c S s g | ]}|� � d �qS )r �r, r� r) r) r* r� v r� z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|� � d �qS )r r� r� r) r) r* r� w r� �rW Zwho)r r r rA r, r� rW �users�subTest� enumeraters Zterminalr$ rX )rY r3 �linesr� Z terminals�idx�ur) r) r* �
test_usersp s
zTestSystemAPIs.test_usersc C s t d�}|�� st�d��d }t�d|�}|r4d}nNt�d|�}|rJd}n8t�d|�}|r`d}n"t�d |�}|r�d}d
d � |D �}|s�t�d | ��| jt�� |d
��J t t�� �D ],\}}t
j
� |j ��
|�}||| ks�J �q�W d � n1 s�0 Y d S )Nr� r� z \d\d\d\d-\d\d-\d\d \d\d:\d\dz%Y-%m-%d %H:%Mz [A-Z][a-z][a-z] \d\d \d\d:\d\dz %b %d %H:%Mz[A-Z][a-z][a-z] \d\dz%b %dz[a-z][a-z][a-z] \d\dc S s g | ] }|� � �qS r) )�
capitalizer� r) r) r* r� � r� z5TestSystemAPIs.test_users_started.<locals>.<listcomp>z(cannot interpret tstamp in who output
%sr� )r r r rA r1 �findallr� rW r� r� r� r� �startedr� )rY r3 Ztstampr� r� r� Z psutil_valuer) r) r* �test_users_started� s:
���z!TestSystemAPIs.test_users_startedc C s~ t jdttjd�d��T}t�t�� tj� t
� � � W d � n1 sH0 Y |j s\J �W d � n1 sp0 Y d S )Nzpsutil._psposix.os.killr0 rd )
r re �OSError�errno�EBADFr r� rW �_psposixZ
pid_existsrp �getpidrg �rY �mr) r) r* �test_pid_exists_let_raise� s � .z(TestSystemAPIs.test_pid_exists_let_raisec C s~ t jdttjd�d��T}t�t�� tj� t
� � � W d � n1 sH0 Y |j s\J �W d � n1 sp0 Y d S )N�psutil._psposix.os.waitpidr0 rd )
r re r� r� r� r r� rW r� �wait_pidrp r� rg r� r) r) r* �test_os_waitpid_let_raise� s � .z(TestSystemAPIs.test_os_waitpid_let_raisec C s� t jdttjd�d��\}t�tjj ��$ tjj
t � � dd� W d � n1 sP0 Y |j
sdJ �W d � n1 sx0 Y d S )Nr� r0 rd g{�G�z�?)�timeout)r re r� r� ZEINTRr r� rW r� �TimeoutExpiredr� rp r� rg r� r) r) r* �test_os_waitpid_eintr� s �2z$TestSystemAPIs.test_os_waitpid_eintrc C sv t jddd��T}t�t�� tj�t� � � W d � n1 s@0 Y |j
sTJ �W d � n1 sh0 Y d S )Nr� )r ���ry ) r re r r� r" rW r� r� rp r� rg r� r) r) r* � test_os_waitpid_bad_ret_status� s � .z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc C s� d}t jdd�D ]�}t �|j�}zt|j�\}}}}W nV ty� } z>t|��� }d|v sjd|v sjd|v rvW Y d }~q� W Y d }~qd }~0 0 t |j
| �|k s�J �t |j | �|k s�J �t |j | �|k s�J �t |j
| �dksJ �qd S )Ni @ F)�allzno such file or directoryzraw devices not supportedzpermission deniedr )rW Zdisk_partitions�
disk_usageZ
mountpointrJ rC r? r r@ �abs�total�used�free�percent) rY Z tolerance�part�usagerF rG rH rI rD r) r) r* �test_disk_usage� s& ��� z TestSystemAPIs.test_disk_usageN)r� r� r� r� r r� r r� r� r r r
r� r� r� r� r� r� r� r r� r) r) r) r* r� G s$
&
r� c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }|dksJ �|t� � ks"J �|tjks0J �d S )Nr )r �resource�mmapZPAGESIZE)rY Zpagesizer) r) r* �test_getpagesize� s zTestMisc.test_getpagesizeN)r� r� r� r� r) r) r) r* r� � s r� )N),r� r� r� rp r1 rO rk rW r r r r r r r Z psutil.testsr r
r r r
r r r r r r r r r� r� Zpsutil._psutil_posixr r r. r4 r7 r8 rJ r� r� rM r� r� r) r) r) r* �<module> sZ
8
- #