OwlCyberSecurity - MANAGER
Edit File: test_posix.cpython-36.pyc
3 ��JZ�? � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd� Z"d d � Z#d d � Z$e j%e d �G d!d"� d"e j&��Z'e j%e d �G d#d$� d$e j&��Z(e)d%k�r�ee*� dS )&zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�OPENBSD)�OSX)�POSIX)�SUNOS)�callable)�PY3)�APPVEYOR)�get_kernel_version)�get_test_subprocess)�mock)� PYTHON_EXE)� reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�skip_on_access_denied)�TRAVIS)�unittest)� wait_for_pid)�whichc C sp t s| jdd�} tr | jdd�} tr0| jdd�} t| �}t sN|jd�d j� }yt|�S tk rj |S X d S ) zkExpects a ps command with a -o argument and parse the result returning only the value of interest. z --no-headers � z-o startz-o stimez-o rssz -o rssize� � N) r �replacer r r �split�strip�int� ValueError)�cmd�output� r# �"/usr/lib64/python3.6/test_posix.py�ps+ s r% c C s& d}t r d}td|| f �jd�d S )N�commandZcommzps --no-headers -o %s -p %sr r )r r% r )�pid�fieldr# r# r$ �ps_nameI s r) c C s d}t s trd}td|| f �S )Nr& �argszps --no-headers -o %s -p %s)r r r% )r' r( r# r# r$ �ps_argsP s r+ z POSIX onlyc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z e � e � dd� ��Ze � e � dd� ��Zdd� Zdd� Zdd� Zdd� Zejep�ed �d d � �Zd d � Zd!d"� Zejed#�eje d$�d%d&� ��Z d'd(� Z d)S )*� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s& t tddgtjd�j| _t| j� d S )Nz-Ez-O)�stdin)r r � subprocess�PIPEr' r )�clsr# r# r$ � setUpClass[ s zTestProcess.setUpClassc C s t � d S )N)r )r0 r# r# r$ � tearDownClassa s zTestProcess.tearDownClassc C s. t d| j �}tj| j�j� }| j||� d S )Nz ps --no-headers -o ppid -p %s)r% r' �psutil�ProcessZppid� assertEqual)�selfZppid_psZ ppid_psutilr# r# r$ � test_ppide s zTestProcess.test_ppidc C s0 t d| j �}tj| j�j� j}| j||� d S )Nz ps --no-headers -o uid -p %s)r% r' r3 r4 �uids�realr5 )r6 Zuid_psZ uid_psutilr# r# r$ �test_uidj s zTestProcess.test_uidc C s0 t d| j �}tj| j�j� j}| j||� d S )Nz ps --no-headers -o rgid -p %s)r% r' r3 r4 Zgidsr9 r5 )r6 Zgid_psZ gid_psutilr# r# r$ �test_gido s zTestProcess.test_gidc C s. t d| j �}tj| j�j� }| j||� d S )Nz ps --no-headers -o user -p %s)r% r' r3 r4 �usernamer5 )r6 Z username_psZusername_psutilr# r# r$ � test_usernamet s zTestProcess.test_usernamec C sJ t j� }tjdtd��*}| j|j� t|j� j �� |j s<t �W d Q R X d S )Nzpsutil.pwd.getpwuid)� side_effect) r3 r4 r �patch�KeyErrorr5 r< �strr8 r9 �called�AssertionError)r6 �pZfunr# r# r$ �test_username_no_resolutiony s z'TestProcess.test_username_no_resolutionc C s@ t jd� td| j �}tj| j�j� d d }| j||� d S )Ng�������?z ps --no-headers -o rss -p %sr i )�time�sleepr% r' r3 r4 � memory_infor5 )r6 Zrss_psZ rss_psutilr# r# r$ �test_rss_memory� s zTestProcess.test_rss_memoryc C s@ t jd� td| j �}tj| j�j� d d }| j||� d S )Ng�������?z ps --no-headers -o vsz -p %sr i )rF rG r% r' r3 r4 rH r5 )r6 Zvsz_psZ vsz_psutilr# r# r$ �test_vsz_memory� s zTestProcess.test_vsz_memoryc C sZ t | j�}tjj|�j� }tj| j�j� j� }t j dd|�}t j dd|�}| j ||� d S )Nz\d.\d� ) r) r' �os�path�basename�lowerr3 r4 �name�re�subr5 )r6 Zname_psZ name_psutilr# r# r$ � test_name� s zTestProcess.test_namec C s^ d}dddg}t jd|d��8 t jd|d�� tj� }| j|j� d� W d Q R X W d Q R X d S )Nzlong-program-namezlong-program-name-extendedZfooZbarz psutil._psplatform.Process.name)� return_valuez"psutil._psplatform.Process.cmdline)r r? r3 r4 r5 rP )r6 rP �cmdlinerD r# r# r$ �test_name_long� s zTestProcess.test_name_longc C s\ d}t jd|d��@ t jdtjdd�d�� tj� }| j|j� d� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rT z"psutil._psplatform.Process.cmdliner rK )r> )r r? r3 � AccessDeniedr4 r5 rP )r6 rP rD r# r# r$ � test_name_long_cmdline_ad_exc� s z)TestProcess.test_name_long_cmdline_ad_excc C s\ d}t jd|d��@ t jdtjdd�d�� tj� }| jtj|j� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rT z"psutil._psplatform.Process.cmdliner rK )r> )r r? r3 Z NoSuchProcessr4 � assertRaisesrP )r6 rP rD r# r# r$ � test_name_long_cmdline_nsp_exc� s z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablec C sh t d| j �jd�d }tj| j�j� }tjj|�jd�}t |�}tjj|�jd�}| j |||g� d S )Nz ps --no-headers -o start -p %sr r z%H:%M:%S) r% r' r r3 r4 Z create_time�datetimeZ fromtimestampZstrftime�round�assertIn)r6 Ztime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr# r# r$ �test_create_time� s z TestProcess.test_create_timec C s` t | j�}tj| j�j� }y| j||� W n0 tk rZ |d t|�� }| j||� Y nX d S )N)r) r' r3 r4 Zexer5 rC �len)r6 Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer# r# r$ �test_exe� s zTestProcess.test_exec C s0 t | j�}djtj| j�j� �}| j||� d S )Nr )r+ r' �joinr3 r4 rU r5 )r6 Z ps_cmdlineZpsutil_cmdliner# r# r$ � test_cmdline� s zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s* t d| j �}tj� j� }| j||� d S )Nz ps --no-headers -o nice -p %s)r% r' r3 r4 �nicer5 )r6 Zps_niceZ psutil_nicer# r# r$ � test_nice� s zTestProcess.test_nicec s � fdd�}t jtj� �}g }ddddddd d d d g }trNt� dk rN|jd� trft� dk rf|jd� x�tt j�D ]�� � jd�sr� |kr�qrqry0|j � }xt d �D ]}||� � q�W |j � }W n t j k r� Y qrX t || �dkrrd� ||f }|j|� qrW |�r| j ddj|� � d S )Nc sB f }t | � d �}|d k r:t|�r:� dkr0tjf}||� n| d S )N�rlimit)�getattrr r3 Z RLIMIT_NOFILE)rD �attrr* )rP r# r$ �call� s z&TestProcess.test_num_fds.<locals>.callZ terminate�killZsuspendZresumerc Z send_signal�waitZchildrenZas_dictZmemory_info_ex� � �$ re � Znum_ctx_switches�_r z@failure while processing Process.%s method (before=%s, after=%s)r )rk rl rm )rk rl rn )r3 r4 rL �getpidr r �append�dir� startswithZnum_fds�rangerW �abs�failra ) r6 rh rD ZfailuresZ ignored_namesZnum1�xZnum2rv r# )rP r$ � test_num_fds� s4 zTestProcess.test_num_fdsN) �__name__� __module__� __qualname__�__doc__� classmethodr1 r2 r7 r: r; r= rE r r rI rJ rS rV rX rZ r �skipIfr r r^ r` rb r r rd rx r# r# r# r$ r, W s, r, c @ s� e Zd ZdZe� dd� �Zejed�eje d�eje d� d�dd � ���Z eje p`e o`e j� d �e� d d � ��Zd d� Zdd� Zdd� Zdd� Zejed�dd� �ZdS )�TestSystemAPIszTest some system APIs.c C s� t strddddg}n ddddg}t|tjd�}|j� d j� }|j� dksPt�t rbt |t j j �}g }x<|jd�d d � D ]&}|rzt|j� d j� �}|j|� qzW |j|j� tj� }|j� |j� ts�tr�d|kr�|jdd� | j||� d S ) Nr% z-Az-or' Zax)�stdoutr r r )r r r r. r/ Z communicater ZpollrC r rA �sysr� �encodingr r rq �remover' r3 Zpids�sortr r �insertr5 )r6 r! rD r" Zpids_ps�liner' Z pids_psutilr# r# r$ � test_pids s( zTestSystemAPIs.test_pidszunreliable on SUNOSzunreliable on TRAVISZifconfigzno ifconfig cmdc C sV t d�}xHtjdd�j� D ]4}x.|j� D ]}|j|�r(P q(W | jd||f � qW d S )Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output %s)r r3 Znet_io_counters�keysr rs rv )r6 r" Znicr� r# r# r$ �test_nic_names? s z TestSystemAPIs.test_nic_namesz unreliable on APPVEYOR or TRAVISc C sx t d�}|jd�}dd� |D �}dd� |D �}| jt|�ttj� �� x,tj� D ] }| j|j|� | j|j|� qPW d S )NZwhor c S s g | ]}|j � d �qS )r )r )�.0rw r# r# r$ � <listcomp>T s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|j � d �qS )r )r )r� rw r# r# r$ r� U s ) r r r5 r_ r3 �usersr] rP Zterminal)r6 �out�linesr� Z terminals�ur# r# r$ � test_usersN s zTestSystemAPIs.test_usersc C sF t jdttjd�d��&}| jttjjt j � � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.killrK )r> ) r r? �OSError�errno�EBADFrY r3 �_psposixZ pid_existsrL rp rB rC )r6 �mr# r# r$ �test_pid_exists_let_raise[ s z(TestSystemAPIs.test_pid_exists_let_raisec C sF t jdttjd�d��&}| jttjjt j � � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.waitpidrK )r> ) r r? r� r� r� rY r3 r� �wait_pidrL rp rB rC )r6 r� r# r# r$ �test_os_waitpid_let_raised s z(TestSystemAPIs.test_os_waitpid_let_raisec C sN t jdttjd�d��.}| jtjjtjj t j � dd� |j s@t �W d Q R X d S )Nzpsutil._psposix.os.waitpidrK )r> g{�G�z�?)Ztimeout)r r? r� r� ZEINTRrY r3 r� ZTimeoutExpiredr� rL rp rB rC )r6 r� r# r# r$ �test_os_waitpid_eintrl s z$TestSystemAPIs.test_os_waitpid_eintrc C s> t jddd��&}| jttjjtj� � |j s0t �W d Q R X d S )Nzpsutil._psposix.os.waitpidr )rT ���)r r� ) r r? rY r r3 r� r� rL rp rB rC )r6 r� r# r# r$ � test_os_waitpid_bad_ret_statusu s z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc C s� dd� }d }x�t jdd�D ]�}t j|j�}y||j�\}}}}W nL tk r� } z0t| �j� } d| ksvd| ksvd | krzwn� W Y d d } ~ X qX | j|j ||d � | j|j ||d � | j|j ||d � | j|j |d d � qW d S )Nc S sv t d| �j� }|jd�d }|j� }t|d �d }t|d �d }t|d �d }t|d jdd ��}||||fS ) Nzdf -k %sr r i rk � � �%rK )r r r r �floatr )�devicer� r� Zfields�total�used�free�percentr# r# r$ �df� s z*TestSystemAPIs.test_disk_usage.<locals>.dfr� i F)�allzno such file or directoryzraw devices not supportedzpermission denied)Zdeltar i i @ ) r3 Zdisk_partitionsZ disk_usageZ mountpointr� � RuntimeErrorrA rO ZassertAlmostEqualr� r� r� r� ) r6 r� Z tolerance�partZusager� r� r� r� �errr# r# r$ �test_disk_usage~ s" z TestSystemAPIs.test_disk_usageN)ry rz r{ r| r r� r r~ r r r r� r r3 r� r� r� r� r� r� r r� r# r# r# r$ r s r �__main__)+r| r[ r� rL rQ r. r� rF r3 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~ ZTestCaser, r ry �__file__r# r# r# r$ �<module> sT E