File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_contracts.cpython-39.pyc
a
��?h!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�tptptksJ �d S )NZ PROCFS_PATH)�hasattr�psutilr r r ��self� r �G/usr/local/lib64/python3.9/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 Zaer 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 �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_VERYLOWr" r$ r r r �test_linux_ioprio_windowsF s
z0TestAvailConstantsAPIs.test_linux_ioprio_windows�%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� d
k�r|ttd�t� t� dk�r0|ttd�t� t� dk�rL|ttd�t� t� dk�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)r+ r, � Z RLIMIT_NICEZ
RLIMIT_RTPRIO)r+ r, � Z
RLIMIT_RTTIMEZRLIMIT_SIGPENDINGZ RLIMIT_SWAPZ
RLIMIT_SBSIZEZ RLIMIT_NPTS)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 r0 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�tksJ �d S )NZwin_service_iter�r r r
r r r r �test_win_service_iterr s z)TestAvailSystemAPIs.test_win_service_iterc C s t td�tksJ �d S )NZwin_service_getr7 r r r r �test_win_service_getu s z(TestAvailSystemAPIs.test_win_service_getc C s& t td�tptptptptks"J �d S )N�cpu_freq)r r r r r
r r r r r r �
test_cpu_freqx s �z!TestAvailSystemAPIs.test_cpu_freqc C s t td�tptksJ �d S )N�sensors_temperatures)r r r r r r r r �test_sensors_temperatures} s z-TestAvailSystemAPIs.test_sensors_temperaturesc C s t td�tksJ �d S )N� sensors_fans)r r r r r r r �test_sensors_fans� s z%TestAvailSystemAPIs.test_sensors_fansc C s" t td�tptptptks J �d S )NZsensors_battery)r r r r
r r r r r r � test_battery� s �z TestAvailSystemAPIs.test_batteryN) r1 r2 r3 r8 r9 r; r= r? r@ r r r r r6 q s r6 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
ks0J �d S )N�environ) r r �Processr r r
r r r r r r r r r � test_environ� s
�������z!TestAvailProcessAPIs.test_environc C s t tjd�tksJ �d S �NZuids�r r rC r r r r r � test_uids� s z TestAvailProcessAPIs.test_uidsc C s t tjd�tksJ �d S rE rF r r r r � test_gids� s z TestAvailProcessAPIs.test_gidsc C s t tjd�tksJ �d S )N�terminalrF r r r r �
test_terminal� s z"TestAvailProcessAPIs.test_terminalc C s t tjd�tptksJ �d S )NZionice)r r rC r r
r r r r � test_ionice� s z TestAvailProcessAPIs.test_ionicer( r) c C s t tjd�tptksJ �d S )NZrlimit)r r rC r r r r r r r0 � s z TestAvailProcessAPIs.test_rlimitc C s" t tjd�}|tpt ks J �d S )NZ io_counters)r r rC r r �r Zhasitr r r �test_io_counters� s z%TestAvailProcessAPIs.test_io_countersc C s t tjd�tksJ �d S )NZnum_fdsrF r r r r � test_num_fds� s z!TestAvailProcessAPIs.test_num_fdsc C s t tjd�tksJ �d S )NZ num_handles)r r rC r
r r r r �test_num_handles� s z%TestAvailProcessAPIs.test_num_handlesc C s t tjd�tptptks J �d S )NZ cpu_affinity)r r rC r r
r r r r r �test_cpu_affinity� s
�z&TestAvailProcessAPIs.test_cpu_affinityc C s t tjd�tptptks J �d S )NZcpu_num)r r rC 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&J �d S )NZ memory_maps)r r rC r r r r rL r r r �test_memory_maps� s z%TestAvailProcessAPIs.test_memory_mapsN)r1 r2 r3 rD rG rH rJ rK r r4 r5 r r r0 rM rN rO rP rQ rR r r r r rA � s �
rA c @ s& e Zd ZdZedd� �Zedfdd�Zdd� Zd d
� Z d d � Z
d
d� Z e j
jeoZe�� 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 �� | _d S �N)r rC �proc)�clsr r r �
setUpClass� s z TestSystemAPITypes.setUpClassTc C s8 t |�s J �|D ]"}t||�s"J �|r|dksJ �qd S )Nr )r �
isinstance)r �nt�type_Zgezero�nr r r �assert_ntuple_of_nums� s
z(TestSystemAPITypes.assert_ntuple_of_numsc C s. | � t�� � tjdd�D ]}| � |� qd S )NT)Zpercpu)r\ r Z cpu_times)r rY r r r �test_cpu_times� s z!TestSystemAPITypes.test_cpu_timesc C s0 t tjd d�t�sJ �t tjdd�t�s,J �d S )N��intervalg�h㈵��>)rX r Z cpu_percent�floatr r r r �test_cpu_percent� s z#TestSystemAPITypes.test_cpu_percentc C s( | � tjd d�� | � tjdd�� d S )Nr^ g-C ��6?)r\ r Zcpu_times_percentr r r r �test_cpu_times_percent� s z)TestSystemAPITypes.test_cpu_times_percentc C s t t�� t�sJ �d S rT )rX r � cpu_count�intr r r r �test_cpu_count� s z!TestSystemAPITypes.test_cpu_count�arm64zskipped due to #1892r) z
not supportedc C s2 t �� d u rt�d��| jt �� tttfd� d S )Nzcpu_freq() returns None�rZ )r r: r �skipr\ r` rd r r r r r r; � s
z TestSystemAPITypes.test_cpu_freqc C s> t jdd��� D ](\}}t|t�s&J �| j|ttfd� qd S )NT)Zperdiskrg )r Zdisk_io_counters�itemsrX �strr\ rd r )r �k�vr r r �test_disk_io_counters� s z(TestSystemAPITypes.test_disk_io_countersc C sR t �� D ]D}t|jt�s J �t|jt�s,J �t|jt�s<J �t|jt�sJ �qd S rT )r Zdisk_partitionsrX Zdevicerj Z
mountpointZfstype�opts)r Zdiskr r r �test_disk_partitions� s
z'TestSystemAPITypes.test_disk_partitionsz
requires rootc C sb t � �H t�d�}t|�tt|��ks*J �|D ]}t|�s.J �q.W d � n1 sT0 Y d S )N�all)r r Znet_connections�len�setr )r �ret�connr r r �test_net_connections s
z'TestSystemAPITypes.test_net_connectionsc C s� t �� �� D ]�\}}t|t�s"J �|D ]t}td urJtsJt|jtj�sZJ �nt|jt �sZJ �t|j
t�sjJ �t|j tt d �f�s�J �t|j
tt d �f�s&J �q&q d S rT )r Z net_if_addrsri rX rj r r �family�IntEnumrd �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� t �� �� D ]r\}}t|t�s"J �t|jt�s2J �td urNt|jtj �s^J �nt|jt
�s^J �t|j t
�snJ �t|j t
�s J �q d S rT )
r Z net_if_statsri rX rj Zisup�boolr Zduplexrw rd �speedZmtu)r r| �infor r r �test_net_if_stats s z$TestSystemAPITypes.test_net_if_statsc C s$ t jdd�D ]}t|t�s J �q d S )NT)Zpernic)r Znet_io_countersrX rj )r r| r r r �test_net_io_counters" s z'TestSystemAPITypes.test_net_io_countersc C s\ t �� �� D ]J\}}t|t�s"J �|D ].}t|jt�s:J �t|jttt d �f�s&J �q&q d S rT )
r r> ri rX rj �label�currentr` rd rz �r �nameZunits�unitr r r r? ( s
z$TestSystemAPITypes.test_sensors_fansc C s� t �� �� D ]~\}}t|t�s"J �|D ]b}t|jt�s:J �t|jttt d �f�sTJ �t|j
ttt d �f�snJ �t|j ttt d �f�s&J �q&q d S rT ) r r< ri rX rj r� r� r` rd rz �high�criticalr� r r r r= 1 s z,TestSystemAPITypes.test_sensors_temperaturesc C s t t�� t�sJ �d S rT )rX r Z boot_timer` r r r r �test_boot_time< s z!TestSystemAPITypes.test_boot_timec C sj t �� D ]\}t|jt�s J �t|jttd �f�s4J �t|jttd �f�sLJ �t|jt td �f�sJ �qd S rT )
r ZusersrX r� rj rI rz �host�pidrd )r �userr r r �
test_users@ s
z TestSystemAPITypes.test_usersN)#r1 r2 r3 �__doc__� classmethodrW r` r\ r] ra rb re r r4 r5 r �platform�machiner
r; rm ro r ru r r r� r r� r r? r r= r� r� r r r r rS � s8
�
rS c @ s& e Zd Zejje dd�dd� �ZdS )�TestProcessWaitTypez not POSIXr) c C s\ t �| �� j�}|�� |�� }|tj ks0J �td urJt |tj
�sXJ �nt |t �sXJ �d S rT ) r rC Zspawn_testprocr� � terminate�wait�signal�SIGTERMr rX rw rd )r �p�coder r r �test_negative_signalJ s z(TestProcessWaitType.test_negative_signalN)r1 r2 r3 r r4 r5 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 r6 rA rS r� r r r r �<module> sB ?>