3 ��JZ2^�@s�dZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl mZddl mZddl m Z ddl mZdd l mZdd l mZddl mZddl mZdd l mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddlm&Z&ddlm'Z'ddl Z Gd d!�d!e$j(�Z)Gd"d#�d#e$j(�Z*Gd$d%�d%e$j(�Z+Gd&d'�d'e$j(�Z,e-d(k�r�e e.�dS))z�Contracts tests. These tests mainly check API sanity in terms of returned types and APIs availability. Some of these are duplicates of tests test_system.py and test_process.py �N)�closing)�AIX)�BSD)�FREEBSD)�LINUX)�NETBSD)�OPENBSD)�OSX)�POSIX)�SUNOS)�WINDOWS)�callable)�long)�bind_unix_socket)�check_connection_ntuple)�get_kernel_version)�HAS_CONNECTIONS_UNIX)� HAS_RLIMIT)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)� is_namedtuple)�run_test_module_by_name)�safe_rmpath)�skip_on_access_denied)�TESTFN)�unittest)�unix_socket_path)�VALID_PROC_STATUSES)�warnc@s�eZdZdZdd�Zdd�Zdd�Zdd �Zd d�Zdd �Z dd�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.S)/�TestAvailabilityzQMake sure code reflects what doc promises in terms of APIs availability. cCs$tp tp t}|jttjd�|�dS)N�cpu_affinity)rrr�assertEqual�hasattr�psutil�Process)�self�hasit�r'�&/usr/lib64/python3.6/test_contracts.py�test_cpu_affinity<sz"TestAvailability.test_cpu_affinitycCs(|jttd�t�|jttd�t�dS)NZwin_service_iterZwin_service_get)r!r"r#r)r%r'r'r(�test_win_service@sz!TestAvailability.test_win_servicecCs|jttd�tptpt�dS)NZPROCFS_PATH)r!r"r#rrr)r%r'r'r(�test_PROCFS_PATHDsz!TestAvailability.test_PROCFS_PATHcCsj|j}|ttd�t�|ttd�t�|ttd�t�|ttd�t�|ttd�t�|ttd�t�dS)NZABOVE_NORMAL_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZIDLE_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZREALTIME_PRIORITY_CLASS)r!r"r#r)r%�aer'r'r(�test_win_priorityHsz"TestAvailability.test_win_prioritycCsJ|j}|ttd�t�|ttd�t�|ttd�t�|ttd�t�dS)NZIOPRIO_CLASS_NONEZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE)r!r"r#r)r%r,r'r'r(�test_linux_ioprioQs z"TestAvailability.test_linux_iopriocCsH|j}tot�dk}|ttjd�|�|ttd�|�|ttd�|�|ttd�|�|ttd�|�|ttd �|�|ttd �|�|ttd�|�|ttd�|�|ttd �|�|ttd�|�|ttd�|�|ttd�|�to�t�dk}|ttd�|�|ttd�|�|ttd�|�|ttd�|�|ttd�|�dS)N��$�rlimitZ RLIM_INFINITYZ RLIMIT_ASZRLIMIT_COREZ RLIMIT_CPUZRLIMIT_DATAZRLIMIT_FSIZEZRLIMIT_LOCKSZRLIMIT_MEMLOCK� RLIMIT_NOFILEZRLIMIT_NPROCZ RLIMIT_RSSZRLIMIT_STACK�rZRLIMIT_MSGQUEUEZRLIMIT_NICEZ RLIMIT_RTPRIOZ RLIMIT_RTTIMEZRLIMIT_SIGPENDING)r/r0r1)r4r)r!rrr"r#r$)r%r,r&r'r'r(�test_linux_rlimitXs*z"TestAvailability.test_linux_rlimitcCs:totjjd�ptjjd�}|jttd�|p2tp2t�dS)Nz/sys/devices/system/cpu/cpufreqz$/sys/devices/system/cpu/cpu0/cpufreqZcpu_freq) r�os�path�existsr!r"r#r r)r%�linuxr'r'r(� test_cpu_freqpszTestAvailability.test_cpu_freqcCs|jttd�t�dS)N�sensors_temperatures)r!r"r#r)r%r'r'r(�test_sensors_temperaturesvsz*TestAvailability.test_sensors_temperaturescCs|jttd�t�dS)N�sensors_fans)r!r"r#r)r%r'r'r(�test_sensors_fansysz"TestAvailability.test_sensors_fanscCs"|jttd�tptptpt�dS)NZsensors_battery)r!r"r#rrrr )r%r'r'r(�test_battery|szTestAvailability.test_batterycCs |jttjd�tptpt�dS)N�environ)r!r"r#r$rr r)r%r'r'r(�test_proc_environ�sz"TestAvailability.test_proc_environcCs|jttjd�t�dS)N�uids)r!r"r#r$r )r%r'r'r(�test_proc_uids�szTestAvailability.test_proc_uidscCs|jttjd�t�dS)NrB)r!r"r#r$r )r%r'r'r(�test_proc_gids�szTestAvailability.test_proc_gidscCs|jttjd�t�dS)N�terminal)r!r"r#r$r )r%r'r'r(�test_proc_terminal�sz#TestAvailability.test_proc_terminalcCs|jttjd�tpt�dS)N�ionice)r!r"r#r$rr)r%r'r'r(�test_proc_ionice�sz!TestAvailability.test_proc_ionicecCs|jttjd�t�dS)Nr2)r!r"r#r$r)r%r'r'r(�test_proc_rlimit�sz!TestAvailability.test_proc_rlimitcCs(ttjd�}|j|tstrdnd�dS)N�io_countersFT)r"r#r$r!r r)r%r&r'r'r(�test_proc_io_counters�sz&TestAvailability.test_proc_io_counterscCs|jttjd�t�dS)N�num_fds)r!r"r#r$r )r%r'r'r(�test_proc_num_fds�sz"TestAvailability.test_proc_num_fdscCs|jttjd�t�dS)N�num_handles)r!r"r#r$r)r%r'r'r(�test_proc_num_handles�sz&TestAvailability.test_proc_num_handlescCs |jttjd�tptpt�dS)Nr )r!r"r#r$rrr)r%r'r'r(�test_proc_cpu_affinity�sz'TestAvailability.test_proc_cpu_affinitycCs |jttjd�tptpt�dS)N�cpu_num)r!r"r#r$rrr)r%r'r'r(�test_proc_cpu_num�sz"TestAvailability.test_proc_cpu_numcCs,ttjd�}|j|tststr"dnd�dS)N�memory_mapsFT)r"r#r$r!rrr)r%r&r'r'r(�test_proc_memory_maps�sz&TestAvailability.test_proc_memory_mapsN)�__name__� __module__�__qualname__�__doc__r)r*r+r-r.r5r:r<r>r?rArCrDrFrHrIrKrMrOrPrRrTr'r'r'r(r7s. rc@seZdZdd�ZdS)�TestDeprecationscCsdtjdd��}tj�j�WdQRX|d}|j|j�t�|jdt |j ��|jdt |j ��dS)NT)�recordrzmemory_info_ex() is deprecatedzuse memory_info() instead)�warnings�catch_warningsr#r$�memory_info_ex�assertIsInstance�category� FutureWarning�assertIn�str�message)r%Zws�wr'r'r(�test_memory_info_ex�sz$TestDeprecations.test_memory_info_exN)rUrVrWrer'r'r'r(rY�srYc@s�eZdZdZedd��Zdd�Zdd�Zdd �Zd d�Z e jed�e je d �eed�dd��Zdd�Zdd�Zdd�Ze jed�dd��Ze jed�dd��Zdd�ZdS)� TestSystemz�Check the return types of system related APIs. Mainly we want to test we never return unicode on Python 2, see: https://github.com/giampaolo/psutil/issues/1039 cCstj�|_dS)N)r#r$�proc)�clsr'r'r(� setUpClass�szTestSystem.setUpClasscCstt�dS)N)rr)r%r'r'r(�tearDown�szTestSystem.tearDowncCs>tj�}t|�st�x$|D]}|j|t�|j|d�qWdS)Nr)r#� cpu_timesr�AssertionErrorr^�float�assertGreaterEqual)r%�ret�nr'r'r(�test_cpu_times�s zTestSystem.test_cpu_timescCs&x tjdd�D]}|j|t�qWdS)NT)Zperdisk)r#Zdisk_io_countersr^rb)r%�kr'r'r(�test_io_counters�szTestSystem.test_io_counterscCsNxHtj�D]<}|j|jt�|j|jt�|j|jt�|j|jt�q WdS)N)r#Zdisk_partitionsr^ZdevicerbZ mountpointZfstypeZopts)r%Zdiskr'r'r(�test_disk_partitions�s zTestSystem.test_disk_partitionsz POSIX onlyzcan't list UNIX sockets)Zonly_ifcCs^t��N}tt|��6tjdd�}|s*t�x|D]}|j|jt�q0WWdQRXWdQRXdS)NZunix)Zkind) rrrr#Znet_connectionsrlr^Zladdrrb)r%�nameZcons�connr'r'r(�test_net_connections�s zTestSystem.test_net_connectionscCsrxltj�j�D]\\}}|j|t�xF|D]>}|j|jt�|j|jttd�f�|j|jttd�f�q(WqWdS)N) r#Znet_if_addrs�itemsr^rbZaddressZnetmask�typeZ broadcast)r%�ifnameZaddrs�addrr'r'r(�test_net_if_addrs�s zTestSystem.test_net_if_addrscCs*x$tj�j�D]\}}|j|t�qWdS)N)r#Znet_if_statsrxr^rb)r%rz�_r'r'r(�test_net_if_stats�szTestSystem.test_net_if_statscCs.x(tjdd�j�D]\}}|j|t�qWdS)NT)Zpernic)r#Znet_io_countersrxr^rb)r%rzr}r'r'r(�test_net_io_counters�szTestSystem.test_net_io_countersz not supportedcCsFx@tj�j�D]0\}}|j|t�x|D]}|j|jt�q(WqWdS)N)r#r=rxr^rb�label)r%ru�units�unitr'r'r(r>�s zTestSystem.test_sensors_fanscCsFx@tj�j�D]0\}}|j|t�x|D]}|j|jt�q(WqWdS)N)r#r;rxr^rbr�)r%rur�r�r'r'r(r<s z$TestSystem.test_sensors_temperaturescCsfx`tj�D]T}|j|jt�|j|jttd�f�|j|jttd�f�|j|jt td�f�q WdS)N) r#�usersr^rurbrEry�host�pid�int)r%�userr'r'r(� test_userss zTestSystem.test_usersN)rUrVrWrX�classmethodrirjrqrsrtrZskipIfr rrr rwr|r~rrr>rr<r�r'r'r'r(rf�s rfc@s(eZdZdZdd�Zdd�Zdd�Zdd �Zd d�Zdd �Z dd�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6d7�Zd8d9�Zd:d;�Z dd?�Z"d@dA�Z#dBdC�Z$dDdE�Z%dFdG�Z&dHS)I�TestFetchAllProcessesz~Test which iterates over all running processes and performs some sanity checks against Process API's returned values. cCsdtr`ddl}ddl}|j�}|j�}tdd�|D��|_tdd�|D��|_tdd�|D��|_dS)NrcSsg|] }|j�qSr')Zpw_uid)�.0�xr'r'r(� (sz/TestFetchAllProcesses.setUp..cSsg|] }|j�qSr')Zpw_name)r�r�r'r'r(r�)scSsg|] }|j�qSr')Zgr_gid)r�r�r'r'r(r�*s) r �pwd�grpZgetpwallZgetgrall�set�all_uids� all_usernames�all_gids)r%r�r�r��groupsr'r'r(�setUp"szTestFetchAllProcesses.setUpcCs�d}tdddddddd d ddg�}tr6tr6|jd �g}x2ttj�D]$}|jd�rVqF||kr`qF|j|�qFWt �}g}�x(tj �D�]}|j��x�|D�]�}|}ydf} i} t||d�}|dk r�t |�r�|d kr�tjf} n|dkr�ddi} || | �}n|}|d7}W�nLtk �r>d|jjd|}t|�Yq�tjtjfk �r�} zJ|j| j|j�| j�r�|j| j|j��t| ��s�t�| j�s�t�WYdd} ~ Xq�tk �rX} z�dddd}|d||f7}||k�r�|dt|�7}|d7}|dd7}|dtj�7}djdd�|j �D��}|d7}|j|�PWYdd} ~ Xq�X|ddgdd ifk�r~|�s~t|��t||�}|||�q�WWdQRXq�W|�r�|j!d j|��|�s�t�dS)!NrZsend_signalZsuspendZresumeZ terminate�kill�waitZas_dict�parentZchildrenr]�oneshotr2r}rSZgroupedF�z&%r was skipped because not implementedz.test_� �=�FzFAIL: test_%s (proc=%sz , ret=%s)z) �-z %scss|]}d|VqdS)� �Nz r')r��ir'r'r(� csz7TestFetchAllProcesses.test_fetch_all..g�)"r�rr�add�dirr#r$� startswith�append�objectZprocess_iterr��getattrr r3�NotImplementedError� __class__rUrZ NoSuchProcessZAccessDeniedr!r�rurbrl�msg� Exception�repr� traceback� format_exc�join� splitlinesZfail)r%Zvalid_procsZexcluded_namesZattrsru�defaultZfailures�pro�args�kwargs�attrr��err�s�methr'r'r(�test_fetch_all,sx z$TestFetchAllProcesses.test_fetch_allcCs*|j|t�x|D]}|j|t�qWdS)N)r^�listrb)r%rorg�partr'r'r(�cmdlinets zTestFetchAllProcesses.cmdlinecCst|j|ttd�f�|s&|j|d�nJtjj|�s:t|��trptjj |�rpt td�rpt td�rptj|tj�spt�dS)Nr��access�X_OK) r^rbryr!r6r7�isabsrlr �isfiler"r�r�)r%rorgr'r'r(�exeyszTestFetchAllProcesses.execCs|j|t�|j|d�dS)Nr)r^r�rn)r%rorgr'r'r(r��szTestFetchAllProcesses.pidcCs |j|ttf�|j|d�dS)Nr)r^r�rrn)r%rorgr'r'r(�ppid�szTestFetchAllProcesses.ppidcCs|j|t�ts|st�dS)N)r^rbrrl)r%rorgr'r'r(ru�szTestFetchAllProcesses.namecCs^|j|t�y|j|d�Wn*tk rFtr@|j�tjkr@n�YnXtj dtj |��dS)Nrz%Y %m %d %H:%M:%S)r^rmrnrlr�statusr#Z STATUS_ZOMBIE�timeZstrftimeZ localtime)r%rorgr'r'r(�create_time�sz!TestFetchAllProcesses.create_timecCsDt|�st�x2|D]*}|j|t�|j|d�|j||j�qWdS)Nr)rrlr^r�rnrar�)r%rorgZuidr'r'r(rB�s zTestFetchAllProcesses.uidscCsPt|�st�x>|D]6}|j|t�trtr|j|d�|j||j�qWdS)Nr) rrlr^r�r rrnrar�)r%rorg�gidr'r'r(�gids�s zTestFetchAllProcesses.gidscCs*|j|t�|st�tr&|j||j�dS)N)r^rbrlr rar�)r%rorgr'r'r(�username�szTestFetchAllProcesses.usernamecCs0|j|t�|st�|j|d�|j|t�dS)N�?)r^rbrlZassertNotEqualrar)r%rorgr'r'r(r��szTestFetchAllProcesses.statuscCsBt|�st�x0|D](}|j|ttf�|dkr|j|d�qWdS)Nr�r��)rrlr^r�rrn)r%rorg�fieldr'r'r(rJ�s z!TestFetchAllProcesses.io_counterscCshtr*t|�st�x|D]}|j|t�qWtrL|j|jd�|j|jd�n|j|d�|j |d�dS)Nrr�r/)rr�r/) r rrlr^r�rrnZioclass�valuera)r%rorgr�r'r'r(rG�s zTestFetchAllProcesses.ionicecCs|j|t�|j|d�dS)Nr�)r^r�rn)r%rorgr'r'r(�num_threads�sz!TestFetchAllProcesses.num_threadscCsr|j|t�x`|D]X}t|�s"t�|j|jd�|j|jd�|j|jd�x|D]}|j|tt f�qRWqWdS)Nr) r^r�rrlrn�idZ user_timeZsystem_timer�rm)r%rorg�tr�r'r'r(�threads�s zTestFetchAllProcesses.threadscCs6t|�st�x$|D]}|j|t�|j|d�qWdS)Nr)rrlr^rmrn)r%rorgrpr'r'r(rk�s zTestFetchAllProcesses.cpu_timescCs0|j|t�d|kodkns,t|��dS)NggY@)r^rmrl)r%rorgr'r'r(�cpu_percent�sz!TestFetchAllProcesses.cpu_percentcCs\|j|t�tr|dkrdS|j|d�tj�dkr@|j|d�|j|tt tj��dS)Nr�rr�) r^r�rrnr#� cpu_countr!rar��range)r%rorgr'r'r(rQ�szTestFetchAllProcesses.cpu_numcCs�t|�st�x(|D] }|j|ttf�|j|d�qWtr�tr�|jdkr�xx|j D](}|dkrRt ||�}|j|j||d�qRWnDtr�|j|j |j�|j|j|j�|j|j|j�|j|j|j�dS)Nr�vms)r�)rrlr^r�rrnr rr��_fieldsr�� assertGreaterrZ peak_wsetZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZ peak_pagefileZpagefile)r%rorgr�rur'r'r(�memory_info�s z!TestFetchAllProcesses.memory_infocCs�t|�st�tj�j}xR|jD]H}t||�}|j|tt f�|j |d||fd�|j|||||fd�qWtr~|j |j |j�dS)Nr)r�)rrlr#Zvirtual_memory�totalr�r�r^r�rrnZassertLessEqualrZpssZuss)r%rorgr�rur�r'r'r(�memory_full_infos z&TestFetchAllProcesses.memory_full_infocCs�|j|t�x�|D]�}|j|jt�|j|jt�trF|j|jd�nhtr�|j|j t�|j|j t�|j|jt�|j|j d�|j |j d �|j|jd�ntr�|jr�qtjj|j�s�t|��tjj|j�st|��qWdS) Nr�r�rrd�a�r+�a+r�)r�rdr�r�r�)r^r��fdr�r7rbrr!rZposition�mode�flagsrnrar�rr6r�rlr�)r%rorg�fr'r'r(� open_filess" z TestFetchAllProcesses.open_filescCs|j|t�|j|d�dS)Nr)r^r�rn)r%rorgr'r'r(rL(szTestFetchAllProcesses.num_fdscCs2|jt|�tt|��x|D]}t|�qWdS)N)r!�lenr�r)r%rorgrvr'r'r(�connections,s z!TestFetchAllProcesses.connectionscCs�|r�|j|t�tjj|�s$t|��ytj|�}WnDtk rv}z(trX|j t jjkrXn|j t j krf�WYdd}~XnXtj|j�s�t�dS)N)r^rbr6r7r�rl�stat�OSErrorr�errnor#Z_psplatformZACCESS_DENIED_SET�ENOENT�S_ISDIR�st_mode)r%rorg�str�r'r'r(�cwd1s zTestFetchAllProcesses.cwdcCs0|j|t�d|kodkns,t|��dS)Nr�d)r^rmrl)r%rorgr'r'r(�memory_percentAsz$TestFetchAllProcesses.memory_percentcCs|j|t�dS)N)r^�bool)r%rorgr'r'r(� is_runningEsz TestFetchAllProcesses.is_runningcCsR|j|t�|gkst|��ttj��}x$|D]}|j|t�|j||�q.WdS)N)r^r�rlr�r#r�r�ra)r%rorgZcpusrpr'r'r(r Hs z"TestFetchAllProcesses.cpu_affinitycCsH|j|ttd�f�|dk rDtjj|�s0t|��tjj|�sDt|��dS)N)r^rbryr6r7r�rlr8)r%rorgr'r'r(rEPszTestFetchAllProcesses.terminalcCs�x�|D]�}|j|jt�|j|jt�|j|jt�xr|jD]h}t||�}|dkrv|jd�s�tjj |j�s�t |j��q<|dkr�|s�t �q<|j|ttf�|j |d�q.i��)r^r�r rlr�r#ra)r%rorgZ prioritiesr'r'r(�nicems "zTestFetchAllProcesses.nicecCs:t|�st�x(|D] }|j|ttf�|j|d�qWdS)Nr)rrlr^r�rrn)r%rorgr�r'r'r(�num_ctx_switchesvs z&TestFetchAllProcesses.num_ctx_switchescCs@|j|t�|jt|�d�|j|dd�|j|dd�dS)Nr/rr�r�r�)r^�tupler!r�rn)r%rorgr'r'r(r2|szTestFetchAllProcesses.rlimitcCs>|j|t�x,|j�D] \}}|j|t�|j|t�qWdS)N)r^�dictrxrb)r%rorgrr�vr'r'r(r@�szTestFetchAllProcesses.environN)'rUrVrWrXr�r�r�r�r�r�rur�rBr�r�r�rJrGr�r�rkr�rQr�r�r�rLr�r�r�r�r rErSrNr�r�r2r@r'r'r'r(r�sH H r��__main__)/rXr�r6r�r�r�r[� contextlibrr#rrrrrrr r rrZpsutil._compatr rZpsutil.testsrrrrrrrrrrrrrrrrZTestCaserrYrfr�rU�__file__r'r'r'r(� sXx^n