File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_memleaks.cpython-39.pyc
a
��?h3< � @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd lm
Z
dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& dd!lm'Z' ej(j)Z)e�*� Z+d"Z,d#d$� Z-G d%d&� d&e �Z.G d'd(� d(e.�Z/e#j0j1e
d)d*�G d+d,� d,e ��Z2G d-d.� d.e �Z3dS )/a� Tests for detecting function memory leaks (typically the ones
implemented in C). It does so by calling a function many times and
checking whether process memory usage keeps increasing between
calls or over time.
Note that this may produce false positives (especially on Windows
for some reason).
PyPy appears to be completely unstable for this framework, probably
because of how its JIT handles memory, so tests are skipped.
� )�print_functionN)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�ProcessLookupError)�super)�HAS_CPU_AFFINITY)� HAS_CPU_FREQ)� HAS_ENVIRON)�
HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_NET_IO_COUNTERS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)�
HAS_RLIMIT)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)� QEMU_USER)�TestMemoryLeak)�create_sockets)�
get_testfn)�process_namespace)�pytest)�skip_on_access_denied)�spawn_testproc)�system_namespace)� terminate� c C s dd� } | S )zsDecorator for those Linux functions which are implemented in pure
Python, and which we want to run faster.
c s t �� �� fdd��}|S )Nc s\ t rB| jj}z(t| j_� | g|�R i |��W || j_S || j_0 n� | g|�R i |��S d S �N)r � __class__�times� FEW_TIMES)�self�args�kwargs�before��fun� �F/usr/local/lib64/python3.9/site-packages/psutil/tests/test_memleaks.py�wrapperD s � z5fewtimes_if_linux.<locals>.decorator.<locals>.wrapper)� functools�wraps)r+ r. r, r* r- � decoratorC s z$fewtimes_if_linux.<locals>.decoratorr, )r1 r, r, r- �fewtimes_if_linux> s r2 c @ s0 e Zd ZdZeZdd� Ze� dd� �Ze� dd� �Z e� dd � �Z
e� d
d � �Z e j
je d d
�e� dd� ��Ze j
je d d
�e� dd� ��Ze� dd� �Zdd� Zdd� Ze j
je dd
�dd� �Ze j
je dd
�dd � �Ze j
je dd
�e� d d � ��Ze j
jed d
�d d!� �Ze� d"d#� �Ze� e e d$�d%d&� ��Z e j
je d'd
�d(d)� �Z e j
je d d
�e� d*d+� ��Z!e� d,d-� �Z"e� e e d$�d.d/� ��Z#e� d0d1� �Z$e� e j
je% dd
�d2d3� ��Z&e� d4d5� �Z'e� d6d7� �Z(e j
je d d
�e� d8d9� ��Z)d:d;� Z*e� d<d=� �Z+e j
je, dd
�d>d?� �Z-e j
je, dd
�d@dA� �Z.e� dBdC� �Z/e j
je0 dd
�e� dDdE� ��Z1e j
je2 dFd
�e j
je3 dd
�dGdH� ��Z4e j
je2 dFd
�e j
je3 dd
�dIdJ� ��Z5e� e j
je dKd
�dLdM� ��Z6e j
je7 dd
�dNdO� �Z8e j
je d'd
�dPdQ� �Z9dRS )S�TestProcessObjectLeaksz$Test leaks of Process class methods.c C s t d �}|�| |j|j � d S r" )r �test_class_coverageZgettersZsetters�r& �nsr, r, r- �
test_coverage_ s z$TestProcessObjectLeaks.test_coveragec C s | � | jj� d S r" )�execute�proc�name�r& r, r, r- � test_namec s z TestProcessObjectLeaks.test_namec C s | � | jj� d S r" )r8 r9 Zcmdliner; r, r, r- � test_cmdlineg s z#TestProcessObjectLeaks.test_cmdlinec C s | � | jj� d S r" )r8 r9 Zexer; r, r, r- �test_exek s z TestProcessObjectLeaks.test_exec C s | � | jj� d S r" )r8 r9 Zppidr; r, r, r- � test_ppido s z TestProcessObjectLeaks.test_ppidz
POSIX only��reasonc C s | � | jj� d S r" )r8 r9 Zuidsr; r, r, r- � test_uidss s z TestProcessObjectLeaks.test_uidsc C s | � | jj� d S r" )r8 r9 �gidsr; r, r, r- � test_gidsx s z TestProcessObjectLeaks.test_gidsc C s | � | jj� d S r" )r8 r9 �statusr; r, r, r- � test_status} s z"TestProcessObjectLeaks.test_statusc C s | � | jj� d S r" )r8 r9 �nicer; r, r, r- � test_nice� s z TestProcessObjectLeaks.test_nicec s t �� � ��� �fdd�� d S )Nc s �j �� �S r" )r9 rG r, �Znicenessr& r, r- �<lambda>� � z6TestProcessObjectLeaks.test_nice_set.<locals>.<lambda>)�thisprocrG r8 r; r, rI r- �
test_nice_set� s z$TestProcessObjectLeaks.test_nice_set�
not supportedc C s | � | jj� d S r" )r8 r9 �ionicer; r, r, r- � test_ionice� s z"TestProcessObjectLeaks.test_ionicec sZ t r"t�� �� �� �fdd�� n4� �� fdd�� t�tjt� � dd�}� �
t |� d S )Nc s � j ���S r" )r9 rO r, �r& �valuer, r- rJ � rK z8TestProcessObjectLeaks.test_ionice_set.<locals>.<lambda>c s � j �tj�S r" )r9 rO �psutilZIOPRIO_CLASS_NONEr, r; r, r- rJ � rK ���r ) r rL rO r8 r/ �partial�cextZproc_ioprio_set�os�getpid�
execute_w_exc�OSError�r& r+ r, rQ r- �test_ionice_set� s z&TestProcessObjectLeaks.test_ionice_setc C s | � | jj� d S r" )r8 r9 Z io_countersr; r, r, r- �test_io_counters� s z'TestProcessObjectLeaks.test_io_counterszworthless on POSIXc C s t �� �� | �| jj� d S r" )rS �Process�usernamer8 r9 r; r, r, r- �
test_username� s z$TestProcessObjectLeaks.test_usernamec C s | � | jj� d S r" )r8 r9 Z create_timer; r, r, r- �test_create_time� s z'TestProcessObjectLeaks.test_create_time)Zonly_ifc C s | � | jj� d S r" )r8 r9 Z num_threadsr; r, r, r- �test_num_threads� s z'TestProcessObjectLeaks.test_num_threads� WINDOWS onlyc C s | � | jj� d S r" )r8 r9 Z num_handlesr; r, r, r- �test_num_handles� s z'TestProcessObjectLeaks.test_num_handlesc C s | � | jj� d S r" )r8 r9 Znum_fdsr; r, r, r- � test_num_fds� s z#TestProcessObjectLeaks.test_num_fdsc C s | � | jj� d S r" )r8 r9 Znum_ctx_switchesr; r, r, r- �test_num_ctx_switches� s z,TestProcessObjectLeaks.test_num_ctx_switchesc C s | � | jj� d S r" )r8 r9 �threadsr; r, r, r- � test_threads� s z#TestProcessObjectLeaks.test_threadsc C s | � | jj� d S r" )r8 r9 � cpu_timesr; r, r, r- �test_cpu_times� s z%TestProcessObjectLeaks.test_cpu_timesc C s | � | jj� d S r" )r8 r9 Zcpu_numr; r, r, r- � test_cpu_num� s z#TestProcessObjectLeaks.test_cpu_numc C s | � | jj� d S r" )r8 r9 Z memory_infor; r, r, r- �test_memory_info� s z'TestProcessObjectLeaks.test_memory_infoc C s | � | jj� d S r" )r8 r9 Zmemory_full_infor; r, r, r- �test_memory_full_info� s z,TestProcessObjectLeaks.test_memory_full_infoc C s | � | jj� d S r" )r8 r9 Zterminalr; r, r, r- �
test_terminal� s z$TestProcessObjectLeaks.test_terminalc C s$ t rtn| j}| j| jj|d� d S )N�r$ )r r% r$ r8 r9 �resume�r& r$ r, r, r- � test_resume� s z"TestProcessObjectLeaks.test_resumec C s | � | jj� d S r" )r8 r9 �cwdr; r, r, r- �test_cwd� s z TestProcessObjectLeaks.test_cwdc C s | � | jj� d S r" )r8 r9 � cpu_affinityr; r, r, r- �test_cpu_affinity� s z(TestProcessObjectLeaks.test_cpu_affinityc s4 t �� � ��� �fdd�� ��t�fdd�� d S )Nc s �j �� �S r" �r9 ru r, �Zaffinityr& r, r- rJ � rK z>TestProcessObjectLeaks.test_cpu_affinity_set.<locals>.<lambda>c s � j �dg�S �NrT rw r, r; r, r- rJ � rK )rL ru r8 rY �
ValueErrorr; r, rx r- �test_cpu_affinity_set� s z,TestProcessObjectLeaks.test_cpu_affinity_setc C s> t t� d�� | �| jj� W d � n1 s00 Y d S )N�w)�openr r8 r9 Z
open_filesr; r, r, r- �test_open_files� s z&TestProcessObjectLeaks.test_open_filesc C s | � | jj� d S r" )r8 r9 Z memory_mapsr; r, r, r- �test_memory_maps� s z'TestProcessObjectLeaks.test_memory_mapsz
LINUX onlyc s � � � fdd�� d S )Nc s � j �tj�S r" �r9 �rlimitrS �
RLIMIT_NOFILEr, r; r, r- rJ � rK z4TestProcessObjectLeaks.test_rlimit.<locals>.<lambda>�r8 r; r, r; r- � test_rlimit� s z"TestProcessObjectLeaks.test_rlimitc s< t �tj�� ��� �fdd�� ��ttf�fdd�� d S )Nc s �j �tj� �S r" r� r, ��limitr& r, r- rJ � rK z8TestProcessObjectLeaks.test_rlimit_set.<locals>.<lambda>c s � j �d�S ry )r9 r� r, r; r, r- rJ � rK )rL r� rS r� r8 rY rZ rz r; r, r� r- �test_rlimit_set� s z&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc sJ t � �0 trdnd� ��� �fdd�� W d � n1 s<0 Y d S )NZinet�allc s �j �� �S r" )r9 �net_connectionsr, ��kindr& r, r- rJ rK z=TestProcessObjectLeaks.test_net_connections.<locals>.<lambda>)r r r8 r; r, r� r- �test_net_connections� s z+TestProcessObjectLeaks.test_net_connectionsc C s | � | jj� d S r" )r8 r9 �environr; r, r, r- � test_environ s z#TestProcessObjectLeaks.test_environc C s | � dd� � d S )Nc S s t �t�� �S r" )rV � proc_inforW rX r, r, r, r- rJ rK z7TestProcessObjectLeaks.test_proc_info.<locals>.<lambda>r� r; r, r, r- �test_proc_info
s z%TestProcessObjectLeaks.test_proc_infoN):�__name__�
__module__� __qualname__�__doc__rL r9 r7 r2 r< r= r> r? r �mark�skipifr rB rD rF rH rM r rP r\ r r] r` ra r r rb r rd re rf rh rj r rk rl rm rn rr rt r rv r{ r~ r r r r r� r� r� r
r� r� r, r, r, r- r3 Z s�
r3 c sp e Zd ZdZe� fdd��Ze� fdd��Zdd� Zerhdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Z� ZS )�TestTerminatedProcessLeaksz�Repeat the tests above looking for leaks occurring when dealing
with terminated processes raising NoSuchProcess exception.
The C functions are still invoked but will follow different code
paths. We'll check those code paths.
c s: t � �� t� | _t�| jj�| _| j�� | j� � d S r" )
r
�
setUpClassr �subprS r^ �pidr9 �kill�wait��cls�r# r, r- r� s
z%TestTerminatedProcessLeaks.setUpClassc s t � �� t| j� d S r" )r
�
tearDownClassr r� r� r� r, r- r� s
z(TestTerminatedProcessLeaks.tearDownClassc C s$ z
|� W n t jy Y n0 d S r" )rS Z
NoSuchProcessr[ r, r, r- �call# s
z TestTerminatedProcessLeaks.callc C s | � | jj� d S r" )r8 r9 r� r; r, r, r- � test_kill+ s z$TestTerminatedProcessLeaks.test_killc C s | � | jj� d S r" )r8 r9 r r; r, r, r- �test_terminate. s z)TestTerminatedProcessLeaks.test_terminatec C s | � | jj� d S r" )r8 r9 Zsuspendr; r, r, r- � test_suspend1 s z'TestTerminatedProcessLeaks.test_suspendc C s | � | jj� d S r" )r8 r9 rp r; r, r, r- rr 4 s z&TestTerminatedProcessLeaks.test_resumec C s | � | jj� d S r" )r8 r9 r� r; r, r, r- � test_wait7 s z$TestTerminatedProcessLeaks.test_waitc s � fdd�}� � |� d S )Nc s( zt �� jj�W S ty" Y n0 d S r" )rV r� r9 r� r r, r; r, r- r� < s z7TestTerminatedProcessLeaks.test_proc_info.<locals>.callr� )r& r� r, r; r- r� : s z)TestTerminatedProcessLeaks.test_proc_info)r� r� r� r� � classmethodr� r� r� r r� r� r� rr r� r� �
__classcell__r, r, r� r- r� s r� rc r@ c @ s e Zd Zdd� Zdd� ZdS )� TestProcessDualImplementationc C s | � dd� � d S )Nc S s t jt�� dd�S )NT�Zuse_peb�rV Z proc_cmdlinerW rX r, r, r, r- rJ H rK zETestProcessDualImplementation.test_cmdline_peb_true.<locals>.<lambda>r� r; r, r, r- �test_cmdline_peb_trueG s z3TestProcessDualImplementation.test_cmdline_peb_truec C s | � dd� � d S )Nc S s t jt�� dd�S )NFr� r� r, r, r, r- rJ K rK zFTestProcessDualImplementation.test_cmdline_peb_false.<locals>.<lambda>r� r; r, r, r- �test_cmdline_peb_falseJ s z4TestProcessDualImplementation.test_cmdline_peb_falseN)r� r� r� r� r� r, r, r, r- r� E s r� c @ sP e Zd ZdZdd� Ze� dd� �Ze� dd� �Ze� dd � �Ze� d
d � �Z e� d d
� �Z
e� e j j
eore�� dkdd�e j j
e dd�dd� ���Ze j j
e dd�dd� �Zdd� Ze j j
edd�dd� �Zd d � Zd d � Ze j j
ed d�d!d"� �Ze j j
e �oe j � d#� d$d�e� d%d&� ��Z e� d'd(� �Z!e� e j j
e" dd�d)d*� ��Z#e� e j j
e�one �$� d+kd,d�d-d.� ��Z%d/d0� Z&e j j
ed d�d1d2� �Z'e� e j j
e( dd�d3d4� ��Z)e� e j j
e* dd�d5d6� ��Z+e� e j j
e, dd�d7d8� ��Z-e� d9d:� �Z.d;d<� Z/d=d>� Z0e�rLd?d@� Z1dAdB� Z2dCdD� Z3dEdF� Z4dGdH� Z5dIS )J�TestModuleFunctionsLeaksz&Test leaks of psutil module functions.c C s t � }|�| |j� d S r" )r r4 r� r5 r, r, r- r7 V s z&TestModuleFunctionsLeaks.test_coveragec C s | � dd� � d S )Nc S s t jdd�S )NT��logical�rS � cpu_countr, r, r, r- rJ ^ rK z9TestModuleFunctionsLeaks.test_cpu_count.<locals>.<lambda>r� r; r, r, r- �test_cpu_count\ s z'TestModuleFunctionsLeaks.test_cpu_countc C s | � dd� � d S )Nc S s t jdd�S )NFr� r� r, r, r, r- rJ b rK z?TestModuleFunctionsLeaks.test_cpu_count_cores.<locals>.<lambda>r� r; r, r, r- �test_cpu_count_cores` s z-TestModuleFunctionsLeaks.test_cpu_count_coresc C s | � tj� d S r" )r8 rS ri r; r, r, r- rj d s z'TestModuleFunctionsLeaks.test_cpu_timesc C s | � dd� � d S )Nc S s t jdd�S )NT)Zpercpu)rS ri r, r, r, r- rJ j rK z=TestModuleFunctionsLeaks.test_per_cpu_times.<locals>.<lambda>r� r; r, r, r- �test_per_cpu_timesh s z+TestModuleFunctionsLeaks.test_per_cpu_timesc C s | � tj� d S r" )r8 rS Z cpu_statsr; r, r, r- �test_cpu_statsl s z'TestModuleFunctionsLeaks.test_cpu_stats�arm64zskipped due to #1892r@ rN c C s | � tj� d S r" )r8 rS Zcpu_freqr; r, r, r- �
test_cpu_freqp s z&TestModuleFunctionsLeaks.test_cpu_freqrc c C s t �� | �t j� d S r" )rS �
getloadavgr8 r; r, r, r- �test_getloadavgy s z(TestModuleFunctionsLeaks.test_getloadavgc C s | � tj� d S r" )r8 rS Zvirtual_memoryr; r, r, r- �test_virtual_memory� s z,TestModuleFunctionsLeaks.test_virtual_memoryz&worthless on SUNOS (uses a subprocess)c C s | � tj� d S r" )r8 rS Z swap_memoryr; r, r, r- �test_swap_memory� s z)TestModuleFunctionsLeaks.test_swap_memoryc C s$ t rtn| j}| jdd� |d� d S )Nc S s t �t�� �S r" )rS Z
pid_existsrW rX r, r, r, r- rJ � rK z:TestModuleFunctionsLeaks.test_pid_exists.<locals>.<lambda>ro �r r% r$ r8 rq r, r, r- �test_pid_exists� s z(TestModuleFunctionsLeaks.test_pid_existsc C s$ t rtn| j}| jdd� |d� d S )Nc S s
t �d�S )N�.)rS �
disk_usager, r, r, r- rJ � rK z:TestModuleFunctionsLeaks.test_disk_usage.<locals>.<lambda>ro r� rq r, r, r- �test_disk_usage� s z(TestModuleFunctionsLeaks.test_disk_usagezQEMU user not supportedc C s | � tj� d S r" )r8 rS Zdisk_partitionsr; r, r, r- �test_disk_partitions� s z-TestModuleFunctionsLeaks.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this Linux versionc C s | � dd� � d S )Nc S s t jdd�S �NF)Znowrap)rS Zdisk_io_countersr, r, r, r- rJ � rK z@TestModuleFunctionsLeaks.test_disk_io_counters.<locals>.<lambda>r� r; r, r, r- �test_disk_io_counters� s z.TestModuleFunctionsLeaks.test_disk_io_countersc C s | � tj� d S r" )r8 rS Zpidsr; r, r, r- � test_pids� s z"TestModuleFunctionsLeaks.test_pidsc C s | � dd� � d S )Nc S s t jdd�S r� )rS Znet_io_countersr, r, r, r- rJ � rK z?TestModuleFunctionsLeaks.test_net_io_counters.<locals>.<lambda>r� r; r, r, r- �test_net_io_counters� s z-TestModuleFunctionsLeaks.test_net_io_countersr zneed root accessc C sD t jdd� t� � | �dd� � W d � n1 s60 Y d S )Nr� �r� c S s t jdd�S )Nr� r� )rS r� r, r, r, r- rJ � rK z?TestModuleFunctionsLeaks.test_net_connections.<locals>.<lambda>)rS r� r r8 r; r, r, r- r� � s z-TestModuleFunctionsLeaks.test_net_connectionsc C s" t rdn| j}| jtj|d� d S )Ni @ )� tolerance)r r� r8 rS Z net_if_addrs)r& r� r, r, r- �test_net_if_addrs� s z*TestModuleFunctionsLeaks.test_net_if_addrsc C s | � tj� d S r" )r8 rS Z net_if_statsr; r, r, r- �test_net_if_stats� s z*TestModuleFunctionsLeaks.test_net_if_statsc C s | � tj� d S r" )r8 rS Zsensors_batteryr; r, r, r- �test_sensors_battery� s z-TestModuleFunctionsLeaks.test_sensors_batteryc C s | � tj� d S r" )r8 rS Zsensors_temperaturesr; r, r, r- �test_sensors_temperatures� s z2TestModuleFunctionsLeaks.test_sensors_temperaturesc C s | � tj� d S r" )r8 rS Z sensors_fansr; r, r, r- �test_sensors_fans� s z*TestModuleFunctionsLeaks.test_sensors_fansc C s | � tj� d S r" )r8 rS Z boot_timer; r, r, r- �test_boot_time� s z'TestModuleFunctionsLeaks.test_boot_timec C s | � tj� d S r" )r8 rS Zusersr; r, r, r- �
test_users� s z#TestModuleFunctionsLeaks.test_usersc C s | � dd� � d S )Nc S s
t �d�S )NF)rS Z
_set_debugr, r, r, r- rJ � rK z9TestModuleFunctionsLeaks.test_set_debug.<locals>.<lambda>r� r; r, r, r- �test_set_debug� s z'TestModuleFunctionsLeaks.test_set_debugc C s | � tj� d S r" )r8 rV Zwinservice_enumerater; r, r, r- �test_win_service_iter� s z.TestModuleFunctionsLeaks.test_win_service_iterc C s d S r" r, r; r, r, r- �test_win_service_get� s z-TestModuleFunctionsLeaks.test_win_service_getc s& t t�� ��� � | �� fdd�� d S )Nc s
t �� �S r" )rV Zwinservice_query_configr, �r: r, r- rJ � rK zFTestModuleFunctionsLeaks.test_win_service_get_config.<locals>.<lambda>��nextrS Zwin_service_iterr: r8 r; r, r� r- �test_win_service_get_config� s z4TestModuleFunctionsLeaks.test_win_service_get_configc s& t t�� ��� � | �� fdd�� d S )Nc s
t �� �S r" )rV Zwinservice_query_statusr, r� r, r- rJ � rK zFTestModuleFunctionsLeaks.test_win_service_get_status.<locals>.<lambda>r� r; r, r� r- �test_win_service_get_status� s z4TestModuleFunctionsLeaks.test_win_service_get_statusc s& t t�� ��� � | �� fdd�� d S )Nc s
t �� �S r" )rV Zwinservice_query_descrr, r� r, r- rJ � rK zKTestModuleFunctionsLeaks.test_win_service_get_description.<locals>.<lambda>r� r; r, r� r- � test_win_service_get_description� s z9TestModuleFunctionsLeaks.test_win_service_get_descriptionN)6r� r� r� r� r7 r2 r� r� rj r� r� r r� r� r �platform�machiner r� r r� r� r r� r� r� r r� r rW �path�existsr� r� r r� �getuidr� r� r� r r� r r� r r� r� r� r� r� r� r� r� r� r, r, r, r- r� S s~
�
�
r� )4r� �
__future__r r/ rW r� rS Zpsutil._commonr 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 r r r r r r Z _psplatformrV r^ rL r% r2 r3 r� r� r� r� r� r, r, r, r- �<module> s\
66