관리-도구
편집 파일: test_posix.cpython-38.pyc
U �Mg 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� |dk r0|�dt|�g� n ts8trD|�d� n |�d� tr^| �dd�} |�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 ) zwWrapper for calling the ps command with a little bit of cross-platform support for a narrow range of features. �psz --no-headersNz-pz-AZax�startZstimez-o� r ) r �append�extend�strr r �replacer � splitlines�strip�int� ValueError)�fmt�pid�cmd�output� all_output�line� r( �G/opt/hc_python/lib64/python3.8/site-packages/psutil/tests/test_posix.pyr / s0 r c C s> d}t r d}t|| ��� }tr6d|d ks.t�|d S |d S )N�commandZcommz /bin/qemu-r r )r r �splitr �AssertionError)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> tk rR } z dt|��� kr@t�d��� W 5 d }~X Y nX |�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(t�d S )N�ppid)r r# �psutil�ProcessrV r, )�selfZppid_psZ ppid_psutilr( r( r) � test_ppid� s zTestProcess.test_ppidc C s. t d| j�}t�| j��� j}||ks*t�d S )N�uid)r r# rW rX �uids�realr, )rY Zuid_psZ uid_psutilr( r( r) �test_uid� s zTestProcess.test_uidc C s. t d| j�}t�| j��� j}||ks*t�d S )NZrgid)r r# rW rX Zgidsr] r, )rY Zgid_psZ gid_psutilr( r( r) �test_gid� s zTestProcess.test_gidc C s, t d| j�}t�| j��� }||ks(t�d S )N�user)r r# rW rX �usernamer, )rY Z username_psZusername_psutilr( r( r) � test_username� s zTestProcess.test_usernamec C sJ t �� }tjdtd��*}|�� t|�� j�ks2t �|j s<t �W 5 Q R X d S )Nzpsutil.pwd.getpwuid�Z side_effect) rW rX r �patch�KeyErrorra r 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 }||ks8t�d S )N皙�����?r r: )�time�sleepr7 r# rW rX � memory_infor, )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 }||ks8t�d S )Nri r r: )rj rk r8 r# rW rX rl r, )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|�}||ksrt �d S )Nz\d.\dr0 z\d) r. r# �os�path�basenamer@ rW rX �namer1 r2 r, )rY Zname_psZ name_psutilr( r( r) � test_name� s zTestProcess.test_namec C s^ d}dddg}t jd|d��8 t jd|d�� t�� }|�� dksFt�W 5 Q R X W 5 Q R X d S )N�long-program-namezlong-program-name-extendedZfoo�bar� psutil._psplatform.Process.name�Z return_value�"psutil._psplatform.Process.cmdline)r rd rW rX rr r, )rY rr �cmdlinerg r( r( r) �test_name_long� s �zTestProcess.test_name_longc C s\ d}t jd|d��@ t jdt�dd�d�� t�� }|�� dksDt�W 5 Q R X W 5 Q R X d S �Nrt rv rw rx r r0 rc )r rd rW Z AccessDeniedrX rr r, �rY rr rg r( r( r) � test_name_long_cmdline_ad_exc� s �z)TestProcess.test_name_long_cmdline_ad_excc C sl d}t jd|d��P t jdt�dd�d��. t�� }t�tj�� |�� W 5 Q R X W 5 Q R X W 5 Q R X d S r{ )r rd rW Z NoSuchProcessrX r �raisesrr 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�}|||fksXt �d S )Nr z%H:%M:%S) r r# rW rX Z create_time�datetime� fromtimestamp�strftime�roundr, )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(t�W n0 tk rZ |d t|�� }||ksVt�Y nX d S rT )r. r# rW rX �exer, �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|�|�sPt �n ||ksPt �d S )N� ) r4 r# �joinrW rX ry r r� � startswithr, )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$t�d S )N�nice)r r# rW rX r� r, )rY Zps_niceZ psutil_nicer( r( r) � test_nice? s zTestProcess.test_niceN) �__name__� __module__� __qualname__�__doc__� classmethodrS rU rZ r^ r_ rb rh r r rm rn rs rz 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� 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>X s z,TestSystemAPIs.test_pids.<locals>.<listcomp>c s g | ]}|� kr|�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�� �ksXt�| j t�� |d��V t t�� �D ]B\}}|j || ks�t�|j || ks�t�|j d k rxt�|j � qxW 5 Q R X d S )N�who -u�no users on this systemr9 c S s g | ]}|� � d �qS )r �r+ r� r( r( r) r� v s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|� � d �qS )r r� r� r( r( r) r� w s �rW Zwho)r r r rA r+ r� rW �usersr, �subTest� enumeraterr 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 ��@ t t�� �D ],\}}t j � |j �� |�}||| ks�t�q�W 5 Q R X 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 %sr� )r r r rA r1 �findallr� rW r� r� r� r� �startedr� r, )rY r3 Ztstampr� r� r� Z psutil_valuer( r( r) �test_users_started� s: ���z!TestSystemAPIs.test_users_startedc C sV t jdttjd�d��6}t�t�� tj� t � � � W 5 Q R X |j sHt �W 5 Q R X d S )Nzpsutil._psposix.os.killr0 rc )r rd �OSError�errno�EBADFr r~ rW �_psposixZ pid_existsro �getpidrf r, �rY �mr( r( r) �test_pid_exists_let_raise� s � z(TestSystemAPIs.test_pid_exists_let_raisec C sV t jdttjd�d��6}t�t�� tj� t � � � W 5 Q R X |j sHt �W 5 Q R X d S )N�psutil._psposix.os.waitpidr0 rc )r rd r� r� r� r r~ rW r� �wait_pidro r� rf r, 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 5 Q R X |j sPt�W 5 Q R X d S )Nr� r0 rc g{�G�z�?)�timeout)r rd r� r� ZEINTRr r~ rW r� �TimeoutExpiredr� ro r� rf r, r� r( r( r) �test_os_waitpid_eintr� s � z$TestSystemAPIs.test_os_waitpid_eintrc C sN t jddd��6}t�t�� tj�t� � � W 5 Q R X |j s@t �W 5 Q R X d S )Nr� )r ���rw ) r rd r r~ r! rW r� r� ro r� rf r, 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 nP tk r� } z2t|��� }d|ksld|ksld|krtW Y �q� W 5 d }~X Y qX t |j | �|k s�t �t |j | �|k s�t �t |j | �|k s�t �t |j| �dkst �qd S )Ni @ F)�allzno such file or directoryzraw devices not supportedzpermission deniedr )rW Zdisk_partitions� disk_usageZ mountpointrJ rC r? r r@ �abs�totalr, �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 � }|dkst�|t� � ks"t�|tjks0t�d S )Nr )r r, �resource�mmap�PAGESIZE)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� ro r1 rO rj 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 - #