File: //opt/osm/venv/lib64/python3.10/site-packages/psutil/tests/__pycache__/test_posix.cpython-310.pyc
o
=i�C � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
Z
ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z er�ddl Z ddl Z d)dd�Z dd� Z dd� Z!dd � 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)�mock)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�AARCH64)�HAS_NET_IO_COUNTERS)�
PYTHON_EXE)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�skip_on_access_denied)�
spawn_subproc)� terminatec C s� dg}t r
|�d� |dur|�dt|�g� nts tr"|�d� n|�d� tr/| �dd�} |�d | g� t|�}t r@|�� n|�� d
d� }g }|D ]}|� � }zt
|�}W n t ya Y nw |�|� qL|du rn|S |d S ) 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& �E/opt/osm/venv/lib/python3.10/site-packages/psutil/tests/test_posix.pyr * s2
� r c C s" d}t rd}t|| ��� }|d S )N�commandZcommr )r r �split)r! �fieldr( r& r& r' �ps_namea s
r+ c C s0 d}t strd}t|| �}t�dd|�}|�� S )Nr( �argsz\(python.*?\)$� )r r r �re�subr )r! r* �outr& r& r' �ps_argsi s
r1 c C � d}t rd}t|| �S )NZrssZrssize�r r �r! r* r& r& r' �ps_rsss �
r5 c C r2 )NZvszZvsizer3 r4 r& r& r' �ps_vszz r6 r7 c
C s� z t d| � ���� }W n ty+ } zdt|��� v r&t�d�W Y d }~S � d }~ww |�d�d }|�� }t|d �d }t|d �d }t|d �d }t |d �
d
d ��}||||fS ) Nzdf -k z device busyzdf returned EBUSY�
r � � � � �%r- ) r r � RuntimeErrorr �lowerr �skipr) r �floatr ) �devicer0 �errr% �fields� sys_total�sys_used�sys_free� sys_percentr& r& r' �df� s �� rI 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� ZejjepPed 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! rP r& r& r' �
tearDownClass� s zTestProcess.tearDownClassc C �, t d| j�}t�| j��� }||ksJ �d S )N�ppid)r r! �psutil�ProcessrV )�selfZppid_psZ ppid_psutilr& r& r' � test_ppid� � zTestProcess.test_ppidc C �. t d| j�}t�| j��� j}||ksJ �d S )N�uid)r r! rW rX �uids�real)rY Zuid_psZ
uid_psutilr& r& r' �test_uid� � zTestProcess.test_uidc C r\ )NZrgid)r r! rW rX �gidsr_ )rY Zgid_psZ
gid_psutilr& r& r' �test_gid� ra zTestProcess.test_gidc C rU )N�user)r r! rW rX �username)rY Z username_psZusername_psutilr& r& r' �
test_username� r[ zTestProcess.test_usernamec C s` t �� }tjdtd��}|�� t|�� j�ksJ �|j s J �W d � d S 1 s)w Y d S )Nzpsutil.pwd.getpwuid�Z side_effect)
rW rX r �patch�KeyErrorre r r^ r_ �called)rY �pZfunr& r& r' �test_username_no_resolution� s
"�z'TestProcess.test_username_no_resolutionc C �< t �d� t| j�}t�| j��� d d }||ks J �d S )N皙�����?r r9 )�time�sleepr5 r! rW rX � memory_info)rY Zrss_psZ
rss_psutilr& r& r' �test_rss_memory� �
zTestProcess.test_rss_memoryc C rm )Nrn r r9 )ro rp r7 r! rW rX rq )rY Zvsz_psZ
vsz_psutilr& r& r' �test_vsz_memory� rs zTestProcess.test_vsz_memoryc C sv t | j�}tj�|��� }t�| j��� �� }t �
dd|�}t �
dd|�}t �
dd|�}t �
dd|�}||ks9J �d S )Nz\d.\dr- z\d) r+ r! �os�path�basenamer? rW rX �namer. r/ )rY Zname_psZ name_psutilr& r& r' � test_name� s
zTestProcess.test_namec C s� d}g d�}t jd|d��4 t jd|d�� t�� }|�� dks"J �W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S )N�long-program-name)�long-program-name-extendedZfoo�bar� psutil._psplatform.Process.name�� return_value�"psutil._psplatform.Process.cmdliner{ )r rh rW rX rx )rY rx �cmdlinerk r& r& r' �test_name_long� s � ��"�zTestProcess.test_name_longc C s� d}t jd|d��8 t jdt�dd�d�� t�� }|�� dks"J �W d � n1 s,w Y W d � d S W d � d S 1 sDw Y d S �Nrz r} r~ r� r r- rg )r rh rW Z AccessDeniedrX rx �rY rx rk r& r& r' � test_name_long_cmdline_ad_exc� s
� ��"�z)TestProcess.test_name_long_cmdline_ad_excc
C s� d}t jd|d��J t jdt�dd�d��& t�� }t�tj�� |�� W d � n1 s/w Y W d � n1 s>w Y W d � d S W d � d S 1 sVw Y d S r� )r rh rW Z
NoSuchProcessrX r �raisesrx r� r& r& r' � test_name_long_cmdline_nsp_exc� s
�
�� ��"�z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablerJ c C s\ t d| j�}t�| j��� }tj�|��d�}t|�}tj�|��d�}|||hv s,J �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 ||ksJ �W d S ty. |d t|�� }||ks+J �Y d S w rS )r+ r! rW rX �exe�AssertionError�len)rY Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer& r& r' �test_exe s
�zTestProcess.test_exec C sV t | j�}d�t�| j��� �}tr#t|�t|�k r#|�|�s!J �d S ||ks)J �d S )N� ) r1 r! �joinrW rX r� r
r� �
startswith)rY Z
ps_cmdlineZpsutil_cmdliner& r& r' � test_cmdline( s
zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}t�� �� }||ksJ �d S )N�nice)r r! rW rX r� )rY Zps_niceZ psutil_nicer& r& r' � test_nice6 s zTestProcess.test_niceN) �__name__�
__module__� __qualname__�__doc__� classmethodrR rT rZ r` rc rf rl r r rr rt ry r� r� r� r �mark�skipifr r r� r� r� r r r� r& r& r& r' rL � s:
rL c @ s� e Zd ZdZe� dd� �Zejje dd�ejje
� d� dd�ejje dd�d d
� ���Z
e� d d � �Ze� d
d� �Zdd� Zdd� Zdd� Zdd� Zejjedd�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� t�� �tstrd� vr� �dd� t� �t�� dkr=� fdd��D ��fdd�� D � }t� dt
|� �S d S )Nr! r r c � g | ]}|� vr|�qS r& r& ��.0�x)�pids_psr& r' �
<listcomp>O � z,TestSystemAPIs.test_pids.<locals>.<listcomp>c r� r& r& r� )� pids_psutilr& r' r� O s z difference: ) �sortedr rW Zpidsr r �insertr� r �failr )rY �
differencer& )r� r� r' � test_pidsB s ��zTestSystemAPIs.test_pidszunreliable on SUNOSrJ �ifconfigzno ifconfig cmdz
not supportedc C sT t d�}tjdd�D ] }|�� D ] }|�|�r nqt�d|� d|� �� S q
d S )Nz ifconfig -aT)Zperniczcouldn't find z nic in 'ifconfig -a' output
)r rW Znet_io_countersr) r� r r� )rY r# Znicr% r& r& r' �test_nic_namesV s
�����z TestSystemAPIs.test_nic_namesc C s$ t d�}|�� s
t�d�S g }|�� D ]>}|�� d }|�� d }ts%trJz
t|�� d �}W n t y@ t|�� d �}Y nw |�
|||f� q|�
||f� qtsVtr`dd� t � � D �}n d d� t � � D �}t
|�t
|�kssJ �t|�t|�ks}J �t � � D ]}|jd ur�|jdks�J �q�d S )
N�who -u�no users on this systemr r ������c S s g | ]
}|j |j|jf�qS r& )rx �terminalr! �r� �ur& r& r' r� y s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|j |jf�qS r& )rx r� r� r& r& r' r� { r� )r r r r@ r r) r r r r r rW �usersr� r� r! )rY r0 Zsusersr% rd r� r! Zpusersr& r& r' �
test_userse s2
�
��zTestSystemAPIs.test_usersc C s t d�}|�� s
t�d�S d }t�d|�}|rd}n't�d|�}|r%d}n t�d|�}|r0d}nt�d |�}|rAd}d
d � |D �}|sKt�d |� ��S | jt�� |d
��& t t�� �D ]\}}t
j
� |j ��
|�}||| ksqJ �q[W d � d S 1 s}w 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� � s z5TestSystemAPIs.test_users_started.<locals>.<listcomp>z&cannot interpret tstamp in who output
)rW Zwho)r r r r@ r. �findallZsubTestrW r� � enumerater� r� �startedr� )rY r0 Ztstampr� �idxr� Z psutil_valuer& r& r' �test_users_started� s:
���"�z!TestSystemAPIs.test_users_startedc C � t jdttjd�d��+}t�t�� tj� t
� � � W d � n1 s$w Y |j s.J �W d � d S 1 s9w Y d S )Nzpsutil._psposix.os.killr- rg )
r rh �OSError�errno�EBADFr r� rW �_psposixZ
pid_existsru �getpidrj �rY �mr& r& r' �test_pid_exists_let_raise� s � � "�z(TestSystemAPIs.test_pid_exists_let_raisec C r� )N�psutil._psposix.os.waitpidr- rg )
r rh r� r� r� r r� rW r� �wait_pidru r� rj 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 s(w Y |j
s2J �W d � d S 1 s=w Y d S )Nr� r- rg g{�G�z�?)�timeout)r rh r� r� ZEINTRr r� rW r� �TimeoutExpiredr� ru r� rj r� r& r& r' �test_os_waitpid_eintr� s � � "�z$TestSystemAPIs.test_os_waitpid_eintrc C sx t jddd��+}t�t�� tj�t� � � W d � n1 s w Y |j
s*J �W d � d S 1 s5w Y d S )Nr� )r r� r~ ) r rh r r� r rW r� r� ru r� rj 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 ]e}t �|j�}z t|j�\}}}}W n% ty@ } zt|��� }d|v s5d|v s5d|v r;W Y d }~q� d }~ww t |j
| �|k sLJ �t |j | �|k sWJ �t |j | �|k sbJ �t |j
| �dksmJ �qd S )Ni @ F)�allzno such file or directoryzraw devices not supportedzpermission deniedr )rW Zdisk_partitions�
disk_usageZ
mountpointrI rB r> r r? �abs�total�used�free�percent) rY Z tolerance�part�usagerE rF rG rH rC r& r& r' �test_disk_usage� s& ��
�z TestSystemAPIs.test_disk_usageN)r� r� r� r� r r� r r� r� r �shutil�whichr r� r� r� r� r� r� r� r r� r& r& r& r' r� > s&
$
r� c @ s e Zd Zdd� ZdS )�TestMiscc C s: t jj�� }|dks J �|t�� ksJ �|tjksJ �d S )Nr )rW Z _psplatformZcextZ getpagesize�resource�mmap�PAGESIZE)rY Zpagesizer& r& r' �test_getpagesize� s zTestMisc.test_getpagesizeN)r� r� r� r� r& r& r& r' r� � s r� rS )*r� r� r� ru r. r� rN ro Zunittestr rW r r r r r r r Z psutil.testsr
r r r
r r r r r r r� r� r r+ r1 r5 r7 rI r� r� rL r� r� r& r& r& r' �<module> sX
7
- .