관리-도구
편집 파일: test_memleaks.cpython-38.pyc
U �Mg3< � @ 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 sJ t r6| jj}z t| j_� | f|�|�W �S || j_X n� | f|�|�S d S �N)r � __class__�times� FEW_TIMES)�self�args�kwargs�before��fun� �J/opt/hc_python/lib64/python3.8/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_coverage�gettersZsetters�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" )r9 r: Zcmdliner< r, r, r- � test_cmdlineg s z#TestProcessObjectLeaks.test_cmdlinec C s | � | jj� d S r" )r9 r: �exer< r, r, r- �test_exek s z TestProcessObjectLeaks.test_exec C s | � | jj� d S r" )r9 r: Zppidr< r, r, r- � test_ppido s z TestProcessObjectLeaks.test_ppidz POSIX only��reasonc C s | � | jj� d S r" )r9 r: Zuidsr< r, r, r- � test_uidss s z TestProcessObjectLeaks.test_uidsc C s | � | jj� d S r" )r9 r: Zgidsr< r, r, r- � test_gidsx s z TestProcessObjectLeaks.test_gidsc C s | � | jj� d S r" )r9 r: �statusr< r, r, r- � test_status} s z"TestProcessObjectLeaks.test_statusc C s | � | jj� d S r" )r9 r: �nicer< r, r, r- � test_nice� s z TestProcessObjectLeaks.test_nicec s t �� � ��� �fdd�� d S )Nc s �j �� �S r" )r: rH r, �Znicenessr& r, r- �<lambda>� � z6TestProcessObjectLeaks.test_nice_set.<locals>.<lambda>)�thisprocrH r9 r< r, rJ r- � test_nice_set� s z$TestProcessObjectLeaks.test_nice_set� not supportedc C s | � | jj� d S r" )r9 r: �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" )r: rP r, �r& �valuer, r- rK � rL z8TestProcessObjectLeaks.test_ionice_set.<locals>.<lambda>c s � j �tj�S r" )r: rP �psutilZIOPRIO_CLASS_NONEr, r< r, r- rK � rL ���r ) r rM rP r9 r/ �partial�cextZproc_ioprio_set�os�getpid� execute_w_exc�OSError�r& r+ r, rR r- �test_ionice_set� s z&TestProcessObjectLeaks.test_ionice_setc C s | � | jj� d S r" )r9 r: Z io_countersr< r, r, r- �test_io_counters� s z'TestProcessObjectLeaks.test_io_counterszworthless on POSIXc C s t �� �� | �| jj� d S r" )rT �Process�usernamer9 r: r< r, r, r- � test_username� s z$TestProcessObjectLeaks.test_usernamec C s | � | jj� d S r" )r9 r: Z create_timer< r, r, r- �test_create_time� s z'TestProcessObjectLeaks.test_create_time)Zonly_ifc C s | � | jj� d S r" )r9 r: Z num_threadsr< r, r, r- �test_num_threads� s z'TestProcessObjectLeaks.test_num_threads� WINDOWS onlyc C s | � | jj� d S r" )r9 r: Z num_handlesr< r, r, r- �test_num_handles� s z'TestProcessObjectLeaks.test_num_handlesc C s | � | jj� d S r" )r9 r: Znum_fdsr< r, r, r- � test_num_fds� s z#TestProcessObjectLeaks.test_num_fdsc C s | � | jj� d S r" )r9 r: Znum_ctx_switchesr< r, r, r- �test_num_ctx_switches� s z,TestProcessObjectLeaks.test_num_ctx_switchesc C s | � | jj� d S r" )r9 r: �threadsr< r, r, r- � test_threads� s z#TestProcessObjectLeaks.test_threadsc C s | � | jj� d S r" )r9 r: � cpu_timesr< r, r, r- �test_cpu_times� s z%TestProcessObjectLeaks.test_cpu_timesc C s | � | jj� d S r" )r9 r: Zcpu_numr< r, r, r- � test_cpu_num� s z#TestProcessObjectLeaks.test_cpu_numc C s | � | jj� d S r" )r9 r: Z memory_infor< r, r, r- �test_memory_info� s z'TestProcessObjectLeaks.test_memory_infoc C s | � | jj� d S r" )r9 r: Zmemory_full_infor< r, r, r- �test_memory_full_info� s z,TestProcessObjectLeaks.test_memory_full_infoc C s | � | jj� d S r" )r9 r: 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$ r9 r: �resume�r& r$ r, r, r- � test_resume� s z"TestProcessObjectLeaks.test_resumec C s | � | jj� d S r" )r9 r: �cwdr< r, r, r- �test_cwd� s z TestProcessObjectLeaks.test_cwdc C s | � | jj� d S r" )r9 r: � 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" �r: rv r, �Zaffinityr& r, r- rK � rL z>TestProcessObjectLeaks.test_cpu_affinity_set.<locals>.<lambda>c s � j �dg�S �NrU rx r, r< r, r- rK � rL )rM rv r9 rZ � ValueErrorr< r, ry r- �test_cpu_affinity_set� s z,TestProcessObjectLeaks.test_cpu_affinity_setc C s* t t� d�� | �| jj� W 5 Q R X d S )N�w)�openr r9 r: Z open_filesr< r, r, r- �test_open_files� s z&TestProcessObjectLeaks.test_open_filesc C s | � | jj� d S r" )r9 r: 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" �r: �rlimitrT � RLIMIT_NOFILEr, r< r, r- rK � rL z4TestProcessObjectLeaks.test_rlimit.<locals>.<lambda>�r9 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- rK � rL z8TestProcessObjectLeaks.test_rlimit_set.<locals>.<lambda>c s � j �d�S rz )r: r� r, r< r, r- rK � rL )rM r� rT r� r9 rZ r[ r{ r< r, r� r- �test_rlimit_set� s z&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc s6 t � �& trdnd� ��� �fdd�� W 5 Q R X d S )NZinet�allc s �j �� �S r" )r: �net_connectionsr, ��kindr& r, r- rK rL z=TestProcessObjectLeaks.test_net_connections.<locals>.<lambda>)r r r9 r< r, r� r- �test_net_connections� s z+TestProcessObjectLeaks.test_net_connectionsc C s | � | jj� d S r" )r9 r: �environr< r, r, r- � test_environ s z#TestProcessObjectLeaks.test_environc C s | � dd� � d S )Nc S s t �t�� �S r" )rW � proc_inforX rY r, r, r, r- rK rL z7TestProcessObjectLeaks.test_proc_info.<locals>.<lambda>r� r< r, r, r- �test_proc_info s z%TestProcessObjectLeaks.test_proc_infoN):�__name__� __module__� __qualname__�__doc__rM r: r8 r2 r= r> r@ rA r �mark�skipifr rD rE rG rI rN r rQ r] r r^ ra rb r r rc r re rf rg ri rk r rl rm rn ro rs ru r rw 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 �subprT r_ �pidr: �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 jk r Y nX d S r" )rT Z NoSuchProcessr\ r, r, r- �call# s z TestTerminatedProcessLeaks.callc C s | � | jj� d S r" )r9 r: r� r< r, r, r- � test_kill+ s z$TestTerminatedProcessLeaks.test_killc C s | � | jj� d S r" )r9 r: r r< r, r, r- �test_terminate. s z)TestTerminatedProcessLeaks.test_terminatec C s | � | jj� d S r" )r9 r: Zsuspendr< r, r, r- � test_suspend1 s z'TestTerminatedProcessLeaks.test_suspendc C s | � | jj� d S r" )r9 r: rq r< r, r, r- rs 4 s z&TestTerminatedProcessLeaks.test_resumec C s | � | jj� d S r" )r9 r: r� r< r, r, r- � test_wait7 s z$TestTerminatedProcessLeaks.test_waitc s � fdd�}� � |� d S )Nc s* zt �� jj�W S tk r$ Y nX d S r" )rW r� r: 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� rs r� r� � __classcell__r, r, r� r- r� s r� rd rB c @ s e Zd Zdd� Zdd� ZdS )� TestProcessDualImplementationc C s | � dd� � d S )Nc S s t jt�� dd�S )NT�Zuse_peb�rW Z proc_cmdlinerX rY r, r, r, r- rK H rL 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- rK K rL 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� r6 r, r, r- r8 V s z&TestModuleFunctionsLeaks.test_coveragec C s | � dd� � d S )Nc S s t jdd�S )NT�Zlogical�rT � cpu_countr, r, r, r- rK ^ rL 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- rK b rL 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" )r9 rT rj r< r, r, r- rk d s z'TestModuleFunctionsLeaks.test_cpu_timesc C s | � dd� � d S )Nc S s t jdd�S )NT)Zpercpu)rT rj r, r, r, r- rK j rL 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" )r9 rT Z cpu_statsr< r, r, r- �test_cpu_statsl s z'TestModuleFunctionsLeaks.test_cpu_stats�arm64zskipped due to #1892rB rO c C s | � tj� d S r" )r9 rT Zcpu_freqr< r, r, r- � test_cpu_freqp s z&TestModuleFunctionsLeaks.test_cpu_freqrd c C s t �� | �t j� d S r" )rT � getloadavgr9 r< r, r, r- �test_getloadavgy s z(TestModuleFunctionsLeaks.test_getloadavgc C s | � tj� d S r" )r9 rT 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" )r9 rT 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" )rT Z pid_existsrX rY r, r, r, r- rK � rL z:TestModuleFunctionsLeaks.test_pid_exists.<locals>.<lambda>rp �r r% r$ r9 rr 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�.)rT � disk_usager, r, r, r- rK � rL z:TestModuleFunctionsLeaks.test_disk_usage.<locals>.<lambda>rp r� rr r, r, r- �test_disk_usage� s z(TestModuleFunctionsLeaks.test_disk_usagezQEMU user not supportedc C s | � tj� d S r" )r9 rT 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)rT Zdisk_io_countersr, r, r, r- rK � rL 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" )r9 rT Zpidsr< r, r, r- � test_pids� s z"TestModuleFunctionsLeaks.test_pidsc C s | � dd� � d S )Nc S s t jdd�S r� )rT Znet_io_countersr, r, r, r- rK � rL 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 s0 t jdd� t� � | �dd� � W 5 Q R X d S )Nr� �r� c S s t jdd�S )Nr� r� )rT r� r, r, r, r- rK � rL z?TestModuleFunctionsLeaks.test_net_connections.<locals>.<lambda>)rT r� r r9 r< r, r, r- r� � s z-TestModuleFunctionsLeaks.test_net_connectionsc C s" t rdn| j}| jtj|d� d S )Ni @ )� tolerance)r r� r9 rT 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" )r9 rT Z net_if_statsr< r, r, r- �test_net_if_stats� s z*TestModuleFunctionsLeaks.test_net_if_statsc C s | � tj� d S r" )r9 rT Zsensors_batteryr< r, r, r- �test_sensors_battery� s z-TestModuleFunctionsLeaks.test_sensors_batteryc C s | � tj� d S r" )r9 rT Zsensors_temperaturesr< r, r, r- �test_sensors_temperatures� s z2TestModuleFunctionsLeaks.test_sensors_temperaturesc C s | � tj� d S r" )r9 rT Z sensors_fansr< r, r, r- �test_sensors_fans� s z*TestModuleFunctionsLeaks.test_sensors_fansc C s | � tj� d S r" )r9 rT Z boot_timer< r, r, r- �test_boot_time� s z'TestModuleFunctionsLeaks.test_boot_timec C s | � tj� d S r" )r9 rT Zusersr< r, r, r- � test_users� s z#TestModuleFunctionsLeaks.test_usersc C s | � dd� � d S )Nc S s t �d�S )NF)rT Z _set_debugr, r, r, r- rK � rL 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" )r9 rW 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" )rW Zwinservice_query_configr, �r; r, r- rK � rL zFTestModuleFunctionsLeaks.test_win_service_get_config.<locals>.<lambda>��nextrT Zwin_service_iterr; r9 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" )rW Zwinservice_query_statusr, r� r, r- rK � rL 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" )rW Zwinservice_query_descrr, r� r, r- rK � rL 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� r8 r2 r� r� rk r� r� r r� r� r �platform�machiner r� r r� r� r r� r� r� r r� r rX �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/ rX r� rT 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 _psplatformrW r_ rM r% r2 r3 r� r� r� r� r� r, r, r, r- �<module> s\ 66