관리-도구
편집 파일: test_process.cpython-38.pyc
U �Mgn� � @ 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�e�D� dk�r�G d5d6� d6eC�ZEG d7d8� d8e4�ZFdS )9� 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 )�� TestProcessr c O sD | j ||�}zt�|j�W S tjk r> | �|j� � Y nX d S �N)�spawn_testproc�psutil�Process�pid� NoSuchProcessZ assertPidGone)�self�args�kwargs�sproc� r? �I/opt/hc_python/lib64/python3.8/site-packages/psutil/tests/test_process.py� spawn_psprocS s zTestProcess.spawn_psprocc C s: t �� }|jt�� kst�t�t�� d|_W 5 Q R X d S )N�! ) r7 r8 r9 �os�getpid�AssertionErrorr, �raises�AttributeError�r; �pr? r? r@ �test_pid] s zTestProcess.test_pidc C sJ | � � }|�� |�� }tr,|tjks<t�n|tj ks<t�| �|� d S r5 ) rA �kill�waitr �signal�SIGTERMrE �SIGKILL�assertProcessGone�r; rI �coder? r? r@ � test_killc s zTestProcess.test_killc C sJ | � � }|�� |�� }tr,|tjks<t�n|tj ks<t�| �|� d S r5 )rA � terminaterL r rM rN rE rP rQ r? r? r@ �test_terminatem s zTestProcess.test_terminatec C sX t r tjntj}| �� }|�|� |�� }tr<||ksJt�n|| ksJt�| � |� d S r5 ) r rM rO rN rA � send_signalrL r rE rP )r; �sigrI rR r? r? r@ �test_send_signalw s z TestProcess.test_send_signalz not POSIX��reasonc C s� t j}| �� }tjdttjd�d��( t� t j �� |� |� W 5 Q R X W 5 Q R X | �� }tjdttj d�d��( t� t j�� |� |� W 5 Q R X W 5 Q R X d S )Nzpsutil.os.kill� �� side_effect)rM rN rA r* �patch�OSError�errnoZESRCHr, rF r7 r: rV �EPERM� AccessDenied)r; rW rI r? r? r@ �test_send_signal_mocked� s � �z#TestProcess.test_send_signal_mockedc C s� t ddg}| �|�}|�� }|dks(t�| �|� t ddg}| j|tjd�}|�� }|dks`t�| �|� t ddg}| �|�}|�� }|dks�t�| �|� t dd g}| �|�}|�� }|dks�t�| �|� d S ) N�-c�passr z1 / 0)�stderr� zimport sys; sys.exit(5);� zimport os; os._exit(5);)r! rA rL rE rP � subprocess�PIPE)r; �cmdrI rR r? r? r@ �test_wait_exited� s( z TestProcess.test_wait_exitedzfails on NETBSDc C s. | � � }tr�|�tj� t�tj�� |j dd� W 5 Q R X |�tj � t�tj�� |j dd� W 5 Q R X |�tj � |� � tj ks�t �|� � tj ks�t �n�|� � t�tj�� |j dd� W 5 Q R X |�� t�tj�� |j dd� W 5 Q R X |�� |� � tj k�st �|� � tj k�s*t �d S )N���MbP?)�timeout)rA r rV rM �SIGSTOPr, rF r7 �TimeoutExpiredrL �SIGCONTrN rE �suspend�resumerT rH r? r? r@ �test_wait_stopped� s( z TestProcess.test_wait_stoppedc C s� | � � \}}t�tj�� |�d� W 5 Q R X t�tj�� |�d� W 5 Q R X |�� |�� |�� }|�� }tr�|tj ks�t �|d ks�t �n |tj ks�t �|tj ks�t �d S )N�{�G�z�?) �spawn_children_pairr, rF r7 rp rL rT r rM rN rE )r; �child� grandchildZ child_retZgrandchild_retr? r? r@ �test_wait_non_children� s z"TestProcess.test_wait_non_childrenc C sx | � � }|�� t�tj�� |�d� W 5 Q R X t�tj�� |�d� W 5 Q R X t�t�� |�d� W 5 Q R X d S )Nru r ���)rA �namer, rF r7 rp rL � ValueErrorrH r? r? r@ �test_wait_timeout� s z TestProcess.test_wait_timeoutc C s� | � � }t�tj�� |�d� W 5 Q R X |�� t�� t }t�� |k rvz|�d�}W q�W q> tjk rr Y q>X q>| � d��t r�|t j ks�t �n|t jks�t �| �|� d S )Nr rn )rA r, rF r7 rp rL rK �timer �failr rM rO rE rN rP )r; rI Zstop_atrR r? r? r@ � test_wait_timeout_nonblocking� s z)TestProcess.test_wait_timeout_nonblockingc C sz t �� }|jdd� |jdd� td�D ]*}|jd d�}t|t�sFt�|dks(t�q(t�t �� |jdd� W 5 Q R X d S )Nrm )�interval�d � rz ) r7 r8 � cpu_percent�range� isinstance�floatrE r, rF r| )r; rI �_�percentr? r? r@ �test_cpu_percent� s z TestProcess.test_cpu_percentc C s4 t jdd d�� }t�� �� |js&t�W 5 Q R X d S )Nzpsutil.cpu_count)Z return_value)r* r^ r7 r8 r� �calledrE )r; �mr? r? r@ � test_cpu_percent_numcpus_none s z)TestProcess.test_cpu_percent_numcpus_nonezQEMU user not supportedc C s� t �� �� }|jdks t|��|jdks0t|��|jdksBt|��|jdksTt|��trj|j dksjt|��|j D ] }t � dt � t||��� qpd S )Nr� z%H:%M:%S)r7 r8 � cpu_times�userrE �system� children_user�children_systemr Ziowait�_fieldsr~ �strftime� localtime�getattr)r; �timesr{ r? r? r@ �test_cpu_times s zTestProcess.test_cpu_timesc C s� t �� �� d d� \}}t�� d d� \}}t||g�t||g� dkrZ| �d||f ��t||g�t||g� dkr�| �d||f ��d S )N� 皙�����?zexpected: %s, found: %s)r7 r8 r� rC r� �max�minr )r; � user_timeZ kernel_time�utimeZktimer? r? r@ �test_cpu_times_2 s z TestProcess.test_cpu_times_2z not supportedc C sP t �� }|�� }|dks t�t �� dkr4|dks4t�|�� tt �� �ksLt�d S �Nr rg )r7 r8 Zcpu_numrE � cpu_countr� )r; rI �numr? r? r@ � test_cpu_num) s zTestProcess.test_cpu_numc C sZ | � � }t�� }|�� }t|| �}|dkr@| �d|||f ��t�dt�|�� �� d S )Nr� z'expected: %s, found: %s, difference: %sz%Y %m %d %H:%M:%S)rA r~ � create_time�absr r� r� )r; rI �nowr� � differencer? r? r@ �test_create_time2 s ��z TestProcess.test_create_timez POSIX onlyc C sX t �� �� }|d k rTztj�td��}W n tk rF t� d��Y nX ||ksTt �d S )N�ttyzcan't rely on `tty` CLI) r7 r8 �terminalrC �path�realpathr0 � RuntimeErrorr, �skiprE )r; r� r� r? r? r@ � test_terminalD s zTestProcess.test_terminal)Zonly_ifc C s� t �� }|�� }ttd��}|�� W 5 Q R X |�� }ts�ts�|j|jksNt �|j |j ks^t �t r�|j |j ksrt �|j |j ks�t �n |j|jks�t �|j|jks�t �|�� }t| �� d��.}tr�|�tdd d�� n|�dd � W 5 Q R X |�� }|j |j k�st �|j|jk�st �|j|jk�s,t �|j|jk�s>t �t �rh|j |j k�sVt �|j |j k�sht �tt|��D ]>}t�r�|dk�r��qt|| dk�s�t �|| dk�stt ��qtd S )N�rb�wb�xi@B �asciir� r )r7 r8 Z io_counters�openr! �readr r Z read_countrE Z write_countr Z read_charsZ write_chars� read_bytes� write_bytes� get_testfnr �write�bytesr� �len)r; rI Zio1�fZio2�ir? r? r@ �test_io_countersP s> z TestProcess.test_io_countersz linux onlyc s� � fdd�}t �� � ts.� �� d t jks.t�t jdks<t�t jdksJt�t jdksXt�t jdksft�� �� }| � ||� � �t j� t � �� �t jdfks�t�t � t �� � jt jdd� W 5 Q R X � �t j� t � �� �t jdfks�t�� jt jdd� t � �� �t jdfk�st�t � t �� � jt jd d� W 5 Q R X z� jt jdd� W n t jk �rj Y nX t j t d d �� � �t jd� W 5 Q R X t j t d d �� � �t jd� W 5 Q R X t j t d d �� � jdd� W 5 Q R X d S ) Nc s& | \}}|t jkrd}� �||� d S �Nr )r7 �IOPRIO_CLASS_NONE�ionice)�initZioclass�value�rI r? r@ �cleanup~ s z.TestProcess.test_ionice_linux.<locals>.cleanupr rg r� � � �r� � zioclass accepts no value��matchz$'ioclass' argument must be specified)r7 r8 r r� r� rE ZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE� addCleanup�tupler, rF r| rb )r; r� r� r? r� r@ �test_ionice_linux{ sB �z TestProcess.test_ionice_linuxz!not supported on this win versionc C s t �� }ts |�� t jks t�|�� }| �|j|� |�t j� |�� t jksRt�|�t j� |�� t jkspt�z|�t j � W n t j k r� Y nX |�� t j ks�t�t j t dd�� |jt jdd� W 5 Q R X t j tdd�� |�t j d � W 5 Q R X d S )Nz&value argument not accepted on Windowsr� rg r� zis not a valid priority)r7 r8 r r� Z IOPRIO_NORMALrE r� ZIOPRIO_VERYLOWZ IOPRIO_LOWZ IOPRIO_HIGHrb r, rF � TypeErrorr| )r; rI r� r? r? r@ �test_ionice_win� s* �zTestProcess.test_ionice_winc C s� dd l }t�t�� �}dd� tt�D �}|s4t|��|D ]�}tt|�}|dksRt�|t|�kr�|t||�kspt�trvq8|� |�|� |�ks�t�q8|� |�}t |�dks�t�|d dks�t�|d dks8t�q8d S )Nr c S s g | ]}|� d �r|�qS )ZRLIMIT�� startswith��.0r� r? r? r@ � <listcomp>� s z/TestProcess.test_rlimit_get.<locals>.<listcomp>r� rz rg ) �resourcer7 r8 rC rD �dirrE r� r �rlimitZ getrlimitr� )r; r� rI �namesr{ r� �retr? r? r@ �test_rlimit_get� s zTestProcess.test_rlimit_getc C s� | � � }|�tjd� |�tj�dks*t�trZtjtdd�� tj � d��d� W 5 Q R X t�t�� |�tjd� W 5 Q R X d S )N)rh rh zcan't use prlimitr� r )rh rh rh ) rA r� r7 Z RLIMIT_NOFILErE r r, rF r| Z _psplatformr8 rH r? r? r@ �test_rlimit_set� s zTestProcess.test_rlimit_setc C s� t �� }| �� }|�t j�\}}z�|�t jd|f� t|d��}|�d� W 5 Q R X t� t ��&}t|d��}|�d� W 5 Q R X W 5 Q R X t r�|j j n|j d t jks�t�W 5 |�t j||f� |�t j�||fks�t�X d S )N� r� s XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXs XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXr )r7 r8 r� r� � RLIMIT_FSIZErE r� r� r, rF �IOErrorr r� r` ZEFBIG)r; rI �testfn�soft�hardr� �excr? r? r@ � test_rlimit� s $zTestProcess.test_rlimitc C s� t �� }|�t j�\}}zN|�t jd|f� |�t jt j|f� t| �� d��}|�d� W 5 Q R X W 5 |�t j||f� |�t j�||fks�t�X d S )Nr� r� s XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX) r7 r8 r� r� rE � RLIM_INFINITYr� r� r� )r; rI r� r� r� r? r? r@ �test_rlimit_infinity� s z TestProcess.test_rlimit_infinityc C s< t �� }|�t j�\}}|t jks&t�|�t j||f� d S r5 )r7 r8 r� r� r� rE )r; rI r� r� r? r? r@ �test_rlimit_infinity_value s z&TestProcess.test_rlimit_infinity_valuec C sr t �� }tr<z |�� }W qD t jk r8 t�d��Y qDX n|�� }t� � |�� }||d ksdt�W 5 Q R X d S )N�$on OpenBSD this requires root accessrg ) r7 r8 r � num_threadsrb r, r� r% rE )r; rI �step1�step2r? r? r@ �test_num_threads s z TestProcess.test_num_threadsz WINDOWS onlyc C s t �� }|�� dkst�d S r� )r7 r8 Z num_handlesrE rH r? r? r@ �test_num_handles* s z TestProcess.test_num_handlesc C s� t �� }tr<z |�� }W qD t jk r8 t�d��Y qDX n|�� }t� �d |�� }t|�t|�d kslt �|d }|j |d ks�t �|j |d ks�t �|j |d ks�t �W 5 Q R X d S )Nr� rg r r� ) r7 r8 r �threadsrb r, r� r% r� rE �idr� � system_time)r; rI r� r� Zathreadr? r? r@ � test_threads0 s zTestProcess.test_threadsc C s� | � � }tr:z |�� W n tjk r8 t�d��Y nX t|�� j t dd� |�� D �� �dk sft �t|�� j t dd� |�� D �� �dk s�t �d S )Nr� c S s g | ] }|j �qS r? )r� r� r? r? r@ r� O s z.TestProcess.test_threads_2.<locals>.<listcomp>r� c S s g | ] }|j �qS r? )r� r� r? r? r@ r� U s ) rA r r� r7 rb r, r� r� r� r� �sumrE r� rH r? r? r@ �test_threads_2D s$ "������zTestProcess.test_threads_2c C s� t �� }|�� d d� \}}|�� }|dks0t�|dks<t�d gd }|�� d d� \}}|�� }||ksnt�||kszt�||ks�t�~tr�|�� } | j| jks�t�| j| j ks�t�|�� } | j D ]} t | | �dks�t�q�d S )Nr� r i`� ) r7 r8 Z memory_info�memory_percentrE r ZrssZwset�vmsZpagefiler� r� ) r; rI Zrss1Zvms1Zpercent1ZmemarrZrss2Zvms2Zpercent2�memr{ r? r? r@ �test_memory_infoZ s& z TestProcess.test_memory_infoc C s� t �� }t �� j}|�� }|jD ]8}t||�}|dks:t�|dkrFts t rLq ||ks t�q t sft sft rt|j dkstt�t r�|j dks�t�|jdks�t�d S )Nr r� )r7 r8 Zvirtual_memory�totalZmemory_full_infor� r� rE r r r r �ussZpssZswap)r; rI r� r� r{ r� r? r? r@ �test_memory_full_infoy s z!TestProcess.test_memory_full_infoc C s� t �� }|�� }t|�tt|��ks(t�|jdd�}|D �]}|j�d�s8trZd|jkrZq8t j� |j�srt|j��t r�z*t j� |j�s�t j� |j�s�t|j��W nH tk r� ts�� n,td��}|�� }W 5 Q R X d|j |kr� Y nX q8dt j�|j�kr8zt �|j�}W n tk �r$ Y q8X t�|j�s8t|j��q8|D ]l}|jD ]^}t||�} |dk�rh�qL|d k�r�| �s�t| ��n&t| ttf��s�t�| d k�sLt| ���qL�qBd S ) NF)�grouped�[z /bin/qemu-z/proc/self/smapsz %s (deleted)Z64r� )�addrZpermsr )r7 r8 � memory_mapsr� �setrE r� r� r# rC �isabsr �exists�islinkr r r� �basename�statr �S_ISREG�st_moder� r� r� �intr ) r; rI �mapsZext_maps�ntr� �data�st�fnamer� r? r? r@ �test_memory_maps� sL �� z TestProcess.test_memory_mapsc sL t �� }t� �4}dd� � � fdd�|�� D �}� |�|ks>t�W 5 Q R X d S )Nc S s t j�t j�| ��S r5 )rC r� r� �normcaser� r? r? r@ �normpath� s z8TestProcess.test_memory_maps_lists_lib.<locals>.normpathc s g | ]}� |j ��qS r? )r� r� �r r? r@ r� � s z:TestProcess.test_memory_maps_lists_lib.<locals>.<listcomp>)r7 r8 r' r� rE )r; rI r� Zlibpathsr? r r@ �test_memory_maps_lists_lib� s z&TestProcess.test_memory_maps_lists_libc C sN t �� }|�� t�t�� |jdd� W 5 Q R X ts>ts>trJ|jdd� d S )Nz?!?)Zmemtyper� ) r7 r8 r� r, rF r| r r r rH r? r? r@ �test_memory_percent� s z TestProcess.test_memory_percentc C sL | � � }|�� st�|�� s t�|�� |�� |�� r<t�|�� rHt�d S r5 )rA � is_runningrE rK rL rH r? r? r@ �test_is_running� s zTestProcess.test_is_runningc C s� | � � }|�� }z|tks t�W n� tk r� trbt|�tt�krbtjj}||�|t�ks�t�nNdt j d t j d f }z |� |d�t� |d�ks�t�W n tk r� Y nX Y nX t |ddg�}|dks�t�d S )Nz%s.%sr rg r[ rd zimport os; print('hey')Zhey) rA �exer! rE r r� rC r� r �sys� version_info�replacer0 )r; rI r r �ver�outr? r? r@ �test_exe� s zTestProcess.test_exec C s� t ddg}| �|�}tr.|�� g kr.t�d��ts:ts:trP|�� d t ks�t�n�t r�t r�|�� d }|t kr�d� |�� dd � �d� |dd � �ks�t�d S t r�d� |�� dd � �d� |�ks�t�d S d� |�� �d� |�ks�t�d S )Nrd �2import time; [time.sleep(0.1) for x in range(100)]�OPENBSD: returned EBUSYr � rg r� ) r! rA r �cmdliner, r� r r rE r r �joinr# )r; r rI �pyexer? r? r@ � test_cmdline� s$ � ,$zTestProcess.test_cmdlinezbroken on PYPYc C s� t g}|�dgd � |�ddg� | �|�}trjz|�� |ksDt�W q� tjk rf t� d��Y q�X nHt r�|�� dd � |ks�t�n*|�� }t r�|g kr�t� d��||ks�t�d S )Nz-v�2 rd r z#OPENBSD: process turned into zombier� r ) r! �extendrA r r rE r7 � ZombieProcessr, r� r# r )r; r rI r� r? r? r@ �test_long_cmdline s" � z TestProcess.test_long_cmdlinec C sH | � � }|�� �� }tj�tj�tj���� }|� |�sDt ||f��d S r5 ) rA r{ �lowerrC r� r r� r � executabler� rE )r; rI r{ r! r? r? r@ � test_name( s zTestProcess.test_namezunreliable on PYPYzunreliable on QEMU userc C s� t | jtjd d��}|ddg}| �|�}tr�z |�� tj� |�ksHt �W q� t k r� |� � t j kr�tj� |��|�� �s�t �n� Y q�X n|�� tj� |�ks�t �d S )Nr� ��suffixrd r )r) r� �string�digitsrA r r{ rC r� r rE �statusr7 � STATUS_ZOMBIEr� �r; r! r rI r? r? r@ �test_long_name. s � zTestProcess.test_long_namezbroken on SUNOSz broken on AIXzbroken on QEMU userc C sp t | jdd��}|ddg}| �|�}|�� |ks4t�|�� tj�|�ksLt�tj� |� � �tj� |�kslt�d S )Nz foo bar )r* rd r ) r) r� rA r rE r{ rC r� r r r r0 r? r? r@ �test_prog_w_funky_nameK s � z"TestProcess.test_prog_w_funky_namec C sX t �� }|�� \}}}|t�� ks&t�|t�� ks6t�ttd�rTt�� |�� ksTt�d S �N� getresuid) r7 r8 �uidsrC �getuidrE �geteuid�hasattrr4 �r; rI �realZ effectiveZ_savedr? r? r@ � test_uids^ s zTestProcess.test_uidsc C sX t �� }|�� \}}}|t�� ks&t�|t�� ks6t�ttd�rTt�� |�� ksTt�d S r3 ) r7 r8 �gidsrC �getgidrE �getegidr8 � getresgidr9 r? r? r@ � test_gidsl s zTestProcess.test_gidsc s� � fdd�}t �� � t�t�� � �d� W 5 Q R X � �� }| �||� tr�d }t jt j t j t j t j t j fD ]�}| j|d��v z� �|� W n t jk r� Y nLX � �� }|t j t j t j fkr�||ks�|d kr�|}||ks�t�n ||ks�t�W 5 Q R X qln�z�ttd��r*t�tjt�� �� �� k�s*t�� �d� � �� dk�sFt�ttd��rrt�tjt�� �� �� k�srt�t�s�� �d� � �� dk�s�t�W n t jk �r� Y nX d S )Nc s* z� � | � W n tjk r$ Y nX d S r5 )�nicer7 rb )r� r� r? r@ r� { s z&TestProcess.test_nice.<locals>.cleanup�str)�prio� getpriorityrg r )r7 r8 r, rF r� rA r� r ZIDLE_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZABOVE_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZREALTIME_PRIORITY_CLASS�subTestrb rE r8 rC rD � PRIO_PROCESSrD r )r; r� r� Z highest_priorC Znew_prior? r� r@ � test_nicez s` �� �� �� zTestProcess.test_nicec C s t �� }|�� t jkst�d S r5 )r7 r8 r. ZSTATUS_RUNNINGrE rH r? r? r@ � test_status� s zTestProcess.test_statusc C s| | � � }|�� }trh|�d�\}}t�� }|�d�r>t�d��||ksJt �dt j krx|t j d ksxt �n|t�� ksxt �d S )N�\�$zrunning as service accountZ USERDOMAIN) rA �usernamer �split�getpass�getuser�endswithr, r� rE rC �environ)r; rI rK �domainZ getpass_userr? r? r@ � test_username� s zTestProcess.test_usernamec C s | � � }|�� t�� ks t�d S r5 )rA �cwdrC �getcwdrE rH r? r? r@ �test_cwd� s zTestProcess.test_cwdc s( t ddg}| �|�� t� fdd�� d S )Nrd zFimport os, time; os.chdir('..'); [time.sleep(0.1) for x in range(100)]c s � � � tj�t�� �kS r5 )rS rC r� �dirnamerT r? r� r? r@ �<lambda>� � z(TestProcess.test_cwd_2.<locals>.<lambda>)r! rA r& )r; rk r? r� r@ � test_cwd_2� s � zTestProcess.test_cwd_2c C s� t �� }|�� }|s t|��| �|j|� ttd�rL|tt�|j ��ksLt�t |�t t |��ksdt�tt t t j dd����}|D ]j}|�|g� |�� |gks�t�ttd�r�|�� tt�|j ��ks�t�t|d�r�|�� d |�� ks�t�q�|�g � t�r|�� |j�� k�s(t�n|�� |k�s(t�ttd��rR|�� tt�|j ��k�sRt�t�t�� |�d� W 5 Q R X |�|� |�t |�� |�t|�� d S )N�sched_getaffinityT�Zpercpu�num_cpur rg )r7 r8 � cpu_affinityrE r� r8 rC �listrZ r9 r� r� r� r� r\ r �_procZ_get_eligible_cpusr, rF r� r� )r; rI �initialZall_cpus�nr? r? r@ �test_cpu_affinity� s4 z TestProcess.test_cpu_affinityc C s� | � � }ttjdd��d g}t�t�� |�|� W 5 Q R X t�t�� |�tdd�� W 5 Q R X t�t �� |�ddg� W 5 Q R X t�t�� |�ddg� W 5 Q R X d S ) NTr[ � i' i�* r �1rz ) rA r� r7 r� r, rF r| r] r� r� )r; rI Z invalid_cpur? r? r@ �test_cpu_affinity_errs s z"TestProcess.test_cpu_affinity_errsc C s� t �� }|�� }|s t|��| �|j|� t|�dkrB|d d� }g }tt|�d �D ](}t�||�D ]}|rf|� t |�� qfqV|D ]&}|�|� t |�� �t |�ks�t�q�d S )N� rg ) r7 r8 r] rE r� r� r� � itertools� combinations�appendr^ �sorted)r; rI r` Zcombosr� ZsubsetZcombor? r? r@ �"test_cpu_affinity_all_combinations s z.TestProcess.test_cpu_affinity_all_combinationsz broken on BSDzunreliable on APPVEYORc s\ t �� �| �� }��� � |� ks$t�t|d��z}|�d� |�� t� �fdd�� ��� � dd� � D �}t j � |�|ks~t�t r�� D ] }|j |kr�|j dks�t�q�W 5 Q R X � D ]}t j �|j �s�t|��q�d| }| �td |g��td �D ]0}d d� ��� D �}||k�r �q6t�d � q�t j � |�|k�s6t�|D ] }t j �|��s:t|���q:d S ) Nr� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxc s t ��� �t � �kS r5 )r� � open_filesr? ��filesrI r? r@ rW 7 rX z-TestProcess.test_open_files.<locals>.<lambda>c S s g | ]}t j�|j��qS r? �rC r� r r� r? r? r@ r� 9 s z/TestProcess.test_open_files.<locals>.<listcomp>r� zIimport time; f = open(r'%s', 'r'); [time.sleep(0.1) for x in range(100)];rd r� c S s g | ]}t j�|j��qS r? ro r� r? r? r@ r� J s ru )r7 r8 r� rl rE r� r� �flushr& rC r� r r �position�isfilerA r! r� r~ �sleep)r; r� r� � filenames�filer r� r? rm r@ �test_open_files+ s< �� zTestProcess.test_open_filesc C s� t �� }tjj}| �� }t|d���}|�� D ]*}||j�||j�ksR|j |� � kr, qnq,| � dt |�� � ��||j�||j�ks�t �tr�|j dks�t �n|j |� � ks�t �|�� d }|d |jks�t �|d |j ks�t �|j|�� ks�t �W 5 Q R X d S )N�wzno file found; files=%srz r rg )r7 r8 rC r� r r� r� rl r{ �fd�filenor �reprrE r )r; rI r r� �fileobjru Zntupler? r? r@ �test_open_files_2T s* � �� z TestProcess.test_open_files_2c C s� t �� }| �� }|�� }t|d�}| �|j� |�� |d ksBt�t�� }| �|j� |�� |d ksjt�|�� |�� |�� |ks�t�d S )Nrw rg r� ) r7 r8 r� Znum_fdsr� r� �closerE �socket)r; rI r� �startru �sockr? r? r@ � test_num_fdss s zTestProcess.test_num_fdsz not reliable on OPENBSD & NETBSDc C sT t �� }t|�� �}td�D ](}t�d� t|�� �}||kr d S q | �d��d S )Nr� g�������?z2num ctx switches still the same after 2 iterations)r7 r8 r� Znum_ctx_switchesr� r~ rs r )r; rI �beforer� �afterr? r? r@ �test_num_ctx_switches� s z!TestProcess.test_num_ctx_switchesc C sF t �� }ttd�r&|�� t�� ks&t�| �� }|�� t�� ksBt�d S )N�getppid) r7 r8 r8 rC �ppidr� rE rA rD rH r? r? r@ � test_ppid� s zTestProcess.test_ppidc C sD | � � }|�� jt�� ks t�t�� d }t�|��� d ks@t�d S r� ) rA �parentr9 rC rD rE r7 �pidsr8 )r; rI Z lowest_pidr? r? r@ � test_parent� s zTestProcess.test_parentc C s8 t �� }| �� \}}|�� |ks$t�|�� |ks4t�d S r5 )r7 r8 rv r� rE �r; r� rw rx r? r? r@ �test_parent_multi� s z TestProcess.test_parent_multic C s` t �� }|�� st�| �� \}}|�� d |ks4t�|�� d |ksHt�|�� d |ks\t�d S r� )r7 r8 �parentsrE rv r� r? r? r@ � test_parents� s zTestProcess.test_parentsc C s� t �� }|�� g kst�|jdd�g ks,t�| jdd�}|�� }|jdd�}||fD ]>}t|�dksht�|d j|jks|t�|d �� |jksTt�qTd S )NT�� recursiver )� creationflagsrg )r7 r8 �childrenrE rA r� r9 r� )r; r� rw Z children1Z children2r� r? r? r@ � test_children� s zTestProcess.test_childrenc C sf t �� }| �� \}}|�� |gks&t�|jdd�||gks>t�|�� |�� |jdd�g ksbt�d S )NTr� )r7 r8 rv r� rE rT rL r� r? r? r@ �test_children_recursive� s z#TestProcess.test_children_recursivec C s� t �t�}t�� D ]4}z||�� d7 < W q tjk rD Y qX qt|�� dd� d�d d }t rz|dkrzt � d��t� |�}z|j dd �}W n tjk r� Y nX t|�tt|��ks�t�d S ) Nrg c S s | d S )Nrg r? )r� r? r? r@ rW � rX z6TestProcess.test_children_duplicates.<locals>.<lambda>)�keyrz r zPID 0Tr� )� collections� defaultdictr r7 � process_iterr� �Errorrj �itemsr r, r� r8 r� rb r� r� rE )r; �tablerI r9 �cr? r? r@ �test_children_duplicates� s z$TestProcess.test_children_duplicatesc C s| t �� }| �� \}}|jdd�}t|�dks0t�|d |ks@t�|d |ksPt�|�� }|d |ksht�|d |ksxt�d S )NTr� r� r rg )r7 r8 rv r� r� rE r� )r; r� rw rx r� r� r? r? r@ �test_parents_and_children� s z%TestProcess.test_parents_and_childrenc C sX | � � }|�� td�D ] }|�� tjkr. q:t�d� q|�� |�� tjksTt �d S )Nr� ru ) rA rr r� r. r7 ZSTATUS_STOPPEDr~ rs rs rE )r; rI r� r? r? r@ �test_suspend_resume� s z TestProcess.test_suspend_resumec C sD t �t�� t�d� W 5 Q R X t �t�� t�d� W 5 Q R X d S )Nrd rz )r, rF r� r7 r8 r| �r; r? r? r@ �test_invalid_pid� s z TestProcess.test_invalid_pidc C s t �� }|jddgd�}t|�� �ddgks0t�t �tt �� ��}|jdgdd�}t|d t �sp|d dkspt�t j ddt j d ��" |jd gd d�d d iks�t�W 5 Q R X t j ddt � |jd�d ��, t�t j �� |jd gd� W 5 Q R X W 5 Q R X t j ddt �|jd�d ��$ |jd gdd�d dik�s0t�W 5 Q R X t j ddtd ��H |�� }d t |�� �k�sjt�t�t�� |jd gd� W 5 Q R X W 5 Q R X t�t�� |�d� W 5 Q R X t�t�� |�dg� W 5 Q R X t�t�� |�dd g� W 5 Q R X d S ) Nr r{ )�attrsZnet_connections�foo)r� Zad_valuezpsutil.Process.niceT)�creater] rA rg �bar)r7 r8 �as_dictrj �keysrE r� r� r� r^ r* r^ rb r: r9 r, rF r% �NotImplementedErrorr� r| )r; rI �dr? r? r@ � test_as_dict� sR �& �" �( � " zTestProcess.test_as_dictc C s� t �� }t�d��8}|�� � |�� |�� W 5 Q R X |jdksFt�W 5 Q R X t�d��}|�� |�� W 5 Q R X |jdks�t�d S )N�$psutil._psplatform.Process.cpu_timesrg r� �r7 r8 r* r^ �oneshotr� Z call_countrE �r; rI r� r? r? r@ � test_oneshot/ s zTestProcess.test_oneshotc C s� t �� }t�d���}t�d��j}|�� �: |�� |�� |�� � |�� |�� W 5 Q R X W 5 Q R X |jdksvt�|jdks�t�W 5 Q R X W 5 Q R X t�d��}|�� |�� W 5 Q R X |jdks�t�d S )Nr� z(psutil._psplatform.Process.oneshot_enterrg r� r� )r; rI �m1�m2r� r? r? r@ �test_oneshot_twice< s " z TestProcess.test_oneshot_twicec C s� | � � \}}|�� }|�� }||ks(t�|�� �& |�� |ksBt�|�� |ksRt�W 5 Q R X |�� �& |�� |ksvt�|�� |ks�t�W 5 Q R X d S r5 )rv r� rE r� )r; �p1Zp2Zp1_ppidZp2_ppidr? r? r@ �test_oneshot_cacheP s z TestProcess.test_oneshot_cachec sn �fdd�}�� � � � �� � �� tr8t� fdd�� ��� � t� �}|�|j�D ]\}}|||� qVd S )Nc s� z | � }W nT t jk r" � Y n` t jk r6 Y nL t jk r^ trX|dkrXY d S � Y n$X trp|dkrpd S � �d| |f ��d S )N)r� r� )r r{ z+%r didn't raise NSP and returned %r instead)r7 r% r: rb r r r )�funZfun_namer� r� r? r@ �assert_raises_nspg s �zFTestProcess.test_halfway_terminated_process.<locals>.assert_raises_nspc s � j t�� kS r5 )r9 r7 r� r? r� r? r@ rW ~ rX z=TestProcess.test_halfway_terminated_process.<locals>.<lambda>) rA rT rL r r& rP r+ �iter�all)r; r� �nsr� r{ r? )rI r; r@ � test_halfway_terminated_process_ s z+TestProcess.test_halfway_terminated_processc C s | � � \}}| �|� d S r5 )Z spawn_zombieZassertProcessZombie)r; Z_parent�zombier? r? r@ �test_zombie_process� s z TestProcess.test_zombie_processc C sB t �� }tjdt �d�d�� }|�� s*t�|js4t�W 5 Q R X d S )Nzpsutil.Processr r\ )r7 r8 r* r^ r% r rE r� r� r? r? r@ �$test_zombie_process_is_running_w_exc� s � z0TestProcess.test_zombie_process_is_running_w_excc C sH t �� }tjdt �d�d��"}|�� t jks0t�|js:t�W 5 Q R X d S )Nz!psutil._psplatform.Process.statusr r\ ) r7 r8 r* r^ r% r. r/ rE r� 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} | �� }t�|j�}|j|�� d f|_t t� � � |jtj ksbt �|� � rnt �tj�tjdd��( t|� ��}t t� � � W 5 Q R X W 5 Q R X d|j |�� ks�t �|jtj ks�t �|t�|j�ks�t �d}t|�}|j|j|j dd �D ]D\}}| j|d ��( tjtj|d �� |� W 5 Q R X W 5 Q R X �q d t |�k�sbt �d t |�k�stt �tjtj|d �� |� � W 5 Q R X tjtj|d �� |� � W 5 Q R X tjtj|d �� |� � W 5 Q R X tjtj|d �� |�!� W 5 Q R X d S ) Nr )�StringIOr� Z PSUTIL_DEBUGTz-refreshing Process instance for reused PID %sz4process no longer exists and its PID has been reusedF)� clear_cache)r{ r� zterminated + PID reused)"r �ior� r6 r7 r8 r9 r� �_identr^ r� Z_pmaprE r r* r^ �objectZ_commonr �getvaluer+ r� �settersZkillersrE r, rF r: rB rz r� r� r� r� ) r; r� ZsubprI r� �msgr� r� r{ r? r? r@ �test_reused_pid� sD �� zTestProcess.test_reused_pidc C s� dt �� krVt�t j�� t �d� W 5 Q R X t �d�r<t�t �d��� dksRt�d S t �d�}t rjt j nt }t�|�� |� � W 5 Q R X t�|�� |� � W 5 Q R X t�|�� |�� W 5 Q R X t�|�� |�� W 5 Q R X t�|�� |�� W 5 Q R X t�|�� |�tj� W 5 Q R X t|�}|�|j|j �D ]�\}}z |� }W n t j k �rj Y nbX |dk�r�|jdk�s�t�nD|dk�r�t �r�dnd}|�� |k�s�t�n|dk�r@|�s@t|���q@t�s�dt �� k�s�t�t �d��s�t�d S )Nr rg )r5 r< rK zNT AUTHORITY\SYSTEM�rootr{ )r7 r� r, rF r: r8 Z pid_existsrE r� r rb r| rL rT rr rs rK rV rM rN r+ r� �gettersr� r: rK r )r; rI r� r� r� r{ r� r� r? r? r@ � test_pid_0� sH zTestProcess.test_pid_0c C sH dd� }d | _ t�� }||�� �}|tj�� �}tsDtrD||ksDt�d S )Nc S sN ddddg}t r"|�ddddg� |D ]}| �|d � q&tdd � | �� D ��S ) N�PLAT�HOMEZPYTEST_CURRENT_TESTZPYTEST_VERSIONZ__CF_USER_TEXT_ENCODINGZ VERSIONER_PYTHON_PREFER_32_BITZVERSIONER_PYTHON_VERSIONc S s8 g | ]0\}}|� d d�� dd�|� d d�� dd�f�qS )� r[ � )r )r� �k�vr? r? r@ r� s ��z@TestProcess.test_environ.<locals>.clean_dict.<locals>.<listcomp>)r r$ �pop�dictr� )r� �excluder{ r? r? r@ � clean_dict� s ��z,TestProcess.test_environ.<locals>.clean_dict) ZmaxDiffr7 r8 rP rC �copyr r rE )r; r� rI 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 �d�}t| �� |d�}| j|gtjtjd�}t�|j �}t |j � |� � sRt �|j �� dksdt �tr�tr�z |�� }W q� tjk r� Y d S X n|�� }|ddd�ks�t �|�� |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); } )Zc_code)�stdinrf rX rd �3)�A�Cr )�textwrap�dedentr( r� r6 ri rj r7 r8 r9 r3 r rE rf r� r r rP rb � communicate� returncode)r; rR Zcexer> rI �envr? r? r@ �test_weird_environ s( � z TestProcess.test_weird_environN)m�__name__� __module__� __qualname__�__doc__rA rJ rS rU rX r, �mark�skipifr rc rl r rt ry r} r� r� r� r# r� r� r r� r� r� r r2 r r� r r� r r� r r� r� r� r� r� r� r� r r� r/ r1 r r� r� r� r r r r r r r" r r& r) r1 r r r2 r; r@ rG rH rR rU rY r rb re rk r r rv 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@ r4 P s& ) + � / B & & � 4 & .. � �r4 c sr e Zd ZdZeed�r&e�� Ze�� 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. r6 c sT t � j||� dd� t| �D �D ].}t| |�� � fdd�}t| |t�|| �� q d S )Nc S s g | ]}|� d �r|�qS )�testr� r� r? r? r@ r� ` s z0LimitedUserTestCase.__init__.<locals>.<listcomp>c s& z � � W n t jk r Y nX d S r5 )r7 rb r� ��methr? r@ �test_c s z+LimitedUserTestCase.__init__.<locals>.test_)r �__init__r� r� �setattr�types� MethodType)r; r< r= �attrr� �� __class__r� r@ r� \ s z LimitedUserTestCase.__init__c s" t � �� t�d� t�d� d S )Ni� )r �setUprC �setegid�seteuidr� r� r? r@ r� k s zLimitedUserTestCase.setUpc s& t �| j� t �| j� t� �� d S r5 )rC r� � PROCESS_UIDr� � PROCESS_GIDr �tearDownr� r� r? r@ r� p s z LimitedUserTestCase.tearDownc C s8 zt �� �d� W n t jk r( Y n X | �d��d S )Nrz zexception not raised)r7 r8 rA rb r r� r? r? r@ rG u s z LimitedUserTestCase.test_niceTzcauses problem as rootrY c C s d S r5 r? r� r? r? r@ r� } s z'LimitedUserTestCase.test_zombie_process)r� r� r� r� r8 rC r6 r� r= r� r� r� r� rG 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 r5 r- )�clsr? r? r@ � tearDownClass� s zTestPopen.tearDownClassc C s� t ddg}tj|tjtjtd��L}|�� |�� |j t |�sDt �t � t �� |j W 5 Q R X |�� W 5 Q R X tr�|�d�tj ks�t �n|�d�tjks�t �d S )Nrd �3import time; [time.sleep(0.1) for x in range(100)];��stdoutrf r� rh )r! r7 �Popenri rj r" r{ r� r� r� rE r, rF rG r� rT r rL rM rN �r; rk �procr? r? r@ � test_misc� s* �� zTestPopen.test_miscc C sh t jtdgtjtjtjtd��}|�� W 5 Q R X |jjs>t �|j jsJt �|j jsVt �|j dksdt �d S )Nz-V)r� rf r� r� r ) r7 r r! ri rj r" r� r� �closedrE rf r� r� )r; r r? r? r@ �test_ctx_manager� s � zTestPopen.test_ctx_managerc C s� t ddg}tj|tjtjtd���}|�� |�� t� tj �� |�� W 5 Q R X t� tj �� |� � W 5 Q R X t� tj �� |� t j� W 5 Q R X tr�t� tj �� |� t j� W 5 Q R X t� tj �� |� t j� W 5 Q R X W 5 Q R X d S )Nrd r� r� )r! r7 r ri rj r" rT rL r, rF r: rK rV rM rN r Z CTRL_C_EVENTZCTRL_BREAK_EVENTr r? r? r@ �test_kill_terminate� s0 ��z TestPopen.test_kill_terminateN) r� r� r� r� � classmethodr� r r r r? r? r? r@ r� � s r� )Gr� r� r` rM rg rC rM r~ r r, ri r r� r~ r� r7 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/ r0 r1 r2 r3 r4 r6 r� r� r? r? r? r@ �<module> s� 6