OwlCyberSecurity - MANAGER
Edit File: test_contracts.cpython-36.opt-1.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 s2 t j� }x$|D ] }| j|t� | j|d� qW d S )Nr )r# � cpu_timesr^ �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 sV t � �F}tt|���. tjdd�}x|D ]}| j|jt� q(W W d Q R X W d Q R X d S )NZunix)Zkind)r r r r# Znet_connectionsr^ 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_statsrw r^ rb )r% ry �_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_countersrw r^ rb )r% ry 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= rw r^ rb �label)r% rt �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; rw r^ rb r )r% rt 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^ rt rb rE rx �host�pid�int)r% �userr' r' r( � test_users s zTestSystem.test_usersN)rU rV rW rX � classmethodri rj rp rr rs r ZskipIfr r r r rv 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 �n2 tk �r> d| jjd | } t| � Y q� tjtjfk �r� } z0| j| j|j� | j�r�| j| j|j� � W Y d d } ~ X q� tk �r> } z�ddd d }|d||f 7 }||k�r�|dt|� 7 }|d7 }|dd 7 }|d tj� 7 }dj d d � |j � D ��}|d7 }|j|� P W Y d d } ~ X q�X |dd g d d i fk�rVt | |�}|||� q�W W d Q R X q�W |�r�| j d j |�� 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� rt � Exception�repr� traceback� format_exc�join� splitlinesZfail)r% Z valid_procsZexcluded_namesZattrsrt �defaultZfailures�prn �args�kwargs�attr�msg�err�s�methr' r' r( �test_fetch_all, st z$TestFetchAllProcesses.test_fetch_allc C s* | j |t� x|D ]}| j |t� qW d S )N)r^ �listrb )r% rn rg �partr' r' r( �cmdlinet s z TestFetchAllProcesses.cmdlinec C sN | j |ttd �f� |s&| j|d� n$trJtjj|�rJttd�rJttd�rJd S )Nr� �access�X_OK) r^ rb rx r! r r6 r7 �isfiler" )r% rn rg r' r' r( �exey s zTestFetchAllProcesses.exec C s | j |t� | j|d� d S )Nr )r^ r� rm )r% rn rg r' r' r( r� � s zTestFetchAllProcesses.pidc C s | j |ttf� | j|d� d S )Nr )r^ r� r rm )r% rn rg r' r' r( �ppid� s zTestFetchAllProcesses.ppidc C s | j |t� tsd S )N)r^ rb r )r% rn rg r' r' r( rt � 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^ rl rm �AssertionErrorr �statusr# Z STATUS_ZOMBIE�timeZstrftimeZ localtime)r% rn rg r' r' r( � create_time� s z!TestFetchAllProcesses.create_timec C s8 x2|D ]*}| j |t� | j|d� | j|| j� qW d S )Nr )r^ r� rm ra r� )r% rn rg Zuidr' r' r( rB � s zTestFetchAllProcesses.uidsc C sD x>|D ]6}| j |t� t rt r| j|d� | j|| j� qW d S )Nr )r^ r� r r rm ra r� )r% rn rg �gidr' r' r( �gids� s zTestFetchAllProcesses.gidsc C s" | j |t� tr | j|| j� d S )N)r^ rb r ra r� )r% rn rg r' r' r( �username� s z TestFetchAllProcesses.usernamec C s( | j |t� | j|d� | j|t� d S )N�?)r^ rb ZassertNotEqualra r )r% rn rg r' r' r( r� � s z TestFetchAllProcesses.statusc C s6 x0|D ](}| j |ttf� |dkr| j|d� qW d S )Nr� r ���)r^ r� r rm )r% rn rg �fieldr' r' r( rJ � s z!TestFetchAllProcesses.io_countersc C s\ t r x|D ]}| j|t� q W tr@| j|jd� | j|jd� n| j|d� | j|d� d S )Nr r� r/ )r r� r/ )r r^ r� r rm Zioclass�valuera )r% rn rg r� r' r' r( rG � s z TestFetchAllProcesses.ionicec C s | j |t� | j|d� d S )Nr� )r^ r� rm )r% rn rg r' r' r( � num_threads� s z!TestFetchAllProcesses.num_threadsc C sf | j |t� xT|D ]L}| j|jd� | j|jd� | j|jd� x |D ]}| j |ttf� qFW qW d S )Nr )r^ r� rm �idZ user_timeZ system_timer� rl )r% rn rg �tr� r' r' r( �threads� s z TestFetchAllProcesses.threadsc C s* x$|D ] }| j |t� | j|d� qW d S )Nr )r^ rl rm )r% rn rg ro r' r' r( rk � s z TestFetchAllProcesses.cpu_timesc C s | j |t� d S )N)r^ rl )r% rn 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 rm r# � cpu_countr! ra r� �range)r% rn rg r' r' r( rQ � s z TestFetchAllProcesses.cpu_numc C s� x(|D ] }| j |ttf� | j|d� qW trtt rt|jdkrtxx|jD ](}|dkrFt||�}| j |j||d� qFW nDt r�| j|j |j � | j|j |j� | j|j|j� | j|j|j� d S )Nr �vms)r� )r^ r� r rm r r r� �_fieldsr� � assertGreaterr Z peak_wsetZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZ peak_pagefileZpagefile)r% rn rg r� rt r' r' r( � memory_info� s z!TestFetchAllProcesses.memory_infoc C sv t j� j}xR|jD ]H}t||�}| j|ttf� | j|d||fd� | j |||||fd� qW t rr| j|j |j � d S )Nr )r� ) r# Zvirtual_memory�totalr� r� r^ r� r rm ZassertLessEqualr ZpssZuss)r% rn rg r� rt 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� qtr�| j |j t� | j |j t� | j |j t� | j |j d� | j |j d � | j|j d� qtr|j rqqW d S ) Nr� r �rrd �a�r+�a+r� )r� rd r� r� r� )r^ r� �fdr� r7 rb r r! r Zposition�mode�flagsrm ra r� r )r% rn rg �fr' r' r( � open_files s z TestFetchAllProcesses.open_filesc C s | j |t� | j|d� d S )Nr )r^ r� rm )r% rn 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% rn rg ru r' r' r( � connections, s z!TestFetchAllProcesses.connectionsc C sh |rd| j |t� ytj|�}W nD tk rb } z(trD|jtjj krDn|jtj krR� W Y d d }~X nX d S )N) r^ rb r6 �stat�OSErrorr �errnor# Z _psplatformZACCESS_DENIED_SET�ENOENT)r% rn rg �str� r' r' r( �cwd1 s zTestFetchAllProcesses.cwdc C s | j |t� d S )N)r^ rl )r% rn rg r' r' r( �memory_percentA s z$TestFetchAllProcesses.memory_percentc C s | j |t� d S )N)r^ �bool)r% rn rg r' r' r( � is_runningE s z TestFetchAllProcesses.is_runningc C sB | j |t� ttj� �}x$|D ] }| j |t� | j||� q W d S )N)r^ r� r� r# r� r� ra )r% rn rg Zcpusro r' r' r( r H s z"TestFetchAllProcesses.cpu_affinityc C s | j |ttd �f� |d k r d S )N)r^ rb rx )r% rn rg r' r' r( rE P s z TestFetchAllProcesses.terminalc C s� x�|D ]�}| j |jt� | j |jt� | j |jt� xR|jD ]H}t||�}|dkr^|jd�s�q<|dkrhq<| j |tt f� | j |d� q<W qW d S )Nr7 �[rz �permsr )rz r� ) r^ rz rb r� r7 r� r� r� r� r rm )r% rn rg �ntZfnamer� r' r' r( rS V s z!TestFetchAllProcesses.memory_mapsc C s | j |t� | j|d� d S )Nr )r^ r� rm )r% rn rg r' r' r( rN i s z!TestFetchAllProcesses.num_handlesc C s4 | j |t� trn dd� tt�D �}| j||� d S )Nc 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>)r^ r� r r� r# ra )r% rn rg Z prioritiesr' r' r( �nicem s zTestFetchAllProcesses.nicec C s. x(|D ] }| j |ttf� | j|d� qW d S )Nr )r^ r� r rm )r% rn 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� rm )r% rn 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^ �dictrw rb )r% rn rg rq �vr' r' r( r@ � s z TestFetchAllProcesses.environN)'rU rV rW rX r� r� r� r� r� r� rt 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