← Back
Editing: test_contracts.cpython-36.pyc
3 ��JZ2^ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddl m Z ddl m Z ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ 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' ddl Z G d d!� d!e$j(�Z)G d"d#� d#e$j(�Z*G d$d%� d%e$j(�Z+G d&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� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � 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. c C s$ t p tp t}| jttjd�|� d S )N� cpu_affinity)r r r � assertEqual�hasattr�psutil�Process)�self�hasit� r' �&/usr/lib64/python3.6/test_contracts.py�test_cpu_affinity< s z"TestAvailability.test_cpu_affinityc C s( | j ttd�t� | j ttd�t� d S )NZwin_service_iterZwin_service_get)r! r" r# r )r% r' r' r( �test_win_service@ s z!TestAvailability.test_win_servicec C s | j ttd�tptpt� d S )NZ PROCFS_PATH)r! r" r# r r r )r% r' r' r( �test_PROCFS_PATHD s z!TestAvailability.test_PROCFS_PATHc C sj | j }|ttd�t� |ttd�t� |ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )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_priorityH s z"TestAvailability.test_win_priorityc C sJ | j }|ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )NZIOPRIO_CLASS_NONEZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE)r! r" r# r )r% r, r' r' r( �test_linux_ioprioQ s z"TestAvailability.test_linux_ioprioc C sH | j }tot� dk}|ttjd�|� |ttd�|� |ttd�|� |ttd�|� |ttd�|� |ttd �|� |ttd �|� |ttd �|� |ttd �|� |ttd �|� |ttd�|� |ttd�|� |ttd�|� to�t� dk}|ttd�|� |ttd�|� |ttd�|� |ttd�|� |ttd�|� d S )N� � �$ �rlimitZ RLIM_INFINITYZ RLIMIT_ASZ RLIMIT_COREZ RLIMIT_CPUZ RLIMIT_DATAZ RLIMIT_FSIZEZ RLIMIT_LOCKSZRLIMIT_MEMLOCK� RLIMIT_NOFILEZ RLIMIT_NPROCZ RLIMIT_RSSZ RLIMIT_STACK� r ZRLIMIT_MSGQUEUEZ RLIMIT_NICEZ RLIMIT_RTPRIOZ RLIMIT_RTTIMEZRLIMIT_SIGPENDING)r/ r0 r1 )r4 r )r! r r r" r# r$ )r% r, r& r' r' r( �test_linux_rlimitX s* z"TestAvailability.test_linux_rlimitc C s: t otjjd�ptjjd�}| jttd�|p2tp2t� d S )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_freqp s z TestAvailability.test_cpu_freqc C s | j ttd�t� d S )N�sensors_temperatures)r! r" r# r )r% r' r' r( �test_sensors_temperaturesv s z*TestAvailability.test_sensors_temperaturesc C s | j ttd�t� d S )N� sensors_fans)r! r" r# r )r% r' r' r( �test_sensors_fansy s z"TestAvailability.test_sensors_fansc C s" | j ttd�tptptpt� d S )NZsensors_battery)r! r" r# r r r r )r% r' r' r( � test_battery| s z TestAvailability.test_batteryc C s | j ttjd�tptpt� d S )N�environ)r! r" r# r$ r r r )r% r' r' r( �test_proc_environ� s z"TestAvailability.test_proc_environc C s | j ttjd�t� d S )N�uids)r! r" r# r$ r )r% r' r' r( �test_proc_uids� s z TestAvailability.test_proc_uidsc C s | j ttjd�t� d S )NrB )r! r" r# r$ r )r% r' r' r( �test_proc_gids� s z TestAvailability.test_proc_gidsc C s | j ttjd�t� d S )N�terminal)r! r" r# r$ r )r% r' r' r( �test_proc_terminal� s z#TestAvailability.test_proc_terminalc C s | j ttjd�tpt� d S )N�ionice)r! r" r# r$ r r )r% r' r' r( �test_proc_ionice� s z!TestAvailability.test_proc_ionicec C s | j ttjd�t� d S )Nr2 )r! r" r# r$ r )r% r' r' r( �test_proc_rlimit� s z!TestAvailability.test_proc_rlimitc C s( t tjd�}| j|tstr dnd� d S )N� io_countersFT)r" r# r$ r! r r )r% r& r' r' r( �test_proc_io_counters� s z&TestAvailability.test_proc_io_countersc C s | j ttjd�t� d S )N�num_fds)r! r" r# r$ r )r% r' r' r( �test_proc_num_fds� s z"TestAvailability.test_proc_num_fdsc C s | j ttjd�t� d S )N� num_handles)r! r" r# r$ r )r% r' r' r( �test_proc_num_handles� s z&TestAvailability.test_proc_num_handlesc C s | j ttjd�tptpt� d S )Nr )r! r" r# r$ r r r )r% r' r' r( �test_proc_cpu_affinity� s z'TestAvailability.test_proc_cpu_affinityc C s | j ttjd�tptpt� d S )N�cpu_num)r! r" r# r$ r r r )r% r' r' r( �test_proc_cpu_num� s z"TestAvailability.test_proc_cpu_numc C s, t tjd�}| j|ts ts tr"dnd� d S )N� memory_mapsFT)r" r# r$ r! r r r )r% r& r' r' r( �test_proc_memory_maps� s z&TestAvailability.test_proc_memory_mapsN)�__name__� __module__� __qualname__�__doc__r) r* r+ r- r. r5 r: r< r> r? rA rC rD rF rH rI rK rM rO rP rR rT r' r' r' r( r 7 s. r c @ s e Zd Zdd� ZdS )�TestDeprecationsc C sd t jdd��}tj� j� W d Q R X |d }| j|j� t� | jdt |j �� | jdt |j �� d S )NT)�recordr z memory_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� s z$TestDeprecations.test_memory_info_exN)rU rV rW re r' r' r' r( rY � s rY c @ s� e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd d � Z e j e d �e j e d �eed�dd� ���Zdd� Zdd� Zdd� Ze j e d�dd� �Ze j e d�dd� �Zd d � Zd S ) � 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 c C s t j� | _d S )N)r# r$ �proc)�clsr' r' r( � setUpClass� s zTestSystem.setUpClassc C s t t� d S )N)r r )r% r' r' r( �tearDown� s zTestSystem.tearDownc C s> t j� }t|�st�x$|D ] }| j|t� | j|d� qW d S )Nr )r# � cpu_timesr �AssertionErrorr^ �float�assertGreaterEqual)r% �ret�nr' r' r( �test_cpu_times� s zTestSystem.test_cpu_timesc C s& x t jdd�D ]}| j|t� qW d S )NT)Zperdisk)r# Zdisk_io_countersr^ rb )r% �kr' r' r( �test_io_counters� s zTestSystem.test_io_countersc C sN xHt j� D ]<}| j|jt� | j|jt� | j|jt� | j|jt� q W d S )N)r# Zdisk_partitionsr^ Zdevicerb Z mountpointZfstypeZopts)r% Zdiskr' r' r( �test_disk_partitions� s z TestSystem.test_disk_partitionsz POSIX onlyzcan't list UNIX sockets)Zonly_ifc C s^ t � �N}tt|���6 tjdd�}|s*t�x|D ]}| j|jt� q0W W d Q R X W d Q R X d S )NZunix)Zkind) r r r r# Znet_connectionsrl r^ Zladdrrb )r% �nameZcons�connr' r' r( �test_net_connections� s z TestSystem.test_net_connectionsc C sr xlt j� j� D ]\\}}| j|t� xF|D ]>}| j|jt� | j|jttd �f� | j|jttd �f� q(W qW d S )N) r# Z net_if_addrs�itemsr^ rb ZaddressZnetmask�typeZ broadcast)r% �ifnameZaddrs�addrr' r' r( �test_net_if_addrs� s z TestSystem.test_net_if_addrsc C s* x$t j� j� D ]\}}| j|t� qW d S )N)r# Z net_if_statsrx r^ rb )r% rz �_r' r' r( �test_net_if_stats� s z TestSystem.test_net_if_statsc C s. x(t jdd�j� D ]\}}| j|t� qW d S )NT)Zpernic)r# Znet_io_countersrx r^ rb )r% rz r} r' r' r( �test_net_io_counters� s z TestSystem.test_net_io_countersz not supportedc C sF x@t j� j� D ]0\}}| j|t� x|D ]}| j|jt� q(W qW d S )N)r# r= rx r^ rb �label)r% ru �units�unitr' r' r( r> � s z TestSystem.test_sensors_fansc C sF x@t j� j� D ]0\}}| j|t� x|D ]}| j|jt� q(W qW d S )N)r# r; rx r^ rb r� )r% ru r� r� r' r' r( r< s z$TestSystem.test_sensors_temperaturesc C sf x`t j� D ]T}| j|jt� | j|jttd �f� | j|jttd �f� | j|jt td �f� q W d S )N) r# �usersr^ ru rb rE ry �host�pid�int)r% �userr' r' r( � test_users s zTestSystem.test_usersN)rU rV rW rX � classmethodri rj rq rs rt r ZskipIfr r r r rw r| r~ r r r> r r< r� r' r' r' r( rf � s rf c @ s( e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� 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. c C sd t r`dd l}dd l}|j� }|j� }tdd� |D ��| _tdd� |D ��| _tdd� |D ��| _d S )Nr c S s g | ] }|j �qS r' )Zpw_uid)�.0�xr' r' r( � <listcomp>( s z/TestFetchAllProcesses.setUp.<locals>.<listcomp>c S s g | ] }|j �qS r' )Zpw_name)r� r� r' r' r( r� ) s c S s g | ] }|j �qS r' )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" s zTestFetchAllProcesses.setUpc C s� d}t dddddddd d d d g �}tr6t r6|jd � g }x2ttj�D ]$}|jd�rVqF||kr`qF|j|� qFW t � }g }�x(tj � D �]}|j � �� �x�|D �]�}|}ydf } i } t ||d �} | d k r�t | �r�|d kr�tjf} n|dkr�ddi} | | | �}n| }|d7 }W �nL tk �r> d| jjd | } t| � Y q� tjtjfk �r� } zJ| j| j|j� | j�r�| j| j|j� � t| ��s�t�| j�s�t�W Y d d } ~ X q� tk �rX } z�ddd d }|d||f 7 }||k�r�|dt |� 7 }|d7 }|dd 7 }|d t j � 7 }dj d d � |j � D ��}|d7 }|j|� P W Y d d } ~ X q�X |dd g d d i fk�r~|�s~t|��t | |�}|||� q�W W d Q R X q�W |�r�| j!d j |�� |�s�t�d S )!Nr Z send_signalZsuspendZresumeZ terminate�kill�waitZas_dict�parentZchildrenr] �oneshotr2 r} rS ZgroupedF� z&%r was skipped because not implementedz.test_� �=�F zFAIL: test_%s (proc=%sz , ret=%s)z) �-z %sc s s | ]}d| V qdS )� � Nz r' )r� �ir' r' r( � <genexpr>c s z7TestFetchAllProcesses.test_fetch_all.<locals>.<genexpr>g � )"r� r r �add�dirr# r$ � startswith�append�objectZ process_iterr� �getattrr r3 �NotImplementedError� __class__rU r Z NoSuchProcessZ AccessDeniedr! r� ru rb rl �msg� Exception�repr� traceback� format_exc�join� splitlinesZfail)r% Z valid_procsZexcluded_namesZattrsru �defaultZfailures�pro �args�kwargs�attrr� �err�s�methr' r' r( �test_fetch_all, sx z$TestFetchAllProcesses.test_fetch_allc C s* | j |t� x|D ]}| j |t� qW d S )N)r^ �listrb )r% ro rg �partr' r' r( �cmdlinet s z TestFetchAllProcesses.cmdlinec C st | j |ttd �f� |s&| j|d� nJtjj|�s:t|��trptjj |�rpt td�rpt td�rptj |tj �spt�d S )Nr� �access�X_OK) r^ rb ry r! r6 r7 �isabsrl r �isfiler" r� r� )r% ro rg r' r' r( �exey s zTestFetchAllProcesses.exec C s | j |t� | j|d� d S )Nr )r^ r� rn )r% ro rg r' r' r( r� � s zTestFetchAllProcesses.pidc C s | j |ttf� | j|d� d S )Nr )r^ r� r rn )r% ro rg r' r' r( �ppid� s zTestFetchAllProcesses.ppidc C s | j |t� ts|st�d S )N)r^ rb r rl )r% ro rg r' r' r( ru � s zTestFetchAllProcesses.namec C s^ | j |t� y| j|d� W n* tk rF tr@|j� tjkr@n� Y nX tj dtj |�� d S )Nr z%Y %m %d %H:%M:%S) r^ rm rn rl r �statusr# Z STATUS_ZOMBIE�timeZstrftimeZ localtime)r% ro rg r' r' r( � create_time� s z!TestFetchAllProcesses.create_timec C sD t |�s t�x2|D ]*}| j|t� | j|d� | j|| j� qW d S )Nr )r rl r^ r� rn ra r� )r% ro rg Zuidr' r' r( rB � s zTestFetchAllProcesses.uidsc C sP t |�s t�x>|D ]6}| j|t� t rt r| j|d� | j|| j� qW d S )Nr ) r rl r^ r� r r rn ra r� )r% ro rg �gidr' r' r( �gids� s zTestFetchAllProcesses.gidsc C s* | j |t� |st�tr&| j|| j� d S )N)r^ rb rl r ra r� )r% ro rg r' r' r( �username� s z TestFetchAllProcesses.usernamec C s0 | j |t� |st�| j|d� | j|t� d S )N�?)r^ rb rl ZassertNotEqualra r )r% ro rg r' r' r( r� � s z TestFetchAllProcesses.statusc C sB t |�s t�x0|D ](}| j|ttf� |dkr| j|d� qW d S )Nr� r ���)r rl r^ r� r rn )r% ro rg �fieldr' r' r( rJ � s z!TestFetchAllProcesses.io_countersc C sh t r*t|�st�x|D ]}| j|t� qW trL| j|jd� | j|jd� n| j|d� | j |d� d S )Nr r� r/ )r r� r/ ) r r rl r^ r� r rn Zioclass�valuera )r% ro rg r� r' r' r( rG � s z TestFetchAllProcesses.ionicec C s | j |t� | j|d� d S )Nr� )r^ r� rn )r% ro rg r' r' r( � num_threads� s z!TestFetchAllProcesses.num_threadsc C sr | j |t� x`|D ]X}t|�s"t�| j|jd� | j|jd� | j|jd� x |D ]}| j |tt f� qRW qW d S )Nr ) r^ r� r rl rn �idZ user_timeZ system_timer� rm )r% ro rg �tr� r' r' r( �threads� s z TestFetchAllProcesses.threadsc C s6 t |�s t�x$|D ] }| j|t� | j|d� qW d S )Nr )r rl r^ rm rn )r% ro rg rp r' r' r( rk � s z TestFetchAllProcesses.cpu_timesc C s0 | j |t� d| ko dkn s,t|��d S )Ng g Y@)r^ rm rl )r% ro rg r' r' r( � cpu_percent� s z!TestFetchAllProcesses.cpu_percentc C s\ | j |t� tr |dkr d S | j|d� tj� dkr@| j|d� | j|tt tj� ��� d S )Nr� r r� ) r^ r� r rn r# � cpu_countr! ra r� �range)r% ro rg r' r' r( rQ � s z TestFetchAllProcesses.cpu_numc C s� t |�s t�x(|D ] }| j|ttf� | j|d� qW tr�t r�|jdkr�xx|j D ](}|dkrRt ||�}| j |j||d� qRW nDt r�| j|j |j� | j|j|j� | j|j|j� | j|j|j� d S )Nr �vms)r� )r rl r^ r� r rn r r r� �_fieldsr� � assertGreaterr Z peak_wsetZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZ peak_pagefileZpagefile)r% ro rg r� ru r' r' r( � memory_info� s z!TestFetchAllProcesses.memory_infoc C s� t |�s t�tj� j}xR|jD ]H}t||�}| j|tt f� | j |d||fd� | j |||||fd� q W t r~| j |j |j� d S )Nr )r� )r rl r# Zvirtual_memory�totalr� r� r^ r� r rn ZassertLessEqualr ZpssZuss)r% ro rg r� ru r� r' r' r( �memory_full_info s z&TestFetchAllProcesses.memory_full_infoc C s� | j |t� x�|D ]�}| j |jt� | j |jt� trF| j|jd� nhtr�| j |j t� | j |j t� | j |j t� | j |j d� | j |j d � | j|j d� ntr�|j r�qtjj|j�s�t|��tjj|j�st|��qW d S ) Nr� r �rrd �a�r+�a+r� )r� rd r� r� r� )r^ r� �fdr� r7 rb r r! r Zposition�mode�flagsrn ra r� r r6 r� rl r� )r% ro rg �fr' r' r( � open_files s" z TestFetchAllProcesses.open_filesc C s | j |t� | j|d� d S )Nr )r^ r� rn )r% ro rg r' r' r( rL ( s z TestFetchAllProcesses.num_fdsc C s2 | j t|�tt|��� x|D ] }t|� q W d S )N)r! �lenr� r )r% ro rg rv r' r' r( � connections, s z!TestFetchAllProcesses.connectionsc C s� |r�| j |t� tjj|�s$t|��ytj|�}W nD tk rv } z(trX|j t j j krXn|j t j krf� W Y d d }~X nX tj|j�s�t�d S )N)r^ rb r6 r7 r� rl �stat�OSErrorr �errnor# Z _psplatformZACCESS_DENIED_SET�ENOENT�S_ISDIR�st_mode)r% ro rg �str� r' r' r( �cwd1 s zTestFetchAllProcesses.cwdc C s0 | j |t� d| ko dkn s,t|��d S )Nr �d )r^ rm rl )r% ro rg r' r' r( �memory_percentA s z$TestFetchAllProcesses.memory_percentc C s | j |t� d S )N)r^ �bool)r% ro rg r' r' r( � is_runningE s z TestFetchAllProcesses.is_runningc C sR | j |t� |g ks t|��ttj� �}x$|D ] }| j |t� | j||� q.W d S )N)r^ r� rl r� r# r� r� ra )r% ro rg Zcpusrp r' r' r( r H s z"TestFetchAllProcesses.cpu_affinityc C sH | j |ttd �f� |d k rDtjj|�s0t|��tjj|�sDt|��d S )N)r^ rb ry r6 r7 r� rl r8 )r% ro rg r' r' r( rE P s z TestFetchAllProcesses.terminalc C s� 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 |t t f� | j |d� q<W qW d S )Nr7 �[r{ �permsr )r{ r� )r^ r{ rb r� r7 r� r� r� r6 r� rl r� r rn )r% ro rg �ntZfnamer� r' r' r( rS V s z!TestFetchAllProcesses.memory_mapsc C s | j |t� | j|d� d S )Nr )r^ r� rn )r% ro rg r' r' r( rN i s z!TestFetchAllProcesses.num_handlesc C sT | j |t� tr2d| ko"dkn sPt|��n dd� tt�D �}| j||� d S )N� c S s g | ]}|j d �rtt|��qS )Z_PRIORITY_CLASS)�endswithr� r# )r� r� r' r' r( r� r s z.TestFetchAllProcesses.nice.<locals>.<listcomp>i��)r^ r� r rl r� r# ra )r% ro rg Z prioritiesr' r' r( �nicem s "zTestFetchAllProcesses.nicec C s: t |�s t�x(|D ] }| j|ttf� | j|d� qW d S )Nr )r rl r^ r� r rn )r% ro rg r� r' r' r( �num_ctx_switchesv s z&TestFetchAllProcesses.num_ctx_switchesc C s@ | j |t� | jt|�d� | j|d d� | j|d d� d S )Nr/ r r� r� r� )r^ �tupler! r� rn )r% ro rg r' r' r( r2 | s z TestFetchAllProcesses.rlimitc C s> | j |t� x,|j� D ] \}}| j |t� | j |t� qW d S )N)r^ �dictrx rb )r% ro rg rr �vr' r' r( r@ � s z TestFetchAllProcesses.environN)'rU rV rW rX r� r� r� r� r� r� ru r� rB r� r� r� rJ rG r� r� rk r� rQ r� r� r� rL r� r� r� r� r rE rS rN r� r� r2 r@ r' r' r' r( r� sH H r� �__main__)/rX r� r6 r� r� r� r[ � contextlibr r# r 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 ZTestCaser rY rf r� rU �__file__r' r' r' r( �<module> sX x^ n
Save File
Cancel