U
�=�_'; � @ 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 e
�rBddl Z ddl Z ddl m Z d*dd�Z dd� Z!dd � Z"d d � Z#d d � Z$e�%e
d!�G d"d#� d#e��Z&e�%e
d!�G d$d%� d%e��Z'e�%e
d!�G d&d'� d'e��Z(e)d(k�r�dd)l*m+Z+ e+e,� dS )+zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�
CI_TESTING)�spawn_testproc)�HAS_NET_IO_COUNTERS)�mock)�PsutilTestCase)�
PYTHON_EXE)�retry_on_failure)�sh)�skip_on_access_denied)� terminate)�unittest)�which)� getpagesizec C s� dg}t r|�d� |dk r0|�dt|�g� n ts8trD|�d� n
|�d� trftd�}|�| | �} |�d| g� t|�}t r�|� � }n|� � d d� }g }|D ]8}|�
� }z t |�}W n t k
r� Y nX |�|� q�|dkr�|S |d
S dS ) z|
Wrapper for calling the ps command with a little bit of cross-platform
support for a narrow range of features.
�psz --no-headersNz-pz-AZax)�command�comm�startZstimez-o� r )
r �append�extend�strr r �set�getr �
splitlines�strip�int�
ValueError)Zfmt�pid�cmdZfmt_map�outputZ
all_output�line� r( �J/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_posix.pyr - s6
r c C s d}t r d}t|| ��� d S )Nr r r )r r �split�r$ Zfieldr( r( r) �ps_namei s r, c C s d}t s trd}t|| �S )Nr �args)r r r r+ r( r( r) �ps_argsp s r. c C s d}t r d}t|| �S )NZrssZrssize�r r r+ r( r( r) �ps_rssw s r0 c C s d}t r d}t|| �S )NZvszZvsizer/ r+ r( r( r) �ps_vsz~ s r1 z
POSIX onlyc @ 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� Ze�ep�ed �d d � �Zd d � Zd!d"� Ze�ed#�e�e 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$ r6 r( r( r) �
tearDownClass� s zTestProcess.tearDownClassc C s, t d| j�}t�| j��� }| �||� d S )N�ppid)r r$ �psutil�Processr; � assertEqual)�selfZppid_psZ ppid_psutilr( r( r) � test_ppid� s zTestProcess.test_ppidc C s. t d| j�}t�| j��� j}| �||� d S )NZuid)r r$ r< r= �uids�realr> )r? Zuid_psZ
uid_psutilr( r( r) �test_uid� s zTestProcess.test_uidc C s. t d| j�}t�| j��� j}| �||� d S )NZrgid)r r$ r< r= ZgidsrB r> )r? Zgid_psZ
gid_psutilr( r( r) �test_gid� s zTestProcess.test_gidc C s, t d| j�}t�| j��� }| �||� d S )N�user)r r$ r< r= �usernamer> )r? Z username_psZusername_psutilr( r( r) �
test_username� s zTestProcess.test_usernamec C sJ t �� }tjdtd��*}| �|�� t|�� j �� |j
s<t �W 5 Q R X d S )Nzpsutil.pwd.getpwuid�Z side_effect) r< r= r �patch�KeyErrorr> rF r rA rB �called�AssertionError)r? �pZfunr( r( r) �test_username_no_resolution� s z'TestProcess.test_username_no_resolutionc C s< t �d� t| j�}t�| j��� d d }| �||� d S )N皙�����?r � )�time�sleepr0 r$ r< r= � memory_infor> )r? Zrss_psZ
rss_psutilr( r( r) �test_rss_memory� s
zTestProcess.test_rss_memoryc C s< t �d� t| j�}t�| j��� d d }| �||� d S )NrO r rP )rQ rR r1 r$ r< r= rS r> )r? 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|�}| � ||� d S )Nz\d.\d� z\d) r, r$ �os�path�basename�lowerr< r= �name�re�subr> )r? Zname_psZ name_psutilr( r( r) � test_name� s
zTestProcess.test_namec
C s^ d}dddg}t jd|d��8 t jd|d�� t�� }| �|�� d� W 5 Q R X W 5 Q R X d S )N�long-program-namezlong-program-name-extendedZfooZbar� psutil._psplatform.Process.name�Z return_value�"psutil._psplatform.Process.cmdline)r rI r< r= r> r[ )r? r[ �cmdlinerM r( r( r) �test_name_long� s
��zTestProcess.test_name_longc
C s\ d}t jd|d��@ t jdt�dd�d�� t�� }| �|�� d� W 5 Q R X W 5 Q R X d S �Nr_ r` ra rb r rV rH )r rI r< Z AccessDeniedr= r> r[ �r? r[ rM 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�� t�� }| �tj|j� W 5 Q R X W 5 Q R X d S re )r rI r< Z
NoSuchProcessr= � assertRaisesr[ rf r( r( r) � test_name_long_cmdline_nsp_exc� s �
�z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablec C s\ t d| j�}t�| j��� }tj�|��d�}t|�}tj�|��d�}| � |||g� d S )Nr z%H:%M:%S)
r r$ r< r= Z create_time�datetimeZ
fromtimestamp�strftime�round�assertIn)r? 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| �||� W n0 tk
rZ |d t|�� }| �||� Y nX d S r9 )r, r$ r< r= Zexer> rL �len)r? Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer( r( r) �test_exe s
zTestProcess.test_exec C s0 t | j�}d�t�| j��� �}| �||� d S )N� )r. r$ �joinr< r= rc r> )r? Z
ps_cmdlineZpsutil_cmdliner( r( r) � test_cmdline s
zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}t�� �� }| �||� d S )N�nice)r r$ r< r= rt r> )r? Zps_niceZ psutil_nicer( r( r) � test_nice s zTestProcess.test_niceN) �__name__�
__module__� __qualname__�__doc__� classmethodr8 r: r@ rC rD rG rN r r rT rU r^ rd rg ri r �skipIfr r rn rp rs r r ru r( r( r( r) r2 � s6
r2 c @ s� e Zd ZdZe� dd� �Ze�ed�e�e d� d�e�e
d�dd � ���Z e�e o^e
�� d
�e� d d � ��Zd
d� Zdd� Zdd� Zdd� Ze�ed�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� t�� �ts$tr0d� kr0� �dd� t� �t�� dkrz� fdd��D ��fdd�� D � }| �dt |� � d S )Nr$ r r c s g | ]}|� kr|�qS r( r( ��.0�x)�pids_psr( r) �
<listcomp>4 s z,TestSystemAPIs.test_pids.<locals>.<listcomp>c s g | ]}|� kr|�qS r( r( r} )� pids_psutilr( r) r� 5 s z difference: )
�sortedr r< Zpidsr r �insertro �failr )r? �
differencer( )r� r� r) � test_pids' s �zTestSystemAPIs.test_pidszunreliable on SUNOSZifconfigzno ifconfig cmdz
not supportedc C sP 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 r< Znet_io_counters�keysr* �
startswithr� )r? r& Znicr' r( r( r) �test_nic_names: s
��z TestSystemAPIs.test_nic_nameszunreliable on CIc C s� t d�}|�� s| �d��|�d�}dd� |D �}dd� |D �}| �t|�tt�� �� t�� D ] }| �|j |� | �|j
|� q`d S )NZwhozno users on this system�
c S s g | ]}|� � d �qS )r �r* r} r( r( r) r� O s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|� � d �qS )r r� r} r( r( r) r� P s ) r r! ZskipTestr* r> ro r< �usersrm r[ Zterminal)r? �out�linesr� Z terminals�ur( r( r) �
test_usersH s
zTestSystemAPIs.test_usersc C sF t jdttjd�d��&}| �ttjjt �
� � |j s8t �W 5 Q R X d S )Nzpsutil._psposix.os.killrV rH )
r rI �OSError�errno�EBADFrh r< �_psposixZ
pid_existsrW �getpidrK rL �r? �mr( r( r) �test_pid_exists_let_raiseV s
�z(TestSystemAPIs.test_pid_exists_let_raisec C sF t jdttjd�d��&}| �ttjjt �
� � |j s8t �W 5 Q R X d S )N�psutil._psposix.os.waitpidrV rH )
r rI r� r� r� rh r< r� �wait_pidrW r� rK rL r� r( r( r) �test_os_waitpid_let_raise_ s
�z(TestSystemAPIs.test_os_waitpid_let_raisec C sN t jdttjd�d��.}| jtjjtjj t
� � dd� |j s@t
�W 5 Q R X d S )Nr� rV rH g{�G�z�?)Ztimeout)r rI r� r� ZEINTRrh r< r� ZTimeoutExpiredr� rW r� rK rL r� r( r( r) �test_os_waitpid_eintrg s
� �z$TestSystemAPIs.test_os_waitpid_eintrc C s> t jddd��&}| �ttjjt�� � |j s0t
�W 5 Q R X d S )Nr� )r ���ra ) r rI rh r# r< r� r� rW r� rK rL r� r( r( r) � test_os_waitpid_bad_ret_statusp s � �z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc
C s� dd� }d}t jdd�D ]�}t �|j�}z||j�\}}}}W nR tk
r� } z4t| ��� } d| kstd| kstd| kr~W Y �
qn� W 5 d } ~ X Y qX | j|j ||d � | j|j
||d � | j|j ||d � | j|j |d
d � qd S ) Nc S sv t d| ��� }|�d�d }|�� }t|d �d }t|d �d }t|d �d }t|d �dd ��}||||fS )
Nzdf -k %sr� r rP � � � �%rV )r r! r* r" �float�replace)�devicer� r' Zfields�total�used�free�percentr( r( r) �df| s z*TestSystemAPIs.test_disk_usage.<locals>.dfi @ F)�allzno such file or directoryzraw devices not supportedzpermission denied)Zdeltar )
r< Zdisk_partitionsZ
disk_usageZ
mountpointr� � RuntimeErrorr rZ ZassertAlmostEqualr� r� r� r� )
r? r� Z tolerance�partZusager� r� r� r� �errr( r( r) �test_disk_usagey s&
��
z TestSystemAPIs.test_disk_usageN)rv rw rx ry r r� r r{ r r r r� r r< r� r� r� r� r� r� r r� r( r( r( r) r| # s"
r| c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }| �|d� | �|t� � � | �|tj� d S )Nr )r Z
assertGreaterr> �resource�mmapZPAGESIZE)r? Zpagesizer( r( r) �test_getpagesize� s zTestMisc.test_getpagesizeN)rv rw rx r� r( r( r( r) r� � s r� �__main__)�
run_from_name)N)-ry rj r� rW r\ r4 rQ r< r r r r r r r Z psutil.testsr r
r r r
r r r r r r r r� r� Zpsutil._psutil_posixr r r, r. r0 r1 r{ r2 r| r� rv Zpsutil.tests.runnerr� �__file__r( r( r( r) �<module> sZ
< y