Advanced Web Server Manager
Complete File Manager & Terminal - Standalone Version
By Sid Gifari | Gifari Industries
Current path:
/
/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
psutil
/
tests
/
__pycache__
✏️
Editing: test_testutils.cpython-36.pyc
3 �̪e�9 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z ddl m Z ddlmZ ddlmZ ddlmZ dd l mZ dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ ddl 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$ 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 m-Z- dd%l m.Z. G d&d'� d'e�Z/G d(d)� d)e�Z0G d*d+� d+e�Z1G d,d-� d-e�Z2G d.d/� d/e�Z3e(G d0d1� d1e��Z4G d2d3� d3e�Z5G d4d5� d5e�Z6e7d6k�r�dd7l8m9Z9 e9e:� dS )8z1Tests for testing utils (psutil.tests namespace).� N)�FREEBSD)�NETBSD)�POSIX)� open_binary)� open_text)� supports_ipv6)� CI_TESTING)�COVERAGE)�HAS_CONNECTIONS_UNIX)� PYTHON_EXE)�PYTHON_EXE_ENV)�PsutilTestCase)�TestMemoryLeak)� bind_socket)�bind_unix_socket)� call_until)�chdir)�create_sockets)�filter_proc_connections)� get_free_port)� is_namedtuple)�mock)�process_namespace)� reap_children)�retry)�retry_on_failure)� safe_mkdir)� safe_rmpath)� serialrun)�system_namespace)�tcp_socketpair)� terminate)�unix_socketpair)� wait_for_file)� wait_for_pidc @ sx e Zd Zejd�dd� �Zejd�dd� �Zejd�dd� �Zejd�dd � �Zejd�d d � �Z ejd�d d � �Z dS )�TestRetryDecoratorz time.sleepc sF t ddd d�� fdd��}ttd��� | j|� d� | j|jd� d S )N� � )�retries�interval�logfunc s x� r� j � dd qW dS )Nr' r )�pop� )�queuer, �G/usr/local/lib64/python3.6/site-packages/psutil/tests/test_testutils.py�fooD 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��� | jt|� | j|jd� d S )Nr& r' )r( r) r* c s x� r� j � dd qW dS )Nr' r )r+ r, )r- r, r. r/ R s z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r r1 r2 � assertRaises�ZeroDivisionErrorr3 r4 )r5 r6 r/ r, )r- r. �test_retry_failureO s z%TestRetryDecorator.test_retry_failurec C s2 t tdd�dd� �}| jt|� | j|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 � ValueErrorr9 r= r3 r4 )r5 r6 r/ r, r, r. �test_exception_arg] s z%TestRetryDecorator.test_exception_argc C s4 t dd d d�dd� �}| jt|� | j|jd� d S )Nr& )r( r) r* c S s dd d S )Nr' r r, r, r, r, r. r/ j s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r r9 r: r3 r4 )r5 r6 r/ r, r, r. �test_no_interval_argf s z'TestRetryDecorator.test_no_interval_argc C s4 t ddd d�dd� �}| jt|� | j|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 r9 r: r3 r4 )r5 r6 r/ r, r, r. �test_retries_argq s z#TestRetryDecorator.test_retries_argc C s | j ttddd� d S )Nr& r' )r( �timeout)r9 r> r )r5 r6 r, r, r. � test_retries_and_timeout_argsz s z0TestRetryDecorator.test_retries_and_timeout_argsN) �__name__� __module__� __qualname__r �patchr7 r; r? r@ rA rC 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 tj� � ttj� �d }tjdtdg�d�� | j tj t |� W d Q R X d S )Ni�� zpsutil.tests.retry.__iter__r )� return_value) r$ �os�getpid�max�psutilZpidsr rG �iterr9 � NoSuchProcess)r5 Znopidr, r, r. �test_wait_for_pid� s z#TestSyncTestUtils.test_wait_for_pidc C sF | j � }t|d��}|jd� W d Q R X t|� tjj|� sBt�d S )N�wr/ )� get_testfn�open�writer# rJ �path�exists�AssertionError)r5 �testfn�fr, r, r. �test_wait_for_file� s z$TestSyncTestUtils.test_wait_for_filec C s@ | j � }t|d�� W d Q R X t|dd� tjj|� s<t�d S )NrQ T)�empty)rR rS r# rJ rU rV rW )r5 rX r, r, r. �test_wait_for_file_empty� s z*TestSyncTestUtils.test_wait_for_file_emptyc C s: | j � }tjdtdg�d�� | jtt|� W d Q R X d S )Nzpsutil.tests.retry.__iter__r )rI )rR r rG rN r9 �IOErrorr# )r5 rX r, r, r. �test_wait_for_file_no_file� s z,TestSyncTestUtils.test_wait_for_file_no_filec C sH | j � }t|d��}|jd� W d Q R X t|dd� tjj|�sDt�d S )NrQ r/ F)�delete)rR rS rT r# rJ rU rV rW )r5 rX rY r, r, r. � test_wait_for_file_no_delete� s z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd� d�}| j|d� d S )Nc S s dS )Nr' r, r, r, r, r. �<lambda>� s z3TestSyncTestUtils.test_call_until.<locals>.<lambda>zret == 1r' )r r3 )r5 �retr, r, r. �test_call_until� s z!TestSyncTestUtils.test_call_untilN) rD rE rF rP rZ r\ r^ r` rc r, r, r, r. rH s rH c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestFSTestUtilsc C s& t t��}| j|jd� W d Q R X d S )N�r)r �__file__r3 �mode)r5 rY r, r, r. �test_open_text� s z TestFSTestUtils.test_open_textc C s& t t��}| j|jd� W d Q R X d S )N�rb)r rf r3 rg )r5 rY r, r, r. �test_open_binary� s z TestFSTestUtils.test_open_binaryc C s< | j � }t|� tjj|�s t�t|� tjj|�s8t�d S )N)rR r rJ rU �isdirrW )r5 rX r, r, r. �test_safe_mkdir� s z TestFSTestUtils.test_safe_mkdirc C s� | j � }t|d�j� t|� tjj|� s0t�t|� tj|� t|� tjj|� s\t�t j dt t j d�d��.}| jt �� t|� W d Q R X |js�t�W d Q R X d S )NrQ zpsutil.tests.os.stat� )� side_effect)rR rS �closer rJ rU rV rW �mkdirr rG �OSError�errno�EINVALr9 �called)r5 rX �mr, r, r. �test_safe_rmpath� s z TestFSTestUtils.test_safe_rmpathc C s\ | j � }tj� }tj|� t|�� | jtj� tjj||�� W d Q R X | jtj� |� d S )N)rR rJ �getcwdrp r r3 rU �join)r5 rX �baser, r, r. � test_chdir� s $zTestFSTestUtils.test_chdirN)rD rE rF rh rj rl rv rz r, r, r, r. rd � s rd c @ s: e Zd Zdd� Zdd� Zeje d�dd� �Zdd � Z d S ) �TestProcessUtilsc C sT | j � }tj|j�}|j� s t�t� |j� s4t�tjj sBt�tjj sPt�d S )N) �spawn_testprocrM �Process�pid� is_runningrW r �testsZ _pids_startedZ_subprocesses_started)r5 Zsubp�pr, r, r. �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� | j � \}}| j|j|j� |j� s(t�|j� s4t�tj� j� }| j||g� tj� jdd�}| jt |�d� | j ||� | j ||� | j|j � t j � � | j|j � |j� t|� |j� s�t�|j� s�t�t|� |j� s�t�d S )NT)� recursive� )Zspawn_children_pair�assertNotEqualr~ r rW rM r} �childrenr3 �lenZassertIn�ppidrJ rK r! )r5 �childZ grandchildr� r, r, r. �test_spawn_children_pair� s" z)TestProcessUtils.test_spawn_children_pairz POSIX onlyc C s" | j � \}}| j|j� tj� d S )N)� spawn_zombier3 �statusrM � STATUS_ZOMBIE)r5 �parent�zombier, r, r. �test_spawn_zombie� s z"TestProcessUtils.test_spawn_zombiec C s� | j � }t|� | j|j� t|� tj| j � j�}t|� | j|j� t|� tddg}tj|tj tj t d�}t|� | j|j� t|� | j � j}t|� | j|j� t|� t r�| j � \}}t|� t|� | j|j� | j|j� d S )Nz-cz import time; time.sleep(60);)�stdout�stderr�env) r| r! Z assertPidGoner~ rM r} r �Popen� subprocess�PIPEr r r� )r5 r� �cmdr~ r� r� r, r, r. �test_terminate� s6 z TestProcessUtils.test_terminateN) rD rE rF r� r� �unittest�skipIfr r� r� r, r, r, r. r{ � s r{ c @ s` e Zd Zdd� Zeje d�dd� �Zdd� Zeje d�eje pDe d�d d � ��Z d d � Z d S )� TestNetUtilsc C s> t � }tjtd|fd���}| j|j� d |� W d Q R X d S )Nrm )�addrr' )r � contextlib�closingr r3 � getsockname)r5 �port�sr, r, r. r s zTestNetUtils.bind_socketz POSIX onlyc C s� | j � }t|�}tj|��\ | j|jtj� | j|jtj � | j|j � |� t j j |�s\t�tjt j|�j�srt�W d Q R X | j � }t|tjd�}tj|�� | j|jtj� W d Q R X d S )N)�type)rR r r� r� r3 �family�socket�AF_UNIXr� � SOCK_STREAMr� rJ rU rV rW �stat�S_ISSOCK�st_mode� SOCK_DGRAM)r5 �name�sockr, r, r. �test_bind_unix_socket s z"TestNetUtils.test_bind_unix_socketc C s| dt � f}ttj|d�\}}tj|��L tj|��6 | j|j� |� | j|j� |� | j |j� |� W d Q R X W d Q R X d S )Nz 127.0.0.1)r� ) r r r� �AF_INETr� r� r3 r� � getpeernamer� )r5 r� �server�clientr, r, r. �tcp_tcp_socketpair0 s z TestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C s� t j� }|j� }| jt|jdd��g � | j� }t|�\}}zztj j |�sNt �t j tj |�j�sdt �| j|j� | d� | jtt|jdd���d� | j|j� |� | j|j� |� W d |j� |j� X d S )N�unix)�kindr� )rM r} �num_fdsr3 r Z connectionsrR r" rJ rU rV rW r� r� r� r� r� r� ro )r5 r� r� r� r� r� r, r, r. �test_unix_socketpair; s z!TestNetUtils.test_unix_socketpairc C s� t � ��}tjt�}tjt�}x:|D ]2}||j d7 < ||jtjtj� d7 < q"W | j |tj d� t � r�| j |tj d� t r�tr�| j |tj d� | j |tj d� | j |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� )r5 ZsocksZfams�typesr� r, r, r. �test_create_socketsT s z TestNetUtils.test_create_socketsN) rD rE rF r r� r� r r� r� r r r� r� r, r, r, r. r� s r� c @ s` e Zd Ze� dd� �Zdd� Ze� ejed�eje d�dd� ���Z d d � Z d d � Z d d� Z dS )�TestMemLeakClassc s8 � fdd�}ddi� | j |ddd� | j� d d� d S ) Nc s � d d7 < d S )N�cntr' r, r, )r� r, r. �funi s z(TestMemLeakClass.test_times.<locals>.funr� r � � )�times� warmup_times� )�executer3 )r5 r� r, )r� r. � test_timesg s zTestMemLeakClass.test_timesc C s| | j t| jdd� dd� | j t| jdd� d d� | 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. ra q s z1TestMemLeakClass.test_param_err.<locals>.<lambda>r )r� c S s dS )Nr r, r, r, r, r. ra r s r' c S s dS )Nr r, r, r, r, r. ra s s )r� c S s dS )Nr r, r, r, r, r. ra t s )� tolerancec S s dS )Nr r, r, r, r, r. ra u s )r( ���r� r� r� )r9 r> r� )r5 r, r, r. �test_param_errp s z TestMemLeakClass.test_param_errz skipped on CIz skipped during test coveragec C s4 g }|fdd�}z| j td| j|dd� W d ~X d S )Nc S s | j dd d � d S )N�x�| i )�append)�lsr, r, r. r� } s z+TestMemLeakClass.test_leak_mem.<locals>.funz extra-mem�2 )r� )�assertRaisesRegexrW r� )r5 r� r� r, r, r. � test_leak_memw s z TestMemLeakClass.test_leak_memc s8 � �fdd�}g � t rdnd}�jtd| �j|� d S )Nc s" t t�} �j| j� � j| � d S )N)rS rf Z addCleanupro r� )rY )�boxr5 r, r. r� � s z1TestMemLeakClass.test_unclosed_files.<locals>.fun�fd�handlez unclosed )r r� rW r� )r5 r� r� r, )r� r5 r. �test_unclosed_files� s z$TestMemLeakClass.test_unclosed_filesc s> � fdd�}g � d}| j ||dd d� | jt� �|d � d S ) Nc s � j dd d � d S )Nr� � i )r� r, )r� r, r. r� � s z,TestMemLeakClass.test_tolerance.<locals>.fun�d r �� i )r� r� r� r' i i � )r� r3 r� )r5 r� r� r, )r� r. �test_tolerance� s z TestMemLeakClass.test_tolerancec C sd dd� }| j t|� | jt�� | j t|� W d Q R X dd� }| jt�� | j t|� W d Q R X d S )Nc S s dd d S )Nr' r r, r, r, r, r. �fun_1� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_1c S s d S )Nr, r, r, r, r. �fun_2� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_2)Z execute_w_excr: r9 rq rW )r5 r� r� r, r, r. �test_execute_w_exc� s z#TestMemLeakClass.test_execute_w_excN)rD rE rF r r� r� r� r� r r r� r� r� r� r, r, r, r. r� e s r� c @ s e Zd Zdd� Zdd� ZdS )�TestTestingUtilsc C sL t j� }t|�}|j� dd� |j|j�D �d d }| j|� |j� � d S )Nc S s g | ]}|d dkr|�qS )r' r� r, )�.0r� r, r, r. � <listcomp>� s z;TestTestingUtils.test_process_namespace.<locals>.<listcomp>r )rM r} r �testrN �gettersr3 r� )r5 r� �nsr� r, r, r. �test_process_namespace� s z'TestTestingUtils.test_process_namespacec C s: t � }dd� |j|j�D �d d }| j|� tj� � 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 rN r� r3 rM r� )r5 r� r� r, r, r. �test_system_namespace� s z&TestTestingUtils.test_system_namespaceN)rD rE rF r� r� r, r, r, r. r� � s r� c @ s e Zd Zdd� ZdS )�TestOtherUtilsc C s0 t tjdd�ddd��s t�t t� � s,t�d S )Nr/ za b cr' r� r0 )r r� � namedtuplerW �tuple)r5 r, r, r. �test_is_namedtuple� s z!TestOtherUtils.test_is_namedtupleN)rD rE rF r� r, r, r, r. r� � s r� �__main__)� run_from_name);�__doc__r� r� rr rJ r� r� r� r� rM Z psutil.testsr r r Zpsutil._commonr 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% rH rd r{ r� r� r� r� rD Zpsutil.tests.runnerr� rf r, r, r, r. �<module> sr @&-HK H
💾 Save Changes
❌ Cancel