3
�[g�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 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��� |� dks4t�|jdksBt�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�fooI s z2TestRetryDecorator.test_retry_success.<locals>.foo� )r �list�range�AssertionError�
call_count)�self�sleepr2 r/ )r0 r1 �test_retry_successE s z%TestRetryDecorator.test_retry_successc sT t ddd d�� fdd��}ttd��� tjt�� |� W d Q R X |jdksPt�d S )Nr) r* )r+ r, r- c s x� r� j � dd qW dS )Nr* r )r. r/ )r0 r/ r1 r2 W s z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r r4 r5 r �raises�ZeroDivisionErrorr7 r6 )r8 r9 r2 r/ )r0 r1 �test_retry_failureT s
z%TestRetryDecorator.test_retry_failurec C sB t tdd�dd� �}tjt�� |� W d Q R X |jdks>t�d S )Nr* )� exceptionr, c S s t �d S )N)� TypeErrorr/ r/ r/ r1 r2 e s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r �
ValueErrorr r< r@ r7 r6 )r8 r9 r2 r/ r/ r1 �test_exception_argc s z%TestRetryDecorator.test_exception_argc C sD t dd d d�dd� �}tjt�� |� W d Q R X |jdks@t�d S )Nr) )r+ r, r- c S s dd d S )Nr* r r/ r/ r/ r/ r1 r2 q s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r r r< r= r7 r6 )r8 r9 r2 r/ r/ r1 �test_no_interval_argm s z'TestRetryDecorator.test_no_interval_argc C sD t ddd d�dd� �}tjt�� |� W d Q R X |jdks@t�d S )Nr) r* )r+ r, r- c S s dd d S )Nr* r r/ r/ r/ r/ r1 r2 { s z0TestRetryDecorator.test_retries_arg.<locals>.foo)r r r< r= r7 r6 )r8 r9 r2 r/ r/ r1 �test_retries_argy s z#TestRetryDecorator.test_retries_argc C s& t jt�� tddd� W d Q R X d S )Nr) r* )r+ �timeout)r r< rA r )r8 r9 r/ r/ r1 � test_retries_and_timeout_args� s z0TestRetryDecorator.test_retries_and_timeout_argsN) �__name__�
__module__� __qualname__r �patchr: r> rB rC rD rF r/ r/ r/ r1 r( D s
r( c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestSyncTestUtilsc C s` t tj� � ttj� �d }tjdtdg�d��& t j
tj �� t |� W d Q R X W d Q R X d S )Ni�� zpsutil.tests.retry.__iter__r )� return_value) r' �os�getpid�max�psutilZpidsr rJ �iterr r< �
NoSuchProcess)r8 Znopidr/ r/ r1 �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�wr2 )�
get_testfn�open�writer& rM �path�existsr6 )r8 �testfn�fr/ r/ r1 �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 )NrT T)�empty)rU rV r& rM rX rY r6 )r8 rZ r/ r/ r1 �test_wait_for_file_empty� s
z*TestSyncTestUtils.test_wait_for_file_emptyc C sJ | j � }tjdtdg�d��$ tjt�� t|� W d Q R X W d Q R X d S )Nzpsutil.tests.retry.__iter__r )rL )rU r rJ rQ r r< �IOErrorr& )r8 rZ r/ r/ r1 �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 )NrT r2 F)�delete)rU rV rW r& rM rX rY r6 )r8 rZ r[ r/ r/ r1 � 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* r/ r/ r/ r/ r1 �<lambda>� s z3TestSyncTestUtils.test_call_until.<locals>.<lambda>)r )r8 r/ r/ r1 �test_call_until� s z!TestSyncTestUtils.test_call_untilN) rG rH rI rS r\ r^ r` rb rd r/ r/ r/ r1 rK � s rK c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestFSTestUtilsc C s& t t��}|jdkst�W d Q R X d S )N�r)r �__file__�moder6 )r8 r[ r/ r/ r1 �test_open_text� s
z TestFSTestUtils.test_open_textc C s& t t��}|jdkst�W d Q R X d S )N�rb)r rg rh r6 )r8 r[ r/ r/ r1 �test_open_binary� s
z TestFSTestUtils.test_open_binaryc C s< | j � }t|� tjj|�s t�t|� tjj|�s8t�d S )N)rU r rM rX �isdirr6 )r8 rZ r/ r/ r1 �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��.}tjt �� t|� W d Q R X |js�t�W d Q R X d S )NrT zpsutil.tests.os.stat� )Z side_effect)rU rV �closer! rM rX rY r6 �mkdirr rJ �OSError�errno�EINVALr r< �called)r8 rZ �mr/ r/ r1 �test_safe_rmpath� s
z TestFSTestUtils.test_safe_rmpathc C s\ | j � }tj� }tj|� t|�� tj� tjj||�ks>t�W d Q R X tj� |ksXt�d S )N)rU rM �getcwdrp r rX �joinr6 )r8 rZ �baser/ r/ r1 �
test_chdir� s
$zTestFSTestUtils.test_chdirN)rG rH rI ri rk rm rv rz r/ r/ r/ r1 re � s
re c @ s> e Zd Zdd� Zdd� Zejje dd�dd� �Z d d
� Z
d S ) �TestProcessUtilsc C sT | j � }tj|j�}|j� s t�t� |j� s4t�tjj sBt�tjj sPt�d S )N)
�spawn_testprocrP �Process�pid�
is_runningr6 r �testsZ
_pids_startedZ_subprocesses_started)r8 �subp�pr/ r/ r1 �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� | j � \}}|j|jks t�|j� s(t�|j� s4t�tj� j� }||gksNt�tj� jdd�}t|�dksnt�||kszt�||ks�t�|j� t j
� ks�t�|j� |jks�t�t |� |j� s�t�|j� s�t�t |� |j� s�t�d S )NT)� recursive� ) Zspawn_children_pairr~ r6 r rP r} �children�len�ppidrM rN r$ )r8 �childZ
grandchildr� r/ r/ r1 �test_spawn_children_pair� s" z)TestProcessUtils.test_spawn_children_pairz
POSIX only)�reasonc C s" | j � \}}|j� tjks t�d S )N)� spawn_zombie�statusrP �
STATUS_ZOMBIEr6 )r8 Z_parent�zombier/ r/ r1 �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-cz3import time; [time.sleep(0.1) for x in range(100)];)�stdout�stderr�env)
r| r$ Z
assertPidGoner~ rP r} r
�Popen�
subprocess�PIPEr r r� )r8 r� �cmdr~ �parentr� r/ r/ r1 �test_terminate s:
z TestProcessUtils.test_terminateN) rG rH rI r� r� r �mark�skipifr r� r� r/ r/ r/ r1 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 s> t � }tjtd|fd���}|j� d |ks0t�W d Q R X d S )Nrn )�addrr* )r �
contextlib�closingr � getsocknamer6 )r8 �port�sr/ r/ r1 r + s zTestNetUtils.bind_socketz
POSIX only)r� c C s� | j � }t|�}tj|��\ |jtjks,t�|jtj ks<t�|j
� |ksLt�t j j
|�s\t�tjt j|�j�srt�W d Q R X | j � }t|tjd�}tj|�� |jtjks�t�W d Q R X d S )N)�type)rU r r� r� �family�socket�AF_UNIXr6 r� � SOCK_STREAMr� rM rX rY �stat�S_ISSOCK�st_mode�
SOCK_DGRAM)r8 �name�sockr/ r/ r1 �test_bind_unix_socket0 s z"TestNetUtils.test_bind_unix_socketc C s| dt � f}ttj|d�\}}tj|��L tj|��6 |j� |ksDt�|j� |ksTt�|j� |ksdt�W d Q R X W d Q R X d S )Nz 127.0.0.1)r� ) r r# r� �AF_INETr� r� r� r6 � getpeername)r8 r� �server�clientr/ r/ r1 �tcp_tcp_socketpair@ s
z TestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C s� t j� }|j� }t|jdd��g ks(t�| j� }t|�\}}zztj j
|�sNt�t j tj |�j
�sdt�|j� | dksxt�tt|jdd���dks�t�|j� |ks�t�|j� |ks�t�W d |j� |j� X d S )N�unix)�kindr� )rP r} �num_fdsr Znet_connectionsr6 rU r% rM rX rY r� r� r� r� r� r� ro )r8 r� r� r� r� r� r/ r/ r1 �test_unix_socketpairK s
z!TestNetUtils.test_unix_socketpairc C s� t � ��}tjt�}tjt�}x:|D ]2}||j d7 < ||jtjtj� d7 < q"W |tj dksjt
�t � r�|tj dks�t
�t
r�tr�|tj dks�t
�|tj dks�t
�|tj dks�t
�W d Q R X d S )Nr* r� )r � collections� defaultdict�intr� �
getsockoptr� �
SOL_SOCKET�SO_TYPEr� r6 r �AF_INET6r r r� r� r� )r8 ZsocksZfams�typesr� r/ r/ r1 �test_create_socketsg s
z TestNetUtils.test_create_socketsN)rG rH rI r r r� r� r r� r� r r r� r� r/ r/ r/ r1 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 dks4t�d S ) Nc s � d d7 < d S )N�cntr* r/ r/ )r� r/ r1 �fun| s z(TestMemLeakClass.test_times.<locals>.funr� r �
� )�times� warmup_times� )�executer6 )r8 r� r/ )r� r1 �
test_timesz s zTestMemLeakClass.test_timesc ' C s� t jt�� | jdd� dd� W d Q R X t jt�� | jdd� d
d� W d Q R X t jt�� | jdd� dd� W d Q R X t jt�� | jd d� dd
� W d Q R X t jt�� | jd d� dd � W d Q R X d S )Nc S s dS )Nr r/ r/ r/ r/ r1 rc � s z1TestMemLeakClass.test_param_err.<locals>.<lambda>r )r� c S s dS )Nr r/ r/ r/ r/ r1 rc � s r* c S s dS )Nr r/ r/ r/ r/ r1 rc � s )r� c S s dS )Nr r/ r/ r/ r/ r1 rc � s )� tolerancec S s dS )Nr r/ r/ r/ r/ r1 rc � s )r+ ���r� r� r� )r r< rA r� )r8 r/ r/ r1 �test_param_err� s z TestMemLeakClass.test_param_errz
skipped on CI)r� z skipped during test coveragec C sF g }|fdd�}z,t jtdd�� | j|dd� W d Q R X W d ~X d S )Nc S s | j dd d � d S )N�x� i )�append)�lsr/ r/ r1 r� � s z+TestMemLeakClass.test_leak_mem.<locals>.funz extra-mem)�match�d )r� )r r< r6 r� )r8 r� r� r/ r/ r1 �
test_leak_mem� s z TestMemLeakClass.test_leak_memc sJ � �fdd�}g � t rdnd}tjtd| d�� �j|� W d Q R X d S )Nc s" t t�} �j| j� � j| � d S )N)rV rg Z
addCleanupro r� )r[ )�boxr8 r/ r1 r� � s z1TestMemLeakClass.test_unclosed_files.<locals>.fun�fd�handlez unclosed )r� )r r r< r6 r� )r8 r� r� r/ )r� r8 r1 �test_unclosed_files� s
z$TestMemLeakClass.test_unclosed_filesc s> � fdd�}g � d}| j ||dd
d� t� �|d ks:t�d S ) Nc s � j dd d � d S )Nr� � i )r� r/ )r� r/ r1 r� � s z,TestMemLeakClass.test_tolerance.<locals>.funr� r �� i )r� r� r� r* i i � )r� r� r6 )r8 r� r� r/ )r� r1 �test_tolerance� s z TestMemLeakClass.test_tolerancec C sd dd� }| j t|� tjt�� | j t|� W d Q R X dd� }tjt�� | j t|� W d Q R X d S )Nc S s dd d S )Nr* r r/ r/ r/ r/ r1 �fun_1� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_1c S s d S )Nr/ r/ r/ r/ r1 �fun_2� s z2TestMemLeakClass.test_execute_w_exc.<locals>.fun_2)Z
execute_w_excr= r r< rq r6 )r8 r� r� r/ r/ r1 �test_execute_w_exc� s z#TestMemLeakClass.test_execute_w_excN)rG rH rI r r� r� r r� r� r r
r� r� r� r� r/ r/ r/ r1 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 j� }|j|� t j� }|j|�}|S )N)�unittestZ TestSuiteZaddTestZTextTestRunner�run)r8 �klass�suiteZrunner�resultr/ r/ r1 �run_test_class� s
z TestFakePytest.run_test_classc C s� t jt��}dd W d Q R X t|jt�s.t�t jtdd��}td��W d Q R X y&t jtdd��}td��W d Q R X W n2 tk
r� } zt|�dks�t�W Y d d }~X n X | jd��d S )Nr* r r2 )r� �barz"foo" does not match "bar"zexception not raised) r r< r= �
isinstance�valuer6 rA �str�fail)r8 �cm�errr/ r/ r1 � test_raises� s "zTestFakePytest.test_raisesc C sV t jjdd�dd� �}|� dks$t�t jjdd�G dd� d��}|� j� dksRt�d S )Nr� )r� c S s dS )Nr* r/ r/ r/ r/ r1 r2 � s z%TestFakePytest.test_mark.<locals>.foor* c @ s e Zd Zdd� ZdS )z%TestFakePytest.test_mark.<locals>.Fooc S s dS )Nr* r/ )r8 r/ r/ r1 r� � s z)TestFakePytest.test_mark.<locals>.Foo.barN)rG rH rI r� r/ r/ r/ r1 �Foo� s r� )r r� � xdist_groupr6 r� )r8 r2 r� r/ r/ r1 � test_mark� s zTestFakePytest.test_markc C s� G dd� dt j�}| j|d��}|j� s,t�t|j�dks>t�|jd d dksTt�G dd� dt j�}| j|d��}|j� s�t�t|j�dks�t�d S )Nc @ s$ e Zd Zejjddd�dd� �ZdS )z,TestFakePytest.test_skipif.<locals>.TestCaseTr� )r� c S s ddks t �d S )Nr* )r6 )r8 r/ r/ r1 r2 � s z0TestFakePytest.test_skipif.<locals>.TestCase.fooN)rG rH rI r r� r� r2 r/ r/ r/ r1 �TestCase� s r� r2 r* r r� c @ s$ e Zd Zejjddd�dd� �ZdS )z,TestFakePytest.test_skipif.<locals>.TestCaseFr� )r� c S s ddks t �d S )Nr* )r6 )r8 r/ r/ r1 r2 � s z0TestFakePytest.test_skipif.<locals>.TestCase.fooN)rG rH rI r r� r� r2 r/ r/ r/ r1 r� � s )r� r� r� �
wasSuccessfulr6 r� �skipped)r8 r� r� r/ r/ r1 � test_skipif� s zTestFakePytest.test_skipifznot PY3)r� c C sX G dd� dt j�}| j|d��}|j� s,t�t|j�dks>t�|jd d dksTt�d S )Nc @ s e Zd Zdd� ZdS )z*TestFakePytest.test_skip.<locals>.TestCasec S s t jd� ddkst�d S )Nr� r* r )r �skipr6 )r8 r/ r/ r1 r2 s
z.TestFakePytest.test_skip.<locals>.TestCase.fooN)rG rH rI r2 r/ r/ r/ r1 r� s r� r2 r* r r� )r� r� r� r� r6 r� r� )r8 r� r� r/ r/ r1 � test_skip� s
zTestFakePytest.test_skipc C s� | j td�}tj|� ttjj|d�d�� W d Q R X ttjj|d�d��}|jtj d�j
� � W d Q R X t j j
tjd|��6 | jtd�� tj� }|j� dks�t�W d Q R X W d Q R X d S ) N)�dirz __init__.pyrT z test_file.pyz� import unittest
class TestCase(unittest.TestCase):
def test_passed(self):
pass
r zFake pytest module was usedr* )rU r rM rp rV rX rx rW �textwrap�dedent�lstripr rJ �objectrP r� ZassertWarnsRegex� UserWarningr �mainZcountTestCasesr6 )r8 Ztmpdirr[ r� r/ r/ r1 � test_main
s
zTestFakePytest.test_mainc - C s� t jt�� tjdtdd� W d Q R X y*t jt�� tjdtdd� W d Q R X W n tk
rd Y n X | jd��t jtdd�� tjdtdd� W d Q R X y.t jtdd�� tjdtdd� W d Q R X W n tk
r� Y n X | jd��d S )Nr2 r* )�
stacklevelzexception not raised)r� r� )r Zwarnsr� �warnings�warn�DeprecationWarningr6 r� )r8 r/ r/ r1 �
test_warns s
zTestFakePytest.test_warnsN)rG rH rI r� r� r� r� r r� r� r r� r� r r/ r/ r/ r1 r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestTestingUtilsc C sL t j� }t|�}|j� dd� |j|j�D �d d }|� |j� ksHt�d S )Nc S s g | ]}|d dkr|�qS )r* r� r/ )�.0r� r/ r/ r1 �
<listcomp>? s z;TestTestingUtils.test_process_namespace.<locals>.<listcomp>r )rP r} r �testrQ �gettersr� r6 )r8 r� �nsr� r/ r/ r1 �test_process_namespace; s
z'TestTestingUtils.test_process_namespacec C s: t � }dd� |j|j�D �d d }|� tj� ks6t�d S )Nc S s g | ]}|d dkr|�qS )r* � net_if_addrsr/ )r r� r/ r/ r1 r D s z:TestTestingUtils.test_system_namespace.<locals>.<listcomp>r )r" rQ r rP r r6 )r8 r
r� r/ r/ r1 �test_system_namespaceB s z&TestTestingUtils.test_system_namespaceN)rG rH rI r r
r/ r/ r/ r1 r : s r c @ s e Zd Zdd� ZdS )�TestOtherUtilsc C s0 t tjdd�ddd��s t�t t� � s,t�d S )Nr2 za b cr* r� r3 )r r� �
namedtupler6 �tuple)r8 r/ r/ r1 �test_is_namedtupleI s z!TestOtherUtils.test_is_namedtupleN)rG rH rI r r/ r/ r/ r1 r H s r )@�__doc__r� r� rr rM r� r� r� r� r� r rP 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( rK re r{ r� r� r� r� r� r r r/ r/ r/ r1 �<module> sx E(-LNKv |