OwlCyberSecurity - MANAGER
Edit File: test_process.cpython-36.pyc
3 ��JZ9� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddlm'Z' ddlm(Z( ddlm)Z) dd lm*Z* dd lm+Z+ dd lm,Z, dd lm-Z- dd lm.Z. dd!lm/Z/ dd"lm0Z0 dd#lm1Z1 dd$lm2Z2 dd%lm3Z3 dd&lm4Z4 dd'lm5Z5 dd(lm6Z6 dd)lm7Z7 dd*lm8Z8 dd+lm9Z9 dd,lm:Z: dd-lm;Z; dd.lm<Z< dd/lm=Z= G d0d1� d1e;j>�Z?e�r�ej@� dk�r�G d2d3� d3e?�ZAG d4d5� d5e;j>�ZBeCd6k�r�e2eD� dS )7z Tests for psutil.Process class.� N)�AIX)�BSD)�LINUX)�NETBSD)�OPENBSD)�OSX)�POSIX)�SUNOS)�WINDOWS)�long)�PY3)�APPVEYOR)� call_until)�copyload_shared_lib)� create_exe)�create_proc_children_pair)�create_zombie_proc)�enum)�get_test_subprocess)� get_winver)�HAS_CPU_AFFINITY)� HAS_ENVIRON)� HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)� HAS_RLIMIT)� HAS_THREADS)�mock)�PYPY)� PYTHON_EXE)� reap_children)�retry_before_failing)�run_test_module_by_name)� safe_rmpath)�sh)�skip_on_access_denied)�skip_on_not_implemented)�TESTFILE_PREFIX)�TESTFN)� ThreadTask)�TRAVIS)�unittest)� wait_for_pid)� WIN_VISTAc @ sr e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zeje d �d d � �Zd d � Zeje d!�ejed"�d#d$� ��Zeje d �eed%�d&d'� ��Z eje d �eje o�e � e k d �d(d)� ��Z!eje d �eje �o e � e k d �d*d+� ��Z"eje# d �d,d-� �Z$eje# d �d.d/� �Z%eje# d �d0d1� �Z&eje# d �d2d3� �Z'eje# d �d4d5� �Z(d6d7� Z)eje d8�d9d:� �Z*eje+ d �d;d<� �Z,e-� e.e/d%�eje+ d �d=d>� ���Z0d?d@� Z1dAdB� Z2eje3 d �dCdD� �Z4eje3 d �dEdF� �Z5dGdH� Z6dIdJ� Z7dKdL� Z8dMdN� Z9dOdP� Z:eje;dQ�eje<dR�dSdT� ��Z=eje d!�dUdV� �Z>eje d!�dWdX� �Z?dYdZ� Z@d[d\� ZAd]d^� ZBd_d`� ZCdadb� ZDejeE d �dcdd� �ZFejeE d �dedf� �ZGejeE d �dgdh� �ZHejeIdi�ejeJdj�dkdl� ��ZKejeIdi�ejeJdj�dmdn� ��ZLeje d!�dodp� �ZMeed%�ejeN�pzeOdq�drds� ��ZPdtdu� ZQdvdw� ZRdxdy� ZSdzd{� ZTd|d}� ZUd~d� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\eje d!�d�d�� �Z]eje d!�d�d�� �Z^eje d!�d�d�� �Z_d�d�� Z`ejea d �d�d�� �Zbejea d �eje d!�d�d�� ��Zcd�S )�� TestProcessz Tests for psutil.Process class.c C s t t� d S )N)r$ r) )�self� r1 �$/usr/lib64/python3.6/test_process.py�setUpN s zTestProcess.setUpc C s t � d S )N)r! )r0 r1 r1 r2 �tearDownQ s zTestProcess.tearDownc C sX t j� }| j|jtj� � t� }| jt j|j�j|j� | jt�� d|_W d Q R X d S )N�! ) �psutil�Process� assertEqual�pid�os�getpidr � assertRaises�AttributeError)r0 �p�sprocr1 r1 r2 �test_pidT s zTestProcess.test_pidc C sN t � }|j}tj|�}|j� |j� }| jtj|�� trJ| j |t j � d S )N) r r9 r6 r7 �kill�wait� assertFalse� pid_existsr r8 �signal�SIGKILL)r0 r? r@ r> �sigr1 r1 r2 � test_kill\ s zTestProcess.test_killc C sN t � }|j}tj|�}|j� |j� }| jtj|�� trJ| j |t j � d S )N) r r9 r6 r7 � terminaterB rC rD r r8 rE �SIGTERM)r0 r? r@ r> rG r1 r1 r2 �test_terminatef s zTestProcess.test_terminatec C sL t r tjntj}t� }tj|j�}|j|� |j � }| j tj |j�� t �rH| j || � t� }tj|j�}|j|� t jdttjd�d��( | jtj�� |j|� W d Q R X W d Q R X t� }tj|j�}|j|� t jdttjd�d��, | jtj�� tj� j|� W d Q R X W d Q R X dtj� k�rHtjd�}| jt|jtj� d S )Nzpsutil.os.kill� )� side_effectr )r rE rF rJ r r6 r7 r9 � send_signalrB rC rD r8 r �patch�OSError�errnoZESRCHr< � NoSuchProcessZEPERM� AccessDenied�pids� ValueError)r0 rG r? r> Zexit_sigr1 r1 r2 �test_send_signalp s2 " z TestProcess.test_send_signalc C s^ t � }tj|j�}|j� |j� }tr8| j|tj � n| j|tj � | j |j � � t � }tj|j�}|j � |j� }tr�| j|tj � n| j|tj � | j |j � � d}t td|g�}tj|j�}| j|j� d� | j |j � � t td|g�}tj|j�}| j|j� d� | j|j� d� t � }tj|j�}|j� | jtj|jd� | jt|jd� d S )Nz0import time, sys; time.sleep(0.01); sys.exit(5);z-c� g{�G�z�?� )rW N���)r r6 r7 r9 rA rB r r8 rE rF rJ rC � is_runningrI r �assertIn�namer< �TimeoutExpiredrU )r0 r? r> �coder1 r1 r2 � test_wait� s< zTestProcess.test_waitc C s� t � \}}| jtj|jd� | jtj|jd� |j� |j� |j� }|j� }trp| j|tj � | j|d � n | j|tj � | j|tj � d S )Ng{�G�z�?) r r< r6 r] rB rI r r8 rE rJ )r0 �p1�p2Zret1Zret2r1 r1 r2 �test_wait_non_children� s z"TestProcess.test_wait_non_childrenc C s� t � }tj|j�}| jtj|jd� |j� tj� d }x:y|jd�}W n$ tjk rl tj� |krh� Y q:X P q:W t r�| j |t j � n| j |t j � | j|j� � d S )Nr � )r r6 r7 r9 r< r] rB rA �timer r8 rE rF rJ rC rZ )r0 r? r> Zstop_atr^ r1 r1 r2 �test_wait_timeout_0� s z TestProcess.test_wait_timeout_0c C s� t j� }|jdd� |jdd� xRtd�D ]F}|jd d�}| j|t� | j|d� tsd| j|d� q*| j|d� q*W | j t �� |jdd� W d Q R X d S )Ng����MbP?)Zinterval�d g g Y@rX rY ) r6 r7 � cpu_percent�range�assertIsInstance�float�assertGreaterEqualr �assertLessEqualr< rU )r0 r> �xZpercentr1 r1 r2 �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 rO r6 r7 rg �called�AssertionError)r0 �mr1 r1 r2 � test_cpu_percent_numcpus_none� s z)TestProcess.test_cpu_percent_numcpus_nonec C sx t j� j� }|jdks(|jdks(t|��|jdks:t|��|jdksLt|��x&|jD ] }t j dt j t ||��� qTW d S )Ng z%H:%M:%S) r6 r7 � cpu_times�user�systemrp � children_user�children_system�_fieldsrd �strftime� localtime�getattr)r0 �timesr\ r1 r1 r2 �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 )Nrc g�������?zexpected: %s, found: %s)r6 r7 rs r: r| �max�min�fail)r0 � user_timeZ kernel_time�utimeZktimer1 r1 r2 �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 rX )r6 r7 Zcpu_numrk � cpu_countr8 r[ rh )r0 r> Znumr1 r1 r2 � test_cpu_num s zTestProcess.test_cpu_numc C sd t � }tj� }tj|j�}|j� }t|| �}|dkrJ| jd|||f � tjdtj |j� �� d S )Nrc z'expected: %s, found: %s, difference: %sz%Y %m %d %H:%M:%S) r rd r6 r7 r9 � create_time�absr� ry rz )r0 r? Znowr> r� � differencer1 r1 r2 �test_create_time s z TestProcess.test_create_timez POSIX onlyznot reliable on TRAVISc C sL t j� j� }tjj� s tjj� r>tjj t d��}| j ||� n | j |� d S )N�tty) r6 r7 �terminal�sys�stdin�isatty�stdoutr: �path�realpathr% r8 � assertIsNone)r0 r� r� r1 r1 r2 � test_terminal- 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� }tjtd��.}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��ql| j|| d� | j|| d� �qlW d S )N�rb)�prefixrm i@B �asciirc r )r6 r7 Z io_counters�openr �readr r � assertGreaterZ read_countr8 Z write_countr Z read_charsZ write_charsrk Z read_bytesZ write_bytes�tempfileZ TemporaryFiler( r �write�bytesrh �len)r0 r> Zio1�fZio2�ir1 r1 r2 �test_io_counters7 s> z TestProcess.test_io_countersc C s� t �r<ddlm}m}m}m} | j|d� | j|d� | j|d� | j|d� tj� }z�|jd� |j� \}}t d k r�| j |t j � | j|d� | j|d� |jd� |j� \}}| j|d� | j|d� |jdd� |j� \}}| j|d� | j|d� |jdd� |j� \}}| j|d� | j|d� W d |j|� X nZtj� }|j� }| j |t � z0d}||k�rld}|j|� | j|j� |� W d |j|� X d S )Nr )�IOPRIO_CLASS_NONE�IOPRIO_CLASS_RT�IOPRIO_CLASS_BE�IOPRIO_CLASS_IDLErX rc � � � ) r r6 r� r� r� r� r8 r7 �ionicer ri �IntEnum�int) r0 r� r� r� r� r> Zioclass�value�originalr1 r1 r2 � test_ionicec sJ zTestProcess.test_ionicec C s� t � }tj|j�}tr�| jt|jdd� | jt|jdd � | jt|jd� | jt|jdd� | j td|jtj d� | j td|jtj d� | j td|jdd � n"| jt|jd � | jt|jdd� d S ) Nrc � rX r� �fooz*can't specify value with IOPRIO_CLASS_NONEz*can't specify value with IOPRIO_CLASS_IDLEz$'ioclass' argument must be specified)r� r� rY ) r r6 r7 r9 r r< rU r� � TypeErrorZassertRaisesRegexr� r� )r0 r? r> r1 r1 r2 �test_ionice_errs� s$ z TestProcess.test_ionice_errsc 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)�.0rm r1 r1 r2 � <listcomp>� s z/TestProcess.test_rlimit_get.<locals>.<listcomp>rc rX rY rY )�resourcer6 r7 r: r; �dirrp r{ rk r8 r �rlimitZ getrlimitr� )r0 r� r> �namesr\ r� �retr1 r1 r2 �test_rlimit_get� s zTestProcess.test_rlimit_getc C s� t � }tj|j�}|jtjd� | j|jtj�d� | jt�� tj jd�jd� W d Q R X | jt�� |jtjd� W d Q R X d S )NrW r )rW rW )rW rW )rW rW rW ) r r6 r7 r9 r� � RLIMIT_NOFILEr8 r< rU Z _psplatform)r0 r? r> r1 r1 r2 �test_rlimit_set� s zTestProcess.test_rlimit_setc C s� t j� }|jt j�\}}z�|jt jd|f� ttd��}|jdd � W d Q R X | jt��*}ttd��}|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 �wb� Xi r )r6 r7 r� � RLIMIT_FSIZEr� r) r� r< �IOErrorr8 r Z exceptionrQ ZEFBIG)r0 r> �soft�hardr� �excr1 r1 r2 � test_rlimit� s " zTestProcess.test_rlimitc C s� t j� }|jt j�\}}zN|jt jd|f� |jt jt j|f� ttd��}|jdd � W d Q R X W d |jt j||f� | j|jt j�||f� X d S )Ni r� r� i ) r6 r7 r� r� � RLIM_INFINITYr� r) r� r8 )r0 r> r� r� r� r1 r1 r2 �test_rlimit_infinity� s z TestProcess.test_rlimit_infinityc C s< t j� }|jt j�\}}| jt j|� |jt j||f� d S )N)r6 r7 r� r� r8 r� )r0 r> r� r� r1 r1 r2 �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 accessrX ) r6 r7 r � num_threadsrS r, �SkipTestr* r8 )r0 r> �step1�step2r1 r1 r2 �test_num_threads� s z TestProcess.test_num_threadsz WINDOWS onlyc C s t j� }| j|j� d� d S )Nr )r6 r7 r� Z num_handles)r0 r> r1 r1 r2 �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� �~ |j� }| jt |�t |�d � t r�| j|d j t j � � |d }| j|j |d � | j|j|d � | j|j|d � W d Q R X d S )Nz$on OpenBSD this requires root accessrX r rc )r6 r7 r �threadsrS r, r� r* r8 r� r �idr: r; r� � system_time)r0 r> r� r� Zathreadr1 r1 r2 � test_threads s zTestProcess.test_threadsc C s� t � }tj|j�}trDy |j� W n tjk rB 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 r1 )r� )r� rm r1 r1 r2 r� 7 s z.TestProcess.test_threads_2.<locals>.<listcomp>g�������?)Zdeltac S s g | ] }|j �qS r1 )r� )r� rm r1 r1 r2 r� : s )r r6 r7 r9 r r� rS r, r� ZassertAlmostEqualrs rt �sumru )r0 r? r> r1 r1 r2 �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 )Nrc r i`� )r6 r7 Z memory_info�memory_percentr� rk r r8 ZrssZwset�vmsZpagefilerx r{ ) r0 r> Zrss1Zvms1Zpercent1ZmemarrZrss2Zvms2Zpercent2�memr\ r1 r1 r2 �test_memory_info< s& z TestProcess.test_memory_infoc C s� t j� j}t j� j� }xB|jD ]8}t||�}| j|d||fd� | j|||||fd� q W t sft sft rt| j|j d� t r�| j|j d� | j|jd� d S )Nr )�msg)r6 Zvirtual_memory�totalr7 Zmemory_full_inforx r{ rk rl r r r �ussZpssZswap)r0 r� r� r\ r� r1 r1 r2 �test_memory_full_infoZ s z!TestProcess.test_memory_full_infoc C s� t j� }|j� }dd� |D �}| jt|�tt|��� |jdd�}x�|D ]�}|jjd�sHtjj |j�spt |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 qHdtjj|j�krHtjj |j�sHt |j��qHW xv|D ]n}xf|jD ]\}t||�} |d k�rH�q*n<|d k�rb| �s�t | ��n"| j| ttf� | d k�s*t | ���q*W �q W d S )Nc S s g | ]}|�qS r1 r1 )r� rm r1 r1 r2 r� k s z0TestProcess.test_memory_maps.<locals>.<listcomp>F)Zgrouped�[z/proc/self/smapsz %s (deleted)Z64r� �addr�permsr )r� r� )r6 r7 � memory_mapsr8 r� �setr� r� r: �isabsrp r �exists�islinkr r� r� �basenamerx r{ ri r� r ) r0 r> �maps�pathsZext_maps�ntr� �dataZfnamer� r1 r1 r2 �test_memory_mapsg s< z TestProcess.test_memory_mapsc sH t � �8}dd� � � fdd�tj� j� D �}| j� |�|� W d Q R X d S )Nc S s t jjt jj| ��S )N)r: r� r� �normcase)r> r1 r1 r2 �normpath� s z8TestProcess.test_memory_maps_lists_lib.<locals>.normpathc s g | ]}� |j ��qS r1 )r� )r� rm )r� r1 r2 r� � s z:TestProcess.test_memory_maps_lists_lib.<locals>.<listcomp>)r r6 r7 r� r[ )r0 r� Zlibpathsr1 )r� r2 �test_memory_maps_lists_lib� s z&TestProcess.test_memory_maps_lists_libc C s� t j� }|j� }d| ko"dkn s0t|��|jdd�}d| koNdkn s\t|��d| kondkn s|t|��| jt|jdd� ts�ts�tr�|jdd�}d| ko�dkn s�t|��d| ko�dkn s�t|��d S )Nr rf r� )Zmemtypez?!?r� ) r6 r7 r� rp r< rU r r r )r0 r> r� r1 r1 r2 �test_memory_percent� s z TestProcess.test_memory_percentc C sZ t � }tj|j�}|j� s t�|j� s*t�|j� |j� |j� sHt�|j� sVt�d S )N)r r6 r7 r9 rZ rp rA rB )r0 r? r> r1 r1 r2 �test_is_running� s zTestProcess.test_is_runningc C s� t � }tj|j�j� }y| j|t� W n� tk r� trht |�t t�krht j j }| j||�|t�� nNdt jd t jd f }y | j|j|d�tj|d�� W n tk r� Y nX Y nX t|ddg�}| j|d� d S )Nz%s.%sr rX rL z-czimport os; print('hey')Zhey)r r6 r7 r9 �exer8 r rp r r� r: r� r� r� � version_info�replacer% )r0 r? r� r� Zver�outr1 r1 r2 �test_exe� s zTestProcess.test_exec C s� t ddg}t|�}y(| jdjtj|j�j� �dj|�� W n@ tk rz t sVt sVt rt| jtj|j�j� d t � n� Y nX d S )Nz-czimport time; time.sleep(60)� r ) r r r8 �joinr6 r7 r9 �cmdlinerp r r r )r0 r� r? r1 r1 r2 � test_cmdline� s zTestProcess.test_cmdlinec C sP t t�}tj|j�j� j� }tjj tjj t j ��j� }|j |�sLt||f��d S )N)r r r6 r7 r9 r\ �lowerr: r� r� r� r� � executabler� rp )r0 r? r\ Zpyexer1 r1 r2 � test_name� s zTestProcess.test_namezbroken on SUNOSz broken on AIXc s� � fdd�}t d � t� � | j|� � ddddddg}t|�}tj|j�}trZt|j� | j |j � |� | j |j � t j j� �� | j t j j|j� �t j j� �� d S ) Nc s& y t � � W n tk r Y nX d S )N)r$ rP r1 )� funky_pathr1 r2 �rm� s z.TestProcess.test_prog_w_funky_name.<locals>.rmz foo bar )z-cz9import time; [time.sleep(0.01) for x in range(3000)];arg1Zarg2rL Zarg3)r) r � addCleanupr r6 r7 r9 r+ r- r8 r� r\ r: r� r� r� r� )r0 r� r� r? r> r1 )r� r2 �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) r6 r7 �uidsr8 r: �getuid�geteuid�hasattrr� )r0 r> �real� effective�savedr1 r1 r2 � test_uids s zTestProcess.test_uidsc C sX t j� }|j� \}}}| j|tj� � | j|tj� � ttd�rT| jtj� |j� � d S )Nr� ) r6 r7 �gidsr8 r: �getgid�getegidr � getresgid)r0 r> r r r r1 r1 r2 � test_gids s zTestProcess.test_gidsc C s� t j� }| jt|jd� tr�z||j� }tjdkr@| j|t j � n | j|t � | j |t j � |jt j� | j |j� t j� |jt j � | j |j� t j � W d |jt j � X n�|j� }z�y�ttd�r�| j tjtjtj� �|j� � |jd� | j |j� d� ttd��r"| j tjtjtj� �|j� � t�sB|jd� | j |j� d� W n t jk �r\ Y nX W d y|j|� W n t jk �r� Y nX X d S )N�strr� r� � getpriorityrX r )r� r� )r6 r7 r< r� �nicer r� r� ri r r� r� r8 �NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSr r: r � PRIO_PROCESSr; r rS )r0 r> ZinitZ first_nicer1 r1 r2 � test_nice" sB zTestProcess.test_nicec C s t j� }| j|j� t j� d S )N)r6 r7 r8 �statusZSTATUS_RUNNING)r0 r> r1 r1 r2 � test_statusK s zTestProcess.test_statusc C sn t � }tj|j�}|j� }trZ|jd�\}}| j|tj � � dt j krj| j|t j d � n| j|tj � � d S )N�\Z USERDOMAIN) r r6 r7 r9 �usernamer �splitr8 �getpassZgetuserr: �environ)r0 r? r> r Zdomainr1 r1 r2 � test_usernameO s zTestProcess.test_usernamec C s* t � }tj|j�}| j|j� tj� � d S )N)r r6 r7 r9 r8 �cwdr: �getcwd)r0 r? r> r1 r1 r2 �test_cwd[ s zTestProcess.test_cwdc C s. t ddg}t|�}tj|j�}t|jd� d S )Nz-cz/import os, time; os.chdir('..'); time.sleep(60)z#ret == os.path.dirname(os.getcwd()))r r r6 r7 r9 r r )r0 �cmdr? r> r1 r1 r2 � 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����}xzts�|n|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��rX| j|j� ttj |j ��� | jt|jd� |j|� |jt |�� |jt|�� d S )N�sched_getaffinityT)�percpu�num_cpur rX )r6 r7 � cpu_affinityrp r� r r: r8 �listr r9 r� r� rh rg r+ r" r �_procZ_get_eligible_cpusr< r� �tuple)r0 r> �initialZall_cpus�nr1 r1 r2 �test_cpu_affinityg s6 z TestProcess.test_cpu_affinityc C sz t � }tj|j�}ttjdd��d g}| jt|j|� | jt|jt dd�� | jt |jddg� | jt|jdd g� d S ) NT)r! r� i' i�* r �1rX rY ) r r6 r7 r9 r� rs r< rU r# rh r� )r0 r? r> Z invalid_cpur1 r1 r2 �test_cpu_affinity_errs� s z"TestProcess.test_cpu_affinity_errsc C s� t j� }|j� }|s t|��| j|j|� g }xBtdt|�d �D ],}x&tj||�D ]}|rT|j t |�� qTW qBW x&|D ] }|j|� | j |j� |� qxW d S )Nr rX ) r6 r7 r# rp r� rh r� � itertools� combinations�appendr$ r8 )r0 r> r' Zcombos�lZsubsetZcombor1 r1 r2 �"test_cpu_affinity_all_combinations� s z.TestProcess.test_cpu_affinity_all_combinationsz broken on BSDzunreliable on APPVEYORc C sR t j� }|j� }| jt|k� ttd��n}|jdd � |j� t|jdt |� �}x<|D ]"}|j tkrZt rz| j |j d� P qZW | jdt|� � W d Q R X x"|D ]}tj j|j �s�t|��q�W dt }ttd|g�}t j|j�}xDtd�D ],}d d � |j� D �}t|k�rP tjd � q�W | jt|� x$|D ] }tj j|��s.t|���q.W d S ) Nr� � xi zlen(ret) != %izno file found; files=%sz2import time; f = open(r'%s', 'r'); time.sleep(60);z-crf c S s g | ] }|j �qS r1 )r� )r� rm r1 r1 r2 r� � s z/TestProcess.test_open_files.<locals>.<listcomp>g{�G�z�?)r6 r7 � open_filesrC r) r� r� �flushr r� r� r r8 Zpositionr� �reprr: �isfilerp r r r9 rh rd �sleepr[ ) r0 r> �filesr� �filer� r? rm � filenamesr1 r1 r2 �test_open_files� s4 zTestProcess.test_open_filesc C s� t td���}tj� }xB|j� D ] }|j|jks<|j|j� kr P q W | j dt |j� � � | j |j|j� t r|| j |jd� 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=%srX r rY )r� r) r6 r7 r2 r� r\ �fd�filenor� r4 r8 r � assertNotIn)r0 Zfileobjr> r8 Zntupler1 r1 r2 �test_open_files_2� s z TestProcess.test_open_files_2c C s� t j� }|j� }ttd�}| j|j� | j|j� |d � tj� }| j|j� | j|j� |d � |j� |j� | j|j� |� d S )Nr; rX rc ) r6 r7 Znum_fdsr� r) r� �closer8 �socket)r0 r> �startr8 Zsockr1 r1 r2 � test_num_fds� s zTestProcess.test_num_fdsz not reliable on OPENBSD & NETBSDc C sL t j� }t|j� �}x(td�D ] }t|j� �}||kr d S q W | jd� d S )Ni � z7num ctx switches still the same after 50.000 iterations)r6 r7 r� Znum_ctx_switchesrh r� )r0 r> Zbeforerm �afterr1 r1 r2 �test_num_ctx_switches� s z!TestProcess.test_num_ctx_switchesc C s� t td�r"| jtj� j� tj� � tj� }t� }tj|j �}| j|j� |� t dd� t r^d S x2tj � D ]&}|j |j krzqh| j |j� ||d� qhW d S )N�getppidT)� recursive)r� )r r: r8 r6 r7 �ppidrF r; r r9 r! r � process_iter�assertNotEqual)r0 � this_parentr? r> r1 r1 r2 � test_ppid s zTestProcess.test_ppidc C s0 t j� }t� }tj|j�}| j|j� j|� d S )N)r: r; r r6 r7 r9 r8 �parent)r0 rK r? r> r1 r1 r2 � test_parent s zTestProcess.test_parentc C sF t � }tj|j�}tjdtjdd�d�� | j|j� � W d Q R X d S )Nzpsutil.Processr r� )rM ) r r6 r7 r9 r rO rR r� rM )r0 r? r> r1 r1 r2 �test_parent_disappeared s z#TestProcess.test_parent_disappearedc C s� t j� }| j|j� g � | j|jdd�g � tdd�}|j� }|jdd�}xL||fD ]@}| jt|�d� | j|d j|j� | j|d j� tj � � qTW d S )NT)rG r )Z creationflagsrX ) r6 r7 r8 �childrenr r� r9 rH r: r; )r0 r> r? Z children1Z children2rP r1 r1 r2 � test_children( s zTestProcess.test_childrenc C sd t � \}}tj� }| j|j� |g� | j|jdd�||g� |j� |j� | j|jdd�g � d S )NT)rG )r r6 r7 r8 rP rI rB )r0 r` ra r> r1 r1 r2 �test_children_recursive7 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 }tj |�}y|j dd�}W n tj k r� Y nX | j t |�t t|��� d S ) NrX c S s | d S )NrX r1 )rm r1 r1 r2 �<lambda>M s z6TestProcess.test_children_duplicates.<locals>.<lambda>)�keyr T)rG rY )� collections� defaultdictr� r6 rI rH �Error�sorted�itemsr7 rP rS r8 r� r� )r0 �tabler> r9 �cr1 r1 r2 �test_children_duplicatesD s z$TestProcess.test_children_duplicatesc C sd t � }tj|j�}|j� x*td�D ] }|j� tjkr8P tj d� q$W |j � | j |j� tj� d S )Nrf g{�G�z�?) r r6 r7 r9 �suspendrh r ZSTATUS_STOPPEDrd r6 �resumerJ )r0 r? r> rm r1 r1 r2 �test_suspend_resumeV s z TestProcess.test_suspend_resumec C s$ | j ttjd� | j ttjd� d S )Nr* rX rY )r< r� r6 r7 rU )r0 r1 r1 r2 �test_invalid_pida 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� r\ )�attrsZ connectionsr� )ra Zad_valuezpsutil.Process.niceT)ZcreaterM r rX Zbar)r6 r7 �as_dictr8 rX �keysr rT � isinstancer$ r rO rS rR r9 r< � ZombieProcess�NotImplementedErrorr> r� rU )r0 r> �dr1 r1 r2 � test_as_dicte s> " " " zTestProcess.test_as_dictc C s� t jd��@}tj� }|j� � |j� |j� W d Q R X | j|jd� W d Q R X t jd��}|j� |j� W d Q R X | j|jd� d S )Nz$psutil._psplatform.Process.cpu_timesrX rc )r rO r6 r7 �oneshotrs r8 � call_count)r0 rq r> r1 r1 r2 � test_oneshot� s zTestProcess.test_oneshotc & C s� t jd���}t jd��r}tj� }|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 t jd��}|j� |j� W d Q R X | j|jd� d S )Nz$psutil._psplatform.Process.cpu_timesz(psutil._psplatform.Process.oneshot_enterrX rc )r rO r6 r7 ri rs r8 rj )r0 �m1�m2r> rq r1 r1 r2 �test_oneshot_twice� s " z TestProcess.test_oneshot_twicec C s� t � }tj|j�}|j� |j� tr8ttjd|j � | j |j � � ddddddg}t rjt rj|j d� �xTt|�D �]F}|jd �sv||kr�qvy�t||�}|d kr�tr�|d �}n |tj�}np|d kr�|� }|d �}nX|dkr�|tj�}|tjd�}n8|dk�r|� }|dg�}n |dk�r(|tj�}n|� }W nz tjk �rV | jd| � Y qv tjk �rl Y qv tjk �r� t�r�|dk�r�n� Y qv tk �r� Y qvX | jd||f � qvW d S )Nz %s not in retr9 rZ rB r� ri Zmemory_info_exr� �_r rX r� rc rW r# r rN z/ZombieProcess for %r was not supposed to happenr� r� z4NoSuchProcess exception not raised for %r, retval=%s)rW rW )r� r� ) r r6 r7 r9 rI rB r r rT rC rZ r r r. r� r� r{ r r r� rE rJ re r� rR rS r rf )r0 r? r> Zexcluded_namesr\ �methr� r1 r1 r2 � test_halfway_terminated_process� s^ z+TestProcess.test_halfway_terminated_processc C s� dd� }t � }| jtdd� tj|�}| j|j� tj� | j|j � � |j � ||j �}|d k rl| j|g � t |d�r�||j tj� ||j tjd� ||j� t |d�r�y||jdg� W n: tk r� } z tr�tr�d t|�kr�n� W Y d d }~X nX ||jd� t |d ��r0t�r$||jd d� n ||jd� t |d��rL||j tjd� ||j � ||j� ||j� ||j� | jtj|�� t �r�t�r�| j |tj � � | j |d d � tj � D �� i t_ | j |dd � tj � D �� d S )Nc _ s, y | ||�S t jt jfk r& Y nX d S )N)r6 re rS )Zfun�args�kwargsr1 r1 r2 �succeed_or_zombie_p_exc� s z@TestProcess.test_zombie_process.<locals>.succeed_or_zombie_p_excT)rG r� rW r# r z not eligibler� rc c S s g | ] }|j �qS r1 )r9 )r� rm r1 r1 r2 r� 9 s z3TestProcess.test_zombie_process.<locals>.<listcomp>c S s g | ] }|j �qS r1 )r9 )r� rm r1 r1 r2 r� ; s )rW rW )rW rW ) r r� r! r6 r7 r8 r � STATUS_ZOMBIE� assertTruerZ rb r] r r� r� rM r# rU r+ r r r r� r^ rI rA rD r r[ rT rI Z_pmap)r0 rt ZzpidZzprocr� �errr1 r1 r2 �test_zombie_process� sP 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 )rM )r6 r7 r rO re rZ rp ro )r0 r> rq r1 r1 r2 �$test_zombie_process_is_running_w_exc= 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 )rM ) r6 r7 r rO re r8 r ru ro rp )r0 r> rq r1 r1 r2 � test_zombie_process_status_w_excG s z,TestProcess.test_zombie_process_status_w_excc C s: dt j� kr"| jt jt jd� d S t jd�}x�t jD ]�}|dkrBq4t||�}y |� }W n t jk rl Y q4X |d kr�| j|j d� q4|dkr�t r�| j|j � d� q�t r�| j|j � d� q4|dkr4|s4t |��q4W t|d ��ry|jt j� W n t jk �r Y nX |j� t�s6| jdt j� � | jt jd�� d S ) Nr r9 r r r �rootzNT AUTHORITY\SYSTEMr\ r� )r r )r6 rT r< rR r7 Z_as_dict_attrnamesr{ rS r8 r r r r rp r r� r� rb r r[ rv rD )r0 r> r\ rp r� r1 r1 r2 � test_pid_0Q s< zTestProcess.test_pid_0c C s@ dd� }d | _ tj� }||j� �}|tjj� �}| j||� d S )Nc S sb | j dd � | j dd � | j dd � trL| j dd � | j dd � | j dd � tdd� | j� D ��S ) NZPSUTIL_TESTINGZPLAT�HOMEZ__CF_USER_TEXT_ENCODINGZ VERSIONER_PYTHON_PREFER_32_BITZVERSIONER_PYTHON_VERSIONc S s$ g | ] \}}|j d �|j d �f�qS )z )�rstrip)r� �k�vr1 r1 r2 r� � s z@TestProcess.test_environ.<locals>.clean_dict.<locals>.<listcomp>)�popr �dictrY )rg r1 r1 r2 � clean_dictz s z,TestProcess.test_environ.<locals>.clean_dict)ZmaxDiffr6 r7 r r: �copyr8 )r0 r� r> Zd1Zd2r1 r1 r2 � test_environx s zTestProcess.test_environc C s� t jd�}t}t||d� | jt|� t|gtjtjd�}t j |j �}t |j � | j |j� � | j|jj� d� | 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); } )Zc_code)r� �stderr� r* �3)�A�Cr )�textwrap�dedentr) r r� r$ r � subprocess�PIPEr6 r7 r9 r- rv rZ r8 r� r� r � communicate� returncode)r0 r^ r� r? r> r1 r1 r2 �test_weird_environ� s z TestProcess.test_weird_environN)d�__name__� __module__� __qualname__�__doc__r3 r4 r@ rH rK rV r_ rb re rn rr r} r� r, ZskipIfr 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 r r r) r+ r0 r r r: r? rC r r rE rL rN rO rQ rR r\ r_ r` rh rk rn rq rx ry rz r| r r� r� r1 r1 r1 r2 r/ K s� / + . " ) ) * % . @I ' r/ c @ sR e Zd ZdZeed�r&ej� Zej� Z dd� Z dd� Z dd� Z d d � Z d d � Zd S )�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 s\ t j| f|�|� xDdd� t| �D �D ].}t| |�� � fdd�}t| |tj|| �� q&W d S )Nc S s g | ]}|j d �r|�qS )Ztest)r� )r� rm r1 r1 r2 r� � s z0LimitedUserTestCase.__init__.<locals>.<listcomp>c s& y � � W n t jk r Y nX d S )N)r6 rS )r0 )rp r1 r2 �test_� s z+LimitedUserTestCase.__init__.<locals>.test_)r/ �__init__r� r{ �setattr�types� MethodType)r0 rr rs �attrr� r1 )rp r2 r� � s z LimitedUserTestCase.__init__c C s* t t� tj| � tjd� tjd� d S )Ni� )r$ r) r/ r3 r: �setegid�seteuid)r0 r1 r1 r2 r3 � s zLimitedUserTestCase.setUpc C s& t j| j� t j| j� tj| � d S )N)r: r� � PROCESS_UIDr� � PROCESS_GIDr/ r4 )r0 r1 r1 r2 r4 � s z LimitedUserTestCase.tearDownc C s8 yt j� jd� W n t jk r( Y n X | jd� d S )NrX zexception not raisedrY )r6 r7 r rS r� )r0 r1 r1 r2 r � s z LimitedUserTestCase.test_nicec C s d S )Nr1 )r0 r1 r1 r2 rx � s z'LimitedUserTestCase.test_zombie_processN)r� r� r� r� r r: r r� r r� r� r3 r4 r rx r1 r1 r1 r2 r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � TestPopenz Tests for psutil.Popen class.c C s t � d S )N)r! )r0 r1 r1 r2 r4 � s zTestPopen.tearDownc C sj t ddg}tj|tjtjd��B}|j� |j� |j | jt |�� | j t t |d� |j � W d Q R X d S )Nz-cz import time; time.sleep(60);)r� r� r� )r r6 �Popenr� r� r\ rs r� rv r� r< r= r{ rI )r0 r �procr1 r1 r2 � test_misc� s zTestPopen.test_miscc C sf t jtdgtjtjtjd��}|j� W d Q R X |jjs<t�|j jsHt�|j jsTt�| j |j d� d S )Nz-V)r� r� r� r ) r6 r� r r� r� r� r� �closedrp r� r� r8 r� )r0 r� r1 r1 r2 �test_ctx_manager� s zTestPopen.test_ctx_managerc C s� t ddg}tj|tjtjd���}|j� |j� | jtj|j� | jtj|j � | jtj|j t j � t r�tjdkr�| jtj|j t j� | jtj|j t j� W d Q R X d S )Nz-cz import time; time.sleep(60);)r� r� rc r� )rc r� )r r6 r� r� r� rI rB r< rR rA rN rE rJ r r� r� Z CTRL_C_EVENTZCTRL_BREAK_EVENT)r0 r r� r1 r1 r2 �test_kill_terminate s z TestPopen.test_kill_terminateN)r� r� r� r� r4 r� r� r� r1 r1 r1 r2 r� � s r� �__main__)Er� rU rQ r r, r: rE rA r� r� r� r� rd r� r6 r r r r r r r r r Zpsutil._compatr 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. ZTestCaser/ r r� r� r� �__file__r1 r1 r1 r2 �<module> s� q53