File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_testutils.cpython-39.pyc
a
��?h�H � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl m
Z
ddl mZ ddl mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
l mZ dd l mZ dd l mZ dd
l mZ ddl mZ ddl mZ ddl m Z ddl m Z ddl m Z ddl m Z ddl m Z ddl m!Z! ddl m"Z" ddl m#Z# ddl m$Z$ ddl m%Z% ddl m&Z& ddl m'Z' dd l m(Z( dd l m)Z) dd l m*Z* dd l m+Z+ dd l m,Z, dd!l m-Z- dd"l m.Z. dd#l m/Z/ dd$l m0Z0 dd%l m1Z1 dd&l m2Z2 dd'l m3Z3 dd(l m4Z4 G d)d*� d*e �Z5G d+d,� d,e �Z6G d-d.� d.e �Z7G d/d0� d0e �Z8G d1d2� d2e �Z9e)j:j;d3d4�G d5d6� d6e ��Z<G d7d8� d8e �Z=G d9d:� d:e �Z>G d;d<� d<e �Z?dS )=z1Tests for testing utils (psutil.tests namespace).� N)�FREEBSD)�NETBSD)�POSIX)� open_binary)� open_text)�
supports_ipv6)�PY3)�
CI_TESTING)�COVERAGE)�HAS_NET_CONNECTIONS_UNIX)�HERE)�
PYTHON_EXE)�PYTHON_EXE_ENV)�PsutilTestCase)�TestMemoryLeak)� bind_socket)�bind_unix_socket��
call_until)�chdir)�create_sockets)� fake_pytest)�filter_proc_net_connections)�
get_free_port)�
is_namedtuple)�mock)�process_namespace)�pytest)�
reap_children)�retry)�retry_on_failure)�
safe_mkdir)� safe_rmpath)�system_namespace)�tcp_socketpair)� terminate)�unix_socketpair)�
wait_for_file)� wait_for_pidc @ sx e Zd Ze�d�dd� �Ze�d�dd� �Ze�d�dd� �Ze�d�dd � �Ze�d�d
d � �Z e�d�d d
� �Z
dS )�TestRetryDecoratorz
time.sleepc sF t ddd d�� fdd��}ttd��� |� dks4J �|jdksBJ �d S )N� � ��retries�intervalZlogfunc s � r� � � dd q dS �Nr+ r ��pop� ��queuer2 �G/usr/local/lib64/python3.9/site-packages/psutil/tests/test_testutils.py�fooI s
z2TestRetryDecorator.test_retry_success.<locals>.foo� )r �list�range�
call_count��self�sleepr6 r2 r3 r5 �test_retry_successE s
z%TestRetryDecorator.test_retry_successc sh t ddd d�� fdd��}ttd��� t�t�� |� W d � n1 sL0 Y |jdksdJ �d S )Nr* r+ r, c s � r� � � dd q dS r/ r0 r2 r3 r2 r5 r6 W s
z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r r8 r9 r �raises�ZeroDivisionErrorr: r; r2 r3 r5 �test_retry_failureT s $z%TestRetryDecorator.test_retry_failurec C sV t tdd�dd� �}t�t�� |� W d � n1 s:0 Y |jdksRJ �d S )Nr+ )� exceptionr. c S s t �d S �N)� TypeErrorr2 r2 r2 r5 r6 e s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r �
ValueErrorr r@ rE r: r; r2 r2 r5 �test_exception_argc s
$z%TestRetryDecorator.test_exception_argc C sX t dd d d�dd� �}t�t�� |� W d � n1 s<0 Y |jdksTJ �d S )Nr* r, c S s dd d S r/ r2 r2 r2 r2 r5 r6 q s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor �r r r@ rA r: r; r2 r2 r5 �test_no_interval_argm s
$z'TestRetryDecorator.test_no_interval_argc C sX t ddd d�dd� �}t�t�� |� W d � n1 s<0 Y |jdksTJ �d S )Nr* r+ r, c S s dd d S r/ r2 r2 r2 r2 r5 r6 { s z0TestRetryDecorator.test_retries_arg.<locals>.foorH r; r2 r2 r5 �test_retries_argy s
$z#TestRetryDecorator.test_retries_argc C s: t �t�� tddd� W d � n1 s,0 Y d S )Nr* r+ )r- �timeout)r r@ rF r )r< r= r2 r2 r5 � test_retries_and_timeout_args� s z0TestRetryDecorator.test_retries_and_timeout_argsN) �__name__�
__module__� __qualname__r �patchr> rB rG rI rJ rL r2 r2 r2 r5 r) D s
r) c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestSyncTestUtilsc C s� t t�� � tt�� �d }tjdtdg�d��D t �
tj �� t |� W d � n1 s\0 Y W d � n1 sz0 Y d S )Ni�� �psutil.tests.retry.__iter__r �Z return_value) r( �os�getpid�max�psutilZpidsr rP �iterr r@ Z
NoSuchProcess)r< Znopidr2 r2 r5 �test_wait_for_pid� s
z#TestSyncTestUtils.test_wait_for_pidc C sX | � � }t|d��}|�d� W d � n1 s20 Y t|� tj�|�rTJ �d S )N�wr6 ��
get_testfn�open�writer' rT �path�exists�r< �testfn�fr2 r2 r5 �test_wait_for_file� s
(z$TestSyncTestUtils.test_wait_for_filec C sR | � � }t|d�� W d � n1 s(0 Y t|dd� tj�|�rNJ �d S )NrZ T)�empty)r\ r] r' rT r_ r` �r< rb r2 r2 r5 �test_wait_for_file_empty� s
z*TestSyncTestUtils.test_wait_for_file_emptyc C sr | � � }tjdtdg�d��B t�t�� t|� W d � n1 sF0 Y W d � n1 sd0 Y d S )NrR r rS )r\ r rP rX r r@ �IOErrorr' rf r2 r2 r5 �test_wait_for_file_no_file� s z,TestSyncTestUtils.test_wait_for_file_no_filec C s\ | � � }t|d��}|�d� W d � n1 s20 Y t|dd� tj�|�sXJ �d S )NrZ r6 F)�deleter[ ra r2 r2 r5 � test_wait_for_file_no_delete� s
( z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd� � d S )Nc S s dS �Nr+ r2 r2 r2 r2 r5 �<lambda>� � z3TestSyncTestUtils.test_call_until.<locals>.<lambda>r �r< r2 r2 r5 �test_call_until� s z!TestSyncTestUtils.test_call_untilN) rM rN rO rY rd rg ri rk rp r2 r2 r2 r5 rQ � s rQ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestFSTestUtilsc C s: t t�� }|jdksJ �W d � n1 s,0 Y d S )N�r)r �__file__�mode�r< rc r2 r2 r5 �test_open_text� s
z TestFSTestUtils.test_open_textc C s: t t�� }|jdksJ �W d � n1 s,0 Y d S )N�rb)r rs rt ru r2 r2 r5 �test_open_binary� s
z TestFSTestUtils.test_open_binaryc C s< | � � }t|� tj�|�s J �t|� tj�|�s8J �d S rD )r\ r! rT r_ �isdirrf r2 r2 r5 �test_safe_mkdir� s
z TestFSTestUtils.test_safe_mkdirc C s� | � � }t|d��� t|� tj�|�r.J �t|� t�|� t|� tj�|�rXJ �tj dt
t j d�d��L}t
�t
�� t|� W d � n1 s�0 Y |js�J �W d � n1 s�0 Y d S )NrZ zpsutil.tests.os.stat� )Z side_effect)r\ r] �closer" rT r_ r` �mkdirr rP �OSError�errno�EINVALr r@ �called)r< rb �mr2 r2 r5 �test_safe_rmpath� s
� &z TestFSTestUtils.test_safe_rmpathc C sp | � � }t�� }t�|� t|��* t�� tj�||�ks>J �W d � n1 sR0 Y t�� |kslJ �d S rD )r\ rT �getcwdr} r r_ �join)r< rb �baser2 r2 r5 �
test_chdir� s
8zTestFSTestUtils.test_chdirN)rM rN rO rv rx rz r� r� r2 r2 r2 r5 rq � s
rq c @ s> e Zd Zdd� Zdd� Zejje dd�dd� �Z d d
� Z
d S ) �TestProcessUtilsc C sN | � � }t�|j�}|�� s J �t� |�� r2J �tjjr>J �tjjrJJ �d S rD ) �spawn_testprocrW �Process�pid�
is_runningr �testsZ
_pids_startedZ_subprocesses_started)r< Zsubp�pr2 r2 r5 �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� | � � \}}|j|jks J �|�� s(J �|�� s4J �t�� �� }||gksNJ �t�� jdd�}t|�dksnJ �||v szJ �||v s�J �|�� t� � ks�J �|�� |jks�J �t
|� |�� r�J �|�� s�J �t
|� |�� r�J �d S )NT)� recursive� ) Zspawn_children_pairr� r� rW r� �children�len�ppidrT rU r% )r< �childZ
grandchildr� r2 r2 r5 �test_spawn_children_pair� s" z)TestProcessUtils.test_spawn_children_pair�
POSIX only��reasonc C s" | � � \}}|�� tjks J �d S rD )� spawn_zombie�statusrW Z
STATUS_ZOMBIE)r< �_parent�zombier2 r2 r5 �test_spawn_zombie� s z"TestProcessUtils.test_spawn_zombiec C s� | � � }t|� | �|j� t|� t�| � � j�}t|� | �|j� t|� tddg}tj|tj tj t
d�}t|� | �|j� t|� | � � j}t|� | �|j� t|� t r�| � � \}}t|� t|� | �|j� | �|j� d S )Nz-cz3import time; [time.sleep(0.1) for x in range(100)];)�stdout�stderr�env)
r� r% Z
assertPidGoner� rW r� r
�Popen�
subprocess�PIPEr r r� )r< r� �cmdr� �parentr� r2 r2 r5 �test_terminate s> ��
z TestProcessUtils.test_terminateN) rM rN rO r� r� r �mark�skipifr r� r� r2 r2 r2 r5 r� � s
r� c @ sl e Zd Zdd� Zejje dd�dd� �Zdd� Z ejje dd�ejje
pNe d d�d
d � ��Z d d
� Z
dS )� TestNetUtilsc C sR t � }t�td|fd���$}|�� d |ks0J �W d � n1 sD0 Y d S )Nr{ ��addrr+ )r �
contextlib�closingr � getsockname)r< �port�sr2 r2 r5 r + s zTestNetUtils.bind_socketr� r� c C s� | � � }t|�}t�|��f |jtjks,J �|jtjks<J �|� � |ksLJ �t
j � |�s\J �t
�t
�
|�j�srJ �W d � n1 s�0 Y | � � }t|tjd�}t�|�� |jtjks�J �W d � n1 s�0 Y d S )N)�type)r\ r r� r� �family�socket�AF_UNIXr� � SOCK_STREAMr� rT r_ r` �stat�S_ISSOCK�st_mode�
SOCK_DGRAM)r< �name�sockr2 r2 r5 �test_bind_unix_socket0 s 4 z"TestNetUtils.test_bind_unix_socketc C s� dt � f}ttj|d�\}}t�|��j t�|��@ |�� |ksDJ �|�� |ksTJ �|�� |ksdJ �W d � n1 sx0 Y W d � n1 s�0 Y d S )Nz 127.0.0.1r� )r r$ r� �AF_INETr� r� r� � getpeername)r< r� �server�clientr2 r2 r5 �tcp_tcp_socketpair@ s
z TestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C s� t �� }|�� }t|jdd��g ks(J �| �� }t|�\}}z�tj� |�sNJ �t
� t�
|�j �sdJ �|�� | dksxJ �t
t|jdd���dks�J �|�� |ks�J �|�� |ks�J �W |�� |�� n|�� |�� 0 d S )N�unix)�kindr� )rW r� �num_fdsr Znet_connectionsr\ r& rT r_ r` r� r� r� r� r� r� r| )r< r� r� r� r� r� r2 r2 r5 �test_unix_socketpairK s, � ���
�z!TestNetUtils.test_unix_socketpairc C s� t � ��}t�t�}t�t�}|D ]2}||j d7 < ||�tjtj� d7 < q |tj dksfJ �t
� r~|tj dks~J �t r�t
r�|tj dks�J �|tj dks�J �|tj dks�J �W d � n1 s�0 Y d S )Nr+ r� )r � collections� defaultdict�intr� �
getsockoptr� �
SOL_SOCKET�SO_TYPEr� r �AF_INET6r r r� r� r� )r< �socksZfams�typesr� r2 r2 r5 �test_create_socketsg s
z TestNetUtils.test_create_socketsN)rM rN rO r r r� r� r r� r� r r r� r� r2 r2 r2 r5 r� * s
� r� �serial�r� c @ sh e Zd Ze� dd� �Zdd� Ze� ejje dd�ejje
dd�dd � ���Z d
d � Z d d
� Z
dd� ZdS )�TestMemLeakClassc s8 � fdd�}ddi� | j |ddd� � d dks4J �d S ) Nc s � d d7 < d S )N�cntr+ r2 r2 �r� r2 r5 �fun| s z(TestMemLeakClass.test_times.<locals>.funr� r �
� )�times� warmup_times� )�execute)r< r� r2 r� r5 �
test_timesz s zTestMemLeakClass.test_timesc C s2 t �t��" | jdd� dd� W d � n1 s20 Y t �t��" | jdd� dd� W d � n1 sn0 Y t �t��" | jdd� dd� W d � n1 s�0 Y t �t��" | jd d� dd
� W d � n1 s�0 Y t �t��" | jd d� dd � W d � n1 �s$0 Y d S )
Nc S s dS �Nr r2 r2 r2 r2 r5 rm � rn z1TestMemLeakClass.test_param_err.<locals>.<lambda>r �r� c S s dS r� r2 r2 r2 r2 r5 rm � rn ���c S s dS r� r2 r2 r2 r2 r5 rm � rn )r� c S s dS r� r2 r2 r2 r2 r5 rm � rn )� tolerancec S s dS r� r2 r2 r2 r2 r5 rm � rn )r- )r r@ rF r� ro r2 r2 r5 �test_param_err� s 0 0 0 0 z TestMemLeakClass.test_param_errz
skipped on CIr� z skipped during test coveragec C s\ g }|fdd�}zBt jtdd�� | j|dd� W d � n1 sD0 Y W ~n~0 d S )Nc S s | � dd � d S )NZ�xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� ��append��lsr2 r2 r5 r� � s z+TestMemLeakClass.test_leak_mem.<locals>.funz extra-mem��match�d r� )r r@ �AssertionErrorr� )r< r� r� r2 r2 r5 �
test_leak_mem� s .z TestMemLeakClass.test_leak_memc s^ � �fdd�}g � t rdnd}tjtd| d�� ��|� W d � n1 sP0 Y d S )Nc s" t t�} ��| j� � �| � d S rD )r] rs Z
addCleanupr| r� )rc �Zboxr< r2 r5 r� � s z1TestMemLeakClass.test_unclosed_files.<locals>.fun�fd�handlez unclosed r� )r r r@ r� r� )r< r� r� r2 r� r5 �test_unclosed_files� s
z$TestMemLeakClass.test_unclosed_filesc s> � fdd�}g � d}| j ||ddd� t� �|d ks:J �d S )Nc s � � dd � d S )NZxxxxxxxxxxxxxxxxxxxxxxxxr� r� r2 r� r2 r5 r� � s z,TestMemLeakClass.test_tolerance.<locals>.funr� r i � )r� r� r� r+ )r� r� )r< r� r� r2 r� r5 �test_tolerance� s �z TestMemLeakClass.test_tolerancec C s� dd� }| � t|� t�t�� | � t|� W d � n1 s@0 Y dd� }t�t�� | � t|� W d � n1 s~0 Y d S )Nc S s dd d S r/ r2 r2 r2 r2 r5 �fun_1� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_1c S s d S rD r2 r2 r2 r2 r5 �fun_2� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_2)Z
execute_w_excrA r r@ r~ r� )r< r� r� r2 r2 r5 �test_execute_w_exc� s * z#TestMemLeakClass.test_execute_w_excN)rM rN rO r r� r� r r� r� r r
r� r� r� r� r2 r2 r2 r5 r� x s
r� c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zejj e
d d
�d d � �Z d
d� Z dd� Z
dS )�TestFakePytestc C s( t �� }|�|� t �� }|�|�}|S rD )�unittestZ TestSuiteZaddTestZTextTestRunner�run)r< �klass�suite�runner�resultr2 r2 r5 �run_test_class� s
z TestFakePytest.run_test_classc
C s� t �t��}dd W d � n1 s(0 Y t|jt�sBJ �t jtdd��}td��W d � n1 sn0 Y z:t jtdd��}td��W d � n1 s�0 Y W n4 ty� } z t|�dks�J �W Y d }~nd }~0 0 | �d��d S )Nr+ r r6 r� �barz"foo" does not match "bar"�exception not raised) r r@ rA �
isinstance�valuerF r� �str�fail)r< �cm�errr2 r2 r5 � test_raises� s &&*&zTestFakePytest.test_raisesc C sV t jjdd�dd� �}|� dks$J �t jjdd�G dd� d��}|� �� dksRJ �d S )Nr� r� c S s dS rl r2 r2 r2 r2 r5 r6 � s z%TestFakePytest.test_mark.<locals>.foor+ c @ s e Zd Zdd� ZdS )z%TestFakePytest.test_mark.<locals>.Fooc S s dS rl r2 ro r2 r2 r5 r� � s z)TestFakePytest.test_mark.<locals>.Foo.barN)rM rN rO r� r2 r2 r2 r5 �Foo� s r )r r� � xdist_groupr� )r< r6 r r2 r2 r5 � test_mark� s
zTestFakePytest.test_markc C s� G dd� dt j�}| �|d��}|�� s,J �t|j�dks>J �|jd d dksTJ �G dd� dt j�}| �|d��}|�� s�J �t|j�dks�J �d S )Nc @ s$ e Zd Zejjddd�dd� �ZdS )�,TestFakePytest.test_skipif.<locals>.TestCaseTr� r� c S s ddks J �d S rl r2 ro r2 r2 r5 r6 � s �0TestFakePytest.test_skipif.<locals>.TestCase.fooN�rM rN rO r r� r� r6 r2 r2 r2 r5 �TestCase� s r r6 r+ r r� c @ s$ e Zd Zejjddd�dd� �ZdS )r Fr� r� c S s ddks J �d S rl r2 ro r2 r2 r5 r6 � s r
Nr r2 r2 r2 r5 r � s �r� r r� Z
wasSuccessfulr� �skipped�r< r r� r2 r2 r5 � test_skipif� s zTestFakePytest.test_skipifznot PY3r� c C sX G dd� dt j�}| �|d��}|�� s,J �t|j�dks>J �|jd d dksTJ �d S )Nc @ s e Zd Zdd� ZdS )z*TestFakePytest.test_skip.<locals>.TestCasec S s t �d� ddksJ �d S )Nr� r+ r )r �skipro r2 r2 r5 r6 s
z.TestFakePytest.test_skip.<locals>.TestCase.fooN)rM rN rO r6 r2 r2 r2 r5 r s r r6 r+ r r� r
r r2 r2 r5 � test_skip� s
zTestFakePytest.test_skipc C s | j td�}t�|� ttj�|d�d�� W d � n1 s@0 Y ttj�|d�d��$}|�t� d��
� � W d � n1 s�0 Y t j �
tjd|��T | �td��( t�� }|�� dks�J �W d � n1 s�0 Y W d � n1 s�0 Y d S ) N)�dirz __init__.pyrZ z test_file.pyz� import unittest
class TestCase(unittest.TestCase):
def test_passed(self):
pass
r zFake pytest module was usedr+ )r\ r rT r} r] r_ r� r^ �textwrap�dedent�lstripr rP �objectrW r� ZassertWarnsRegex� UserWarningr �mainZcountTestCases)r< Ztmpdirrc r� r2 r2 r5 � test_main
s
2�zTestFakePytest.test_mainc C s< t �t�� tjdtdd� W d � n1 s00 Y z>t �t�� tjdtdd� W d � n1 sl0 Y W n ty� Y n 0 | �d��t jtdd�� tjdtdd� W d � n1 s�0 Y zDt jtdd�� tjdtdd� W d � n1 �s 0 Y W n t�y, Y n 0 | �d��d S )Nr6 r+ )�
stacklevelr� r� r� )r Zwarnsr �warnings�warn�DeprecationWarningr� r ro r2 r2 r5 �
test_warns s . 2
.4zTestFakePytest.test_warnsN)rM rN rO r� r r r r r� r� r r r r r2 r2 r2 r5 r� � s
r� c @ s e Zd Zdd� Zdd� ZdS )�TestTestingUtilsc C sL t �� }t|�}|�� dd� |�|j�D �d d }|� |�� ksHJ �d S )Nc S s g | ]}|d dkr|�qS )r+ r� r2 ��.0�xr2 r2 r5 �
<listcomp>? rn z;TestTestingUtils.test_process_namespace.<locals>.<listcomp>r )rW r� r �testrX �gettersr� )r< r� �nsr� r2 r2 r5 �test_process_namespace; s
z'TestTestingUtils.test_process_namespacec C s: t � }dd� |�|j�D �d d }|� t�� ks6J �d S )Nc S s g | ]}|d dkr|�qS )r+ � net_if_addrsr2 r! r2 r2 r5 r$ D rn z:TestTestingUtils.test_system_namespace.<locals>.<listcomp>r )r# rX r& rW r) )r< r' r� r2 r2 r5 �test_system_namespaceB s z&TestTestingUtils.test_system_namespaceN)rM rN rO r( r* r2 r2 r2 r5 r : s r c @ s e Zd Zdd� ZdS )�TestOtherUtilsc C s. t t�dd�ddd��s J �t t� �r*J �d S )Nr6 za b cr+ r� r7 )r r� �
namedtuple�tuplero r2 r2 r5 �test_is_namedtupleI s z!TestOtherUtils.test_is_namedtupleN)rM rN rO r. r2 r2 r2 r5 r+ H s r+ )@�__doc__r� r� r rT r� r� r� r r� r rW Z psutil.testsr r r Zpsutil._commonr r r Zpsutil._compatr 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$ r% r& r' r( r) rQ rq r� r� r� r r� r� r r+ r2 r2 r2 r5 �<module> sx E(-LN Kv