Edit File: test_testutils.cpython-37.pyc
B xe�_T8 � @ s~ d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl m Z ddl mZ ddl mZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z 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) dd"lm*Z* dd#lm+Z+ ddlZddlZG d$d%� 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 �Z0e#G d.d/� d/e'��Z1G d0d1� d1e �Z2G d2d3� d3e �Z3e4d4k�rzdd5l5m6Z6 e6e7� dS )6z3 Tests for testing utils (psutil.tests namespace). � N)�FREEBSD)�NETBSD)�POSIX)� open_binary)� open_text)� supports_ipv6)� bind_socket)�bind_unix_socket)� call_until)�chdir)� CI_TESTING)�create_sockets)� get_free_port)�HAS_CONNECTIONS_UNIX)� is_namedtuple)�mock)�process_namespace)�PsutilTestCase)� PYTHON_EXE)� reap_children)�retry)�retry_on_failure)� safe_mkdir)� safe_rmpath)� serialrun)�system_namespace)�tcp_socketpair)� terminate)�TestMemoryLeak)�unittest)�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��� | �|� d� | �|jd� d S )N� � )�retries�interval�logfunc s x� r� � � dd qW dS )Nr% r )�pop� )�queuer* �N/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_testutils.py�fooC s z2TestRetryDecorator.test_retry_success.<locals>.foo� )r �list�range� assertEqual� call_count)�self�sleepr- r* )r+ r, �test_retry_success? s z%TestRetryDecorator.test_retry_successc sD t ddd d�� fdd��}ttd��� | �t|� | �|jd� d S )Nr$ r% )r& r' r( c s x� r� � � dd qW dS )Nr% r )r) r* )r+ r* r, r- Q s z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r r/ r0 � assertRaises�ZeroDivisionErrorr1 r2 )r3 r4 r- r* )r+ r, �test_retry_failureN s z%TestRetryDecorator.test_retry_failurec C s2 t tdd�dd� �}| �t|� | �|jd� d S )Nr% )� exceptionr' c S s t �d S )N)� TypeErrorr* r* r* r, r- ^ s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r � ValueErrorr7 r; r1 r2 )r3 r4 r- r* r* r, �test_exception_arg\ s z%TestRetryDecorator.test_exception_argc C s4 t dd d d�dd� �}| �t|� | �|jd� d S )Nr$ )r& r' r( c S s dd d S )Nr% r r* r* r* r* r, r- i s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r r7 r8 r1 r2 )r3 r4 r- r* r* r, �test_no_interval_arge s z'TestRetryDecorator.test_no_interval_argc C s4 t ddd d�dd� �}| �t|� | �|jd� d S )Nr$ r% )r& r' r( c S s dd d S )Nr% r r* r* r* r* r, r- s s z0TestRetryDecorator.test_retries_arg.<locals>.foo)r r7 r8 r1 r2 )r3 r4 r- r* r* r, �test_retries_argp s z#TestRetryDecorator.test_retries_argc C s | j ttddd� d S )Nr$ r% )r& �timeout)r7 r< r )r3 r4 r* r* r, � test_retries_and_timeout_argsz s z0TestRetryDecorator.test_retries_and_timeout_argsN) �__name__� __module__� __qualname__r �patchr5 r9 r= r> r? rA r* r* r* r, r# = s r# c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestSyncTestUtilsc C sP t t�� � tt�� �d }tjdtdg�d�� | � tj t |� W d Q R X d S )Ni�� zpsutil.tests.retry.__iter__r )� return_value) r" �os�getpid�max�psutilZpidsr rE �iterr7 � NoSuchProcess)r3 Znopidr* r* r, �test_wait_for_pid� s z#TestSyncTestUtils.test_wait_for_pidc C sD | � � }t|d��}|�d� W d Q R X t|� tj�|�r@t�d S )N�wr- )� get_testfn�open�writer! rH �path�exists�AssertionError)r3 �testfn�fr* r* r, �test_wait_for_file� s z$TestSyncTestUtils.test_wait_for_filec C s> | � � }t|d�� W d Q R X t|dd� tj�|�r:t�d S )NrO T)�empty)rP rQ r! rH rS rT rU )r3 rV r* r* r, �test_wait_for_file_empty� s z*TestSyncTestUtils.test_wait_for_file_emptyc C s: | � � }tjdtdg�d�� | �tt|� W d Q R X d S )Nzpsutil.tests.retry.__iter__r )rG )rP r rE rL r7 �IOErrorr! )r3 rV r* r* r, �test_wait_for_file_no_file� s z,TestSyncTestUtils.test_wait_for_file_no_filec C sH | � � }t|d��}|�d� W d Q R X t|dd� tj�|�sDt�d S )NrO r- F)�delete)rP rQ rR r! rH rS rT rU )r3 rV rW r* r* r, � test_wait_for_file_no_delete� s z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd� d�}| �|d� d S )Nc S s dS )Nr% r* r* r* r* r, �<lambda>� � z3TestSyncTestUtils.test_call_until.<locals>.<lambda>zret == 1r% )r r1 )r3 �retr* r* r, �test_call_until� s z!TestSyncTestUtils.test_call_untilN) rB rC rD rN rX rZ r\ r^ rb r* r* r* r, rF s rF c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestFSTestUtilsc C s& t t��}| �|jd� W d Q R X d S )N�rt)r �__file__r1 �mode)r3 rW r* r* r, �test_open_text� s z TestFSTestUtils.test_open_textc C s& t t��}| �|jd� W d Q R X d S )N�rb)r re r1 rf )r3 rW r* r* r, �test_open_binary� s z TestFSTestUtils.test_open_binaryc C s< | � � }t|� tj�|�s t�t|� tj�|�s8t�d S )N)rP r rH rS �isdirrU )r3 rV r* r* r, �test_safe_mkdir� s z TestFSTestUtils.test_safe_mkdirc C s� | � � }t|d��� t|� tj�|�r.t�t|� t�|� t|� tj�|�rXt�t j dt t j d�d��.}| �t �� t|� W d Q R X |js�t�W d Q R X d S )NrO zpsutil.tests.os.stat� )� side_effect)rP rQ �closer rH rS rT rU �mkdirr rE �OSError�errno�EINVALr7 �called)r3 rV �mr* r* r, �test_safe_rmpath� s z TestFSTestUtils.test_safe_rmpathc C s\ | � � }t�� }t�|� t|�� | �t�� tj�||�� W d Q R X | �t�� |� d S )N)rP rH �getcwdro r r1 rS �join)r3 rV �baser* r* r, � test_chdir� s $zTestFSTestUtils.test_chdirN)rB rC rD rg ri rk ru ry r* r* r* r, rc � s rc c @ s: e Zd Zdd� Zdd� Ze�e d�dd� �Zdd � Z d S ) �TestProcessUtilsc C sN | � � }t�|j�}|�� s t�t� |�� r2t�tjjr>t�tjj rJt�d S )N) �spawn_testprocrK �Process�pid� is_runningrU r �testsZ _pids_startedZ_subprocesses_started)r3 Zsubp�pr* r* r, �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� | � � \}}| �|j|j� |�� s(t�|�� s4t�t�� �� }| �||g� t�� jdd�}| �t |�d� | � ||� | � ||� | �|� � t � � � | �|� � |j� t|� |�� r�t�|�� s�t�t|� |�� r�t�d S )NT)� recursive� )Zspawn_children_pair�assertNotEqualr} r~ rU rK r| �childrenr1 �lenZassertIn�ppidrH rI r )r3 �childZ grandchildr� r* r* r, �test_spawn_children_pair� s" z)TestProcessUtils.test_spawn_children_pairz POSIX onlyc C s" | � � \}}| �|�� tj� d S )N)� spawn_zombier1 �statusrK � STATUS_ZOMBIE)r3 �parent�zombier* r* r, �test_spawn_zombie� s z"TestProcessUtils.test_spawn_zombiec C s� | � � }t|� | �|� t|� t�| � � j�}t|� | �|� t|� tddg}tj|tj tj d�}t|� | �|� t|� | � � j}t|� | �|� t|� t r�| � � \}}t|� t|� | �|� | �|� d S )Nz-cz import time; time.sleep(60);)�stdout�stderr) r{ r ZassertProcessGonerK r| r} r �Popen� subprocess�PIPEr r� )r3 r� �cmdr} r� r� r* r* r, �test_terminate� s. z TestProcessUtils.test_terminateN) rB rC rD r� r� r �skipIfr r� r� r* r* r* r, rz � s rz c @ s` e Zd Zdd� Ze�e d�dd� �Zdd� Ze�e d�e�e pDe d�d d � ��Z d d � Z d S )� TestNetUtilsc C s> t � }t�td|fd���}| �|�� d |� W d Q R X d S )Nrl )�addrr% )r � contextlib�closingr r1 � getsockname)r3 �port�sr* r* r, r s zTestNetUtils.bind_socketz POSIX onlyc C s� | � � }t|�}t�|��\ | �|jtj� | �|jtj � | �|� � |� t j � |�s\t�t�t �|�j�srt�W d Q R X | � � }t|tjd�}t�|�� | �|jtj� W d Q R X d S )N)�type)rP r r� r� r1 �family�socket�AF_UNIXr� � SOCK_STREAMr� rH rS rT rU �stat�S_ISSOCK�st_mode� SOCK_DGRAM)r3 �name�sockr* r* r, �test_bind_unix_socket s z"TestNetUtils.test_bind_unix_socketc C s| dt � f}ttj|d�\}}t�|��L t�|��6 | �|�� |� | �|�� |� | � |�� |� W d Q R X W d Q R X d S )Nz 127.0.0.1)r� ) r r r� �AF_INETr� r� r1 r� � getpeernamer� )r3 r� �server�clientr* r* r, �tcp_tcp_socketpair. s z TestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C s� t �� }|�� }|jdd�r t�| �� }t|�\}}zvtj� |�sFt�t � t� |�j �s\t�| � |�� | d� | � t|jdd��d� | � |�� |� | � |�� |� W d |�� |�� X d S )N�unix)�kindr� )rK r| �num_fdsZ connectionsrU rP r rH rS rT r� r� r� r1 r� r� r� rn )r3 r� r� r� r� r� r* r* r, �test_unix_socketpair9 s z!TestNetUtils.test_unix_socketpairc C s� t � ��}t�t�}t�t�}x:|D ]2}||j d7 < ||�tjtj� d7 < q"W | � |tj d� t � r�| � |tj d� t r�tr�| � |tj d� | � |tj d� | � |tj d� W d Q R X d S )Nr% r� )r � collections� defaultdict�intr� � getsockoptr� � SOL_SOCKET�SO_TYPEZassertGreaterEqualr� r �AF_INET6r r r� r� r� )r3 ZsocksZfams�typesr� r* r* r, �test_create_socketsM s z TestNetUtils.test_create_socketsN) rB rC rD r r r� r r� r� r r r� r� r* r* r* r, r� s r� c @ sN e Zd Zdd� Zdd� Ze� e�ed�dd� ��Z dd � Z d d � Z d d � Z dS )�TestMemLeakClassc s8 � fdd�}ddi� | j |ddd� | �� d d� d S ) Nc s � d d7 < d S )N�cntr% r* r* )r� r* r, �funb s z(TestMemLeakClass.test_times.<locals>.funr� r � � )�times� warmup_times� )�executer1 )r3 r� r* )r� r, � test_timesa s zTestMemLeakClass.test_timesc C s| | j t| jdd� dd� | j t| jdd� dd� | j t| jdd� dd� | j t| jd d� dd � | j t| jd d� dd � d S ) Nc S s dS )Nr r* r* r* r* r, r_ i r` z1TestMemLeakClass.test_param_err.<locals>.<lambda>r )r� c S s dS )Nr r* r* r* r* r, r_ j r` ���c S s dS )Nr r* r* r* r* r, r_ k r` )r� c S s dS )Nr r* r* r* r* r, r_ l r` )� tolerancec S s dS )Nr r* r* r* r* r, r_ m r` )r& )r7 r< r� )r3 r* r* r, �test_param_errh s z TestMemLeakClass.test_param_errz skipped on CIc C s4 g }|fdd�}z| j td| j|dd� W d ~X d S )Nc S s | � dd � d S )N�xxxxxxxxxxxxxxxxxxxxxxxxi )�append)�lsr* r* r, r� t s z+TestMemLeakClass.test_leak_mem.<locals>.funz extra-mem�2 )r� )�assertRaisesRegexrU r� )r3 r� r� r* r* r, � test_leak_memo s z TestMemLeakClass.test_leak_memc s8 � �fdd�}g � t rdnd}��td| �j|� d S )Nc s" t t�} ��| j� � �| � d S )N)rQ re Z addCleanuprn r� )rW )�boxr3 r* r, r� s z1TestMemLeakClass.test_unclosed_files.<locals>.fun�fd�handlez unclosed )r r� rU r� )r3 r� r� r* )r� r3 r, �test_unclosed_files~ s z$TestMemLeakClass.test_unclosed_filesc s> � fdd�}g � d}| j ||ddd� | �t� �|d � d S )Nc s � � dd � d S )Nr� i )r� r* )r� r* r, r� � s z,TestMemLeakClass.test_tolerance.<locals>.fun�d r i � )r� r� r� r% )r� r1 r� )r3 r� r� r* )r� r, �test_tolerance� s z TestMemLeakClass.test_tolerancec C sd dd� }| � t|� | �t�� | � t|� W d Q R X dd� }| �t�� | � t|� W d Q R X d S )Nc S s dd d S )Nr% r r* r* r* r* r, r� � s z0TestMemLeakClass.test_execute_w_exc.<locals>.func S s d S )Nr* r* r* r* r, r� � s )Z execute_w_excr8 r7 rp rU )r3 r� r* r* r, �test_execute_w_exc� s z#TestMemLeakClass.test_execute_w_excN) rB rC rD r� r� r r r� r r� r� r� r� r* r* r* r, r� ^ s r� c @ s e Zd Zdd� Zdd� ZdS )�TestTestingUtilsc C sL t �� }t|�}|�� dd� |�|j�D �d d }| �|� |�� � d S )Nc S s g | ]}|d dkr|�qS )r% r� r* )�.0�xr* r* r, � <listcomp>� s z;TestTestingUtils.test_process_namespace.<locals>.<listcomp>r )rK r| r �testrL �gettersr1 r� )r3 r� �nsr� r* r* r, �test_process_namespace� s z'TestTestingUtils.test_process_namespacec C s: t � }dd� |�|j�D �d d }| �|� t�� � d S )Nc S s g | ]}|d dkr|�qS )r% � net_if_addrsr* )r� r� r* r* r, r� � s z:TestTestingUtils.test_system_namespace.<locals>.<listcomp>r )r rL r� r1 rK r� )r3 r� r� r* r* r, �test_system_namespace� s z&TestTestingUtils.test_system_namespaceN)rB rC rD r� r� r* r* r* r, r� � s r� c @ s e Zd Zdd� ZdS )�TestOtherUtilsc C s. t t�dd�ddd��s t�t t� �r*t�d S )Nr- za b cr% r� r. )r r� � namedtuplerU �tuple)r3 r* r* r, �test_is_namedtuple� s z!TestOtherUtils.test_is_namedtupleN)rB rC rD r� r* r* r* r, r� � s r� �__main__)� run_from_name)8�__doc__r� r� rq rH r� r� r� rK r r r Zpsutil._commonr r 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 r r r r! r" r# rF rc rz r� r� r� r� rB Zpsutil.tests.runnerr� re r* r* r* r, �<module> sl B'-DG @