관리-도구
편집 파일: test_process_all.cpython-38.pyc
U �Mg�H � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# e �o�e �o�e Z$d d � Z%G d d � d e �Z&G d d!� d!e �Z'dS )"zeIterate over all process PIDs and for each one of them invoke and test all psutil.Process() methods. � N)�AIX)�BSD)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�OSX)�POSIX)�WINDOWS)�PY3)�FileNotFoundError)�long)�unicode)� CI_TESTING)�PYTEST_PARALLEL)� QEMU_USER)�VALID_PROC_STATUSES)�PsutilTestCase)�check_connection_ntuple)�create_sockets)� is_namedtuple)�is_win_secure_system_proc)�process_namespace)�pytestc s( t � ���fdd�� � ����fdd�}zt����W n$ tjk rZ ���� i Y S X z��ddg�}W n tjk r� ���� Y n�X |d |d ��d�ji}t��}|j |j dd �D ]T\}}z|� ||< W q� tj k �r } z� |���� W Y �q�W 5 d }~X Y q�X q�|� |S d S ) Nc s� �� | j� � | jd k r&�� | j|� t| tj�rd��|� | jd k rz��| jd� �� | j|� nt| tj �rz�� |� t | � t | � d S �Nr ) Z assertEqual�pid�name� isinstance�psutilZ ZombieProcessZassertProcessZombie�ppidZassertGreaterEqual� NoSuchProcess�assertProcessGone�str�repr)�exc�procr r )r �tcase� �M/opt/hc_python/lib64/python3.8/site-packages/psutil/tests/test_process_all.py�check_exception7 s z"proc_info.<locals>.check_exceptionc sN �dkrJz�� d� W n2 tjk rH } z� | ���� W 5 d } ~ X Y nX d S r )�waitr �Error)r% )r* r r r r& r( r) �do_waitE s zproc_info.<locals>.do_waitr r r F)� clear_cache) r r �Processr! Z assertPidGone�as_dictr" r r �iter�gettersr, )r r- �d�info�nsZfunZfun_namer% r( )r* r r r r& r' r) � proc_info4 s. r6 c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLS )M�TestFetchAllProcessesz�Test which iterates over all running processes and performs some sanity checks against Process API's returned values. Uses a process pool to get info about all processes. c C s t �d� trt�� | _d S �NF)r � _set_debug� USE_PROC_POOL�multiprocessingZPool�pool��selfr( r( r) �setUpk s zTestFetchAllProcesses.setUpc C s& t �d� tr"| j�� | j�� d S �NT)r r9 r: r< � terminate�joinr= r( r( r) �tearDownr s z TestFetchAllProcesses.tearDownc C sJ ddl m} tr"| j�|t�� �S g }t�� D ]}|�||�� q.|S d S )Nr )r6 )Z psutil.tests.test_process_allr6 r: r< Zimap_unorderedr �pids�append)r> r6 Zlsr r( r( r) �iter_proc_infox s z$TestFetchAllProcesses.iter_proc_infoc C s� g }| � � D ]�}|�� D ]�\}}t| |�}z|||� W nr tk r� d}|d||d t|�|f 7 }|d7 }|dt�� 7 }d�dd� |�� D ��d }|� |� Y qX |d d g d d i fkr|st |��qq |r�| � d �|���d S ) NzH ====================================================================== z+FAIL: name=test_%s, pid=%s, ret=%s info=%s r zF----------------------------------------------------------------------z %s� c s s | ]}d | V qdS )z Nr( )�.0�ir( r( r) � <genexpr>� s z1TestFetchAllProcesses.test_all.<locals>.<genexpr>r � � ) rF �items�getattr� Exceptionr$ � traceback� format_excrB � splitlinesrE �AssertionErrorZfail)r> Zfailuresr4 r �value�meth�sr( r( r) �test_all� s, � z TestFetchAllProcesses.test_allc C s* t |t�st�|D ]}t |t�st�qd S �N)r �listrS r# )r> �retr4 �partr( r( r) �cmdline� s z TestFetchAllProcesses.cmdlinec C s� t |ttf�st�|�� |ks"t�|r�tr8|�d�s8d S tj� |�sLt|��t r�tj� |�r�t td�r�t td�r�zt� |tj�s�t�W n& tk r� tj�|�r�ts�� Y nX d S )Nz.exe�access�X_OK)r r# r rS �stripr �endswith�os�path�isabsr �isfile�hasattrr] r^ �existsr �r> rZ r4 r( r( r) �exe� s zTestFetchAllProcesses.exec C s t |t�st�|dkst�d S r �r �intrS rg r( r( r) r � s zTestFetchAllProcesses.pidc C s* t |ttf�st�|dks t�t|� d S r )r rj r rS r6 rg r( r( r) r � s zTestFetchAllProcesses.ppidc C sB t |ttf�st�tr*|s*t|d �r*d S ts>|s>tt|���d S )Nr )r r# r rS r r r r$ rg r( r( r) r � s zTestFetchAllProcesses.namec C s` t |t�st�z|dks t�W n* tk rH trB|d tjkrBn� Y nX t�dt�|�� d S )Nr �statusz%Y %m %d %H:%M:%S) r �floatrS r r Z STATUS_ZOMBIE�time�strftime� localtimerg r( r( r) � create_time� s z!TestFetchAllProcesses.create_timec C s4 t |�s t�|D ] }t|t�s"t�|dkst�qd S r )r rS r rj )r> rZ r4 �uidr( r( r) �uids� s zTestFetchAllProcesses.uidsc C s< t |�s t�|D ]&}t|t�s"t�tsts|dkst�qd S r )r rS r rj r r )r> rZ r4 �gidr( r( r) �gids� s zTestFetchAllProcesses.gidsc C s. t |t�st�|�� |ks t�|�� s*t�d S rX )r r# rS r_ rg r( r( r) �username� s z TestFetchAllProcesses.usernamec C s> t |t�st�|st|��tr"d S |dks.t�|tks:t�d S )N�?)r r# rS r r rg r( r( r) rk � s z TestFetchAllProcesses.statusc C s@ t |�s t�|D ]*}t|ttf�s&t�|dkr|dkst�qd S )N���r �r rS r rj r )r> rZ r4 �fieldr( r( r) � io_counters� s z!TestFetchAllProcesses.io_countersc C s� t rBt|jt�st�t|jt�s$t�|jdks2t�|jdks|t�n:tjtjtj tj g}t|t�sdt�|dkspt�||ks|t�d S r ) r r Zioclassrj rS rT r ZIOPRIO_VERYLOWZ IOPRIO_LOWZ IOPRIO_NORMALZ IOPRIO_HIGH)r> rZ r4 �choicesr( r( r) �ionice� s � z TestFetchAllProcesses.ionicec C s: t |t�st�tr*|dkr*t|d �r*d S |dks6t�d S )Nr r � )r rj rS r r rg r( r( r) � num_threads s z!TestFetchAllProcesses.num_threadsc C sn t |t�st�|D ]V}t|�s"t�|jdks0t�|jdks>t�|jdksLt�|D ]}t |ttf�sPt�qPqd S r ) r rY rS r �idZ user_timeZ system_timerj rl )r> rZ r4 �try r( r( r) �threads s z TestFetchAllProcesses.threadsc C s4 t |�s t�|D ] }t|t�s"t�|dkst�qd S r )r rS r rl )r> rZ r4 �nr( r( r) � cpu_times s z TestFetchAllProcesses.cpu_timesc C s0 t |t�st�d| kr"dks,n t|��d S )NrK g Y@�r rl rS rg r( r( r) � cpu_percent% s z!TestFetchAllProcesses.cpu_percentc C s^ t |t�st�tr |dkr d S |dks*t�t�� dkrB|dksBt�|ttt�� ��ksZt�d S )Nrw r r} )r rj rS r r � cpu_countrY �rangerg r( r( r) �cpu_num) s z TestFetchAllProcesses.cpu_numc C s| t |�s t�|D ]"}t|ttf�s&t�|dkst�qtrx|j|jksHt�|j|j ksXt�|j |j ksht�|j |j ksxt�d S r )r rS r rj r r Z peak_wsetZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZ peak_pagefileZpagefile�r> rZ r4 rT r( r( r) � memory_info2 s z!TestFetchAllProcesses.memory_infoc C s� t |�s t�t�� j}|jD ]N}t||�}t|tt f�s<t�|dksHt�t s t rZ|dkrZq ||ks t|��q t r�|j |j ks�t�d S )Nr )Zvms�data)r rS r Zvirtual_memory�total�_fieldsrN r rj r r r ZpssZuss)r> rZ r4 r� r rT r( r( r) �memory_full_info= s z&TestFetchAllProcesses.memory_full_infoc C s$ t |t�st�|D �] }t |jt�s(t�t |jt�s8t�|j�� |jksLt�tr`|jdks�t�nlt r�t |j t�stt�t |j t�s�t�t |j t�s�t�|j dks�t�|j dks�t�|j dks�t�n t r�|js�qtj�|j�s�t|��zt�|j�}W n tk �r Y qX t�|j�st|��qd S )Nrw r )�r�w�azr+za+)r rY rS �fdrj rb r# r_ r r �position�mode�flagsr ra rc �statr �S_ISREG�st_mode)r> rZ r4 �f�str( r( r) � open_filesM s, z TestFetchAllProcesses.open_filesc C s t |t�st�|dkst�d S r ri rg r( r( r) �num_fdsg s z TestFetchAllProcesses.num_fdsc C sL t � �< t|�tt|��ks t�|D ]}t|�s4t�t|� q$W 5 Q R X d S rX )r �len�setrS r r )r> rZ r4 �connr( r( r) �net_connectionsk s z%TestFetchAllProcesses.net_connectionsc C s� t |ttf�st�|�� |ks"t�|r�tj�|�s:t|��zt�|�}W nB t k r� } z$t rlt j � |�rln|jtjkrz� W 5 d }~X Y nX t�|j�s�t�d S rX )r r# r rS r_ ra rb rc r� �OSErrorr r Z _psplatformZis_permission_err�errno�ENOENT�S_ISDIRr� )r> rZ r4 r� �errr( r( r) �cwdr s zTestFetchAllProcesses.cwdc C s0 t |t�st�d| kr"dks,n t|��d S )Nr �d r� rg r( r( r) �memory_percent� s z$TestFetchAllProcesses.memory_percentc C s t |t�st�d S rX )r �boolrS rg r( r( r) � is_running� s z TestFetchAllProcesses.is_runningc C sR t |t�st�|g kst�ttt�� ��}|D ] }t |t�s@t�||ks.t�q.d S rX )r rY rS r� r r� rj )r> rZ r4 Zcpusr� r( r( r) � cpu_affinity� s z"TestFetchAllProcesses.cpu_affinityc C sJ t |ttd �f�st�|d k rFtj�|�s2t|��tj�|�sFt|��d S rX )r r# �typerS ra rb rc rf rg r( r( r) �terminal� s z TestFetchAllProcesses.terminalc C s� |D ]�}t |jt�st�t |jt�s(t�t |jt�s8t�|jD ]�}t||�}|dkrx|�d�s�t j� |j�s�t|j��q>|dkr�|s�tt |���q>|dkr�t s�|s�tt |���q>t |t tf�s�t�|dks>t�q>qd S )Nrb )�[z anon_inode:�addr�permsr )r r� r# rS r� rb r� rN � startswithra rc r$ r rj r )r> rZ r4 �nt�fnamerT r( r( r) � memory_maps� s z!TestFetchAllProcesses.memory_mapsc C s t |t�st�|dkst�d S r ri rg r( r( r) � num_handles� s z!TestFetchAllProcesses.num_handlesc C sx t |t�st�tr2d| kr&dkstn t|��nBdd� tt�D �}||ksPt�trft |tj�stt�nt |t�stt�d S )Ni��� c S s g | ]}|� d �rtt|��qS )Z_PRIORITY_CLASS)r` rN r )rH �xr( r( r) � <listcomp>� s �z.TestFetchAllProcesses.nice.<locals>.<listcomp>) r rj rS r �dirr r �enum�IntEnum)r> rZ r4 Z prioritiesr( r( r) �nice� s � zTestFetchAllProcesses.nicec C s8 t |�s t�|D ]"}t|ttf�s&t�|dkst�qd S r rx r� r( r( r) �num_ctx_switches� s z&TestFetchAllProcesses.num_ctx_switchesc C sB t |t�st�t|�dks t�|d dks.t�|d dks>t�d S )N� r rw r} )r �tuplerS r� rg r( r( r) �rlimit� s z TestFetchAllProcesses.rlimitc C s@ t |t�st�|�� D ]$\}}t |t�s,t�t |t�st�qd S rX )r �dictrS rM r# )r> rZ r4 �k�vr( r( r) �environ� s z TestFetchAllProcesses.environN))�__name__� __module__� __qualname__�__doc__r? rC rF rW r\ rh r r r rp rr rt ru rk rz 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) r7 e sL r7 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � TestPidsRangea@ Given pid_exists() return value for a range of PIDs which may or may not exist, make sure that psutil.Process() and psutil.pids() agree with pid_exists(). This guarantees that the 3 APIs are all consistent with each other. See: https://github.com/giampaolo/psutil/issues/2359 XXX - Note about Windows: it turns out there are some "hidden" PIDs which are not returned by psutil.pids() and are also not revealed by taskmgr.exe and ProcessHacker, still they can be instantiated by psutil.Process() and queried. One of such PIDs is "conhost.exe". Running as_dict() for it reveals that some Process() APIs erroneously raise NoSuchProcess, so we know we have problem there. Let's ignore this for now, since it's quite a corner case (who even imagined hidden PIDs existed on Windows?). c C s t �d� d S r8 �r r9 r= r( r( r) r? � s zTestPidsRange.setUpc C s t �d� d S r@ r� r= r( r( r) rC � s zTestPidsRange.tearDownc C sR dd� }dd� }t dd�D ]2}tr,||�r,q| j|d�� ||� W 5 Q R X qd S )Nc S s� zt d| d�}W n tk r( Y dS X |�J |D ]6}|�d�r4t|�� d �}|| k W 5 Q R � S q4td��W 5 Q R X d S )Nz/proc/%s/status�rbFs Tgid:r} z'Tgid' line not found)�openr r� rj �split� ValueError)r r� �lineZtgidr( r( r) � is_linux_tid� s z+TestPidsRange.test_it.<locals>.is_linux_tidc S s� d}t �| �}zb|r4t �| � tsn| t �� ksnt�n:tsZt�t j �� t �| � W 5 Q R X tsn| t �� ksnt�W n, t j tfk r� |d8 }|dkr�� Y qX d S qd S )N� r} r ) r Z pid_existsr/ r rD rS r r Zraisesr! r, )r r� rf r( r( r) �check� s" z$TestPidsRange.test_it.<locals>.checkr} i� )r )r� r ZsubTest)r> r� r� r r( r( r) �test_it� s zTestPidsRange.test_itN)r� r� r� r� r? rC r� r( r( r( r) r� � s r� )(r� r� r� r; ra r� rm rP r r r r r r r r r r r Zpsutil._compatr r r r Z psutil.testsr r r r r r r r r r r r: r6 r7 r� r( r( r( r) �<module> sN 1 q