ruạṛ
3 �^gn� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddl m Z ddl m Z ddl m Z ddl m Z ddl m!Z! ddl"m#Z# ddl"m$Z$ ddl"m%Z% ddl"m&Z& ddl"m'Z' ddl"m(Z( ddl"m)Z) ddl"m*Z* ddl"m+Z+ dd l"m,Z, dd l"m-Z- dd l"m.Z. dd l"m/Z/ dd l"m0Z0 dd!l"m1Z1 dd"l"m2Z2 dd#l"m3Z3 dd$l"m4Z4 dd%l"m5Z5 dd&l"m6Z6 dd'l"m7Z7 dd(l"m8Z8 dd)l"m9Z9 dd*l"m:Z: dd+l"m;Z; dd,l"m<Z< dd-l"m=Z= dd.l"m>Z> dd/l"m?Z? dd0l"m@Z@ dd1l"mAZA dd2l"mBZB G d3d4� d4e4�ZCe�r�ejD� dk�r�G d5d6� d6eC�ZEG d7d8� d8e4�ZFdS )9z Tests for psutil.Process class.� N)�AIX)�BSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�OSX)�POSIX)�SUNOS)�WINDOWS)� open_text)�PY3)�FileNotFoundError)�long)�redirect_stderr)�super)�APPVEYOR)� CI_TESTING)�GITHUB_ACTIONS)�GLOBAL_TIMEOUT)�HAS_CPU_AFFINITY)� HAS_ENVIRON)� HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)� HAS_RLIMIT)� HAS_THREADS)� MACOS_11PLUS)�PYPY)� PYTHON_EXE)�PYTHON_EXE_ENV)� QEMU_USER)�PsutilTestCase)� ThreadTask)� call_until)�copyload_shared_lib)� create_c_exe)� create_py_exe)�mock)�process_namespace)�pytest)� reap_children)�retry_on_failure)�sh)�skip_on_access_denied)�skip_on_not_implemented)� wait_for_pidc @ s e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Ze j j e d d �dd� �Z dd� Ze j j edd �dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Ze j j ed d �d d!� �Ze j j ed d �d"d#� �Ze j j e d$d �d%d&� �Zd'd(� Ze j j e d)d �d*d+� �Z e j j e d$d �e e d,�d-d.� ��Z e j j e! d$d �e j j e d/d �d0d1� ��Z"e j j e! d$d �e j j e# d2d �d3d4� ��Z$e j j e% d$d �d5d6� �Z&e j j e% d$d �d7d8� �Z'e j j e% d$d �d9d:� �Z(e j j e% d$d �d;d<� �Z)e j j e% d$d �d=d>� �Z*d?d@� Z+e j j e# dAd �dBdC� �Z,e j j e- d$d �dDdE� �Z.e/� e0e1d,�e j j e- d$d �dFdG� ���Z2e/� dHdI� �Z3dJdK� Z4e j j e5 d$d �dLdM� �Z6e j j e5 d$d �dNdO� �Z7dPdQ� Z8dRdS� Z9e j j ed d �dTdU� �Z:dVdW� Z;e j j e<dXd �dYdZ� �Z=d[d\� Z>e j j e<�ped]d �e j j ed^d �d_d`� ��Z?e j j e@dad �e j j eAdbd �e j j e<dXd �e j j edcd �ddde� ����ZBe j j e d)d �dfdg� �ZCe j j e d)d �dhdi� �ZDdjdk� ZEe j j ed d �dldm� �ZFdndo� ZGdpdq� ZHdrds� ZIe j j eJ d$d �dtdu� �ZKe j j eJ d$d �dvdw� �ZLe j j eJ d$d �dxdy� �ZMe j j eNdzd �e j j eOd{d �d|d}� ��ZPe j j eNdzd �e j j eOd{d �d~d� ��ZQe j j e d)d �d�d�� �ZRe e d,�e j j eS�p�ed�d �d�d�� ��ZTd�d�� ZUd�d�� ZVd�d�� ZWe j j ed d �e/� d�d�� ��ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zce j j e d)d �d�d�� �Zde j j e d)d �d�d�� �Zee j j e d)d �d�d�� �Zfd�d�� Zgd�d�� Zhe j j ei d$d �d�d�� �Zje j j ei d$d �e j j e d)d �e j j ekd�d �e j j ed�d �d�d�� ����Zld�S )�� TestProcessz Tests for psutil.Process class.c O sB | j ||�}y tj|j�S tjk r< | j|j� � Y nX d S )N)�spawn_testproc�psutil�Process�pid� NoSuchProcessZ assertPidGone)�self�args�kwargs�sproc� r< �E/usr/local/lib64/python3.6/site-packages/psutil/tests/test_process.py� spawn_psprocS s zTestProcess.spawn_psprocc C s: t j� }|jtj� kst�tjt�� d|_W d Q R X d S )N�! ) r4 r5 r6 �os�getpid�AssertionErrorr+ �raises�AttributeError)r8 �pr< r< r= �test_pid] s zTestProcess.test_pidc C sJ | j � }|j� |j� }tr,|tjks<t�n|tj ks<t�| j|� d S )N) r> �kill�waitr �signal�SIGTERMrB �SIGKILL�assertProcessGone)r8 rE �coder< r< r= � test_killc s zTestProcess.test_killc C sJ | j � }|j� |j� }tr,|tjks<t�n|tj ks<t�| j|� d S )N)r> � terminaterH r rI rJ rB rL )r8 rE rM r< r< r= �test_terminatem s zTestProcess.test_terminatec C sX t r tjntj}| j� }|j|� |j� }tr<||ksJt�n|| ksJt�| j |� d S )N) r rI rK rJ r> � send_signalrH r rB rL )r8 �sigrE rM r< r< r= �test_send_signalw s z TestProcess.test_send_signalz not POSIX)�reasonc C s� t j}| j� }tjdttjd�d��( tj t j �� |j |� W d Q R X W d Q R X | j� }tjdttj d�d��( tj t j�� |j |� W d Q R X W d Q R X d S )Nzpsutil.os.kill� )� side_effect)rI rJ r> r) �patch�OSError�errnoZESRCHr+ rC r4 r7 rQ �EPERM� AccessDenied)r8 rR rE r< r< r= �test_send_signal_mocked� s z#TestProcess.test_send_signal_mockedc C s� t ddg}| j|�}|j� }|dks(t�| j|� t ddg}| j|tjd�}|j� }|dks`t�| j|� t ddg}| j|�}|j� }|dks�t�| j|� t dd g}| j|�}|j� }|dks�t�| j|� d S ) Nz-c�passr z1 / 0)�stderr� zimport sys; sys.exit(5);� zimport os; os._exit(5);)r r> rH rB rL � subprocess�PIPE)r8 �cmdrE rM r< r< r= �test_wait_exited� s( z TestProcess.test_wait_exitedzfails on NETBSDc C s. | j � }tr�|jtj� tjtj�� |j dd� W d Q R X |jtj � tjtj�� |j dd� W d Q R X |jtj � |j � tj ks�t �|j � tj ks�t �n�|j � tjtj�� |j dd� W d Q R X |j� tjtj�� |j dd� W d Q R X |j� |j � tj k�st �|j � tj k�s*t �d S )Ng����MbP?)�timeout)r> r rQ rI �SIGSTOPr+ rC r4 �TimeoutExpiredrH �SIGCONTrJ rB �suspend�resumerO )r8 rE r< r< r= �test_wait_stopped� s( z TestProcess.test_wait_stoppedc C s� | j � \}}tjtj�� |jd� W d Q R X tjtj�� |jd� W d Q R X |j� |j� |j� }|j� }tr�|tj ks�t �|d ks�t �n |tj ks�t �|tj ks�t �d S )Ng{�G�z�?) �spawn_children_pairr+ rC r4 rg rH rO r rI rJ rB )r8 �child� grandchildZ child_retZgrandchild_retr< r< r= �test_wait_non_children� s z"TestProcess.test_wait_non_childrenc C sx | j � }|j� tjtj�� |jd� W d Q R X tjtj�� |jd� W d Q R X tjt�� |jd� W d Q R X d S )Ng{�G�z�?r r_ ���)r> �namer+ rC r4 rg rH � ValueError)r8 rE r< r< r= �test_wait_timeout� s z TestProcess.test_wait_timeoutc C s� | j � }tjtj�� |jd� W d Q R X |j� tj� t }xBtj� |k rvy|jd�}P W q@ tjk rr Y q@X q@W | j d��t r�|t j ks�t �n|t jks�t �| j|� d S )Nr re )r> r+ rC r4 rg rH rG �timer �failr rI rK rB rJ rL )r8 rE Zstop_atrM r< r< r= � test_wait_timeout_nonblocking� s z)TestProcess.test_wait_timeout_nonblockingc C s~ t j� }|jdd� |jdd� x6td�D ]*}|jd d�}t|t�sHt�|dks*t�q*W tjt �� |jdd� W d Q R X d S )Ng����MbP?)�interval�d g r_ rp ) r4 r5 � cpu_percent�range� isinstance�floatrB r+ rC rr )r8 rE �_�percentr< r< r= �test_cpu_percent� s z TestProcess.test_cpu_percentc C s4 t jdd d�� }tj� j� |js&t�W d Q R X d S )Nzpsutil.cpu_count)Z return_value)r) rW r4 r5 ry �calledrB )r8 �mr< r< r= � test_cpu_percent_numcpus_none s z)TestProcess.test_cpu_percent_numcpus_nonezQEMU user not supportedc C s� t j� j� }|jdks t|��|jdks0t|��|jdksBt|��|jdksTt|��trj|j dksjt|��x&|j D ] }t j dt j t||��� qrW d S )Ng z%H:%M:%S)r4 r5 � cpu_times�userrB �system� children_user�children_systemr Ziowait�_fieldsrt �strftime� localtime�getattr)r8 �timesrq r< r< r= �test_cpu_times s zTestProcess.test_cpu_timesc C s� t j� j� d d� \}}tj� d d� \}}t||g�t||g� dkrZ| jd||f ��t||g�t||g� dkr�| jd||f ��d S )N� g�������?zexpected: %s, found: %s)r4 r5 r� r@ r� �max�minru )r8 � user_timeZ kernel_time�utimeZktimer< r< r= �test_cpu_times_2 s z TestProcess.test_cpu_times_2z not supportedc C sP t j� }|j� }|dks t�t j� dkr4|dks4t�|j� tt j� �ksLt�d S )Nr r_ )r4 r5 Zcpu_numrB � cpu_countrz )r8 rE �numr< r< r= � test_cpu_num) s zTestProcess.test_cpu_numc C sZ | j � }tj� }|j� }t|| �}|dkr@| jd|||f ��tjdtj|j� �� d S )Nr� z'expected: %s, found: %s, difference: %sz%Y %m %d %H:%M:%S)r> rt � create_time�absru r� r� )r8 rE �nowr� � differencer< r< r= �test_create_time2 s z TestProcess.test_create_timez POSIX onlyc C sX t j� j� }|d k rTytjjtd��}W n tk rF tj d��Y nX ||ksTt �d S )N�ttyzcan't rely on `tty` CLI) r4 r5 �terminalr@ �path�realpathr. � RuntimeErrorr+ �skiprB )r8 r� r� r< r< r= � test_terminalD s zTestProcess.test_terminal)Zonly_ifc C s� t j� }|j� }ttd��}|j� W d Q R X |j� }t r�t r�|j|jksRt �|j |j ksbt �t r�|j |j ksvt �|j |j ks�t �n |j|jks�t �|j|jks�t �|j� }t| j� d��.}tr�|jtdd d�� n|jdd � W d Q R X |j� }|j |j k�s t �|j|jk�s t �|j|jk�s0t �|j|jk�sBt �t �rl|j |j k�sZt �|j |j k�slt �xNtt|��D ]>}t�r�|dk�r��qz|| dk�s�t �|| dk�szt ��qzW d S )N�rb�wb�xi@B �asciir� r )r4 r5 Z io_counters�openr �readr r � read_countrB � write_countr Z read_charsZ write_chars� read_bytes� write_bytes� get_testfnr �write�bytesrz �len)r8 rE Zio1�fZio2�ir< r< r= �test_io_countersP s> z TestProcess.test_io_countersz linux onlyc . s� � fdd�}t j� � ts.� j� d t jks.t�t jdks<t�t jdksJt�t jdksXt�t jdksft�� j� }| j ||� � jt j� t � j� �t jdfks�t�t j t �� � jt jdd� W d Q R X � jt j� t � j� �t jdfks�t�� jt jdd� t � j� �t jdfk�st�t j t �� � jt jd d� W d Q R X y� jt jdd� W n t jk �rj Y nX t j t d d �� � jt jd� W d Q R X t j t d d �� � jt jd� W d Q R X t j t d d �� � jdd� W d Q R X d S ) Nc s& | \}}|t jkrd}� j||� d S )Nr )r4 �IOPRIO_CLASS_NONE�ionice)�init�ioclass�value)rE r< r= �cleanup~ s z.TestProcess.test_ionice_linux.<locals>.cleanupr r_ r� � � )r� � zioclass accepts no value)�matchz$'ioclass' argument must be specified)r4 r5 r r� r� rB ZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE� addCleanup�tupler+ rC rr r[ )r8 r� r� r< )rE r= �test_ionice_linux{ s> z TestProcess.test_ionice_linuxz!not supported on this win versionc C s t j� }ts |j� t jks t�|j� }| j|j|� |jt j� |j� t jksRt�|jt j� |j� t jkspt�y|jt j � W n t j k r� Y nX |j� t j ks�t�t j t dd�� |jt jdd� W d Q R X t j tdd�� |jt j d � W d Q R X d S )Nz&value argument not accepted on Windows)r� r_ )r� zis not a valid priority)r4 r5 r r� Z IOPRIO_NORMALrB r� ZIOPRIO_VERYLOWZ IOPRIO_LOWZ IOPRIO_HIGHr[ r+ rC � TypeErrorrr )r8 rE r� r< r< r= �test_ionice_win� s& zTestProcess.test_ionice_winc C s� dd l }tjtj� �}dd� tt�D �}|s4t|��x�|D ]�}tt|�}|dksTt�|t|�kr�|t||�ksrt�trxq:|j |�|j |�ks�t�q:|j |�}t |�dks�t�|d dks�t�|d dks:t�q:W d S )Nr c S s g | ]}|j d �r|�qS )ZRLIMIT)� startswith)�.0r� r< r< r= � <listcomp>� s z/TestProcess.test_rlimit_get.<locals>.<listcomp>r� r_ rp rp ) �resourcer4 r5 r@ rA �dirrB r� r �rlimitZ getrlimitr� )r8 r� rE �namesrq r� �retr< r< r= �test_rlimit_get� s zTestProcess.test_rlimit_getc C s� | j � }|jtjd� |jtj�dks*t�trZtjtdd�� tj j d�jd� W d Q R X tjt�� |jtjd� W d Q R X d S )Nr` zcan't use prlimit)r� r )r` r` )r` r` )r` r` r` ) r> r� r4 Z RLIMIT_NOFILErB r r+ rC rr Z _psplatformr5 )r8 rE r< r< r= �test_rlimit_set� s zTestProcess.test_rlimit_setc C s� t j� }| j� }|jt j�\}}z�|jt jd|f� t|d��}|jdd � W d Q R X tjt ��*}t|d��}|jdd � W d Q R X W d Q R X t r�|j j n|j d t j ks�t�W d |jt j||f� |jt j�||fks�t�X d S )Ni r� � Xi r )r4 r5 r� r� � RLIMIT_FSIZEr� r� r+ rC �IOErrorr r� rY ZEFBIGrB )r8 rE �testfn�soft�hardr� �excr< r< r= � test_rlimit� s "$zTestProcess.test_rlimitc C s� t j� }|jt j�\}}zR|jt jd|f� |jt jt j|f� t| j� d��}|jdd � W d Q R X W d |jt j||f� |jt j�||fks�t�X d S )Ni r� r� i ) r4 r5 r� r� � RLIM_INFINITYr� r� r� rB )r8 rE r� r� r� r< r< r= �test_rlimit_infinity� s z TestProcess.test_rlimit_infinityc C s< t j� }|jt j�\}}|t jks&t�|jt j||f� d S )N)r4 r5 r� r� r� rB )r8 rE r� r� r< r< r= �test_rlimit_infinity_value s z&TestProcess.test_rlimit_infinity_valuec C sr t j� }tr<y |j� }W qD t jk r8 tjd��Y qDX n|j� }t� � |j� }||d ksdt�W d Q R X d S )Nz$on OpenBSD this requires root accessr_ ) r4 r5 r � num_threadsr[ r+ r� r$ rB )r8 rE �step1�step2r< r< r= �test_num_threads s z TestProcess.test_num_threadsz WINDOWS onlyc C s t j� }|j� dkst�d S )Nr )r4 r5 Z num_handlesrB )r8 rE r< r< r= �test_num_handles* s z TestProcess.test_num_handlesc C s� t j� }tr<y |j� }W qD t jk r8 tjd��Y qDX n|j� }t� �d |j� }t|�t|�d kslt �|d }|j |d ks�t �|j |d ks�t �|j |d ks�t �W d Q R X d S )Nz$on OpenBSD this requires root accessr_ r r� ) r4 r5 r �threadsr[ r+ r� r$ r� rB �idr� � system_time)r8 rE r� r� Zathreadr< r< r= � test_threads0 s zTestProcess.test_threadsc C s� | j � }tr:y |j� W n tjk r8 tjd��Y nX t|j� j t dd� |j� D �� �dk sft �t|j� j t dd� |j� D �� �dk s�t �d S )Nz$on OpenBSD this requires root accessc S s g | ] }|j �qS r< )r� )r� r� r< r< r= r� O s z.TestProcess.test_threads_2.<locals>.<listcomp>g�������?c S s g | ] }|j �qS r< )r� )r� r� r< r< r= r� U s ) r> r r� r4 r[ r+ r� r� r� r� �sumrB r� )r8 rE r< r< r= �test_threads_2D s " zTestProcess.test_threads_2c C s� t j� }|j� d d� \}}|j� }|dks0t�|dks<t�d gd }|j� d d� \}}|j� }||ksnt�||kszt�||ks�t�~tr�|j� } | j| jks�t�| j| j ks�t�|j� } x | j D ]} t | | �dks�t�q�W d S )Nr� r i`� ) r4 r5 Z memory_info�memory_percentrB r ZrssZwset�vmsZpagefiler� r� ) r8 rE Zrss1Zvms1Zpercent1ZmemarrZrss2Zvms2Zpercent2�memrq r< r< r= �test_memory_infoZ s& z TestProcess.test_memory_infoc C s� t j� }t j� j}|j� }xB|jD ]8}t||�}|dks<t�|dkrHts"t rNq"||ks"t�q"W t sjt sjt rx|j dksxt�t r�|j dks�t�|jdks�t�d S )Nr r� )r4 r5 Zvirtual_memory�totalZmemory_full_infor� r� rB r r r r �ussZpssZswap)r8 rE r� r� rq r� r< r< r= �test_memory_full_infoy s z!TestProcess.test_memory_full_infoc C s� t j� }|j� }t|�tt|��ks(t�|jdd�}�x |D �]}|jjd�s<tr^d|jkr^q<t jj |j�svt|j��t r�y*t jj |j�s�t jj |j�s�t|j��W nH tk r� ts�� n,td��}|j� }W d Q R X d|j |kr� Y nX q<dt jj|j�kr<yt j|j�}W n tk �r( Y q<X tj|j�s<t|j��q<W xx|D ]p}xh|jD ]^}t||�} |dk�rr�qV|d k�r�| �s�t| ��n&t| ttf��s�t�| d k�sVt| ���qVW �qJW d S ) NF)Zgrouped�[z /bin/qemu-z/proc/self/smapsz %s (deleted)Z64r� �addr�permsr )r� r� )r4 r5 � memory_mapsr� �setrB r� r� r" r@ �isabsr �exists�islinkr r r� �basename�statr �S_ISREG�st_moder� r� r{ �intr ) r8 rE �mapsZext_maps�ntr� �data�st�fnamer� r< r< r= �test_memory_maps� sH z TestProcess.test_memory_mapsc sL t j� }t� �4}dd� � � fdd�|j� D �}� |�|ks>t�W d Q R X d S )Nc S s t jjt jj| ��S )N)r@ r� r� �normcase)rE r< r< r= �normpath� s z8TestProcess.test_memory_maps_lists_lib.<locals>.normpathc s g | ]}� |j ��qS r< )r� )r� r� )r� r< r= r� � s z:TestProcess.test_memory_maps_lists_lib.<locals>.<listcomp>)r4 r5 r& r� rB )r8 rE r� Zlibpathsr< )r� r= �test_memory_maps_lists_lib� s z&TestProcess.test_memory_maps_lists_libc C sN t j� }|j� tjt�� |jdd� W d Q R X ts>ts>trJ|jdd� d S )Nz?!?)Zmemtyper� ) r4 r5 r� r+ rC rr r r r )r8 rE r< r< r= �test_memory_percent� s z TestProcess.test_memory_percentc C sP | j � }|j� st�|j� s t�|j� |j� |j� s>t�|j� sLt�d S )N)r> � is_runningrB rG rH )r8 rE r< r< r= �test_is_running� s zTestProcess.test_is_runningc C s� | j � }|j� }y|tks t�W n� tk r� trbt|�tt�krbtjj}||�|t�ks�t�nNdt j d t j d f }y |j |d�tj |d�ks�t�W n tk r� Y nX Y nX t |ddg�}|dks�t�d S )Nz%s.%sr r_ rU z-czimport os; print('hey')Zhey) r> �exer rB r r� r@ r� r� �sys� version_info�replacer. )r8 rE r r� �ver�outr< r< r= �test_exe� s zTestProcess.test_exec C s� t ddg}| j|�}tr.|j� g kr.tjd��ts:ts:trP|j� d t ks�t�n�t r�t r�|j� d }|t kr�dj |j� dd � �dj |dd � �ks�t�d S t r�dj |j� dd � �dj |�ks�t�d S dj |j� �dj |�ks�t�d S )Nz-cz2import time; [time.sleep(0.1) for x in range(100)]zOPENBSD: returned EBUSYr � r_ r� ) r r> r �cmdliner+ r� r r rB r r �joinr" )r8 r rE �pyexer< r< r= � test_cmdline� s" ,$zTestProcess.test_cmdlinezbroken on PYPYc C s� t g}|jdgd � |jddg� | j|�}trjy|j� |ksDt�W q� tjk rf tj d��Y q�X nHt r�|j� dd � |ks�t�n*|j� }t r�|g kr�tj d��||ks�t�d S )Nz-v�2 z-cz2import time; [time.sleep(0.1) for x in range(100)]z#OPENBSD: process turned into zombier� zOPENBSD: returned EBUSY) r �extendr> r r rB r4 � ZombieProcessr+ r� r" r )r8 r rE r� r< r< r= �test_long_cmdline s z TestProcess.test_long_cmdlinec C sH | j � }|j� j� }tjjtjjtj��j� }|j |�sDt ||f��d S )N) r> rq �lowerr@ r� r� r� r � executabler� rB )r8 rE rq r r< r< r= � test_name( s zTestProcess.test_namezunreliable on PYPYzunreliable on QEMU userc C s� t | jtjd d��}|ddg}| j|�}tr�y |j� tjj |�ksHt �W q� t k r� |j � t j kr�tjj |�j|j� �s�t �n� Y q�X n|j� tjj |�ks�t �d S )Nr� )�suffixz-cz2import time; [time.sleep(0.1) for x in range(100)])r( r� �string�digitsr> r rq r@ r� r� rB �statusr4 � STATUS_ZOMBIEr� )r8 r r rE r< r< r= �test_long_name. s zTestProcess.test_long_namezbroken on SUNOSz broken on AIXzbroken on QEMU userc C sp t | jdd��}|ddg}| j|�}|j� |ks4t�|j� tjj|�ksLt�tjj |j � �tjj |�kslt�d S )Nz foo bar ))r z-cz2import time; [time.sleep(0.1) for x in range(100)]) r( r� r> r rB rq r@ r� r� r� r )r8 r r rE r< r< r= �test_prog_w_funky_nameK s z"TestProcess.test_prog_w_funky_namec C sX t j� }|j� \}}}|tj� ks&t�|tj� ks6t�ttd�rTtj� |j� ksTt�d S )N� getresuid) r4 r5 �uidsr@ �getuidrB �geteuid�hasattrr )r8 rE �real� effective�_savedr< r< r= � test_uids^ s zTestProcess.test_uidsc C sX t j� }|j� \}}}|tj� ks&t�|tj� ks6t�ttd�rTtj� |j� ksTt�d S )Nr ) r4 r5 �gidsr@ �getgidrB �getegidr! � getresgid)r8 rE r" r# r$ r< r< r= � test_gidsl s zTestProcess.test_gidsc s� � fdd�}t j� � tjt�� � jd� W d Q R X � j� }| j||� t�rd }x�t jt j t j t j t j t j gD ]�}| j|d��v y� j|� W n t jk r� Y nLX � j� }|t j t j t j fkr�||ks�|d kr�|}||ks�t�n ||ks�t�W d Q R X qpW n�y�ttd��r0tjtjtj� �� j� k�s0t�� jd� � j� dk�sLt�ttd��rxtjtjtj� �� j� k�sxt�t�s�� jd� � j� dk�s�t�W n t jk �r� Y nX d S )Nc s* y� j | � W n tjk r$ Y nX d S )N)�nicer4 r[ )r� )rE r< r= r� { s z&TestProcess.test_nice.<locals>.cleanup�str)�prio� getpriorityr_ r )r4 r5 r+ rC r� r+ r� r ZIDLE_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZABOVE_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZREALTIME_PRIORITY_CLASS�subTestr[ rB r! r@ r. � PRIO_PROCESSrA r )r8 r� r� Z highest_prior- Znew_prior< )rE r= � test_nicez sV zTestProcess.test_nicec C s t j� }|j� t jkst�d S )N)r4 r5 r �STATUS_RUNNINGrB )r8 rE r< r< r= � test_status� s zTestProcess.test_statusc C s| | j � }|j� }trh|jd�\}}tj� }|jd�r>tjd��||ksJt �dt j krx|t j d ksxt �n|tj� ksxt �d S )N�\�$zrunning as service accountZ USERDOMAIN) r> �usernamer �split�getpassZgetuser�endswithr+ r� rB r@ �environ)r8 rE r6 �domainZ getpass_userr< r< r= � test_username� s zTestProcess.test_usernamec C s | j � }|j� tj� ks t�d S )N)r> �cwdr@ �getcwdrB )r8 rE r< r< r= �test_cwd� s zTestProcess.test_cwdc s( t ddg}| j|�� t� fdd�� d S )Nz-czFimport os, time; os.chdir('..'); [time.sleep(0.1) for x in range(100)]c s � j � tjjtj� �kS )N)r= r@ r� �dirnamer> r< )rE r< r= �<lambda>� s z(TestProcess.test_cwd_2.<locals>.<lambda>)r r> r% )r8 rc r< )rE r= � test_cwd_2� s zTestProcess.test_cwd_2c C s� t j� }|j� }|s t|��| j|j|� ttd�rL|ttj|j ��ksLt�t |�t t |��ksdt�tt t t j dd����}xr|D ]j}|j|g� |j� |gks�t�ttd�r�|j� ttj|j ��ks�t�t|d�r�|j� d |j� ks�t�q�W |jg � t�r|j� |jj� k�s,t�n|j� |k�s,t�ttd��rV|j� ttj|j ��k�sVt�tjt�� |jd� W d Q R X |j|� |jt |�� |jt|�� d S )N�sched_getaffinityT)�percpu�num_cpur r_ )r4 r5 � cpu_affinityrB r� r! r@ �listrC r6 r� r� rz ry rE r �_procZ_get_eligible_cpusr+ rC r� r� )r8 rE �initialZall_cpus�nr< r< r= �test_cpu_affinity� s4 z TestProcess.test_cpu_affinityc C s� | j � }ttjdd��d g}tjt�� |j|� W d Q R X tjt�� |jtdd�� W d Q R X tjt �� |jddg� W d Q R X tjt�� |jdd g� W d Q R X d S ) NT)rD � i' i�* r �1r_ rp ) r> r� r4 r� r+ rC rr rF rz r� )r8 rE Z invalid_cpur< r< r= �test_cpu_affinity_errs s z"TestProcess.test_cpu_affinity_errsc C s� t j� }|j� }|s t|��| j|j|� t|�dkrB|d d� }g }x@tt|�d �D ],}x&tj||�D ]}|rj|j t |�� qjW qXW x.|D ]&}|j|� t |j� �t |�ks�t�q�W d S )N� r_ ) r4 r5 rF rB r� r� rz � itertools� combinations�appendrG �sorted)r8 rE rI Zcombosr� ZsubsetZcombor< r< r= �"test_cpu_affinity_all_combinations s z.TestProcess.test_cpu_affinity_all_combinationsz broken on BSDzunreliable on APPVEYORc sl t j� �| j� }�j� � |� ks$t�t|d���}|jdd � |j� t� �fdd�� �j� � dd� � D �}t j j |�|ks�t�t r�x$� D ] }|j |kr�|j dks�t�q�W W d Q R X x"� D ]}t j j|j �s�t|��q�W d| }| jtd |g��xNtd �D ],}d d� �j� D �}||k�r P tjd � q�W t j j |�|k�sBt�x$|D ] }t j j|��sHt|���qHW d S ) Nr� � xi c s t �j� �t � �kS )N)r� � open_filesr< )�filesrE r< r= rA 7 s z-TestProcess.test_open_files.<locals>.<lambda>c S s g | ]}t jj|j��qS r< )r@ r� r� )r� r� r< r< r= r� 9 s z/TestProcess.test_open_files.<locals>.<listcomp>zIimport time; f = open(r'%s', 'r'); [time.sleep(0.1) for x in range(100)];z-crx c S s g | ]}t jj|j��qS r< )r@ r� r� )r� r� r< r< r= r� J s g{�G�z�?)r4 r5 r� rV rB r� r� �flushr% r@ r� r� r �position�isfiler> r rz rt �sleep)r8 r� r� � filenames�filer r� r< )rW rE r= �test_open_files+ s8 zTestProcess.test_open_filesc C s� t j� }tjj}| j� }t|d���}xJ|j� D ](}||j�||j�ksT|j |j � kr.P q.W | j dt |j� � ��||j�||j�ks�t �tr�|j dks�t �n|j |j � ks�t �|j� d }|d |jks�t �|d |j ks�t �|j|j� ks�t �W d Q R X d S )N�wzno file found; files=%sr_ r rp )r4 r5 r@ r� r� r� r� rV rq �fd�filenoru �reprrB r )r8 rE r� r� �fileobjr] Zntupler< r< r= �test_open_files_2T s$ z TestProcess.test_open_files_2c C s� t j� }| j� }|j� }t|d�}| j|j� |j� |d ksBt�tj� }| j|j� |j� |d ksjt�|j� |j� |j� |ks�t�d S )Nr_ r_ r� ) r4 r5 r� Znum_fdsr� r� �closerB �socket)r8 rE r� �startr] �sockr< r< r= � test_num_fdss s zTestProcess.test_num_fdsz not reliable on OPENBSD & NETBSDc C sV t j� }t|j� �}x2td�D ]&}tjd� t|j� �}||kr d S q W | jd��d S )Nr� g�������?z2num ctx switches still the same after 2 iterations)r4 r5 r� Znum_ctx_switchesrz rt r[ ru )r8 rE �beforer} �afterr< r< r= �test_num_ctx_switches� s z!TestProcess.test_num_ctx_switchesc C sF t j� }ttd�r&|j� tj� ks&t�| j� }|j� tj� ksBt�d S )N�getppid) r4 r5 r! r@ �ppidrm rB r> rA )r8 rE r< r< r= � test_ppid� s zTestProcess.test_ppidc C sD | j � }|j� jtj� ks t�tj� d }tj|�j� d ks@t�d S )Nr ) r> �parentr6 r@ rA rB r4 �pidsr5 )r8 rE Z lowest_pidr< r< r= � test_parent� s zTestProcess.test_parentc C s8 t j� }| j� \}}|j� |ks$t�|j� |ks4t�d S )N)r4 r5 rl rp rB )r8 rp rm rn r< r< r= �test_parent_multi� s z TestProcess.test_parent_multic C s` t j� }|j� st�| j� \}}|j� d |ks4t�|j� d |ksHt�|j� d |ks\t�d S )Nr r_ )r4 r5 �parentsrB rl )r8 rp rm rn r< r< r= � test_parents� s zTestProcess.test_parentsc C s� t j� }|j� g kst�|jdd�g ks,t�| jdd�}|j� }|jdd�}xJ||fD ]>}t|�dksjt�|d j|jks~t�|d j� |jksVt�qVW d S )NT)� recursiver )� creationflagsr_ )r4 r5 �childrenrB r> r� r6 rn )r8 rp rm Z children1Z children2rx r< r< r= � test_children� s zTestProcess.test_childrenc C sf t j� }| j� \}}|j� |gks&t�|jdd�||gks>t�|j� |j� |jdd�g ksbt�d S )NT)rv )r4 r5 rl rx rB rO rH )r8 rp rm rn r< r< r= �test_children_recursive� s z#TestProcess.test_children_recursivec C s� t jt�}x@tj� D ]4}y||j� d7 < W q tjk rF Y qX qW t|j� dd� d�d d }t r~|dkr~t j d��tj |�}y|j dd�}W n tjk r� Y nX t|�tt|��ks�t�d S ) Nr_ c S s | d S )Nr_ r< )r� r< r< r= rA � s z6TestProcess.test_children_duplicates.<locals>.<lambda>)�keyr zPID 0T)rv rp )� collections� defaultdictr� r4 � process_iterrn �ErrorrS �itemsr r+ r� r5 rx r[ r� r� rB )r8 �tablerE r6 �cr< r< r= �test_children_duplicates� s z$TestProcess.test_children_duplicatesc C s| t j� }| j� \}}|jdd�}t|�dks0t�|d |ks@t�|d |ksPt�|j� }|d |ksht�|d |ksxt�d S )NT)rv r� r r_ )r4 r5 rl rx r� rB rt )r8 rp rm rn rx rt r< r< r= �test_parents_and_children� s z%TestProcess.test_parents_and_childrenc C sZ | j � }|j� x*td�D ] }|j� tjkr.P tjd� qW |j� |j� tjksVt �d S )Nrx g{�G�z�?) r> ri rz r r4 �STATUS_STOPPEDrt r[ rj rB )r8 rE r} r< r< r= �test_suspend_resume� s z TestProcess.test_suspend_resumec C sD t jt�� tjd� W d Q R X t jt�� tjd� W d Q R X d S )NrM r_ rp )r+ rC r� r4 r5 rr )r8 r< r< r= �test_invalid_pid� s z TestProcess.test_invalid_pidc B C s t j� }|jddgd�}t|j� �ddgks0t�t jtt j� ��}|jdgdd�}t|d t �sp|d dkspt�t j ddt j d ��" |jd gd d�d d iks�t�W d Q R X t j ddt j |jd�d ��, tjt j �� |jd gd� W d Q R X W d Q R X t j ddt j|jd�d ��$ |jd gdd�d dik�s0t�W d Q R X t j ddtd ��H |j� }d t |j� �k�sjt�tjt�� |jd gd� W d Q R X W d Q R X tjt�� |jd� W d Q R X tjt�� |jdg� W d Q R X tjt�� |jdd g� W d Q R X d S ) Nr rq )�attrsZnet_connections�foo)r� Zad_valuezpsutil.Process.niceT)�createrV r+ r_ Zbar)r4 r5 Zas_dictrS �keysrB r� rq r{ rG r) rW r[ r7 r6 r+ rC r �NotImplementedErrorr� rr )r8 rE �dr< r< r= � test_as_dict� sB &"( " zTestProcess.test_as_dictc C s� t j� }tjd��8}|j� � |j� |j� W d Q R X |jdksFt�W d Q R X tjd��}|j� |j� W d Q R X |jdks�t�d S )Nz$psutil._psplatform.Process.cpu_timesr_ r� )r4 r5 r) rW �oneshotr� � call_countrB )r8 rE r� r< r< r= � test_oneshot/ s zTestProcess.test_oneshotc % C s� t j� }tjd���}tjd��j}|j� �: |j� |j� |j� � |j� |j� W d Q R X W d Q R X |jdksvt�|jdks�t�W d Q R X W d Q R X tjd��}|j� |j� W d Q R X |jdks�t�d S )Nz$psutil._psplatform.Process.cpu_timesz(psutil._psplatform.Process.oneshot_enterr_ r� )r4 r5 r) rW r� r� r� rB )r8 rE �m1�m2r� r< r< r= �test_oneshot_twice< s " z TestProcess.test_oneshot_twicec C s� | j � \}}|j� }|j� }||ks(t�|j� �& |j� |ksBt�|j� |ksRt�W d Q R X |j� �& |j� |ksvt�|j� |ks�t�W d Q R X d S )N)rl rn rB r� )r8 �p1�p2Zp1_ppidZp2_ppidr< r< r= �test_oneshot_cacheP s z TestProcess.test_oneshot_cachec sr �fdd�}�j � � � j� � j� tr8t� fdd�� �j� � t� �}x"|j|j�D ]\}}|||� qXW d S )Nc s� y | � }W nR t jk r" � Y n^ t jk r6 Y nJ t jk r\ trV|dkrVd S � Y n$X trn|dkrnd S � jd| |f ��d S )Nr� r� r rq z+%r didn't raise NSP and returned %r instead)r� r� )r rq )r4 r r7 r[ r r ru )�funZfun_namer� )r8 r< r= �assert_raises_nspg s zFTestProcess.test_halfway_terminated_process.<locals>.assert_raises_nspc s � j tj� kS )N)r6 r4 rq r< )rE r< r= rA ~ s z=TestProcess.test_halfway_terminated_process.<locals>.<lambda>) r> rO rH r r% rL r* �iter�all)r8 r� �nsr� rq r< )rE r8 r= � test_halfway_terminated_process_ s z+TestProcess.test_halfway_terminated_processc C s | j � \}}| j|� d S )N)Z spawn_zombieZassertProcessZombie)r8 Z_parent�zombier< r< r= �test_zombie_process� s z TestProcess.test_zombie_processc C sB t j� }tjdt jd�d�� }|j� s*t�|js4t�W d Q R X d S )Nzpsutil.Processr )rV )r4 r5 r) rW r r rB r� )r8 rE r� r< r< r= �$test_zombie_process_is_running_w_exc� s z0TestProcess.test_zombie_process_is_running_w_excc C sH t j� }tjdt jd�d��"}|j� t jks0t�|js:t�W d Q R X d S )Nz!psutil._psplatform.Process.statusr )rV ) r4 r5 r) rW r r r rB r� )r8 rE r� r< r< r= � test_zombie_process_status_w_exc� s z,TestProcess.test_zombie_process_status_w_excc , C s t rddlm} n ddlm} | j� }tj|j�}|j|j� d f|_t tj � � |jtj ksbt �|j � spt �tjjtjdd��( t|� ��}t tj � � W d Q R X W d Q R X d|j |j� ks�t �|jtj ks�t �|tj|j�ks�t �d}t|�}x^|j|j|j dd �D ]D\}}| j|d ��( tjtj|d �� |� W d Q R X W d Q R X �qW d t |�k�sht �d t |�k�szt �tjtj|d �� |j � W d Q R X tjtj|d �� |j � W d Q R X tjtj|d �� |j � W d Q R X tjtj|d �� |j!� W d Q R X d S ) Nr )�StringIOrx � PSUTIL_DEBUGTz-refreshing Process instance for reused PID %sz4process no longer exists and its PID has been reusedF)� clear_cache)rq )r� zterminated + PID reused)"r �ior� r3 r4 r5 r6 r� �_identrG r~ Z_pmaprB r r) rW �object�_commonr �getvaluer* r� �settersZkillersr/ r+ rC r7 r, rb rn rp rt rx ) r8 r� �subprE r� �msgr� r� rq r< r< r= �test_reused_pid� s@ zTestProcess.test_reused_pidc = C s dt j� krXtjt j�� t jd� W d Q R X t jd� s>t�t jd�j� dksTt�d S t jd�}t rlt j nt }tj|�� |j � W d Q R X tj|�� |j � W d Q R X tj|�� |j� W d Q R X tj|�� |j� W d Q R X tj|�� |j� W d Q R X tj|�� |jtj� W d Q R X t|�}x�|j|j|j �D ]�\}}y |� }W n t j k �rn Y nbX |d k�r�|jdk�s�t�nD|dk�r�t �r�dnd}|j� |k�s�t�n|dk�rD|�sDt|���qDW t�s�dt j� k�s�t�t jd��s�t�d S ) Nr r_ r r& r6 zNT AUTHORITY\SYSTEM�rootrq )r r& )r4 rq r+ rC r7 r5 Z pid_existsrB rn r r[ rr rH rO ri rj rG rQ rI rJ r* r� Zgettersr� r" r6 r )r8 rE r� r� r� rq r� r� r< r< r= � test_pid_0� sH zTestProcess.test_pid_0c C sJ dd� }d | _ tj� }||j� �}|tjj� �}t rFtrF||ksFt�d S )Nc S sR ddddg}t r"|jddddg� x|D ]}| j|d � q(W tdd � | j� D ��S ) N�PLAT�HOMEZPYTEST_CURRENT_TESTZPYTEST_VERSIONZ__CF_USER_TEXT_ENCODINGZ VERSIONER_PYTHON_PREFER_32_BITZVERSIONER_PYTHON_VERSIONc S s8 g | ]0\}}|j d d�j dd�|j d d�j dd�f�qS )� rU � )r )r� �k�vr< r< r= r� s z@TestProcess.test_environ.<locals>.clean_dict.<locals>.<listcomp>)r r �pop�dictr� )r� �excluderq r< r< r= � clean_dict� s z,TestProcess.test_environ.<locals>.clean_dict) ZmaxDiffr4 r5 r: r@ �copyr r rB )r8 r� rE Zd1Zd2r< r< r= � test_environ� s zTestProcess.test_environz<macOS 11+ can't get another process environment, issue #2084z(sometimes fails on `assert is_running()`c C s� t jd�}t| j� |d�}| j|gtjtjd�}tj|j �}t |j � |j � sRt �|j j� dksdt �tr�tr�y |j� }W q� tjk r� d S X n|j� }|ddd�ks�t �|j� |jdks�t �d S ) Na� #include <unistd.h> #include <fcntl.h> char * const argv[] = {"cat", 0}; char * const envp[] = {"A=1", "X", "C=3", 0}; int main(void) { // Close stderr on exec so parent can wait for the // execve to finish. if (fcntl(2, F_SETFD, FD_CLOEXEC) != 0) return 0; return execve("/bin/cat", argv, envp); } )�c_code)�stdinr^ � rM �3)�A�Cr )�textwrap�dedentr' r� r3 ra rb r4 r5 r6 r1 r rB r^ r� r r r: r[ � communicate� returncode)r8 rM Zcexer; rE �envr< r< r= �test_weird_environ s$ z TestProcess.test_weird_environN)m�__name__� __module__� __qualname__�__doc__r> rF rN rP rS r+ �mark�skipifr r\ rd r rk ro rs rv r r� r" r� r� r r� r� r� r r0 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 r r r r r r r r r r% r* r1 r3 r< r? rB r rK rN rT r r r^ rd ri r rl ro rr rs ru ry rz r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r r� r< r< r< r= r2 P s� * , 0 B ' ' 4 & .. r2 c sr e Zd ZdZeed�r&ej� Zej� Z � fdd�Z � fdd�Z � fdd�Z d d � Z ejjd d d �dd� �Z� ZS )�LimitedUserTestCasez�Repeat the previous tests by using a limited user. Executed only on UNIX and only if the user who run the test script is root. r c sX t � j||� xDdd� t| �D �D ].}t| |�� � fdd�}t| |tj|| �� q"W d S )Nc S s g | ]}|j d �r|�qS )�test)r� )r� r� r< r< r= r� ` s z0LimitedUserTestCase.__init__.<locals>.<listcomp>c s& y � � W n t jk r Y nX d S )N)r4 r[ )r8 )�methr< r= �test_c s z+LimitedUserTestCase.__init__.<locals>.test_)r �__init__r� r� �setattr�types� MethodType)r8 r9 r: �attrr� )� __class__)r� r= r� \ s z LimitedUserTestCase.__init__c s" t � j� tjd� tjd� d S )Ni� )r �setUpr@ �setegid�seteuid)r8 )r� r< r= r� k s zLimitedUserTestCase.setUpc s& t j| j� t j| j� t� j� d S )N)r@ r� � PROCESS_UIDr� � PROCESS_GIDr �tearDown)r8 )r� r< r= r� p s z LimitedUserTestCase.tearDownc C s8 yt j� jd� W n t jk r( Y n X | jd��d S )Nr_ zexception not raisedrp )r4 r5 r+ r[ ru )r8 r< r< r= r1 u s z LimitedUserTestCase.test_niceTzcauses problem as root)rT c C s d S )Nr< )r8 r< r< r= r� } s z'LimitedUserTestCase.test_zombie_process)r� r� r� r� r! r@ r r� r' r� r� r� r� r1 r+ r� r� r� � __classcell__r< r< )r� r= r� Q s r� c @ s4 e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd S ) � TestPopenz Tests for psutil.Popen class.c C s t � d S )N)r, )�clsr< r< r= � tearDownClass� s zTestPopen.tearDownClassc C s� t ddg}tj|tjtjtd��L}|j� |j� |j t |�sDt �t j t �� |j W d Q R X |j� W d Q R X tr�|jd�tj ks�t �n|jd�tjks�t �d S )Nz-cz3import time; [time.sleep(0.1) for x in range(100)];)�stdoutr^ r� r` )r r4 �Popenra rb r! rq r� r� r� rB r+ rC rD r� rO r rH rI rJ )r8 rc �procr< r< r= � test_misc� s& zTestPopen.test_miscc C sh t jtdgtjtjtjtd��}|j� W d Q R X |jjs>t �|j jsJt �|j jsVt �|j dksdt �d S )Nz-V)r� r^ r� r� r ) r4 r� r ra rb r! r� r� �closedrB r^ r� r� )r8 r� r< r< r= �test_ctx_manager� s zTestPopen.test_ctx_managerc , C s� t ddg}tj|tjtjtd���}|j� |j� tj tj �� |j� W d Q R X tj tj �� |j � W d Q R X tj tj �� |j t j� W d Q R X tr�tj tj �� |j t j� W d Q R X tj tj �� |j t j� W d Q R X W d Q R X d S )Nz-cz3import time; [time.sleep(0.1) for x in range(100)];)r� r^ r� )r r4 r� ra rb r! rO rH r+ rC r7 rG rQ rI rJ r Z CTRL_C_EVENTZCTRL_BREAK_EVENT)r8 rc r� r< r< r= �test_kill_terminate� s, z TestPopen.test_kill_terminateN) r� r� r� r� � classmethodr� r� r� r� r< r< r< r= r� � s r� )Gr� r| rY r8 rP r@ rI rf r� r ra r r� rt r� r4 r r r r r r r r r r Zpsutil._commonr Zpsutil._compatr r r r r Z psutil.testsr 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- r. r/ r0 r1 r2 r r� r� r< r< r< r= �<module> s� 6
cải xoăn