3
�[g�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�}ytj���W n tjk
rV �j�� i S X y�jddg�}W n tjk
r� �j�� Y n�X |d |d ��d�ji}t��}xb|j |j
dd �D ]N\}}y|� ||< W q� tj k
�r } z� |���� w�W Y d d }~X q�X q�W |� |S d S )
Nc s� �j | j� � | jd k r&�j | j|� t| tj�rd�j|� | jd k rz�j| jd� �j | j|� nt| tj �rz�j
|� t | � t | � d S )Nr )
Z assertEqual�pid�name�
isinstance�psutil�
ZombieProcessZassertProcessZombie�ppidZassertGreaterEqual�
NoSuchProcess�assertProcessGone�str�repr)�exc�procr r )r �tcase� �I/usr/local/lib64/python3.6/site-packages/psutil/tests/test_process_all.py�check_exception7 s
z"proc_info.<locals>.check_exceptionc sN �dkrJy�j d� W n2 tjk
rH } z� | ���� W Y d d } ~ X nX d S )Nr )�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�ns�funZfun_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 jd� trtj� | _d S )NF)r �
_set_debug�
USE_PROC_POOL�multiprocessingZPool�pool)�selfr( r( r) �setUpk s
zTestFetchAllProcesses.setUpc C s& t jd� tr"| jj� | jj� d S )NT)r r7 r8 r: � terminate�join)r; r( r( r) �tearDownr s
z TestFetchAllProcesses.tearDownc C sN ddl m} tr"| jj|tj� �S g }x tj� D ]}|j||�� q0W |S d S )Nr )r5 )Z psutil.tests.test_process_allr5 r8 r: Zimap_unorderedr �pids�append)r; r5 Zlsr r( r( r) �iter_proc_infox s z$TestFetchAllProcesses.iter_proc_infoc C s g }x�| j � D ]�}x�|j� D ]�\}}t| |�}y|||� W n� tk
r� ddd d }|d||d t|�|f 7 }|dd 7 }|dtj� 7 }djdd � |j� D ��d }|j |� Y q X |d
d g d d i fkr |s t
|��q W qW |r�| j d j|���d S )
N�
�=�F z+FAIL: name=test_%s, pid=%s, ret=%s
info=%s
r �-z
%sc s s | ]}d| V qdS )� � Nz r( )�.0�ir( r( r) � <genexpr>� s z1TestFetchAllProcesses.test_all.<locals>.<genexpr>r g � ) rB �items�getattr� Exceptionr$ � traceback�
format_excr> �
splitlinesrA �AssertionErrorZfail)r; Zfailuresr2 r �value�meth�sr( r( r) �test_all� s*
z TestFetchAllProcesses.test_allc C s. t |t�st�x|D ]}t |t�st�qW d S )N)r �listrS r# )r; �retr2 �partr( r( r) �cmdline� s
z TestFetchAllProcesses.cmdlinec C s� t |ttf�st�|j� |ks"t�|r�tr:|jd� r:d S tjj |�sNt|��t
r�tjj |�r�t td�r�t td�r�ytj
|tj�s�t�W n( tk
r� tjj|�r�t r�� Y nX d S )Nz.exe�access�X_OK)r r# r rS �stripr �endswith�os�path�isabsr
�isfile�hasattrr\ r] �existsr )r; rY r2 r( r( r) �exe� s zTestFetchAllProcesses.exec C s t |t�st�|dkst�d S )Nr )r �intrS )r; rY r2 r( r( r) r � s zTestFetchAllProcesses.pidc C s* t |ttf�st�|dks t�t|� d S )Nr )r rg r rS r5 )r; rY r2 r( r( r) r � s zTestFetchAllProcesses.ppidc C sD t |ttf�st�tr,| r,t|d �r,d S ts@|s@tt|���d S )Nr )r r# r rS r r r r$ )r; rY r2 r( r( r) r � s
zTestFetchAllProcesses.namec C s` t |t�st�y|dks t�W n* tk
rH trB|d tjkrBn� Y nX tjdtj|�� d S )Nr �statusz%Y %m %d %H:%M:%S) r �floatrS r r �
STATUS_ZOMBIE�time�strftime� localtime)r; rY r2 r( r( r) � create_time� s z!TestFetchAllProcesses.create_timec C s8 t |�s t�x&|D ] }t|t�s$t�|dkst�qW d S )Nr )r rS r rg )r; rY r2 �uidr( r( r) �uids� s
zTestFetchAllProcesses.uidsc C sD t |�s t�x2|D ]*}t|t�s$t�t rt r|dkst�qW d S )Nr )r rS r rg r r )r; rY r2 �gidr( r( r) �gids� s
zTestFetchAllProcesses.gidsc C s. t |t�st�|j� |ks t�|j� s*t�d S )N)r r# rS r^ )r; rY r2 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 )r; rY r2 r( r( r) rh � s z TestFetchAllProcesses.statusc C sD t |�s t�x2|D ]*}t|ttf�s(t�|dkr|dkst�qW d S )N� r ���)r rS r rg r )r; rY r2 �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 )Nr ) r r �ioclassrg rS rT r ZIOPRIO_VERYLOWZ
IOPRIO_LOWZ
IOPRIO_NORMALZ IOPRIO_HIGH)r; rY r2 �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 ru )r rg rS r r )r; rY r2 r( r( r) � num_threads
s z!TestFetchAllProcesses.num_threadsc C sv t |t�st�xb|D ]Z}t|�s$t�|jdks2t�|jdks@t�|jdksNt�x |D ]}t |ttf�sTt�qTW qW d S )Nr ) r rX rS r �id� user_time� system_timerg ri )r; rY r2 �trw r( r( r) �threads s
z TestFetchAllProcesses.threadsc C s8 t |�s t�x&|D ] }t|t�s$t�|dkst�qW d S )Nr )r rS r ri )r; rY r2 �nr( r( r) � cpu_times s
z TestFetchAllProcesses.cpu_timesc C s2 t |t�st�d| ko dkn s.t|��d S )Ng g Y@)r ri rS )r; rY r2 r( r( r) � cpu_percent% s z!TestFetchAllProcesses.cpu_percentc C s^ t |t�st�tr |dkr d S |dks*t�tj� dkrB|dksBt�|tttj� ��ksZt�d S )Nru r rv )r rg rS r r � cpu_countrX �range)r; rY r2 r( r( r) �cpu_num) s z TestFetchAllProcesses.cpu_numc C s� t |�s t�x*|D ]"}t|ttf�s(t�|dkst�qW tr||j|jksLt�|j|j ks\t�|j
|j kslt�|j |j
ks|t�d S )Nr )r rS r rg r r Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZ
peak_pagefileZpagefile)r; rY r2 rT r( r( r) � memory_info2 s
z!TestFetchAllProcesses.memory_infoc C s� t |�s t�tj� j}xX|jD ]N}t||�}t|tt f�s>t�|dksJt�t
s t r\|dkr\q ||ks t|��q W t
r�|j |j
ks�t�d S )Nr �vms�data)r� r� )r rS r Zvirtual_memory�total�_fieldsrN r rg r r r ZpssZuss)r; rY r2 r� r rT r( r( r) �memory_full_info= s
z&TestFetchAllProcesses.memory_full_infoc C s, t |t�st��x|D �] }t |jt�s,t�t |jt�s<t�|jj� |jksPt�trd|jdks�t�nnt r�t |j
t�sxt�t |j t�s�t�t |j t�s�t�|j
dks�t�|j d ks�t�|j dks�t�nt
r�|j r�qtjj|j�s�t|��ytj|j�}W n tk
�r Y qX tj|j�st|��qW d S )
Nru r �r�w�a�r+�a+rv )r� r� r� r� r� )r rX rS �fdrg ra r# r^ r r �position�mode�flagsr r` rb �statr
�S_ISREG�st_mode)r; rY r2 �f�str( r( r) �
open_filesM s, z TestFetchAllProcesses.open_filesc C s t |t�st�|dkst�d S )Nr )r rg rS )r; rY r2 r( r( r) �num_fdsg s z TestFetchAllProcesses.num_fdsc C sP t � �@ t|�tt|��ks t�x |D ]}t|�s6t�t|� q&W W d Q R X d S )N)r �len�setrS r r )r; rY r2 Zconnr( r( r) �net_connectionsk s
z%TestFetchAllProcesses.net_connectionsc C s� t |ttf�st�|j� |ks"t�|r�tjj|�s:t|��ytj|�}W nB t k
r� } z&t
rlt j j
|�rln|jtjkrz� W Y d d }~X nX tj|j�s�t�d S )N)r r# r rS r^ r` ra rb r� �OSErrorr r Z _psplatformZis_permission_err�errno�ENOENT�S_ISDIRr� )r; rY r2 r� �errr( r( r) �cwdr s zTestFetchAllProcesses.cwdc C s2 t |t�st�d| ko dkn s.t|��d S )Nr �d )r ri rS )r; rY r2 r( r( r) �memory_percent� s z$TestFetchAllProcesses.memory_percentc C s t |t�st�d S )N)r �boolrS )r; rY r2 r( r( r) �
is_running� s z TestFetchAllProcesses.is_runningc C sV t |t�st�|g kst�tttj� ��}x&|D ] }t |t�sBt�||ks0t�q0W d S )N)r rX rS r� r r� rg )r; rY r2 Zcpusr� r( r( r) � cpu_affinity� s
z"TestFetchAllProcesses.cpu_affinityc C sJ t |ttd �f�st�|d k rFtjj|�s2t|��tjj|�sFt|��d S )N)r r# �typerS r` ra rb re )r; rY r2 r( r( r) �terminal� s z TestFetchAllProcesses.terminalc C s� x�|D ]�}t |jt�st�t |jt�s*t�t |jt�s:t�x�|jD ]�}t||�}|dkr||jd�s�t jj
|j�s�t|j��qB|dkr�|s�tt |���qB|dkr�t s�|s�tt |���qBt |t
tf�s�t�|dksBt�qBW qW d S )Nra �[� anon_inode:�addr�permsr )r� r� )r r� r# rS r� ra r� rN �
startswithr` rb r$ r rg r )r; rY r2 �nt�fnamerT r( r( r) � memory_maps� s
z!TestFetchAllProcesses.memory_mapsc C s t |t�st�|dkst�d S )Nr )r rg rS )r; rY r2 r( r( r) � num_handles� s z!TestFetchAllProcesses.num_handlesc C sz t |t�st�tr4d| ko$dkn svt|��nBdd� tt�D �}||ksRt�trht |tj�svt�nt |t�svt�d S )N� c S s g | ]}|j d �rtt|��qS )Z_PRIORITY_CLASS)r_ rN r )rI �xr( r( r) �
<listcomp>� s z.TestFetchAllProcesses.nice.<locals>.<listcomp>i��) r rg rS r
�dirr r �enum�IntEnum)r; rY r2 Z
prioritiesr( r( r) �nice� s " zTestFetchAllProcesses.nicec C s< t |�s t�x*|D ]"}t|ttf�s(t�|dkst�qW d S )Nr )r rS r rg r )r; rY r2 rT 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 ru rv rv )r �tuplerS r� )r; rY r2 r( r( r) �rlimit� s z TestFetchAllProcesses.rlimitc C sD t |t�st�x0|j� D ]$\}}t |t�s.t�t |t�st�qW d S )N)r �dictrS rM r# )r; rY r2 �k�vr( r( r) �environ� s z TestFetchAllProcesses.environN))�__name__�
__module__� __qualname__�__doc__r< r? rB rW r[ rf r r r rn rp rr rs rh rx 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 jd� d S )NF)r r7 )r; r( r( r) r< � s zTestPidsRange.setUpc C s t jd� d S )NT)r r7 )r; r( r( r) r? � s zTestPidsRange.tearDownc
C sV dd� }dd� }x@t dd�D ]2}tr.||�r.q | j|d�� ||� W d Q R X q W d S )Nc S st yt d| d�}W n tk
r& dS X |�> x.|D ]&}|jd�r4t|j� d �}|| kS q4W td��W d Q R X d S )Nz/proc/%s/status�rbFs Tgid:ru z'Tgid' line not found)�openr
r� rg �split�
ValueError)r r� �lineZtgidr( r( r) � is_linux_tid� s
z+TestPidsRange.test_it.<locals>.is_linux_tidc S s� d}x�t j| �}yb|r6t j| � tsp| t j� kspt�n:ts\tjt j �� t j| � W d Q R X tsp| t j� kspt�W n, t j
tfk
r� |d8 }|dkr�� Y qX d S qW d S )N� ru r ) r Z
pid_existsr/ r r@ rS r r Zraisesr! r, )r r� re r( r( r) �check� s$
z$TestPidsRange.test_it.<locals>.checkru i� )r )r� r ZsubTest)r; r� r� r r( r( r) �test_it� s zTestPidsRange.test_itN)r� r� r� r� r< r? r� r( r( r( r) r� � s r� )(r� r� r� r9 r` r� rk 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 r8 r5 r6 r� r( r( r( r) �<module> sN 1 q |