3
�[g!1 � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
Z
dd
lm Z dd lm Z dd l
mZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z G dd � d e�Z G d d � d e�Z G d d � d e�Z G d!d"� d"e�Z!G d#d$� d$e�Z"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)�AIX)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�long)�GITHUB_ACTIONS)� HAS_CPU_FREQ)�HAS_NET_IO_COUNTERS)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�PYPY)� QEMU_USER)� SKIP_SYSCONS)�PsutilTestCase)�create_sockets)�enum)�
is_namedtuple)�kernel_version)�pytestc @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zejj e
o4e d d
�d d � �Z d
S )�TestAvailConstantsAPIsc C s t td�tptptkst�d S )N� PROCFS_PATH)�hasattr�psutilr r r �AssertionError)�self� r �G/usr/local/lib64/python3.6/site-packages/psutil/tests/test_contracts.py�test_PROCFS_PATH3 s z'TestAvailConstantsAPIs.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)� assertEqualr r r
)r �aer r r! �test_win_priority6 s z(TestAvailConstantsAPIs.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_ioprio_linux? s
z.TestAvailConstantsAPIs.test_linux_ioprio_linuxc C sJ | j }|ttd�t� |ttd�t� |ttd�t� |ttd�t� d S )NZ IOPRIO_HIGHZ
IOPRIO_NORMALZ
IOPRIO_LOWZIOPRIO_VERYLOW)r# r r r
)r r$ r r r! �test_linux_ioprio_windowsF s
z0TestAvailConstantsAPIs.test_linux_ioprio_windowsz%unsupported on GITHUB_ACTIONS + LINUX)�reasonc C s� | j }|ttd�tpt� |ttd�tp*t� |ttd�tp>t� |ttd�tpRt� |ttd�tpft� |ttd�tpzt� |ttd�tp�t� |ttd�tp�t� |ttd �tp�t� |ttd
�tp�t� |ttd �tp�t� |ttd �t� t�r�t� dk�r|ttd�t� t� dk�r0|ttd�t� t� d k�rL|ttd�t� t� d k�rh|ttd�t� t� d k�r�|ttd�t� |ttd�t� |ttd�t� |ttd�t� d S ) NZ
RLIM_INFINITYZ RLIMIT_ASZ RLIMIT_COREZ
RLIMIT_CPUZ RLIMIT_DATAZ RLIMIT_FSIZEZRLIMIT_MEMLOCKZ
RLIMIT_NOFILEZ RLIMIT_NPROCZ
RLIMIT_RSSZ RLIMIT_STACKZ RLIMIT_LOCKS� � � ZRLIMIT_MSGQUEUE� Z RLIMIT_NICEZ
RLIMIT_RTPRIO� Z
RLIMIT_RTTIMEZRLIMIT_SIGPENDINGZ RLIMIT_SWAPZ
RLIMIT_SBSIZEZ RLIMIT_NPTS)r) r* r+ )r) r* r, )r) r* r, )r) r* r- )r) r* r+ )r# r r r r r r )r r$ r r r! � test_rlimitM s6 z"TestAvailConstantsAPIs.test_rlimitN)
�__name__�
__module__� __qualname__r" r% r& r' r �mark�skipifr r r. r r r r! r 2 s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestAvailSystemAPIsc C s t td�tkst�d S )NZwin_service_iter)r r r
r )r r r r! �test_win_service_iterr s z)TestAvailSystemAPIs.test_win_service_iterc C s t td�tkst�d S )NZwin_service_get)r r r
r )r r r r! �test_win_service_getu s z(TestAvailSystemAPIs.test_win_service_getc C s& t td�tptptptptks"t�d S )N�cpu_freq)r r r r r
r r r )r r r r! �
test_cpu_freqx s z!TestAvailSystemAPIs.test_cpu_freqc C s t td�tptkst�d S )N�sensors_temperatures)r r r r r )r r r r! �test_sensors_temperatures} s z-TestAvailSystemAPIs.test_sensors_temperaturesc C s t td�tkst�d S )N� sensors_fans)r r r r )r r r r! �test_sensors_fans� s z%TestAvailSystemAPIs.test_sensors_fansc C s" t td�tptptptks t�d S )NZsensors_battery)r r r r
r r r )r r r r! � test_battery� s z TestAvailSystemAPIs.test_batteryN) r/ r0 r1 r5 r6 r8 r: r< r= r r r r! r4 q s r4 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej j
e o<e d d �d
d� �Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) �TestAvailProcessAPIsc C s4 t tjd�tp(tp(tp(tp(tp(tp(t p(t
ks0t �d S )N�environ) r r �Processr r r
r r r r r r )r r r r! � test_environ� s
z!TestAvailProcessAPIs.test_environc C s t tjd�tkst�d S )N�uids)r r r@ r r )r r r r! � test_uids� s z TestAvailProcessAPIs.test_uidsc C s t tjd�tkst�d S )NrB )r r r@ r r )r r r r! � test_gids� s z TestAvailProcessAPIs.test_gidsc C s t tjd�tkst�d S )N�terminal)r r r@ r r )r r r r! �
test_terminal� s z"TestAvailProcessAPIs.test_terminalc C s t tjd�tptkst�d S )N�ionice)r r r@ r r
r )r r r r! � test_ionice� s z TestAvailProcessAPIs.test_ionicez%unsupported on GITHUB_ACTIONS + LINUX)r( c C s t tjd�tptkst�d S )NZrlimit)r r r@ r r r )r r r r! r. � s z TestAvailProcessAPIs.test_rlimitc C s" t tjd�}|tpt ks t�d S )NZ io_counters)r r r@ r r r )r �hasitr r r! �test_io_counters� s z%TestAvailProcessAPIs.test_io_countersc C s t tjd�tkst�d S )NZnum_fds)r r r@ r r )r r r r! � test_num_fds� s z!TestAvailProcessAPIs.test_num_fdsc C s t tjd�tkst�d S )NZ num_handles)r r r@ r
r )r r r r! �test_num_handles� s z%TestAvailProcessAPIs.test_num_handlesc C s t tjd�tptptks t�d S )NZ cpu_affinity)r r r@ r r
r r )r r r r! �test_cpu_affinity� s
z&TestAvailProcessAPIs.test_cpu_affinityc C s t tjd�tptptks t�d S )NZcpu_num)r r r@ r r r r )r r r r! � test_cpu_num� s
z!TestAvailProcessAPIs.test_cpu_numc C s* t tjd�}|tp tp tp t ks&t�d S )NZ memory_maps)r r r@ r r r r r )r rI r r r! �test_memory_maps� s z%TestAvailProcessAPIs.test_memory_mapsN)r/ r0 r1 rA rC rD rF rH r r2 r3 r r r. rJ rK rL rM rN rO r r r r! r> � s r> c @ s& e Zd ZdZedd� �Zedfdd�Zdd� Zd d
� Z d d � Z
d
d� Z e j
jeoZej� dkdd�e j
je dd�dd� ��Zdd� Zdd� Ze j
jedd�dd� �Zd d � Ze j
jed d�d d � �Ze j
je dd�d!d"� �Z e j
je dd�d#d$� �Z e j
je dd�d%d&� �Z d'd(� Z!d)d*� Z"d+S ),�TestSystemAPITypesz�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 z TestSystemAPITypes.setUpClassTc C s< t |�s t�x*|D ]"}t||�s$t�|r|dkst�qW d S )Nr )r r �
isinstance)r �nt�type_Zgezero�nr r r! �assert_ntuple_of_nums� s
z(TestSystemAPITypes.assert_ntuple_of_numsc C s2 | j tj� � x tjdd�D ]}| j |� q W d S )NT)Zpercpu)rX r Z cpu_times)r rU r r r! �test_cpu_times� s z!TestSystemAPITypes.test_cpu_timesc C s0 t tjd d�t�st�t tjdd�t�s,t�d S )N)�intervalg�h㈵��>)rT r Z cpu_percent�floatr )r r r r! �test_cpu_percent� s z#TestSystemAPITypes.test_cpu_percentc C s( | j tjd d�� | j tjdd�� d S )N)rZ g-C ��6?)rX r Zcpu_times_percent)r r r r! �test_cpu_times_percent� s z)TestSystemAPITypes.test_cpu_times_percentc C s t tj� t�st�d S )N)rT r � cpu_count�intr )r r r r! �test_cpu_count� s z!TestSystemAPITypes.test_cpu_countZarm64zskipped due to #1892)r( z
not supportedc C s2 t j� d krtjd��| jt j� tttfd� d S )Nzcpu_freq() returns None)rV )r r7 r �skiprX r[ r_ r )r r r r! r8 � s
z TestSystemAPITypes.test_cpu_freqc C sB x<t jdd�j� D ](\}}t|t�s(t�| j|ttfd� qW d S )NT)Zperdisk)rV ) r Zdisk_io_counters�itemsrT �strr rX r_ r )r �k�vr r r! �test_disk_io_counters� s z(TestSystemAPITypes.test_disk_io_countersc C sV xPt j� D ]D}t|jt�s t�t|jt�s.t�t|jt�s>t�t|jt�s
t�q
W d S )N) r Zdisk_partitionsrT �devicerc r �
mountpoint�fstype�opts)r Zdiskr r r! �test_disk_partitions� s
z'TestSystemAPITypes.test_disk_partitionsz
requires rootc C sR t � �B tjd�}t|�tt|��ks*t�x|D ]}t|�s0t�q0W W d Q R X d S )N�all)r r Znet_connections�len�setr r )r �retZconnr r r! �test_net_connections s
z'TestSystemAPITypes.test_net_connectionsc C s� x�t j� j� D ]�\}}t|t�s$t�x~|D ]v}td k rPt rPt|jtj �s`t�nt|jt
�s`t�t|j t�spt�t|j tt
d �f�s�t�t|jtt
d �f�s*t�q*W qW d S )N)r Z net_if_addrsrb rT rc r r r �family�IntEnumr_ �address�netmask�type� broadcast)r �ifname�addrs�addrr r r! �test_net_if_addrs s
z$TestSystemAPITypes.test_net_if_addrszQEMU user not supportedc C s� x�t j� j� D ]r\}}t|t�s$t�t|jt�s4t�td k rPt|j tj
�s`t�nt|j t �s`t�t|j t �spt�t|j
t �st�qW d S )N)r Z net_if_statsrb rT rc r �isup�boolr �duplexrr r_ �speed�mtu)r rw �infor r r! �test_net_if_stats s z$TestSystemAPITypes.test_net_if_statsc C s( x"t jdd�D ]}t|t�st�qW d S )NT)Zpernic)r Znet_io_countersrT rc r )r rw r r r! �test_net_io_counters" s z'TestSystemAPITypes.test_net_io_countersc C sd x^t j� j� D ]N\}}t|t�s$t�x6|D ].}t|jt�s>t�t|jtt t
d �f�s*t�q*W qW d S )N) r r; rb rT rc r �label�currentr[ r_ ru )r �name�units�unitr r r! r< ( s
z$TestSystemAPITypes.test_sensors_fansc C s� x�t j� j� D ]�\}}t|t�s$t�xj|D ]b}t|jt�s>t�t|jtt t
d �f�sXt�t|j tt t
d �f�srt�t|j tt t
d �f�s*t�q*W qW d S )N)
r r9 rb rT rc r r� r� r[ r_ ru �high�critical)r r� r� r� r r r! r: 1 s
z,TestSystemAPITypes.test_sensors_temperaturesc C s t tj� t�st�d S )N)rT r Z boot_timer[ r )r r r r! �test_boot_time< s z!TestSystemAPITypes.test_boot_timec C sn xht j� D ]\}t|jt�s t�t|jttd �f�s6t�t|jttd �f�sNt�t|j t
td �f�s
t�q
W d S )N) r ZusersrT r� rc r rE ru �host�pidr_ )r �userr r r! �
test_users@ s
z TestSystemAPITypes.test_usersN)#r/ r0 r1 �__doc__� classmethodrS r[ rX rY r\ r] r` r r2 r3 r �platform�machiner
r8 rf rk r rp rz r r� r r� r r< r r: r� r� r r r r! rP � s(
rP c @ s& e Zd Zejje dd�dd� �ZdS )�TestProcessWaitTypez not POSIX)r( c C s\ t j| j� j�}|j� |j� }|tj ks0t�t d k rJt
|t j �sXt�nt
|t �sXt�d S )N)
r r@ Zspawn_testprocr� � terminate�wait�signal�SIGTERMr r rT rr r_ )r �p�coder r r! �test_negative_signalJ s z(TestProcessWaitType.test_negative_signalN)r/ r0 r1 r r2 r3 r r� r r r r! r� I s r� )#r� r� r� r r r r r r r r r r
Zpsutil._compatr Z psutil.testsr r
r r r r r r r r r r r r r r4 r> rP r� r r r r! �<module>
sB ?> |