관리-도구
편집 파일: test_contracts.cpython-38.pyc
U �Mg!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 )NZ PROCFS_PATH)�hasattr�psutilr r r �AssertionError��self� r �K/opt/hc_python/lib64/python3.8/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 r1 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_getr8 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) r2 r3 r4 r9 r: r< r> r@ rA r r r r! r7 q s r7 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 �NZuids�r r rD r r r r r r! � test_uids� s z TestAvailProcessAPIs.test_uidsc C s t tjd�tkst�d S rF rG r r r r! � test_gids� s z TestAvailProcessAPIs.test_gidsc C s t tjd�tkst�d S )N�terminalrG r r r r! � test_terminal� s z"TestAvailProcessAPIs.test_terminalc C s t tjd�tptkst�d S )NZionice)r r rD r r r r r r r! � test_ionice� s z TestAvailProcessAPIs.test_ionicer) r* c C s t tjd�tptkst�d S )NZrlimit)r r rD r r r r r r r! r1 � s z TestAvailProcessAPIs.test_rlimitc C s" t tjd�}|tpt ks t�d S )NZ io_counters)r r rD r r r �r Zhasitr r r! �test_io_counters� s z%TestAvailProcessAPIs.test_io_countersc C s t tjd�tkst�d S )NZnum_fdsrG 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 rD 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 rD 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 rD 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 rD r r r r r rM r r r! �test_memory_maps� s z%TestAvailProcessAPIs.test_memory_mapsN)r2 r3 r4 rE rH rI rK rL r r5 r6 r r r1 rN rO rP rQ rR rS r r r r! rB � s � rB 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 rD �proc)�clsr r r! � setUpClass� s z TestSystemAPITypes.setUpClassTc C s8 t |�s t�|D ]"}t||�s"t�|r|dkst�qd S )Nr )r 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 rZ 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㈵��>)rY r Z cpu_percent�floatr r 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�st�d S rU )rY r � cpu_count�intr r r r r! �test_cpu_count� s z!TestSystemAPITypes.test_cpu_count�arm64zskipped due to #1892r* z not supportedc C s2 t �� d krt�d��| jt �� tttfd� d S )Nzcpu_freq() returns None�r[ )r r; r �skipr] ra re r r r r r! r< � s z TestSystemAPITypes.test_cpu_freqc C s> t jdd��� D ](\}}t|t�s&t�| j|ttfd� qd S )NT)Zperdiskrh ) r Zdisk_io_counters�itemsrY �strr r] re 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 t�t|jt�s,t�t|jt�s<t�t|jt�st�qd S rU ) r Zdisk_partitionsrY Zdevicerk r Z mountpointZfstype�opts)r Zdiskr r r! �test_disk_partitions� s z'TestSystemAPITypes.test_disk_partitionsz requires rootc C sN t � �> t�d�}t|�tt|��ks*t�|D ]}t|�s.t�q.W 5 Q R X d S )N�all)r r Znet_connections�len�setr r )r �ret�connr r r! �test_net_connections s z'TestSystemAPITypes.test_net_connectionsc C s� t �� �� D ]�\}}t|t�s"t�|D ]t}td k rJtsJt|jtj �sZt�nt|jt �sZt�t|j t�sjt�t|j tt d �f�s�t�t|jtt d �f�s&t�q&q d S rU )r Z net_if_addrsrj rY rk r r r �family�IntEnumre �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"t�t|jt�s2t�td k rNt|j tj �s^t�nt|j t �s^t�t|j t �snt�t|j t �s t�q d S rU )r Z net_if_statsrj rY rk r Zisup�boolr Zduplexrx re �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 t�q d S )NT)Zpernic)r Znet_io_countersrY rk r )r r} r r r! �test_net_io_counters" s z'TestSystemAPITypes.test_net_io_countersc C s\ t �� �� D ]J\}}t|t�s"t�|D ].}t|jt�s:t�t|jtt t d �f�s&t�q&q d S rU ) r r? rj rY rk r �label�currentra re r{ �r �nameZunits�unitr r r! r@ ( s z$TestSystemAPITypes.test_sensors_fansc C s� t �� �� D ]~\}}t|t�s"t�|D ]b}t|jt�s:t�t|jtt t d �f�sTt�t|j tt t d �f�snt�t|j tt t d �f�s&t�q&q d S rU ) r r= rj rY rk r r� r� ra re r{ �high�criticalr� r r r! r> 1 s z,TestSystemAPITypes.test_sensors_temperaturesc C s t t�� t�st�d S rU )rY r Z boot_timera r r r r r! �test_boot_time< s z!TestSystemAPITypes.test_boot_timec C sj t �� D ]\}t|jt�s t�t|jttd �f�s4t�t|jttd �f�sLt�t|j t td �f�st�qd S rU ) r ZusersrY r� rk r rJ r{ �host�pidre )r �userr r r! � test_users@ s z TestSystemAPITypes.test_usersN)#r2 r3 r4 �__doc__� classmethodrX ra r] r^ rb rc rf r r5 r6 r �platform�machiner r< rn rp r rv r� r r� r r� r r@ r r> r� r� r r r r! rT � s: � rT c @ s& e Zd Zejje dd�dd� �ZdS )�TestProcessWaitTypez not POSIXr* c C s\ t �| �� j�}|�� |�� }|tj ks0t�t d k rJt |t j �sXt�nt |t �sXt�d S rU ) r rD Zspawn_testprocr� � terminate�wait�signal�SIGTERMr r rY rx re )r �p�coder r r! �test_negative_signalJ s z(TestProcessWaitType.test_negative_signalN)r2 r3 r4 r r5 r6 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 r7 rB rT r� r r r r! �<module> sB ?>