Advanced Web Server Manager
Complete File Manager & Terminal - Standalone Version
By Sid Gifari | Gifari Industries
Current path:
/
/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
psutil
/
tests
/
__pycache__
✏️
Editing: test_process.cpython-36.pyc
3 �̪e_� � @ 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? G d0d1� d1e2�Z@e�r�ejA� dk�r�G d2d3� d3e@�ZBG d4d5� d5e2�ZCeDd6k�r�dd7lEmFZF eFeG� dS )8z Tests for psutil.Process class.� N)�AIX)�BSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�OSX)�POSIX)�SUNOS)�WINDOWS)� open_text)�PY3)�FileNotFoundError)�long)�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)�PsutilTestCase)� ThreadTask)� call_until)�copyload_shared_lib)� create_c_exe)� create_py_exe)�mock)�process_namespace)� 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 e d �d d� �Z dd� Z e j ed�dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Ze j e d"�d#d$� �Zd%d&� Ze j e d'�d(d)� �Ze j e d"�e e d*�d+d,� ��Z e j e d"�e j e d-�d.d/� ��Z e j e d"�e j e! d0�d1d2� ��Z"e j e# d"�d3d4� �Z$e j e# d"�d5d6� �Z%e j e# d"�d7d8� �Z&e j e# d"�d9d:� �Z'e j e# d"�d;d<� �Z(d=d>� Z)e j e! d?�d@dA� �Z*e j e+ d"�dBdC� �Z,e-� e.e/d*�e j e+ d"�dDdE� ���Z0e-� dFdG� �Z1dHdI� Z2e j e3 d"�dJdK� �Z4e j e3 d"�dLdM� �Z5dNdO� Z6dPdQ� Z7dRdS� Z8dTdU� Z9e j e:dV�dWdX� �Z;dYdZ� Z<e j e:d[�d\d]� �Z=e j e>d^�e j e?d_�e j e:dV�d`da� ���Z@e j e d'�dbdc� �ZAe j e d'�ddde� �ZBdfdg� ZCdhdi� ZDdjdk� ZEdldm� ZFdndo� ZGe j eH d"�dpdq� �ZIe j eH d"�drds� �ZJe j eH d"�dtdu� �ZKe j eLdv�e j eMdw�dxdy� ��ZNe j eLdv�e j eMdw�dzd{� ��ZOe j e d'�d|d}� �ZPe e d*�e j eQ�p�ed~�dd�� ��ZRd�d�� ZSd�d�� ZTd�d�� ZUe-� d�d�� �ZVd�d�� ZWd�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�� Zae j e d'�d�d�� �Zbe j e d'�d�d�� �Zce j e d'�d�d�� �Zdd�d�� Zed�d�� Zfe j eg d"�d�d�� �Zhe j eg d"�e j e d'�e j eid��d�d�� ���Zjd�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� r9 �E/usr/local/lib64/python3.6/site-packages/psutil/tests/test_process.py� spawn_psprocP s zTestProcess.spawn_psprocc C s: t j� }| j|jtj� � | jt�� d|_W d Q R X d S )N�! )r1 r2 � assertEqualr3 �os�getpid� assertRaises�AttributeError)r5 �pr9 r9 r: �test_pidZ s zTestProcess.test_pidc C sJ | j � }|j� |j� }tr,| j|tj� n| j|tj � | j|� d S )N) r; �kill�waitr r= �signal�SIGTERM�SIGKILL�assertProcessGone)r5 rB �coder9 r9 r: � test_kill` s zTestProcess.test_killc C sJ | j � }|j� |j� }tr,| j|tj� n| j|tj � | j|� d S )N)r; � terminaterE r r= rF rG rI )r5 rB rJ r9 r9 r: �test_terminatej s zTestProcess.test_terminatec C sX t r tjntj}| j� }|j|� |j� }tr<| j||� n| j|| � | j |� d S )N) r rF rH rG r; � send_signalrE r r= rI )r5 �sigrB rJ r9 r9 r: �test_send_signalt s z TestProcess.test_send_signalz not POSIXc C s� t j}| j� }tjdttjd�d�� | jt j |j |� W d Q R X | j� }tjdttj d�d�� | jt j |j |� W d Q R X d S )Nzpsutil.os.kill� )� side_effect)rF rG r; r' �patch�OSError�errnoZESRCHr@ r1 r4 rN �EPERM� AccessDenied)r5 rO rB r9 r9 r: �test_send_signal_mocked s z#TestProcess.test_send_signal_mockedc C s� t ddg}| j|�}|j� }| j|d� | j|� t ddg}| j|tjd�}|j� }| j|d� | j|� t ddg}| j|�}|j� }| j|d� | j|� t dd g}| j|�}|j� }| j|d� | j|� d S ) Nz-c�passr z1 / 0)�stderr� zimport sys; sys.exit(5);� zimport os; os._exit(5);)r r; rE r= rI � subprocess�PIPE)r5 �cmdrB rJ r9 r9 r: �test_wait_exited� s( z TestProcess.test_wait_exitedzfails on NETBSDc C s� | j � }tr�|jtj� | jtj|jdd� |jtj � | jtj|jdd� |jtj � | j |j� tj � | j |j� tj � nd|j � | jtj|jdd� |j � | jtj|jdd� |j� | j |j� tj � | j |j� tj � d S )Ng����MbP?)�timeout)r; r rN rF �SIGSTOPr@ r1 �TimeoutExpiredrE �SIGCONTrG r= �suspend�resumerL )r5 rB r9 r9 r: �test_wait_stopped� s z TestProcess.test_wait_stoppedc C s� | j � \}}| jtj|jd� | jtj|jd� |j� |j� |j� }|j� }trr| j|tj � | j|d � n | j|tj � | j|tj � d S )Ng{�G�z�?) �spawn_children_pairr@ r1 rc rE rL r r= rF rG )r5 �child� grandchildZ child_retZgrandchild_retr9 r9 r: �test_wait_non_children� s z"TestProcess.test_wait_non_childrenc C sH | j � }|j� | jtj|jd� | jtj|jd� | jt|jd� d S )Ng{�G�z�?r r[ ���)r; �namer@ r1 rc rE � ValueError)r5 rB r9 r9 r: �test_wait_timeout� s z TestProcess.test_wait_timeoutc C s� | j � }| jtj|jd� |j� tj� t }xBtj� |k rfy|jd�}P W q0 tjk rb Y q0X q0W | jd��t r�| j |t j � n| j |t j � | j|� d S )Nr ra )r; r@ r1 rc rE rD �timer �failr r= rF rH rG rI )r5 rB Zstop_atrJ r9 r9 r: � test_wait_timeout_nonblocking� s z)TestProcess.test_wait_timeout_nonblockingc C s| t j� }|jdd� |jdd� x4td�D ](}|jd d�}| j|t� | j|d� q*W | jt�� |jdd� W d Q R X d S )Ng����MbP?)�interval�d g r[ rl ) r1 r2 � cpu_percent�range�assertIsInstance�float�assertGreaterEqualr@ rn )r5 rB �_�percentr9 r9 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' rS r1 r2 ru �called�AssertionError)r5 �mr9 r9 r: � test_cpu_percent_numcpus_none� s z)TestProcess.test_cpu_percent_numcpus_nonec C s� t j� j� }|jdks(|jdks(t|��|jdks:t|��|jdksLt|��trb|j dksbt|��x&|j D ] }t j dt j t||��� qjW d S )Ng z%H:%M:%S)r1 r2 � cpu_times�user�systemr~ � children_user�children_systemr Ziowait�_fieldsrp �strftime� localtime�getattr)r5 �timesrm r9 r9 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)r1 r2 r� r> r� �max�minrq )r5 � user_timeZ kernel_time�utimeZktimer9 r9 r: �test_cpu_times_2 s z TestProcess.test_cpu_times_2z not supportedc C sP t j� }|j� }| j|d� t j� dkr4| j|d� | j|j� tt j� �� d S )Nr r[ )r1 r2 Zcpu_numry � cpu_countr= �assertInrv )r5 rB �numr9 r9 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; rp � create_time�absrq r� r� )r5 rB �nowr� � differencer9 r9 r: �test_create_time s z TestProcess.test_create_timez POSIX onlyc C s4 t j� j� }|d k r0tjjtd��}| j||� d S )N�tty)r1 r2 �terminalr> �path�realpathr+ r= )r5 r� r� r9 r9 r: � test_terminal2 s zTestProcess.test_terminal)Zonly_ifc C s� t j� }|j� }ttd��}|j� W d Q R X |j� }t r�t r�| j|j |j � | j |j |j � t r�| j|j |j � | j |j|j� n | j|j|j� | j|j|j� |j� }t| j� d��.}tr�|jtdd d�� n|jdd � W d Q R X |j� }| j|j |j � | j|j|j� | j|j |j � | j|j|j� t �r`| j|j|j� | j|j |j � xJtt|��D ]:}t�r�|dk�r��qn| j|| d� | j|| d� �qnW d S )N�rb�wb�xi@B �asciir� r )r1 r2 Z io_counters�openr �readr r � assertGreater� read_countr= � write_countr Z read_charsZ write_charsry � read_bytes� write_bytes� get_testfnr �write�bytesrv �len)r5 rB Zio1�fZio2�ir9 r9 r: �test_io_counters9 s> z TestProcess.test_io_countersz linux onlyc . s� � fdd�}t j� � ts.| j� j� d t j� | jt jd� | jt jd� | jt jd� | jt jd� � j� }| j ||� � jt j� | jt � j� �t jdf� | j t �� � jt jdd� W d Q R X � jt j� | jt � j� �t jdf� � jt jdd� | jt � j� �t jdf� | j t �� � jt jd d� W d Q R X y� jt jdd� W n t j k �rh Y nX | jt d �� � jt jd� W d Q R X | jt d �� � jt jd� W d Q R X | jt d �� � jdd� W d Q R X d S ) Nc s& | \}}|t jkrd}� j||� d S )Nr )r1 �IOPRIO_CLASS_NONE�ionice)�init�ioclass�value)rB r9 r: �cleanupg s z.TestProcess.test_ionice_linux.<locals>.cleanupr r[ r� � � )r� � zioclass accepts no valuez$'ioclass' argument must be specified)r1 r2 r r= r� r� ZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE� addCleanup�tupler@ rn rW �assertRaisesRegex)r5 r� r� r9 )rB r: �test_ionice_linuxd s> z TestProcess.test_ionice_linuxz!not supported on this win versionc C s� t j� }ts | j|j� t j� |j� }| j|j|� |jt j� | j|j� t j� |jt j� | j|j� t j� y|jt j � W n t j k r� Y nX | j|j� t j � | j t d�� |jt jdd� W d Q R X | j t d�� |jt j d � W d Q R X d S )Nz&value argument not accepted on Windowsr[ )r� zis not a valid priority)r1 r2 r r= r� Z IOPRIO_NORMALr� ZIOPRIO_VERYLOWZ IOPRIO_LOWZ IOPRIO_HIGHrW r� � TypeErrorrn )r5 rB r� r9 r9 r: �test_ionice_win� s& zTestProcess.test_ionice_winc C s� dd l }tjtj� �}dd� tt�D �}|s4t|��x�|D ]�}tt|�}| j|d� |t|�kr�| j |t||�� t rxq:| j |j |�|j |�� q:|j |�}| j t |�d� | j|d d� | j|d d� q:W d S )Nr c S s g | ]}|j d �r|�qS )ZRLIMIT)� startswith)�.0r� r9 r9 r: � <listcomp>� s z/TestProcess.test_rlimit_get.<locals>.<listcomp>r� r[ rl rl )�resourcer1 r2 r> r? �dirr~ r� ry r= r �rlimitZ getrlimitr� )r5 r� rB �namesrm r� �retr9 r9 r: �test_rlimit_get� s zTestProcess.test_rlimit_getc C s� | j � }|jtjd� | j|jtj�d� trX| jtd�� tjj d�jd� W d Q R X | j t�� |jtjd� W d Q R X d S )Nr\ zcan't use prlimitr )r\ r\ )r\ r\ )r\ r\ r\ ) r; r� r1 Z RLIMIT_NOFILEr= r r� rn Z _psplatformr2 r@ )r5 rB r9 r9 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 | jt��*}t|d��}|jdd � W d Q R X W d Q R X | j t r�|j j n|j d t j � W d |jt j||f� | j |jt j�||f� X d S )Ni r� � Xi r )r1 r2 r� r� � RLIMIT_FSIZEr� r� r@ �IOErrorr= r � exceptionrU ZEFBIG)r5 rB �testfn�soft�hardr� �excr9 r9 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� | j|jt j�||f� X d S )Ni r� r� i ) r1 r2 r� r� � RLIM_INFINITYr� r� r� r= )r5 rB r� r� r� r9 r9 r: �test_rlimit_infinity� s z TestProcess.test_rlimit_infinityc C s< t j� }|jt j�\}}| jt j|� |jt j||f� d S )N)r1 r2 r� r� r= r� )r5 rB r� r� r9 r9 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� }| j||d � W d Q R X d S )Nz$on OpenBSD this requires root accessr[ ) r1 r2 r � num_threadsrW �unittest�SkipTestr" r= )r5 rB �step1�step2r9 r9 r: �test_num_threads s z TestProcess.test_num_threadsz WINDOWS onlyc C s t j� }| j|j� d� d S )Nr )r1 r2 r� Z num_handles)r5 rB r9 r9 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� }| jt |�t |�d � |d }| j|j |d � | j|j |d � | j|j |d � W d Q R X d S )Nz$on OpenBSD this requires root accessr[ r r� ) r1 r2 r �threadsrW r� r� r" r= r� �idr� � system_time)r5 rB r� r� Zathreadr9 r9 r: � test_threads s zTestProcess.test_threadsc C s� | j � }tr:y |j� W n tjk r8 tjd��Y nX | j|j� j t dd� |j� D ��dd� | j|j� j t dd� |j� D ��dd� d S )Nz$on OpenBSD this requires root accessc S s g | ] }|j �qS r9 )r� )r� r� r9 r9 r: r� 9 s z.TestProcess.test_threads_2.<locals>.<listcomp>g�������?)�deltac S s g | ] }|j �qS r9 )r� )r� r� r9 r9 r: r� > s ) r; r r� r1 rW r� r� ZassertAlmostEqualr� r� �sumr� )r5 rB r9 r9 r: �test_threads_2- s zTestProcess.test_threads_2c C s� t j� }|j� d d� \}}|j� }| j|d� | j|d� d gd }|j� d d� \}}|j� }| j||� | j||� | j||� ~tr�|j� } | j| j| j � | j| j | j � |j� } x | j D ]} | jt | | �d� q�W d S )Nr� r i`� )r1 r2 Z memory_info�memory_percentr� ry r r= ZrssZwset�vmsZpagefiler� r� ) r5 rB Zrss1Zvms1Zpercent1ZmemarrZrss2Zvms2Zpercent2�memrm r9 r9 r: �test_memory_infoB s& z TestProcess.test_memory_infoc C s� t j� }t j� j}|j� }xT|jD ]J}t||�}| j|d||fd� |dkrPts"t rVq"| j |||||fd� q"W t s|t s|t r�| j|j d� t r�| j|jd� | j|jd� d S )Nr )�msgr� )r1 r2 Zvirtual_memory�totalZmemory_full_infor� r� ry r r ZassertLessEqualr r �ussZpssZswap)r5 rB r� r� rm r� r9 r9 r: �test_memory_full_infoa s z!TestProcess.test_memory_full_infoc C s� t j� }|j� }| jt|�tt|��� |jdd�}� x�|D ]�}|jjd�s<tjj |j�sdt |j��t r�y*tjj |j�s�tjj |j�s�t |j��W nH t k r� ts�� n,td��}|j� }W d Q R X d|j |krւ Y nX q<dtjj|j�kr<ytj|j�}W n tk �r Y q<X tj|j�s<t |j��q<W xt|D ]l}xd|jD ]Z}t||�} |dk�r`�qD|d k�rz| �s�t | ��n"| j| ttf� | d k�sDt | ���qDW �q8W d S ) NF)Zgrouped�[z/proc/self/smapsz %s (deleted)Z64r� �addr�permsr )r� r� )r1 r2 � memory_mapsr= r� �setr� r� r> �isabsr~ r �exists�islinkr r r� �basename�statr �S_ISREG�st_moder� r� rw �intr ) r5 rB �mapsZext_maps�ntr� �data�st�fnamer� r9 r9 r: �test_memory_mapsq sD z TestProcess.test_memory_mapsc sL t j� }t� �4}dd� � � fdd�|j� D �}| j� |�|� W d Q R X d S )Nc S s t jjt jj| ��S )N)r> r� r� �normcase)rB r9 r9 r: �normpath� s z8TestProcess.test_memory_maps_lists_lib.<locals>.normpathc s g | ]}� |j ��qS r9 )r� )r� r� )r r9 r: r� � s z:TestProcess.test_memory_maps_lists_lib.<locals>.<listcomp>)r1 r2 r$ r� r� )r5 rB r� Zlibpathsr9 )r r: �test_memory_maps_lists_lib� s z&TestProcess.test_memory_maps_lists_libc C s> t j� }|j� | jt|jdd� ts.ts.tr:|jdd� d S )Nz?!?)Zmemtyper� )r1 r2 r� r@ rn r r r )r5 rB r9 r9 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_runningr~ rD rE )r5 rB r9 r9 r: �test_is_running� s zTestProcess.test_is_runningc C s� | j � }|j� }y| j|t� W n� tk r� trbt|�tt�krbtjj }| j||�|t�� nNdt j d t j d f }y | j|j |d�tj |d�� W n tk r� Y nX Y nX t |ddg�}| j|d� d S )Nz%s.%sr r[ rQ z-czimport os; print('hey')Zhey)r; �exer= r r~ r r� r> r� r �sys� version_info�replacer+ )r5 rB r r �ver�outr9 r9 r: �test_exe� s zTestProcess.test_exec C s� t ddg}| j|�}ts ts tr6| j|j� d t � nhtr�tr�|j� d }|t kr�| jdj |j� dd � �dj |dd � �� d S | jdj |j� �dj |�� d S )Nz-czimport time; time.sleep(60)r � r[ ) r r; r r r r= �cmdliner r �join)r5 r rB �pyexer9 r9 r: � test_cmdline� s (zTestProcess.test_cmdlinezbroken on PYPYc C s~ t g}|jdgd � |jddg� | j|�}trjy| j|j� |� W qz tjk rf | jd��Y qzX n| j|j� |� d S )Nz-v�2 z-czimport time; time.sleep(10)z#OPENBSD: process turned into zombie) r �extendr; r r= r r1 � ZombieProcess�skipTest)r5 r rB r9 r9 r: �test_long_cmdline� s z TestProcess.test_long_cmdlinec C sH | j � }|j� j� }tjjtjjtj��j� }|j |�sDt ||f��d S )N) r; rm �lowerr> r� r� r� r � executabler� r~ )r5 rB rm r r9 r9 r: � test_name� s zTestProcess.test_namezunreliable on PYPYc C s� t | jdd��}|ddg}| j|�}tr�y | j|j� tjj|�� W q� t k r� |j � t j kr~tjj|�j |j� �s�t �n� Y q�X n| j|j� tjj|�� d S )N� 0123456789r� )�suffixz-czimport time; time.sleep(10)Z01234567890123456789)r& r� r; r r= rm r> r� r� r~ �statusr1 � STATUS_ZOMBIEr� )r5 r r rB r9 r9 r: �test_long_name s zTestProcess.test_long_namezbroken on SUNOSz broken on AIXc C sp t | jdd��}|ddg}| j|�}| j|j� |� | j|j� tjj|�� | jtjj |j � �tjj |�� d S )Nz foo bar ))r z-czimport time; time.sleep(10)) r& r� r; r= r rm r> r� r� r r )r5 r r rB r9 r9 r: �test_prog_w_funky_name s z"TestProcess.test_prog_w_funky_namec C sX t j� }|j� \}}}| j|tj� � | j|tj� � ttd�rT| jtj� |j� � d S )N� getresuid) r1 r2 �uidsr= r> �getuid�geteuid�hasattrr )r5 rB �real� effective�savedr9 r9 r: � test_uids( s zTestProcess.test_uidsc C sX t j� }|j� \}}}| j|tj� � | j|tj� � ttd�rT| jtj� |j� � d S )Nr ) r1 r2 �gidsr= r> �getgid�getegidr$ � getresgid)r5 rB r% r& r' r9 r9 r: � test_gids6 s zTestProcess.test_gidsc s� � fdd�}t j� � | jt� jd� � j� }| j||� tr�d }x�t jt jt 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�|}| j||� n | j||� W d Q R X q^W n�y�ttd��r | jtjtjtj� �� j� � � jd� | j� j� d� ttd��r`| jtjtjtj� �� j� � t�s�� jd� | j� j� d� W n t jk �r� Y nX d S )Nc s* y� j | � W n tjk r$ Y nX d S )N)�nicer1 rW )r� )rB r9 r: r� E s z&TestProcess.test_nice.<locals>.cleanup�str)�prio� getpriorityr[ r )r1 r2 r@ r� r. r� r ZIDLE_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZABOVE_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZREALTIME_PRIORITY_CLASS�subTestrW r= r$ r> r1 � PRIO_PROCESSr? r )r5 r� r� Z highest_prior0 Znew_prior9 )rB r: � test_niceD sT zTestProcess.test_nicec C s t j� }| j|j� t j� d S )N)r1 r2 r= r �STATUS_RUNNING)r5 rB r9 r9 r: � test_status� s zTestProcess.test_statusc C s| | j � }|j� }trh|jd�\}}tj� }|jd�r>tjd��| j ||� dt j krx| j |t j d � n| j |tj� � d S )N�\�$zrunning as service accountZ USERDOMAIN) r; �usernamer �split�getpassZgetuser�endswithr� r� r= r> �environ)r5 rB r9 �domainZ getpass_userr9 r9 r: � test_username� s zTestProcess.test_usernamec C s | j � }| j|j� tj� � d S )N)r; r= �cwdr> �getcwd)r5 rB r9 r9 r: �test_cwd� s zTestProcess.test_cwdc C s$ t ddg}| j|�}t|jd� d S )Nz-cz/import os, time; os.chdir('..'); time.sleep(60)z#ret == os.path.dirname(os.getcwd()))r r; r# r@ )r5 r_ rB r9 r9 r: � test_cwd_2� s zTestProcess.test_cwd_2c C s� t j� }|j� }|s t|��| j|j|� ttd�rL| j|ttj |j ��� | jt |�t t |��� tt t t jdd����}xr|D ]j}|j|g� | j|j� |g� ttd�r�| j|j� ttj |j ��� t|d�r�| j|j� d |j� � q�W |jg � t�r| j|j� |jj� � n| j|j� |� ttd��rP| j|j� ttj |j ��� | jt|jd� |j|� |jt |�� |jt|�� d S )N�sched_getaffinityT)�percpu�num_cpur r[ )r1 r2 � cpu_affinityr~ r� r$ r> r= �listrD r3 r� r� rv ru rF r �_procZ_get_eligible_cpusr@ r� r� )r5 rB �initialZall_cpus�nr9 r9 r: �test_cpu_affinity� s6 z TestProcess.test_cpu_affinityc C sp | j � }ttjdd��d g}| jt|j|� | jt|jtdd�� | jt|jddg� | jt|jdd g� d S ) NT)rE � i' i�* r �1r[ rl ) r; r� r1 r� r@ rn rG rv r� )r5 rB Z invalid_cpur9 r9 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|� | j t |j� �t |�� q�W d S )N� r[ ) r1 r2 rG r~ r� r� rv � itertools� combinations�appendrH r= �sorted)r5 rB rJ Zcombosr� Zsubset�combor9 r9 r: �"test_cpu_affinity_all_combinations� s z.TestProcess.test_cpu_affinity_all_combinationsz broken on BSDzunreliable on APPVEYORc C sd t j� }| j� }|j� }| j||� t|d��|}|jdd � |j� t|jdt |� �}dd� |D �}| j t j j |�|� tr�x$|D ] }|j |kr�| j|jd� q�W W d Q R X x"|D ]}t j j|j �s�t|��q�W d| }| jtd|g�}xLtd �D ],}d d� |j� D �}||k�rP tjd � q�W | j t j j |�|� x$|D ] }t j j|��s@t|���q@W d S ) Nr� � xi zlen(ret) != %ic S s g | ]}t jj|j��qS r9 )r> r� r )r� r� r9 r9 r: r� � s z/TestProcess.test_open_files.<locals>.<listcomp>z2import time; f = open(r'%s', 'r'); time.sleep(60);z-crt c S s g | ]}t jj|j��qS r9 )r> r� r )r� r� r9 r9 r: r� s g{�G�z�?)r1 r2 r� � open_files� assertNotInr� r� �flushr# r� r� r> r� r r r= �position�isfiler~ r; r rv rp �sleep) r5 rB r� �filesr� � filenames�filer r� r9 r9 r: �test_open_files� s4 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�||j�� tr�| j |j d� n| j |j |j � � |j� d }| j |d |j� | j |d |j � | j|j|j� � W d Q R X d S )N�wzno file found; files=%sr[ r rl )r1 r2 r> r� r r� r� rX rm �fd�filenorq �reprr= r rY )r5 rB r r� �fileobjr` Zntupler9 r9 r: �test_open_files_2 s$ z TestProcess.test_open_files_2c C s� t j� }| j� }|j� }t|d�}| j|j� | j|j� |d � tj� }| j|j� | j|j� |d � |j� |j� | j|j� |� d S )Nrb r[ r� ) r1 r2 r� Znum_fdsr� r� �closer= �socket)r5 rB r� �startr` �sockr9 r9 r: � test_num_fds1 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)r1 r2 r� Znum_ctx_switchesrv rp r] rq )r5 rB �beforerz �afterr9 r9 r: �test_num_ctx_switches@ s z!TestProcess.test_num_ctx_switchesc C sF t j� }ttd�r&| j|j� tj� � | j� }| j|j� tj� � d S )N�getppid) r1 r2 r$ r> r= �ppidrp r; r? )r5 rB r9 r9 r: � test_ppidL s zTestProcess.test_ppidc C sB | j � }| j|j� jtj� � tj� d }| jtj |�j� � d S )Nr ) r; r= �parentr3 r> r? r1 �pidsZ assertIsNoner2 )r5 rB Z lowest_pidr9 r9 r: � test_parentS s zTestProcess.test_parentc C s8 t j� }| j� \}}| j|j� |� | j|j� |� d S )N)r1 r2 rh r= rs )r5 rs ri rj r9 r9 r: �test_parent_multiZ s z TestProcess.test_parent_multic C s` t j� }|j� st�| j� \}}| j|j� d |� | j|j� d |� | j|j� d |� d S )Nr r[ )r1 r2 �parentsr~ rh r= )r5 rs ri rj r9 r9 r: � test_parents` s zTestProcess.test_parentsc C s� t j� }| j|j� g � | j|jdd�g � | jdd�}|j� }|jdd�}xJ||fD ]>}| jt|�d� | j|d j|j� | j|d j� |j� qVW d S )NT)� recursiver )� creationflagsr[ )r1 r2 r= �childrenr; r� r3 rq )r5 rs ri Z children1Z children2r{ r9 r9 r: � test_childreni s zTestProcess.test_childrenc C sf t j� }| j� \}}| j|j� |g� | j|jdd�||g� |j� |j� | j|jdd�g � d S )NT)ry )r1 r2 rh r= r{ rL rE )r5 rs ri rj r9 r9 r: �test_children_recursivex 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~| j d��tj |�}y|j dd�}W n tj k r� Y nX | jt|�tt|��� d S ) Nr[ c S s | d S )Nr[ r9 )r� r9 r9 r: �<lambda>� s z6TestProcess.test_children_duplicates.<locals>.<lambda>)�keyr zPID 0T)ry rl )� collections� defaultdictr� r1 Z process_iterrq �ErrorrT �itemsr r r2 r{ rW r= r� r� )r5 �tablerB r3 �cr9 r9 r: �test_children_duplicates� s z$TestProcess.test_children_duplicatesc C s| t j� }| j� \}}|jdd�}| jt|�d� | j|d |� | j|d |� |j� }| j|d |� | j|d |� d S )NT)ry r� r r[ )r1 r2 rh r{ r= r� rw )r5 rs ri rj r{ rw r9 r9 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 |j� tj� d S )Nrt g{�G�z�?) r; re rv r r1 �STATUS_STOPPEDrp r] rf �assertNotEqual)r5 rB rz r9 r9 r: �test_suspend_resume� s z TestProcess.test_suspend_resumec C s$ | j ttjd� | j ttjd� d S )NrN r[ rl )r@ r� r1 r2 rn )r5 r9 r9 r: �test_invalid_pid� s z TestProcess.test_invalid_pidc ; C s� t j� }|jddgd�}| jt|j� �ddg� t jtt j� ��}|jdgdd�}t|d t �sp| j|d d� t j ddt j d ��" | j|jd gd d�d d i� W d Q R X t j ddt j |jd�d �� | jt j |jd gd� W d Q R X t j ddt j|jd�d ��" | j|jd gdd�d di� W d Q R X t j ddtd ��F |j� }| jd t |j� �� | jt�� |jd gd� W d Q R X W d Q R X | jt�� |jd� W d Q R X | jt�� |jdg� W d Q R X | jt�� |jdd g� W d Q R X d S ) Nr rm )�attrsZ connections�foo)r� Zad_valuezpsutil.Process.niceT)�createrR r. r[ Zbar)r1 r2 Zas_dictr= rT �keysr� rt � isinstancerH r' rS rW r4 r3 r@ r �NotImplementedErrorrY r� rn )r5 rB �dr9 r9 r: � test_as_dict� sD " " " zTestProcess.test_as_dictc C s� t j� }tjd��8}|j� � |j� |j� W d Q R X | j|jd� W d Q R X tjd��}|j� |j� W d Q R X | j|jd� d S )Nz$psutil._psplatform.Process.cpu_timesr[ r� )r1 r2 r' rS �oneshotr� r= � call_count)r5 rB r r9 r9 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 | j|jd� | j|jd� W d Q R X W d Q R X tjd��}|j� |j� W d Q R X | j|jd� d S )Nz$psutil._psplatform.Process.cpu_timesz(psutil._psplatform.Process.oneshot_enterr[ r� )r1 r2 r' rS r� r� r= r� )r5 rB �m1�m2r r9 r9 r: �test_oneshot_twice� s " z TestProcess.test_oneshot_twicec C s� | j � \}}|j� }|j� }| j||� |j� �& | j|j� |� | j|j� |� W d Q R X |j� �& | j|j� |� | j|j� |� W d Q R X d S )N)rh rq r� r� r= )r5 �p1�p2Zp1_ppidZp2_ppidr9 r9 r: �test_oneshot_cache s z TestProcess.test_oneshot_cachec st � fdd�}� j � }|j� |j� tr:ttjd|j � � j|� t |�}x"|j |j �D ]\}}|||� qZW 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 rm z+%r didn't raise NSP and returned %r instead)r� r� )r rm )r1 r r4 rW r r rq )�funZfun_namer� )r5 r9 r: �assert_raises_nsp# s zFTestProcess.test_halfway_terminated_process.<locals>.assert_raises_nspz %s not in ret) r; rL rE r r# r1 rt r3 rI r( �iter�all)r5 r� rB �nsr� rm r9 )r5 r: � test_halfway_terminated_process s z+TestProcess.test_halfway_terminated_processc C s | j � \}}| j|� d S )N)Z spawn_zombieZassertProcessZombie)r5 rs �zombier9 r9 r: �test_zombie_processA 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 )rR )r1 r2 r' rS r r r~ r} )r5 rB r r9 r9 r: �$test_zombie_process_is_running_w_excF s z0TestProcess.test_zombie_process_is_running_w_excc C sH t j� }tjdt jd�d��"}| j|j� t j� |js:t �W d Q R X d S )Nz!psutil._psplatform.Process.statusr )rR ) r1 r2 r' rS r r= r r r} r~ )r5 rB r r9 r9 r: � test_zombie_process_status_w_excQ s z,TestProcess.test_zombie_process_status_w_excc C s� | j � }tj|j�}|j|j� d f|_|j� s6t�|tj|j�ksJt�d}t|�}xJ|j |j |j dd�D ]0\}}| j |d�� | j tj||� W d Q R X qnW | j tj||j� | j tj||j� | j tj||j� | j tj||j� d S )Nrt z4process no longer exists and its PID has been reusedF)� clear_cache)rm )r0 r1 r2 r3 r� �_identr r~ r( r� �settersZkillersr2 r� r4 rq rs rw r{ )r5 ZsubprB r� r� r� rm r9 r9 r: �test_reused_pid] s zTestProcess.test_reused_pidc C s� dt j� krH| jt jt jd� t jd� s.t�| jt jd�j� d� d S t jd�}t r\t j nt }| j||j � | j||j � | j||j� | j||j� | j||j� | j||jtj� t|�}x�|j|j|j �D ]�\}}y |� }W n t j k r� Y q�X |d k�r| j|jd� q�|dk�rBt �r,dnd}| j|j� |� q�|dkr�|s�t|��q�W t�s�| jdt j� � t jd��s�t�d S ) Nr r[ r! r) r9 zNT AUTHORITY\SYSTEM�rootrm )r! r) ) r1 rt r@ r4 r2 Z pid_existsr~ r= rq r rW rn rE rL re rf rD rN rF rG r( r� Zgettersr� r% r9 r r� )r5 rB r� r� r� rm r� r� r9 r9 r: � test_pid_0n s: zTestProcess.test_pid_0c C sJ dd� }d | _ tj� }||j� �}|tjj� �}t rFtrF| j||� d S )Nc S sV | j dd � | j dd � tr@| j dd � | j dd � | j dd � tdd� | j� D ��S )N�PLAT�HOMEZ__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 )� rQ � )r )r� �k�vr9 r9 r: r� � s z@TestProcess.test_environ.<locals>.clean_dict.<locals>.<listcomp>)�popr �dictr� )r� r9 r9 r: � clean_dict� s z,TestProcess.test_environ.<locals>.clean_dict) ZmaxDiffr1 r2 r= r> �copyr r r= )r5 r� rB Zd1Zd2r9 r9 r: � test_environ� s zTestProcess.test_environz<macOS 11+ can't get another process environment, issue #2084c C s� t jd�}t| j� |d�}| j|gtjtjd�}tj|j �}t |j � |j � sRt �| j |jj� d� tr�tr�y |j� }W q� tjk r� d S X n|j� }| j |ddd�� |j� | j |jd� 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)�stdinrZ � rN �3)�A�Cr )�textwrap�dedentr% r� r0 r] r^ r1 r2 r3 r. r r~ r= rZ r� r r r= rW � communicate� returncode)r5 rJ Zcexer8 rB �envr9 r9 r: �test_weird_environ� s$ z TestProcess.test_weird_environN)k�__name__� __module__� __qualname__�__doc__r; rC rK rM rP r� �skipIfr rX r` r rg rk ro rr 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 r r r r r r r r r( r- r4 r6 r? rB rC r rL rO rV r r ra rg rl r ro rr ru rv rx r| r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r r� r9 r9 r9 r: r/ M s� * , . ? * # 7 & ' r/ c sn e Zd ZdZeed�r&ej� Zej� Z � fdd�Z � fdd�Z � fdd�Z d d � Z ejd d �d d� �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� r9 r9 r: r� � s z0LimitedUserTestCase.__init__.<locals>.<listcomp>c s& y � � W n t jk r Y nX d S )N)r1 rW )r5 )�methr9 r: �test_� s z+LimitedUserTestCase.__init__.<locals>.test_)r �__init__r� r� �setattr�types� MethodType)r5 r6 r7 �attrr� )� __class__)r� r: r� � s z LimitedUserTestCase.__init__c s" t � j� tjd� tjd� d S )Ni� )r �setUpr> �setegid�seteuid)r5 )r� r9 r: r� � s zLimitedUserTestCase.setUpc s& t j| j� t j| j� t� j� d S )N)r> r� � PROCESS_UIDr� � PROCESS_GIDr �tearDown)r5 )r� r9 r: r� s z LimitedUserTestCase.tearDownc C s8 yt j� jd� W n t jk r( Y n X | jd��d S )Nr[ zexception not raisedrl )r1 r2 r. rW rq )r5 r9 r9 r: r4 s z LimitedUserTestCase.test_nicer[ zcauses problem as rootc C s d S )Nr9 )r5 r9 r9 r: r� s z'LimitedUserTestCase.test_zombie_process)r� r� r� r� r$ r> r" r� r* r� r� r� r� r4 r� r� r� � __classcell__r9 r9 )r� r: r� � 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) )�clsr9 r9 r: � tearDownClass s zTestPopen.tearDownClassc C s� t ddg}tj|tjtjtd��B}|j� |j� |j | j t |�� | j t t |d� |j� W d Q R X tr�| j|jd�tj � n| j|jd�tj� d S )Nz-cz import time; time.sleep(60);)�stdoutrZ r� r� r\ )r r1 �Popenr] r^ r rm r� r� Z assertTruer� r@ rA r� rL r r= rE rF rG )r5 r_ �procr9 r9 r: � test_misc s zTestPopen.test_miscc C sh t jtdgtjtjtjtd��}|j� W d Q R X |jjs>t �|j jsJt �|j jsVt �| j |j d� d S )Nz-V)r� rZ r� r� r )r1 r� r r] r^ r r� r� �closedr~ rZ r� r= r� )r5 r� r9 r9 r: �test_ctx_manager6 s zTestPopen.test_ctx_managerc C s� t ddg}tj|tjtjtd��v}|j� |j� | jtj |j� | jtj |j � | jtj |j t j � tr�| jtj |j t j� | jtj |j t j� W d Q R X d S )Nz-cz import time; time.sleep(60);)r� rZ r� )r r1 r� r] r^ r rL rE r@ r4 rD rN rF rG r Z CTRL_C_EVENTZCTRL_BREAK_EVENT)r5 r_ r� r9 r9 r: �test_kill_terminateD s( z TestPopen.test_kill_terminateN) r� r� r� r� � classmethodr� r� r� r� r9 r9 r9 r: r� s r� �__main__)� run_from_name)Hr� r� rU r; rQ r> rF ri r� r] r r� rp r� r� r1 r r r r r r r r r r Zpsutil._commonr Zpsutil._compatr 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. r/ r" r� r� r� Zpsutil.tests.runnerr� �__file__r9 r9 r9 r: �<module> s� 6H
💾 Save Changes
❌ Cancel