File: //opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/__pycache__/test_posix.cpython-38.pyc
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)�fmt�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% �fieldr) 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 r0 c C s d}t r d}t|| �S )NZrssZrssize�r r r, r) r) r* �ps_rssw s r2 c C s d}t r d}t|| �S )NZvszZvsizer1 r, r) r) r* �ps_vsz~ s r3 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% r8 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 )N�uid)r r% r> r? �uids�realr@ )rA 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? ZgidsrE r@ )rA 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@ )rA 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�� side_effect) r> r? r �patch�KeyErrorr@ rI r rD rE �called�AssertionError)rA �p�funr) 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�sleepr2 r% r> r? � memory_infor@ )rA 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 )NrT r rU )rV rW r3 r% r> r? rX r@ )rA 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@ )rA 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 rM r> r? r@ r` )rA r` �cmdlinerQ 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 �Nrd re rf rg r r[ rK )r rM r> � AccessDeniedr? r@ r` �rA r` rQ 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 rj )r rM r> �
NoSuchProcessr? � assertRaisesr` rl 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�datetime�
fromtimestamp�strftime�round�assertIn)rA 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 r; )r. r% r> r? Zexer@ rP �len)rA Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer) r) r* �test_exe s
zTestProcess.test_exec C s0 t | j�}d�t�| j��� �}| �||� d S )N� )r0 r% �joinr> r? rh r@ )rA 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? r| r@ )rA Zps_niceZ psutil_nicer) r) r* � test_nice s zTestProcess.test_niceN) �__name__�
__module__� __qualname__�__doc__� classmethodr: r< rB rF rG rJ rS r r rY rZ rc ri rm rp r �skipIfr r rv rx r{ r r r} r) r) r) r* r4 � s6
r4 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 �insertrw �failr )rA �
differencer) )r� r� r* � test_pids' s �zTestSystemAPIs.test_pidszunreliable on SUNOS�ifconfigzno 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� )rA 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 )N�whozno 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@ rw r> �usersru r` �terminal)rA �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.killr[ rK )
r rM �OSError�errno�EBADFro r> �_psposixZ
pid_existsr\ �getpidrO rP �rA �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.waitpidr[ rK )
r rM r� r� r� ro r> r� �wait_pidr\ r� rO rP 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� r[ rK g{�G�z�?)�timeout)r rM r� r� ZEINTRro r> r� �TimeoutExpiredr� r\ r� rO rP 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 ���rf ) r rM ro r# r> r� r� r\ r� rO rP 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 rU � � � �%r[ )r r! r+ r" �float�replace)�devicer� r( �fields�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)�deltar )
r> Zdisk_partitions�
disk_usage�
mountpointr� � RuntimeErrorr r_ ZassertAlmostEqualr� r� r� r� )
rA r� Z tolerance�part�usager� r� r� r� �errr) r) r* �test_disk_usagey 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* r� # s"
r� c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }| �|d� | �|t� � � | �|tj� d S )Nr )r Z
assertGreaterr@ �resource�mmapZPAGESIZE)rA Zpagesizer) r) r* �test_getpagesize� s zTestMisc.test_getpagesizeN)r~ r r� r� r) r) r) r* r� � s r� �__main__)�
run_from_name)N)-r� rq r� r\ ra r6 rV 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. r0 r2 r3 r� r4 r� r� r~ Zpsutil.tests.runnerr� �__file__r) r) r) r* �<module> sZ
< y