File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_process_all.cpython-39.pyc
a
��?h�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" tjyX ���� i Y S 0 z��ddg�}W n tjy� ���� Y n�0 |d |d ��d�ji}t��}|j |j
dd �D ]Z\}}z|� ||< W q� tj �y } z&� |���� W Y d }~q�W Y d }~q�d }~0 0 q�|� |S d S )
Nc s� �� | j� � | jd ur&�� | j|� t| tj�rd��|� | jd urz��| jd� �� | j|� nt| tj �rz��
|� t | � t | � d S �Nr )
� assertEqual�pid�name�
isinstance�psutilZ
ZombieProcessZassertProcessZombie�ppidZassertGreaterEqual�
NoSuchProcess�assertProcessGone�str�repr)�exc�procr r! )r �tcase� �I/usr/local/lib64/python3.9/site-packages/psutil/tests/test_process_all.py�check_exception7 s
z"proc_info.<locals>.check_exceptionc
sP �dkrLz�� d� W n4 tjyJ } z� | ���� W Y d } ~ n
d } ~ 0 0 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
assertPidGoneZas_dictr# r r �iterZgettersr- )r r. �d�info�nsZfunZfun_namer& r) )r+ r r r! r' r( r* � proc_info4 s.
$r5 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 r8 r9 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 )r5 )Z psutil.tests.test_process_allr5 r9 r; Zimap_unorderedr �pids�append)r= r5 Zlsr r) r) r* �iter_proc_infox s z$TestFetchAllProcesses.iter_proc_infoc C s� g }| � � D ]�}|�� D ]�\}}t| |�}z|||� W np ty� d}|d||d t|�|f 7 }|d7 }|dt�� 7 }d�dd� |�� D ��d }|� |� Y q0 |d d
g d d i fvr|sJ |��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>� � z1TestFetchAllProcesses.test_all.<locals>.<genexpr>r � � ) rE �items�getattr� Exceptionr% � traceback�
format_excrA �
splitlinesrD Zfail)r= Zfailuresr3 r �value�meth�sr) r) r* �test_all� s,
� z TestFetchAllProcesses.test_allc C s* t |t�sJ �|D ]}t |t�sJ �qd S �N)r �listr$ )r= �retr3 �partr) r) r* �cmdline� s z TestFetchAllProcesses.cmdlinec C s� t |ttf�sJ �|�� |ks"J �|r�tr8|�d�s8d S tj�|�sLJ |��t r�tj�
|�r�t td�r�t td�r�zt� |tj
�s�J �W n$ ty� tj�|�r�ts�� Y n0 d S )Nz.exe�access�X_OK)r r$ r �stripr �endswith�os�path�isabsr
�isfile�hasattrr\ r] �AssertionError�existsr �r= rY r3 r) r) r* �exe� s zTestFetchAllProcesses.exec C s t |t�sJ �|dksJ �d S r �r �intrg r) r) r* r � s zTestFetchAllProcesses.pidc C s* t |ttf�sJ �|dks J �t|� d S r )r rj r r5 rg r) r) r* r! � s zTestFetchAllProcesses.ppidc C sB t |ttf�sJ �tr*|s*t|d �r*d S ts>|s>J t|���d S )Nr )r r$ r r r r r% rg r) r) r* r � s
zTestFetchAllProcesses.namec C s^ t |t�sJ �z|dks J �W n( tyF tr@|d tjkr@n� Y n0 t�dt�|�� d S )Nr �statusz%Y %m %d %H:%M:%S) r �floatre r r Z
STATUS_ZOMBIE�time�strftime� localtimerg r) r) r* � create_time� s z!TestFetchAllProcesses.create_timec C s4 t |�s J �|D ] }t|t�s"J �|dksJ �qd S r )r r rj )r= rY r3 �uidr) r) r* �uids� s zTestFetchAllProcesses.uidsc C s< t |�s J �|D ]&}t|t�s"J �tsts|dksJ �qd S r )r r rj r r )r= rY r3 �gidr) r) r* �gids� s
zTestFetchAllProcesses.gidsc C s. t |t�sJ �|�� |ks J �|�� s*J �d S rW )r r$ r^ rg r) r) r* �username� s z TestFetchAllProcesses.usernamec C s> t |t�sJ �|sJ |��tr"d S |dks.J �|tv s:J �d S )N�?)r r$ r r rg r) r) r* rk � s z TestFetchAllProcesses.statusc C s@ t |�s J �|D ]*}t|ttf�s&J �|dkr|dksJ �qd S )N���r �r r rj r )r= rY r3 �fieldr) r) r* � io_counters� s
z!TestFetchAllProcesses.io_countersc C s� t rBt|jt�sJ �t|jt�s$J �|jdks2J �|jdks|J �n:tjtjtjtj g}t|t�sdJ �|dkspJ �||v s|J �d S r )
r r Zioclassrj rS r ZIOPRIO_VERYLOWZ
IOPRIO_LOWZ
IOPRIO_NORMALZ IOPRIO_HIGH)r= rY r3 �choicesr) r) r* �ionice� s � z TestFetchAllProcesses.ionicec C s: t |t�sJ �tr*|dkr*t|d �r*d S |dks6J �d S )Nr r � )r rj r r rg r) r) r* � num_threads
s z!TestFetchAllProcesses.num_threadsc C sn t |t�sJ �|D ]V}t|�s"J �|jdks0J �|jdks>J �|jdksLJ �|D ]}t |ttf�sPJ �qPqd S r )r rX r �idZ user_timeZ system_timerj rl )r= rY r3 �try r) r) r* �threads s z TestFetchAllProcesses.threadsc C s4 t |�s J �|D ] }t|t�s"J �|dksJ �qd S r )r r rl )r= rY r3 �nr) r) r* � cpu_times s z TestFetchAllProcesses.cpu_timesc C s0 t |t�sJ �d| kr"dks,n J |��d S )NrK g Y@�r rl rg r) r) r* � cpu_percent% s z!TestFetchAllProcesses.cpu_percentc C s^ t |t�sJ �tr |dkr d S |dks*J �t�� dkrB|dksBJ �|ttt�� ��v sZJ �d S )Nrw r r} )r rj r r � cpu_countrX �rangerg r) r) r* �cpu_num) s z TestFetchAllProcesses.cpu_numc C s| t |�s J �|D ]"}t|ttf�s&J �|dksJ �qtrx|j|jksHJ �|j|jksXJ �|j |j
kshJ �|j |j ksxJ �d S r )
r r rj r r Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZ
peak_pagefileZpagefile�r= rY r3 rS r) r) r* � memory_info2 s z!TestFetchAllProcesses.memory_infoc C s� t |�s J �t�� j}|jD ]N}t||�}t|ttf�s<J �|dksHJ �t s t
rZ|dv rZq ||ks J |��q t r�|j |j ks�J �d S )Nr )Zvms�data)
r r Zvirtual_memory�total�_fieldsrN r rj r r r ZpssZuss)r= rY r3 r� r rS r) r) r* �memory_full_info= s
z&TestFetchAllProcesses.memory_full_infoc C s" t |t�sJ �|D �]}t |jt�s(J �t |jt�s8J �|j�� |jksLJ �tr`|jdks�J �nltr�t |j t�stJ �t |j
t�s�J �t |j t�s�J �|j dks�J �|j
dv s�J �|j dks�J �n t r�|js�qt
j�|j�s�J |��zt
�|j�}W n t�y Y q0 t�|j�sJ |��qd S )Nrw r )�r�w�azr+za+)r rX �fdrj ra r$ r^ r r �position�mode�flagsr r` rb �statr
�S_ISREG�st_mode)r= rY r3 �f�str) r) r* �
open_filesM s,
z TestFetchAllProcesses.open_filesc C s t |t�sJ �|dksJ �d S r ri rg r) r) r* �num_fdsg s z TestFetchAllProcesses.num_fdsc C s` t � �F t|�tt|��ks J �|D ]}t|�s4J �t|� q$W d � n1 sR0 Y d S rW )r �len�setr r )r= rY r3 �connr) r) r* �net_connectionsk s
z%TestFetchAllProcesses.net_connectionsc
C s� t |ttf�sJ �|�� |ks"J �|r�tj�|�s:J |��zt�|�}W nD ty� } z,t rjt
j � |�rjn|j
t
jkrx� W Y d }~nd }~0 0 t�|j�s�J �d S rW )r r$ r r^ r` ra rb r� �OSErrorr r Z _psplatformZis_permission_err�errno�ENOENT�S_ISDIRr� )r= rY r3 r� �errr) r) r* �cwdr s zTestFetchAllProcesses.cwdc C s0 t |t�sJ �d| kr"dks,n J |��d S )Nr �d r� rg r) r) r* �memory_percent� s z$TestFetchAllProcesses.memory_percentc C s t |t�sJ �d S rW )r �boolrg r) r) r* �
is_running� s z TestFetchAllProcesses.is_runningc C sR t |t�sJ �|g ksJ �ttt�� ��}|D ] }t |t�s@J �||v s.J �q.d S rW )r rX r� r r� rj )r= rY r3 Zcpusr� r) r) r* � cpu_affinity� s z"TestFetchAllProcesses.cpu_affinityc C sJ t |ttd �f�sJ �|d urFtj�|�s2J |��tj�|�sFJ |��d S rW )r r$ �typer` ra rb rf rg r) r) r* �terminal� s z TestFetchAllProcesses.terminalc C s� |D ]�}t |jt�sJ �t |jt�s(J �t |jt�s8J �|jD ]�}t||�}|dkrx|�d�s�tj� |j�s�J |j��q>|dkr�|s�J t
|���q>|dkr�t s�|s�J t
|���q>t |t t
f�s�J �|dks>J �q>qd S )Nra )�[z anon_inode:�addr�permsr )r r� r$ r� ra r� rN �
startswithr` rb r% r rj r )r= rY r3 �nt�fnamerS r) r) r* � memory_maps� s
z!TestFetchAllProcesses.memory_mapsc C s t |t�sJ �|dksJ �d S r ri rg r) r) r* � num_handles� s z!TestFetchAllProcesses.num_handlesc C sx t |t�sJ �tr2d| kr&dkstn J |��nBdd� tt�D �}||v sPJ �trft |tj�stJ �nt |t�stJ �d S )Ni��� c S s g | ]}|� d �rtt|��qS )Z_PRIORITY_CLASS)r_ rN r )rG �xr) r) r* �
<listcomp>� s
�z.TestFetchAllProcesses.nice.<locals>.<listcomp>)r rj r
�dirr r �enum�IntEnum)r= rY r3 Z
prioritiesr) r) r* �nice� s � zTestFetchAllProcesses.nicec C s8 t |�s J �|D ]"}t|ttf�s&J �|dksJ �qd S r rx r� r) r) r* �num_ctx_switches� s z&TestFetchAllProcesses.num_ctx_switchesc C sB t |t�sJ �t|�dks J �|d dks.J �|d dks>J �d S )N� r rw r} )r �tupler� rg r) r) r* �rlimit� s z TestFetchAllProcesses.rlimitc C s@ t |t�sJ �|�� D ]$\}}t |t�s,J �t |t�sJ �qd S rW )r �dictrM r$ )r= rY r3 �k�vr) r) r* �environ� s z TestFetchAllProcesses.environN))�__name__�
__module__� __qualname__�__doc__r> rB rE rV 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* r6 e sL
r6 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 r7 �r r8 r< r) r) r* r> � s zTestPidsRange.setUpc C s t �d� d S r? r� r< r) r) r* rB � s zTestPidsRange.tearDownc C sf dd� }dd� }t dd�D ]F}tr,||�r,q| j|d�� ||� W d � q1 sV0 Y qd S )Nc S s� zt d| d�}W n ty& Y dS 0 |�V |D ]8}|�d�r2t|�� d �}|| k W d � S q2td��W d � n1 s�0 Y 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 �| �}zv|r4t �| � ts�| t �� v s�J �nNtsnt�t j�� t �| � W d � n1 sd0 Y ts�| t �� vs�J �W n* t j t
fy� |d8 }|dkr�� Y q0 d S qd S )N� r} r ) r Z
pid_existsr0 r rC r r Zraisesr" r- re )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> rB r� r) r) r) r* r� � s r� )(r� r� r� r: r` 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 r9 r5 r6 r� r) r) r) r* �<module> sN 1 q