Edit File: test_posix.cpython-37.pyc
B �=�_'; � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ e �rBddl Z ddl Z ddl m Z d*dd�Z dd� Z!dd � Z"d d � Z#d d � Z$e�%e d!�G d"d#� d#e��Z&e�%e d!�G d$d%� d%e��Z'e�%e d!�G d&d'� d'e��Z(e)d(k�r�dd)l*m+Z+ e+e,� dS )+zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)� CI_TESTING)�spawn_testproc)�HAS_NET_IO_COUNTERS)�mock)�PsutilTestCase)� PYTHON_EXE)�retry_on_failure)�sh)�skip_on_access_denied)� terminate)�unittest)�which)� getpagesizec C s� dg}t r|�d� |dk r0|�dt|�g� n ts8trD|�d� n |�d� trftd�}|�| | �} |�d| g� t|�}t r�|� � }n|� � d d� }g }x@|D ]8}|� � }y t |�}W n t k r� Y nX |�|� q�W |dkr�|S |d S dS ) z| Wrapper for calling the ps command with a little bit of cross-platform support for a narrow range of features. �psz --no-headersNz-pz-AZax)�command�comm�startZstimez-o� r ) r �append�extend�strr r �set�getr � splitlines�strip�int� ValueError)�fmt�pid�cmdZfmt_map�outputZ all_output�line� r) �J/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_posix.pyr - s6 r c C s d}t r d}t|| ��� d S )Nr r r )r r �split)r% �fieldr) r) r* �ps_namei s r- c C s d}t s trd}t|| �S )Nr �args)r r r )r% r, r) r) r* �ps_argsp s r/ c C s d}t r d}t|| �S )NZrssZrssize)r r )r% r, r) r) r* �ps_rssw s r0 c C s d}t r d}t|| �S )NZvszZvsize)r r )r% r, r) r) r* �ps_vsz~ s r1 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� Ze�ep�ed �d d � �Zd d � Zd!d"� Ze�ed#�e�e d$�d%d&� ��Z d'S )(� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s t tddgtjd�j| _d S )Nz-Ez-O)�stdin)r r � subprocess�PIPEr% )�clsr) r) r* � setUpClass� s zTestProcess.setUpClassc C s t | j� d S )N)r r% )r6 r) r) r* � tearDownClass� s zTestProcess.tearDownClassc C s, t d| j�}t�| j��� }| �||� d S )N�ppid)r r% �psutil�Processr9 � assertEqual)�selfZppid_psZ ppid_psutilr) r) r* � test_ppid� s zTestProcess.test_ppidc C s. t d| j�}t�| j��� j}| �||� d S )N�uid)r r% r: r; �uids�realr< )r= Zuid_psZ uid_psutilr) r) r* �test_uid� s zTestProcess.test_uidc C s. t d| j�}t�| j��� j}| �||� d S )NZrgid)r r% r: r; ZgidsrA r< )r= Zgid_psZ gid_psutilr) r) r* �test_gid� s zTestProcess.test_gidc C s, t d| j�}t�| j��� }| �||� d S )N�user)r r% r: r; �usernamer< )r= Z username_psZusername_psutilr) r) r* � test_username� s zTestProcess.test_usernamec C sJ t �� }tjdtd��*}| �|�� t|�� j �� |j s<t �W d Q R X d S )Nzpsutil.pwd.getpwuid)� side_effect) r: r; r �patch�KeyErrorr< rE r r@ rA �called�AssertionError)r= �p�funr) r) r* �test_username_no_resolution� s z'TestProcess.test_username_no_resolutionc C s< t �d� t| j�}t�| j��� d d }| �||� d S )Ng�������?r i )�time�sleepr0 r% r: r; � memory_infor< )r= Zrss_psZ rss_psutilr) r) r* �test_rss_memory� s zTestProcess.test_rss_memoryc C s< t �d� t| j�}t�| j��� d d }| �||� d S )Ng�������?r i )rO rP r1 r% r: r; rQ r< )r= Zvsz_psZ vsz_psutilr) r) r* �test_vsz_memory� s zTestProcess.test_vsz_memoryc C sv t | j�}tj�|��� }t�| j��� �� }t � dd|�}t � dd|�}t � dd|�}t � dd|�}| � ||� d S )Nz\d.\d� z\d) r- r% �os�path�basename�lowerr: r; �name�re�subr< )r= Zname_psZ name_psutilr) r) r* � test_name� s zTestProcess.test_namec C s^ d}dddg}t jd|d��8 t jd|d�� t�� }| �|�� 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 rH r: r; r< rY )r= rY �cmdlinerL r) r) r* �test_name_long� s zTestProcess.test_name_longc C s\ d}t jd|d��@ t jdt�dd�d�� t�� }| �|�� d� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)r] z"psutil._psplatform.Process.cmdliner rT )rG )r rH r: � AccessDeniedr; r< rY )r= rY rL 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 jdt�dd�d�� t�� }| �tj|j� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)r] z"psutil._psplatform.Process.cmdliner rT )rG )r rH r: � NoSuchProcessr; � assertRaisesrY )r= rY rL r) r) r* � test_name_long_cmdline_nsp_exc� s z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablec C s\ t d| j�}t�| j��� }tj�|��d�}t|�}tj�|��d�}| � |||g� d S )Nr z%H:%M:%S) r r% r: r; Z create_time�datetime� fromtimestamp�strftime�round�assertIn)r= 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�}t�| j��� }y| �||� W n0 tk rZ |d t|�� }| �||� Y nX d S )N)r- r% r: r; Zexer< rK �len)r= Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer) r) r* �test_exe s zTestProcess.test_exec C s0 t | j�}d�t�| j��� �}| �||� d S )N� )r/ r% �joinr: r; r^ r< )r= Z ps_cmdlineZpsutil_cmdliner) r) r* � test_cmdline s zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}t�� �� }| �||� d S )N�nice)r r% r: r; rp r< )r= Zps_niceZ psutil_nicer) r) r* � test_nice s zTestProcess.test_niceN) �__name__� __module__� __qualname__�__doc__� classmethodr7 r8 r> rB rC rF rN r r rR rS r\ r_ ra rd r �skipIfr r rj rl ro r r rq r) r) r) r* r2 � s* r2 c @ s� e Zd ZdZe� dd� �Ze�ed�e�e d� d�e�e d�dd � ���Z e�e o^e �� d �e� d d � ��Zd d� Zdd� Zdd� Zdd� Ze�ed�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� t�� �ts$tr0d� kr0� �dd� t� �t�� dkrz� fdd��D ��fdd�� D � }| �dt |� � d S )Nr% r r c s g | ]}|� kr|�qS r) r) )�.0�x)�pids_psr) r* � <listcomp>4 s z,TestSystemAPIs.test_pids.<locals>.<listcomp>c s g | ]}|� kr|�qS r) r) )ry rz )� pids_psutilr) r* r| 5 s z difference: ) �sortedr r: Zpidsr r �insertrk �failr )r= � differencer) )r{ r} r* � test_pids' s zTestSystemAPIs.test_pidszunreliable on SUNOS�ifconfigzno ifconfig cmdz not supportedc C sV t d�}xHtjdd��� D ]4}x.|�� D ]}|�|�r(P q(W | �d||f � qW d S )Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output %s)r r: Znet_io_counters�keysr+ � startswithr� )r= r' Znicr( r) r) r* �test_nic_names: s z TestSystemAPIs.test_nic_nameszunreliable on CIc C s� t d�}|�� s| �d��|�d�}dd� |D �}dd� |D �}| �t|�tt�� �� x,t�� D ] }| �|j |� | �|j |� qbW d S )N�whozno users on this system� c S s g | ]}|� � d �qS )r )r+ )ry rz r) r) r* r| O s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|� � d �qS )r )r+ )ry rz r) r) r* r| P s ) r r! ZskipTestr+ r< rk r: �usersri rY �terminal)r= �out�linesr� Z terminals�ur) r) r* � test_usersH s zTestSystemAPIs.test_usersc C sF t jdttjd�d��&}| �ttjjt � � � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.killrT )rG ) r rH �OSError�errno�EBADFrc r: �_psposixZ pid_existsrU �getpidrJ rK )r= �mr) r) r* �test_pid_exists_let_raiseV s z(TestSystemAPIs.test_pid_exists_let_raisec C sF t jdttjd�d��&}| �ttjjt � � � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.waitpidrT )rG ) r rH r� r� r� rc r: r� �wait_pidrU r� rJ rK )r= r� r) r) r* �test_os_waitpid_let_raise_ s z(TestSystemAPIs.test_os_waitpid_let_raisec C sN t jdttjd�d��.}| jtjjtjj t � � dd� |j s@t �W d Q R X d S )Nzpsutil._psposix.os.waitpidrT )rG g{�G�z�?)�timeout)r rH r� r� ZEINTRrc r: r� �TimeoutExpiredr� rU r� rJ rK )r= r� r) r) r* �test_os_waitpid_eintrg s z$TestSystemAPIs.test_os_waitpid_eintrc C s> t jddd��&}| �ttjjt�� � |j s0t �W d Q R X d S )Nzpsutil._psposix.os.waitpid)r ���)r] ) r rH rc r# r: r� r� rU r� rJ rK )r= r� r) r) r* � test_os_waitpid_bad_ret_statusp s z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc C s� dd� }d}x�t jdd�D ]�}t �|j�}y||j�\}}}}W nL tk r� } z.t| ��� } d| ksvd| ksvd| krzwn� W d d } ~ X Y qX | j|j ||d � | j|j ||d � | j|j ||d � | j|j |d d � qW d S ) Nc S sv t d| ��� }|�d�d }|�� }t|d �d }t|d �d }t|d �d }t|d �dd ��}||||fS ) Nzdf -k %sr� r i � � � �%rT )r r! r+ r" �float�replace)�devicer� r( �fields�total�used�free�percentr) r) r* �df| s z*TestSystemAPIs.test_disk_usage.<locals>.dfi @ F)�allzno such file or directoryzraw devices not supportedzpermission denied)�deltar ) r: Zdisk_partitions� disk_usage� mountpointr� � RuntimeErrorr rX ZassertAlmostEqualr� r� r� r� ) r= r� Z tolerance�part�usager� r� r� r� �errr) r) r* �test_disk_usagey s" z TestSystemAPIs.test_disk_usageN)rr rs rt ru r r� r rw r r r r� r r: r� r� r� r� r� r� r r� r) r) r) r* rx # s rx c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }| �|d� | �|t� � � | �|tj� d S )Nr )r Z assertGreaterr< �resource�mmapZPAGESIZE)r= Zpagesizer) r) r* �test_getpagesize� s zTestMisc.test_getpagesizeN)rr rs rt r� r) r) r) r* r� � s r� �__main__)� run_from_name)N)-ru re r� rU rZ r4 rO r: r r r r r r r Z psutil.testsr r r r r r r r r r r r r� r� Zpsutil._psutil_posixr r r- r/ r0 r1 rw r2 rx r� rr Zpsutil.tests.runnerr� �__file__r) r) r) r* �<module> sZ < y