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.tearDowncCs2tj�}x$|D]}|j|t�|j|d�qWdS)Nr)r#� cpu_timesr^�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_ifcCsVt��F}tt|��.tjdd�}x|D]}|j|jt�q(WWdQRXWdQRXdS)NZunix)Zkind)rrrr#Znet_connectionsr^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_statsrwr^rb)r%ry�_r'r'r(�test_net_if_stats�szTestSystem.test_net_if_statscCs.x(tjdd�j�D]\}}|j|t�qWdS)NT)Zpernic)r#Znet_io_countersrwr^rb)r%ryr|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=rwr^rb�label)r%rt�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;rwr^rbr)r%rtr�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^rtrbrErx�host�pid�int)r%�userr'r'r(� test_userss zTestSystem.test_usersN)rUrVrWrX�classmethodrirjrprrrsrZskipIfr rrr rvr{r}r~rr>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}�xtj �D�]�}|j��x�|D�]�}|}ydf} i} t||d�}|dk r�t |�r�|d kr�tjf} n|dkr�ddi} || | �}n|}|d7}W�n2tk �r>d|jjd|}t|�Yq�tjtjfk �r�} z0|j| j|j�| j�r�|j| j|j��WYdd} ~ Xq�tk �r>} z�dddd}|d||f7}||k�r�|dt|�7}|d7}|dd7}|dtj�7}djdd�|j�D��}|d7}|j|�PWYdd} ~ Xq�X|ddgdd ifk�rVt||�}|||�q�WWdQRXq�W|�r�|jd j|��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�rt� Exception�repr� traceback� format_exc�join� splitlinesZfail)r%Zvalid_procsZexcluded_namesZattrsrt�defaultZfailures�prn�args�kwargs�attr�msg�err�s�methr'r'r(�test_fetch_all,st z$TestFetchAllProcesses.test_fetch_allcCs*|j|t�x|D]}|j|t�qWdS)N)r^�listrb)r%rnrg�partr'r'r(�cmdlinets zTestFetchAllProcesses.cmdlinecCsN|j|ttd�f�|s&|j|d�n$trJtjj|�rJttd�rJttd�rJdS)Nr��access�X_OK) r^rbrxr!r r6r7�isfiler")r%rnrgr'r'r(�exeyszTestFetchAllProcesses.execCs|j|t�|j|d�dS)Nr)r^r�rm)r%rnrgr'r'r(r��szTestFetchAllProcesses.pidcCs |j|ttf�|j|d�dS)Nr)r^r�rrm)r%rnrgr'r'r(�ppid�szTestFetchAllProcesses.ppidcCs|j|t�tsdS)N)r^rbr)r%rnrgr'r'r(rt�szTestFetchAllProcesses.namecCs^|j|t�y|j|d�Wn*tk rFtr@|j�tjkr@n�YnXtj dtj |��dS)Nrz%Y %m %d %H:%M:%S)r^rlrm�AssertionErrorr�statusr#Z STATUS_ZOMBIE�timeZstrftimeZ localtime)r%rnrgr'r'r(�create_time�sz!TestFetchAllProcesses.create_timecCs8x2|D]*}|j|t�|j|d�|j||j�qWdS)Nr)r^r�rmrar�)r%rnrgZuidr'r'r(rB�s zTestFetchAllProcesses.uidscCsDx>|D]6}|j|t�trtr|j|d�|j||j�qWdS)Nr)r^r�r rrmrar�)r%rnrg�gidr'r'r(�gids�s zTestFetchAllProcesses.gidscCs"|j|t�tr|j||j�dS)N)r^rbr rar�)r%rnrgr'r'r(�username�szTestFetchAllProcesses.usernamecCs(|j|t�|j|d�|j|t�dS)N�?)r^rbZassertNotEqualrar)r%rnrgr'r'r(r��szTestFetchAllProcesses.statuscCs6x0|D](}|j|ttf�|dkr|j|d�qWdS)Nr�r��)r^r�rrm)r%rnrg�fieldr'r'r(rJ�s z!TestFetchAllProcesses.io_counterscCs\trx|D]}|j|t�q Wtr@|j|jd�|j|jd�n|j|d�|j|d�dS)Nrr�r/)rr�r/)r r^r�rrmZioclass�valuera)r%rnrgr�r'r'r(rG�s zTestFetchAllProcesses.ionicecCs|j|t�|j|d�dS)Nr�)r^r�rm)r%rnrgr'r'r(�num_threads�sz!TestFetchAllProcesses.num_threadscCsf|j|t�xT|D]L}|j|jd�|j|jd�|j|jd�x|D]}|j|ttf�qFWqWdS)Nr)r^r�rm�idZ user_timeZsystem_timer�rl)r%rnrg�tr�r'r'r(�threads�s zTestFetchAllProcesses.threadscCs*x$|D]}|j|t�|j|d�qWdS)Nr)r^rlrm)r%rnrgror'r'r(rk�s zTestFetchAllProcesses.cpu_timescCs|j|t�dS)N)r^rl)r%rnrgr'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�rrmr#� cpu_countr!rar��range)r%rnrgr'r'r(rQ�szTestFetchAllProcesses.cpu_numcCs�x(|D] }|j|ttf�|j|d�qWtrttrt|jdkrtxx|jD](}|dkrFt||�}|j |j||d�qFWnDt r�|j|j|j�|j|j |j�|j|j|j�|j|j|j�dS)Nr�vms)r�)r^r�rrmr rr��_fieldsr�� assertGreaterrZ peak_wsetZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZ peak_pagefileZpagefile)r%rnrgr�rtr'r'r(�memory_info�s z!TestFetchAllProcesses.memory_infocCsvtj�j}xR|jD]H}t||�}|j|ttf�|j|d||fd�|j |||||fd�qWt rr|j|j|j�dS)Nr)r�) r#Zvirtual_memory�totalr�r�r^r�rrmZassertLessEqualrZpssZuss)r%rnrgr�rtr�r'r'r(�memory_full_infos z&TestFetchAllProcesses.memory_full_infocCs�|j|t�x�|D]�}|j|jt�|j|jt�trF|j|jd�qtr�|j|j t�|j|j t�|j|jt�|j|j d�|j |j d �|j|jd�qtr|jrqqWdS) Nr�r�rrd�a�r+�a+r�)r�rdr�r�r�)r^r��fdr�r7rbrr!rZposition�mode�flagsrmrar�r)r%rnrg�fr'r'r(� open_filess z TestFetchAllProcesses.open_filescCs|j|t�|j|d�dS)Nr)r^r�rm)r%rnrgr'r'r(rL(szTestFetchAllProcesses.num_fdscCs2|jt|�tt|��x|D]}t|�qWdS)N)r!�lenr�r)r%rnrgrur'r'r(�connections,s z!TestFetchAllProcesses.connectionscCsh|rd|j|t�ytj|�}WnDtk rb}z(trD|jtjj krDn|jtj krR�WYdd}~XnXdS)N)r^rbr6�stat�OSErrorr�errnor#Z_psplatformZACCESS_DENIED_SET�ENOENT)r%rnrg�str�r'r'r(�cwd1s zTestFetchAllProcesses.cwdcCs|j|t�dS)N)r^rl)r%rnrgr'r'r(�memory_percentAsz$TestFetchAllProcesses.memory_percentcCs|j|t�dS)N)r^�bool)r%rnrgr'r'r(� is_runningEsz TestFetchAllProcesses.is_runningcCsB|j|t�ttj��}x$|D]}|j|t�|j||�qWdS)N)r^r�r�r#r�r�ra)r%rnrgZcpusror'r'r(r Hs z"TestFetchAllProcesses.cpu_affinitycCs |j|ttd�f�|dk rdS)N)r^rbrx)r%rnrgr'r'r(rEPszTestFetchAllProcesses.terminalcCs�x�|D]�}|j|jt�|j|jt�|j|jt�xR|jD]H}t||�}|dkr^|jd�s�q<|dkrhq<|j|tt f�|j |d�q.)r^r�r r�r#ra)r%rnrgZ prioritiesr'r'r(�nicems zTestFetchAllProcesses.nicecCs.x(|D] }|j|ttf�|j|d�qWdS)Nr)r^r�rrm)r%rnrgr�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�rm)r%rnrgr'r'r(r2|szTestFetchAllProcesses.rlimitcCs>|j|t�x,|j�D] \}}|j|t�|j|t�qWdS)N)r^�dictrwrb)r%rnrgrq�vr'r'r(r@�szTestFetchAllProcesses.environN)'rUrVrWrXr�r�r�r�r�r�rtr�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