관리-도구
편집 파일: test_misc.cpython-36.pyc
3 ��JZ�� � @ s| d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z 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. dd$lm/Z/ dd%lm0Z0 dd&lm1Z1 dd'lm2Z2 dd(lm3Z3 dd)lm4Z4 dd*lm5Z5 dd+lm6Z6 dd,lm7Z7 dd-lm8Z8 dd.lm9Z9 dd/lm:Z: dd0lm;Z; dd1lm<Z< ddl Z ddlZ G d2d3� d3e8j=�Z>ej?d4d5�Z@G d6d7� d7e8j=�ZAe8jBe6d8�e8jBe7�o�ejCjDe2� d9�G d:d;� d;e8j=���ZEG d<d=� d=e8j=�ZFG d>d?� d?e8j=�ZGG d@dA� dAe8j=�ZHG dBdC� dCe8j=�ZIG dDdE� dEe8j=�ZJG dFdG� dGe8j=�ZKeLdHk�rxe0eM� dS )Iz Miscellaneous tests. � N)�LINUX)�POSIX)�WINDOWS)�memoize)�memoize_when_activated)� supports_ipv6)� wrap_numbers)�PY3)�APPVEYOR)� bind_socket)�bind_unix_socket)� call_until)�chdir)�create_proc_children_pair)�create_sockets)�create_zombie_proc)�DEVNULL)� get_free_port)�get_test_subprocess)� HAS_BATTERY)�HAS_CONNECTIONS_UNIX)�HAS_MEMORY_FULL_INFO)�HAS_MEMORY_MAPS)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�import_module_by_path)� is_namedtuple)�mock)� PYTHON_EXE)� reap_children)� reload_module)�retry)�ROOT_DIR)�run_test_module_by_name)� safe_rmpath)� SCRIPTS_DIR)�sh)�tcp_socketpair)�TESTFN)�TOX)�TRAVIS)�unittest)�unix_socket_path)�unix_socketpair)� wait_for_file)� wait_for_pidc @ s� e Zd Zefdd�Zdd� Zefdd�Zefdd�Zefd d �Zefd d �Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*�TestMiscc C sn t j� }||�}| jd|� | jd|j |� | jd|� | j|j� |� tjjt jdt jt j � �d��@ t j� }||�}| jd|j |� | jd|� | j d|� W d Q R X tjjt jdt j t j � �d��@ t j� }||�}| jd|j |� | jd|� | j d|� W d Q R X tjjt jdt j t j � �d��4 t j� }||�}| jd|j |� | j d|� W d Q R X d S )Nzpsutil.Processzpid=%szname=�name)� side_effectZzombieZ terminated)�psutil�Process�assertIn�pidr2 r �patch�object� ZombieProcess�os�getpid� assertNotIn� NoSuchProcess� AccessDenied)�self�func�p�r� rD �!/usr/lib64/python3.6/test_misc.py�test_process__repr__O s4 z TestMisc.test_process__repr__c C s | j td� d S )N)rA )rF �str)r@ rD rD rE �test_process__str__k s z TestMisc.test_process__str__c C sN | j ttjd��d� | j ttjddd��d� | j ttjddd��d� d S )NiA z7psutil.NoSuchProcess process no longer exists (pid=321)�foo)r2 zCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo'))�msgzpsutil.NoSuchProcess foo)� assertEqual�reprr4 r> )r@ rA rD rD rE � test_no_such_process__repr__n s z%TestMisc.test_no_such_process__repr__c C sj | j ttjd��d� | j ttjddd��d� | j ttjdddd��d� | j ttjddd ��d � d S ) NiA zEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)rI )r2 zQpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo')� )r2 �ppidzYpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo', ppid=1))rJ zpsutil.ZombieProcess foo)rK rL r4 r: )r@ rA rD rD rE �test_zombie_process__repr__z s z$TestMisc.test_zombie_process__repr__c C sN | j ttjd��d� | j ttjddd��d� | j ttjddd��d� d S )NiA z psutil.AccessDenied (pid=321)rI )r2 z)psutil.AccessDenied (pid=321, name='foo'))rJ zpsutil.AccessDenied foo)rK rL r4 r? )r@ rA rD rD rE �test_access_denied__repr__� s z#TestMisc.test_access_denied__repr__c C sP | j ttjd��d� | j ttjddd��d� | j ttjdddd��d� d S ) NiA z/psutil.TimeoutExpired timeout after 321 seconds�o )r7 z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)rI )r7 r2 zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))rK rL r4 ZTimeoutExpired)r@ rA rD rD rE � test_timeout_expired__repr__� s z%TestMisc.test_timeout_expired__repr__c C s> t j� }t j� }| j||� d|_| j||� | j|d� d S )Nr rI )r r )r4 r5 rK Z_ident�assertNotEqual)r@ �p1�p2rD rD rE �test_process__eq__� s zTestMisc.test_process__eq__c C s( t tj� tj� g�}| jt|�d� d S )NrN )�setr4 r5 rK �len)r@ �srD rD rE �test_process__hash__� s z TestMisc.test_process__hash__c C s� t t�}x�|D ]~}|d kr q|jd �sy t|� W q tk r� |tjkr�tt|�}|d kr`w|jd k r�d |jj� kr�| j d | � Y qX qW xtjD ]}| j ||� q�W d S )N�callable�error� namedtuple�tests�long�test�NUM_CPUS� BOOT_TIME� TOTAL_PHYMEM�_Z deprecatedz%r not in psutil.__all__) r\ r] r^ r_ r` ra rb rc rd ) �dirr4 � startswith� __import__� ImportError�__all__�getattr�__doc__�lower�failr6 )r@ Z dir_psutilr2 ZfunrD rD rE � test__all__� s& zTestMisc.test__all__c C s$ | j djdd� tjD ��tj� d S )N�.c S s g | ] }t |��qS rD )rG )�.0�xrD rD rE � <listcomp>� s z)TestMisc.test_version.<locals>.<listcomp>)rK �joinr4 � version_info� __version__)r@ rD rD rE � test_version� s zTestMisc.test_versionc C s" t j� }d|_| jd|j� � d S )N�1rI )r4 r5 rI r= �as_dict)r@ rB rD rD rE �!test_process_as_dict_no_new_names� s z*TestMisc.test_process_as_dict_no_new_namesc s t � fdd��}g � x:td�D ].}|� }f i f}| j||� | jt� �d� q W x<td�D ]0}|d�}d i f}| j||� | jt� �d� qZW xDtd�D ]8}|ddd�}d ddif}| j||� | jt� �d� q�W |j� |� }f i f}| j||� | jt� �d� | j|jd � d S ) Nc s � j d� | |fS )z foo docstringN)�append)�args�kwargs)�callsrD rE rI � s z"TestMisc.test_memoize.<locals>.foo� rN )�barr� � � z foo docstring)rN )rN )r �rangerK rY � cache_clearrl )r@ rI rr �retZexpectedrD )r~ rE � test_memoize� s. zTestMisc.test_memoizec s� G � fdd�d�}|� }g � |j � |j � | jt� �d� g � |j j� |j � |j � | jt� �d� g � |j j� |j � |j � | jt� �d� d S )Nc s e Zd Ze� fdd��ZdS )z1TestMisc.test_memoize_when_activated.<locals>.Fooc s � j d � d S )N)r{ )r@ )r~ rD rE rI � s z5TestMisc.test_memoize_when_activated.<locals>.Foo.fooN)�__name__� __module__� __qualname__r rI rD )r~ rD rE �Foo� s r� r rN )rI rK rY Zcache_activateZcache_deactivate)r@ r� �frD )r~ rE �test_memoize_when_activated� s z$TestMisc.test_memoize_when_activatedc C s� ddl m} dd� }| j|d�|d�di� | j|d�|d�d|d �d i� | j|d �|d�d|d �d i� | j|d �|d�d|d �d i� | j|d�|d�di� | j|d�|d�di� | j|d�|d�di� d S )Nr )�parse_environ_blockc S s t r | j� S | S )N)r �upper)rZ rD rD rE �k s z,TestMisc.test_parse_environ_block.<locals>.kza=1 �arx z a=1 b=2 �b�2za=1 b= � z a=1 b=2 c=3 zxxx a=1 z a=1 =b=2 za=1 b=2)�psutil._commonr� rK )r@ r� r� rD rD rE �test_parse_environ_block s z!TestMisc.test_parse_environ_blockc C s8 | j tj� t� �rtjd�� }d|_tj� t� s:t�W d Q R X tj� tjdtjd�� }t� sjt�|j stt�W d Q R X tj� tjdtj d��$}t� s�t�tj� |j s�t�W d Q R X tj� tjdtj d��$}t� s�t�tj� |j s�t�W d Q R X n0| j t �� tjtj tj�}|jd� W d Q R X d S ) Nzpsutil._common.socketFz psutil._common.socket.socket)r3 z!psutil._common.socket.socket.bind�::1r )r� r )� addCleanupr r� r r8 Zhas_ipv6�AssertionError�socketr] �calledZgaierror� assertRaises� Exception�AF_INET6� SOCK_STREAMZbind)r@ rZ �sockrD rD rE �test_supports_ipv6% s4 zTestMisc.test_supports_ipv6c C s� ddl m} tjjt�}||�s$t�|tjj|�� s:t�tj dt t j d�d�� | j t ||� W d Q R X tj dt t jd�d�� | j t ||� W d Q R X tj dt t jd�d�� ||� s�t�W d Q R X tj ddd�� ||� s�t�W d Q R X d S ) Nr )� isfile_strictzpsutil._common.os.statrI )r3 zpsutil._common.stat.S_ISREGF)� return_value)r� r� r; �path�abspath�__file__r� �dirnamer r8 �OSError�errnoZEPERMr� ZEACCES�EINVAL)r@ r� Z this_filerD rD rE �test_isfile_strictE s zTestMisc.test_isfile_strictc s� � fdd�}|t j� j� � |t j� � |t j� � |t j� � |t jdd�� |t j� � trpt j j d� rpnt s�|t j � � |t j� � |t jt j� �� |t j� � d S )Nc s< t d k rt jt j| �� tj| �}tj|�}� j| |� d S )N)�json�loads�dumps�picklerK )r� r� r� )r@ rD rE �checkW s z*TestMisc.test_serialization.<locals>.checkr )�intervalz/proc/diskstats)r4 r5 ry Zvirtual_memoryZ swap_memoryZ cpu_timesZcpu_times_percent�net_io_countersr r; r� �existsr �disk_io_countersZdisk_partitionsZ disk_usage�getcwd�users)r@ r� rD )r@ rE �test_serializationV s zTestMisc.test_serializationc C sV t jjtd�}tr*t jj|� r*| jd�S t|�}| jt |j � | j |j � t j� d S )Nzsetup.pyzcan't find setup.py)r; r� rt r# r+ r� ZskipTestr r� � SystemExitZsetuprK Z get_versionr4 rv )r@ Zsetup_py�modulerD rD rE �test_setup_scriptm s zTestMisc.test_setup_scriptc C s� t jjtjdtjd��}tj� |js*t�W d Q R X t jjtjdtjd�d��}tj� |jsbt�W d Q R X t jjtjdt d��.}| j t �� tj� W d Q R X |js�t�W d Q R X d S )NZ create_time)r3 rN ) r r8 r9 r4 r5 r? r� r� r: � ValueErrorr� )r@ �methrD rD rE �test_ad_on_process_creationu s z$TestMisc.test_ad_on_process_creationc C sR t jddd��: | jt��}tt� W d Q R X | jdt|j�j � � W d Q R X d S )Nz psutil._psplatform.cext.versionz0.0.0)r� zversion conflict) r r8 r� ri r! r4 r6 rG � exceptionrm )r@ �cmrD rD rE �test_sanity_version_check� s z"TestMisc.test_sanity_version_checkN)r� r� r� rL rF rH rM rP rQ rS rW r[ ro rw rz r� r� r� r� r� r� r� r� r� rD rD rD rE r1 M s( # r1 rI za b cc @ s� e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zejej� p�ej� d�d d � �Zd S ) �TestWrapNumbersc C s t j� d S )N)r r� )r@ rD rD rE �setUp� s zTestWrapNumbers.setUpc C s& dt ddd�i}| jt|d�|� d S )N�disk1� �disk_io)�ntrK r )r@ �inputrD rD rE �test_first_call� s z TestWrapNumbers.test_first_callc C s8 dt ddd�i}| jt|d�|� | jt|d�|� d S )Nr� r� r� )r� rK r )r@ r� rD rD rE �test_input_hasnt_changed� s z(TestWrapNumbers.test_input_hasnt_changedc C s� dt ddd�i}| jt|d�|� dt ddd�i}| jt|d�|� dt ddd�i}| jt|d�|� dt ddd�i}| jt|d�|� d S ) Nr� r� r� � � � � � )r� rK r )r@ r� rD rD rE �test_increase_but_no_wrap� s z)TestWrapNumbers.test_increase_but_no_wrapc C s� dt ddd�i}| jt|d�|� dt ddd�i}| jt|d�dt ddd�i� dt ddd�i}| jt|d�dt ddd�i� dt ddd�i}| jt|d�dt ddd�i� dt ddd�i}| jt|d�dt ddd �i� dt ddd�i}| jt|d�dt ddd �i� dt d dd�i}| jt|d�dt d dd �i� dt d dd�i}| jt|d�dt ddd �i� dt d dd�i}| jt|d�dt ddd �i� d S ) Nr� �d r� r� �n �Z � r� �� �2 � �( )r� rK r )r@ r� rD rD rE � test_wrap� s4 zTestWrapNumbers.test_wrapc C st dt ddd�i}| jt|d�|� t ddd�t ddd�d�}| jt|d�|� dt ddd�i}| jt|d�|� d S )Nr� r� r� � )r� �disk2� )r� rK r )r@ r� rD rD rE �test_changing_keys� s z"TestWrapNumbers.test_changing_keysc C s. t ddd�t ddd�d�}| jt|d�|� t ddd�t ddd�d�}| jt|d�t ddd�t ddd�d�� dt ddd�i}| jt|d�|� t ddd�t ddd�d�}| jt|d�|� t ddd�t ddd�d�}| jt|d�|� t ddd�t ddd�d�}| jt|d�t ddd�t ddd�d�� d S )Nr� r� )r� r� r� r� r� r� )r� rK r )r@ r� rD rD rE �test_changing_keys_w_wrap� s* z)TestWrapNumbers.test_changing_keys_w_wrapc C sb d'd(d)d*d"�}| j t|d#�|� | j t|d#�|� d+d,d-d.d"�}t|d#�}| j |d% d d&� d S )/N�, � � �# �R �� � �� �ȷ � r � tU � � r r� �6 � �$ � �N r� � r� � � �U � �" r� �� �4 �� � �\ )�nvme0n1Z nvme0n1p1Z nvme0n1p2Z nvme0n1p3r� r� r� i� ) 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� 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� r� r� r� r� r� r� r� r� )rK r )r@ �d�outrD rD rE �test_real_data s z TestWrapNumbers.test_real_datac C sb dt ddd�i}t|d� tj� }| j|d d|i� | j|d di i� | j|d di i� d S )Nr� r� r� r rN r )r� r � cache_inforK )r@ r� �cacherD rD rE �test_cache_first_call s z%TestWrapNumbers.test_cache_first_callc C s� dt ddd�i}t|d� dt ddd�i}t|d� tj� }| j|d d|i� | j|d dddd dd dii� | j|d di i� d S ) Nr� r� r� r� r rN r )r� r )r� rN )r� r )r� r r� rK )r@ r� r� rD rD rE �test_cache_call_twice s z%TestWrapNumbers.test_cache_call_twicec s� dt ddd�i}t|d� dt ddd�i}t|d� tj� }� j|d d|i� � j|d dd dddddii� � j|d ddtdg�ii� � fdd �}dt ddd�i}t|d� tj� }� j|d d|i� |� dt ddd �i}t|d� tj� }� j|d d|i� |� dt ddd �i}t|d� tj� }� j|d d|i� � j|d ddddddd ii� � j|d ddtdg�ii� d S )Nr� r� r� r� r rN r c sJ t j� } � j| d dddddd dii� � j| d ddtd g�ii� d S ) NrN r� r� r r r� )r� r )r� rN )r� r )r� r )r r� rK rX )r� )r@ rD rE �assert_4 s z0TestWrapNumbers.test_cache_wrap.<locals>.assert_r� r� r� )r� r )r� rN )r� r )r� r )r� r )r� rN )r� r )r� r )r� r r� rK rX )r@ r� r� r rD )r@ rE �test_cache_wrap% s: z TestWrapNumbers.test_cache_wrapc C s� dt ddd�i}t|d� t ddd�t ddd�d�}t|d� tj� }| j|d d|i� | j|d dd dd dd dii� | j|d di i� d S ) Nr� r� r� r� )r� r� r rN r )r� r )r� rN )r� r )r� r r� rK )r@ r� r� rD rD rE �test_cache_changing_keysU s z(TestWrapNumbers.test_cache_changing_keysc C s\ dt ddd�i}t|d� t|d� tjd� | jtj� i i i f� tjd� tjd� d S )Nr� r� r� z?!?)r� r r� rK r� )r@ r� rD rD rE �test_cache_clearb s z TestWrapNumbers.test_cache_clearzno disks or NICs availablec C s� t j� t j� tj� }x$|D ] }| jd|� | jd|� q W t jj� tj� }x$|D ] }| jd|� | jd|� qVW t jj� tj� }| j|i i i f� d S )Nzpsutil.disk_io_counterszpsutil.net_io_counters) r4 r� r� r r� r6 r� r= rK )r@ Zcachesr� rD rD rE � test_cache_clear_public_apisk s z,TestWrapNumbers.test_cache_clear_public_apisN)r� r� r� r� �tearDownr� r� r� r� r� r� r� r� r� r r r r, �skipIfr4 r� r� r rD rD rD rE r� � s" % 0 r� zcan't test on TOXz can't locate scripts directoryc @ s� e Zd ZdZedd� �Zed=dd��Zdd� Zej e d �d d � �Z d d � Z dd� Z dd� Zdd� Zej epxeoxej� d�dd� �Zdd� Zdd� Zdd � Zej ed �d d � �Zej e d �d!d"� �Zej e d �d#d$� �Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z!ej e" d/�d0d1� �Z#d2d3� Z$ej e% d �ej ed �d4d5� ��Z&ej e' d �ej ed �d6d7� ��Z(ej e) d �ej e* d8�d9d:� ��Z+d;d<� Z,dS )>� TestScriptsz-Tests for scripts in the "scripts" directory.c O s� dt jjt| � } t| g}x|D ]}|j|� q W yt|f|�j� }W n8 tk r~ } z dt |�krlt |�S � W Y d d }~X nX |s�t |��|S )Nz%sr? ) r; r� rt r& r r{ r' �strip� RuntimeErrorrG r� )�exer| r} �cmd�argr� �errrD rD rE � assert_stdout� s zTestScripts.assert_stdoutNc C sR t jjt| �} tr"t| ddd�}n t| d�}|� |j� }W d Q R X tj|� d S )NZrt�utf8)�encoding) r; r� rt r& r �open�read�ast�parse)r r| r� �srcrD rD rE � assert_syntax� s zTestScripts.assert_syntaxc C sZ t | �}xLtjt�D ]>}|jd�rdtjj|�d |kr| jdtjjt|� � qW d S )Nz.pyZtest_r z no test defined for %r script) rf r; �listdirr& �endswithr� �splitextrn rt )r@ Zmethsr2 rD rD rE � test_coverage� s zTestScripts.test_coveragez POSIX onlyc C sT xNt jt�D ]@}|jd�r t jjt|�}tjt j|�tj @ s | j d| � q W d S )Nz.pyz%r is not executable) r; r r& r r� rt �stat�S_IXUSR�ST_MODErn )r@ r2 r� rD rD rE �test_executable� s zTestScripts.test_executablec C s | j d� d S )Nz disk_usage.py)r )r@ rD rD rE �test_disk_usage� s zTestScripts.test_disk_usagec C s | j d� d S )Nzfree.py)r )r@ rD rD rE � test_free� s zTestScripts.test_freec C s | j d� d S )Nz meminfo.py)r )r@ rD rD rE � test_meminfo� s zTestScripts.test_meminfoc C s | j dttj� �� d S )Nz procinfo.py)r rG r; r< )r@ rD rD rE � test_procinfo� s zTestScripts.test_procinfoz unreliable on APPVEYOR or TRAVISc C s | j d� d S )Nzwho.py)r )r@ rD rD rE �test_who� s zTestScripts.test_whoc C s | j d� d S )Nzps.py)r )r@ rD rD rE �test_ps� s zTestScripts.test_psc C s | j d� d S )Nz pstree.py)r )r@ rD rD rE � test_pstree� s zTestScripts.test_pstreec C s | j d� d S )Nz netstat.py)r )r@ rD rD rE � test_netstat� s zTestScripts.test_netstatzunreliable on TRAVISc C s | j d� d S )Nz ifconfig.py)r )r@ rD rD rE � test_ifconfig� s zTestScripts.test_ifconfigz not supportedc C s | j dttj� �� d S )Nzpmap.py)r rG r; r< )r@ rD rD rE � test_pmap� s zTestScripts.test_pmapc C s | j dtd� d S )Nz procsmem.py)�stderr)r r )r@ rD rD rE � test_procsmem� s zTestScripts.test_procsmemc C s | j d� d S )Nz killall.py)r )r@ rD rD rE � test_killall� s zTestScripts.test_killallc C s | j d� d S )Nz nettop.py)r )r@ rD rD rE � test_nettop� s zTestScripts.test_nettopc C s | j d� d S )Nztop.py)r )r@ rD rD rE �test_top� s zTestScripts.test_topc C s | j d� d S )Nziotop.py)r )r@ rD rD rE � test_iotop� s zTestScripts.test_iotopc C s, | j dtj� j� �}| jttj� �|� d S )Nzpidof.py)r r4 r5 r2 r6 rG r; r< )r@ �outputrD rD rE � test_pidof� s zTestScripts.test_pidofz WINDOWS onlyc C s | j d� d S )Nzwinservices.py)r )r@ rD rD rE �test_winservices� s z TestScripts.test_winservicesc C s | j d� d S )Nzcpu_distribution.py)r )r@ rD rD rE �test_cpu_distribution� s z!TestScripts.test_cpu_distributionc C s | j d� d S )Nztemperatures.py)r )r@ rD rD rE �test_temperatures� s z TestScripts.test_temperaturesc C s | j d� d S )Nzfans.py)r )r@ rD rD rE � test_fans� s zTestScripts.test_fansz no batteryc C s | j d� d S )Nz battery.py)r )r@ rD rD rE � test_battery s zTestScripts.test_batteryc C s | j d� d S )Nz sensors.py)r )r@ rD rD rE � test_sensors s zTestScripts.test_sensors)N)-r� r� r� rl � staticmethodr r r r, r r r r r r! r" r r+ r4 r� r# r$ r% r& r' r r( r r* r+ r, r- r. r0 r r1 r2 r r3 r r4 r r r5 r6 rD rD rD rE r � s@ r c @ 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 )Nr� rN )�retriesr� �logfunc s x� r� j � dd qW dS )NrN r )�poprD )�queuerD rE rI s z2TestRetryDecorator.test_retry_success.<locals>.foor� )r" �listr� rK � call_count)r@ �sleeprI rD )r< rE �test_retry_success s z%TestRetryDecorator.test_retry_successc sD t ddd d�� fdd��}ttd��� | jt|� | j|jd� d S )Nr� rN )r9 r� r: c s x� r� j � dd qW dS )NrN r )r; rD )r< rD rE rI & s z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r" r= r� r� �ZeroDivisionErrorrK r> )r@ r? rI rD )r< rE �test_retry_failure" s z%TestRetryDecorator.test_retry_failurec C s2 t tdd�dd� �}| jt|� | j|jd� d S )NrN )r� r� c S s t �d S )N)� TypeErrorrD rD rD rE rI 3 s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r" r� r� rD rK r> )r@ r? rI rD rD rE �test_exception_arg1 s z%TestRetryDecorator.test_exception_argc C s4 t dd d d�dd� �}| jt|� | j|jd� d S )Nr� )r9 r� r: c S s dd d S )NrN r rD rD rD rD rE rI > s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r" r� rB rK r> )r@ r? rI rD rD rE �test_no_interval_arg: s z'TestRetryDecorator.test_no_interval_argc C s4 t ddd d�dd� �}| jt|� | j|jd� d S )Nr� rN )r9 r� r: c S s dd d S )NrN r rD rD rD rD rE rI H s z0TestRetryDecorator.test_retries_arg.<locals>.foo)r" r� rB rK r> )r@ r? rI rD rD rE �test_retries_argE s z#TestRetryDecorator.test_retries_argc C s | j ttddd� d S )Nr� rN )r9 Ztimeout)r� r� r" )r@ r? rD rD rE � test_retries_and_timeout_argsO s z0TestRetryDecorator.test_retries_and_timeout_argsN) r� r� r� r r8 r@ rC rE rF rG rH rD rD rD rE r8 s r8 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestSyncTestUtilsc C s t t� d S )N)r% r) )r@ rD rD rE r V s zTestSyncTestUtils.tearDownc C sP t tj� � ttj� �d }tjdtdg�d�� | j tj t |� W d Q R X d S )Ni�� zpsutil.tests.retry.__iter__r )r� ) r0 r; r< �maxr4 Zpidsr r8 �iterr� r> )r@ ZnopidrD rD rE �test_wait_for_pidY s z#TestSyncTestUtils.test_wait_for_pidc C s> t td��}|jd� W d Q R X tt� tjjt� s:t�d S )N�wrI )r r) �writer/ r; r� r� r� )r@ r� rD rD rE �test_wait_for_file_ s z$TestSyncTestUtils.test_wait_for_filec C s8 t td�� W d Q R X ttdd� tjjt� s4t�d S )NrM T)�empty)r r) r/ r; r� r� r� )r@ rD rD rE �test_wait_for_file_emptye s z*TestSyncTestUtils.test_wait_for_file_emptyc C s2 t jdtdg�d�� | jttt� W d Q R X d S )Nzpsutil.tests.retry.__iter__r )r� )r r8 rK r� �IOErrorr/ r) )r@ rD rD rE �test_wait_for_file_no_filek s z,TestSyncTestUtils.test_wait_for_file_no_filec C s@ t td��}|jd� W d Q R X ttdd� tjjt�s<t�d S )NrM rI F)�delete)r r) rN r/ r; r� r� r� )r@ r� rD rD rE � test_wait_for_file_no_deleteo s z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd� d�}| j|d� d S )Nc S s dS )NrN rD rD rD rD rE �<lambda>v s z3TestSyncTestUtils.test_call_until.<locals>.<lambda>zret == 1rN )r rK )r@ r� rD rD rE �test_call_untilu s z!TestSyncTestUtils.test_call_untilN) r� r� r� r rL rO rQ rS rU rW rD rD rD rE rI T s rI c @ s( e Zd Zdd� ZeZdd� Zdd� ZdS )�TestFSTestUtilsc C s t t� d S )N)r% r) )r@ rD rD rE r� | s zTestFSTestUtils.setUpc C s� t td�j� tt� tjjt� s(t�tt� tjt� tt� tjjt� sTt�t j dt t j d�d��.}| jt �� tt� W d Q R X |js�t�W d Q R X d S )NrM zpsutil.tests.os.statr� )r3 )r r) �closer% r; r� r� r� �mkdirr r8 r� r� r� r� r� )r@ �mrD rD rE �test_safe_rmpath� s z TestFSTestUtils.test_safe_rmpathc C sT t j� }t jt� tt�� | jt j� t jj|t�� W d Q R X | jt j� |� d S )N)r; r� rZ r) r rK r� rt )r@ �baserD rD rE � test_chdir� s $zTestFSTestUtils.test_chdirN)r� r� r� r� r r\ r^ rD rD rD rE rX z s rX c @ s2 e Zd Zdd� Zdd� Zeje d�dd� �ZdS ) �TestProcessUtilsc C sR t � }tj|j�}|j� s t�t� |j� s2t�tjj s@t�tjj sNt�d S )N) r r4 r5 r7 � is_runningr� r r_ � _pids_started�_subprocesses_started)r@ ZsubprB rD rD rE �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s� t � \}}| j|j|j� |j� s&t�|j� s2t�tj� jdd�}| jt |�d� | j ||� | j ||� | j|j � t j � � | j|j � |j� t� |j� s�t�|j� s�t�tjj s�t�tjj s�t�d S )NT)� recursiver )r rT r7 r` r� r4 r5 �childrenrK rY r6 rO r; r< r r_ ra rb )r@ rU rV re rD rD rE � test_create_proc_children_pair� s z/TestProcessUtils.test_create_proc_children_pairz POSIX onlyc C s4 t � }| jtdd� tj|�}| j|j� tj� d S )NT)rd )r r� r r4 r5 rK ZstatusZ STATUS_ZOMBIE)r@ ZzpidrB rD rD rE �test_create_zombie_proc� s z(TestProcessUtils.test_create_zombie_procN) r� r� r� rc rf r, r r rg rD rD rD rE r_ � s r_ c @ sP e Zd Zdd� Zeje d�dd� �Zdd� Zeje d�dd � �Z d d � Z d S ) � TestNetUtilsc C s> t � }tjtd|fd���}| j|j� d |� W d Q R X d S )Nr� )�addrrN )r � contextlib�closingr rK � getsockname)r@ ZportrZ rD rD rE r � s zTestNetUtils.bind_socketz POSIX onlyc C s� t � �z}t|�}tj|��\ | j|jtj� | j|jtj � | j|j � |� t j j |�s\t�tjt j|�j�srt�W d Q R X W d Q R X t � �:}t|tjd�}tj|�� | j|jtj� W d Q R X W d Q R X d S )N)�type)r- r rj rk rK �familyr� �AF_UNIXrm r� rl r; r� r� r� r �S_ISSOCK�st_mode� SOCK_DGRAM)r@ r2 r� rD rD rE �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)ri ) r r( r� �AF_INETrj rk rK rl � getpeernamerT )r@ ri �server�clientrD rD rE �tcp_tcp_socketpair� s z TestNetUtils.tcp_tcp_socketpairc C s� t j� }|j� }|jdd� s"t�t� ��}t|�\}}zvtjj |�sHt�t j tj |�j �s^t�| j |j� | d� | j t|jdd��d� | j |j� |� | j |j� |� W d |j� |j� X W d Q R X d S )NZunix)Zkindr )r4 r5 �num_fdsZ connectionsr� r- r. r; r� r� r rp rq rK rY rl ru rY )r@ rB ry r2 rv rw rD rD rE �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 )NrN r )r � collections� defaultdict�intrn Z getsockoptr� Z SOL_SOCKETZSO_TYPEZassertGreaterEqualrt r r� r r ro r� rr )r@ ZsocksZfams�typesrZ rD rD rE �test_create_sockets� s z TestNetUtils.test_create_socketsN) r� r� r� r r, r r rs rx rz r rD rD rD rE rh � s rh c @ s e Zd Zdd� ZdS )�TestOtherUtilsc C s0 t tjdd�ddd��s t�t t� � s,t�d S )NrI za b crN r r� )r r{ r^ r� �tuple)r@ rD rD rE �test_is_namedtuple s z!TestOtherUtils.test_is_namedtupleN)r� r� r� r� rD rD rD rE r� s r� �__main__)Nrl r r{ rj r� r� r; r� r� r r4 r r r r� r r r r Zpsutil._compatr 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% r& r' r( r) r* r+ r, r- r. r/ r0 ZTestCaser1 r^ r� r� r r� r� r r8 rI rX r_ rh r� r� r� rD rD rD rE �<module> s� I p C&!&E