3 ��JZ��@s|dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl mZddl mZddl m Z ddlmZddlmZddlmZdd lmZdd lmZddlmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm 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%lm0Z0dd&lm1Z1dd'lm2Z2dd(lm3Z3dd)lm4Z4dd*lm5Z5dd+lm6Z6dd,lm7Z7dd-lm8Z8dd.lm9Z9dd/lm:Z:dd0lm;Z;dd1lmej?d4d5�Z@Gd6d7�d7e8j=�ZAe8jBe6d8�e8jBe7�o�ejCjDe2�d9�Gd:d;�d;e8j=��ZEGdd?�d?e8j=�ZGGd@dA�dAe8j=�ZHGdBdC�dCe8j=�ZIGdDdE�dEe8j=�ZJGdFdG�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�eZdZefdd�Zdd�Zefdd�Zefdd�Zefd d �Zefdd�Z d d�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)S)*�TestMisccCsntj�}||�}|jd|�|jd|j|�|jd|�|j|j�|�tjjtjdtjt j ��d��@tj�}||�}|jd|j|�|jd|�|jd|�WdQRXtjjtjdtjt j ��d��@tj�}||�}|jd|j|�|jd|�|jd|�WdQRXtjjtjdtj t j ��d��4tj�}||�}|jd|j|�|jd|�WdQRXdS)Nzpsutil.Processzpid=%szname=�name)�side_effectZzombieZ terminated)�psutil�Process�assertIn�pidr2r�patch�object� ZombieProcess�os�getpid�assertNotIn� NoSuchProcess�AccessDenied)�self�func�p�r�rD�!/usr/lib64/python3.6/test_misc.py�test_process__repr__Os4zTestMisc.test_process__repr__cCs|jtd�dS)N)rA)rF�str)r@rDrDrE�test_process__str__kszTestMisc.test_process__str__cCsN|jttjd��d�|jttjddd��d�|jttjddd��d�dS)NiAz7psutil.NoSuchProcess process no longer exists (pid=321)�foo)r2zCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo'))�msgzpsutil.NoSuchProcess foo)�assertEqual�reprr4r>)r@rArDrDrE�test_no_such_process__repr__nsz%TestMisc.test_no_such_process__repr__cCsj|jttjd��d�|jttjddd��d�|jttjdddd��d�|jttjddd ��d �dS)NiAzEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)rI)r2zQpsutil.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))rJzpsutil.ZombieProcess foo)rKrLr4r:)r@rArDrDrE�test_zombie_process__repr__zsz$TestMisc.test_zombie_process__repr__cCsN|jttjd��d�|jttjddd��d�|jttjddd��d�dS)NiAzpsutil.AccessDenied (pid=321)rI)r2z)psutil.AccessDenied (pid=321, name='foo'))rJzpsutil.AccessDenied foo)rKrLr4r?)r@rArDrDrE�test_access_denied__repr__�sz#TestMisc.test_access_denied__repr__cCsP|jttjd��d�|jttjddd��d�|jttjdddd��d�dS) NiAz/psutil.TimeoutExpired timeout after 321 seconds�o)r7z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)rI)r7r2zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))rKrLr4ZTimeoutExpired)r@rArDrDrE�test_timeout_expired__repr__�sz%TestMisc.test_timeout_expired__repr__cCs>tj�}tj�}|j||�d|_|j||�|j|d�dS)NrrI)rr)r4r5rKZ_ident�assertNotEqual)r@�p1�p2rDrDrE�test_process__eq__�szTestMisc.test_process__eq__cCs(ttj�tj�g�}|jt|�d�dS)NrN)�setr4r5rK�len)r@�srDrDrE�test_process__hash__�szTestMisc.test_process__hash__cCs�tt�}x�|D]~}|d krq|jd �syt|�Wqtk r�|tjkr�tt|�}|dkr`w|jdk r�d|jj�kr�|j d|�YqXqWxtjD]}|j ||�q�WdS)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`rarbrcrd)�dirr4� startswith� __import__�ImportError�__all__�getattr�__doc__�lower�failr6)r@Z dir_psutilr2ZfunrDrDrE�test__all__�s& zTestMisc.test__all__cCs$|jdjdd�tjD��tj�dS)N�.cSsg|]}t|��qSrD)rG)�.0�xrDrDrE� �sz)TestMisc.test_version..)rK�joinr4�version_info�__version__)r@rDrDrE�test_version�szTestMisc.test_versioncCs"tj�}d|_|jd|j��dS)N�1rI)r4r5rIr=�as_dict)r@rBrDrDrE�!test_process_as_dict_no_new_names�sz*TestMisc.test_process_as_dict_no_new_namescst�fdd��}g�x:td�D].}|�}fif}|j||�|jt��d�qWx.foo�rN)�barr��z foo docstring)rN)rN)r�rangerKrY�cache_clearrl)r@rIrr�retZexpectedrD)r~rE�test_memoize�s.zTestMisc.test_memoizecs�G�fdd�d�}|�}g�|j�|j�|jt��d�g�|jj�|j�|j�|jt��d�g�|jj�|j�|j�|jt��d�dS)NcseZdZe�fdd��ZdS)z1TestMisc.test_memoize_when_activated..Foocs�jd�dS)N)r{)r@)r~rDrErI�sz5TestMisc.test_memoize_when_activated..Foo.fooN)�__name__� __module__�__qualname__rrIrD)r~rDrE�Foo�sr�rrN)rIrKrYZcache_activateZcache_deactivate)r@r��frD)r~rE�test_memoize_when_activated�s z$TestMisc.test_memoize_when_activatedcCs�ddlm}dd�}|j|d�|d�di�|j|d�|d�d|d �d i�|j|d�|d�d|d �di�|j|d �|d�d|d �d i�|j|d�|d�di�|j|d�|d�di�|j|d�|d�di�dS)Nr)�parse_environ_blockcSstr|j�S|S)N)r�upper)rZrDrDrE�ksz,TestMisc.test_parse_environ_block..kza=1�arxz a=1b=2�b�2za=1b=�z a=1b=2c=3zxxxa=1z a=1=b=2za=1b=2)�psutil._commonr�rK)r@r�r�rDrDrE�test_parse_environ_blocks z!TestMisc.test_parse_environ_blockcCs�|jtj�t�r�tjd��}d|_tj�WdQRXtj�tjdtjd��}WdQRXtj�tjdtjd��}tj�WdQRXtj�tjdtjd��}tj�WdQRXn0|j t �� tjtjtj�}|j d�WdQRXdS) Nzpsutil._common.socketFzpsutil._common.socket.socket)r3z!psutil._common.socket.socket.bind�::1r)r�r)� addCleanuprr�rr8Zhas_ipv6�socketr]Zgaierror�assertRaises� Exception�AF_INET6�SOCK_STREAMZbind)r@rZ�sockrDrDrE�test_supports_ipv6%s. zTestMisc.test_supports_ipv6cCs�ddlm}tjjt�}tjdtt j d�d��|jt||�WdQRXtjdtt jd�d��|jt||�WdQRXtjdtt j d�d��WdQRXtjddd��WdQRXdS) Nr)� isfile_strictzpsutil._common.os.statrI)r3zpsutil._common.stat.S_ISREGF)�return_value)r�r�r;�path�abspath�__file__rr8�OSError�errnoZEPERMr�ZEACCES�EINVAL)r@r�Z this_filerDrDrE�test_isfile_strictEs zTestMisc.test_isfile_strictcs��fdd�}|tj�j��|tj��|tj��|tj��|tjdd��|tj��trpt j jd�rpnts�|tj ��|tj��|tjt j��|tj��dS)Ncs<tdk rtjtj|��tj|�}tj|�}�j||�dS)N)�json�loads�dumps�picklerK)r�r�r�)r@rDrE�checkWs z*TestMisc.test_serialization..checkr)�intervalz/proc/diskstats)r4r5ryZvirtual_memoryZswap_memoryZ cpu_timesZcpu_times_percent�net_io_countersrr;r��existsr �disk_io_countersZdisk_partitionsZ disk_usage�getcwd�users)r@r�rD)r@rE�test_serializationVszTestMisc.test_serializationcCsVtjjtd�}tr*tjj|�r*|jd�St|�}|jt |j �|j|j�t j�dS)Nzsetup.pyzcan't find setup.py)r;r�rtr#r+r�ZskipTestrr�� SystemExitZsetuprKZget_versionr4rv)r@Zsetup_py�modulerDrDrE�test_setup_scriptms zTestMisc.test_setup_scriptcCs�tjjtjdtjd��}tj�WdQRXtjjtjdtjd�d��}tj�WdQRXtjjtjdtd��$}|jt��tj�WdQRXWdQRXdS)NZcreate_time)r3rN) rr8r9r4r5r?r:� ValueErrorr�)r@�methrDrDrE�test_ad_on_process_creationus z$TestMisc.test_ad_on_process_creationcCsRtjddd��:|jt��}tt�WdQRX|jdt|j�j ��WdQRXdS)Nzpsutil._psplatform.cext.versionz0.0.0)r�zversion conflict) rr8r�rir!r4r6rG� exceptionrm)r@�cmrDrDrE�test_sanity_version_check�s z"TestMisc.test_sanity_version_checkN)r�r�r�rLrFrHrMrPrQrSrWr[rorwrzr�r�r�r�r�r�r�r�r�rDrDrDrEr1Ms(# r1rIza b cc@s�eZdZdd�ZeZdd�Zdd�Zdd�Zd d �Zdd�Z d d�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zejej�p�ej�d�dd��ZdS)�TestWrapNumberscCstj�dS)N)rr�)r@rDrDrE�setUp�szTestWrapNumbers.setUpcCs&dtddd�i}|jt|d�|�dS)N�disk1��disk_io)�ntrKr)r@�inputrDrDrE�test_first_call�szTestWrapNumbers.test_first_callcCs8dtddd�i}|jt|d�|�|jt|d�|�dS)Nr�r�r�)r�rKr)r@r�rDrDrE�test_input_hasnt_changed�sz(TestWrapNumbers.test_input_hasnt_changedcCs�dtddd�i}|jt|d�|�dtddd�i}|jt|d�|�dtddd�i}|jt|d�|�dtddd�i}|jt|d�|�dS) Nr�r�r�� )r�rKr)r@r�rDrDrE�test_increase_but_no_wrap�sz)TestWrapNumbers.test_increase_but_no_wrapcCs�dtddd�i}|jt|d�|�dtddd�i}|jt|d�dtddd�i�dtddd�i}|jt|d�dtddd�i�dtddd�i}|jt|d�dtddd�i�dtddd�i}|jt|d�dtddd �i�dtddd�i}|jt|d�dtddd �i�dtd dd�i}|jt|d�dtddd �i�dtddd�i}|jt|d�dtddd �i�dtddd�i}|jt|d�dtddd �i�dS) Nr��dr�r��n�Z�r��2��()r�rKr)r@r�rDrDrE� test_wrap�s4zTestWrapNumbers.test_wrapcCstdtddd�i}|jt|d�|�tddd�tddd�d�}|jt|d�|�dtddd�i}|jt|d�|�dS)Nr�r�r��)r��disk2�)r�rKr)r@r�rDrDrE�test_changing_keys�s z"TestWrapNumbers.test_changing_keyscCs.tddd�tddd�d�}|jt|d�|�tddd�tddd�d�}|jt|d�tddd�tddd�d��dtddd�i}|jt|d�|�tddd�tddd�d�}|jt|d�|�tddd�tddd�d�}|jt|d�|�tddd�tddd�d�}|jt|d�tddd�tddd�d��dS)Nr�r�)r�r�r�r�r�r�)r�rKr)r@r�rDrDrE�test_changing_keys_w_wrap�s* z)TestWrapNumbers.test_changing_keys_w_wrapcCsbd'd(d)d*d"�}|jt|d#�|�|jt|d#�|�d+d,d-d.d"�}t|d#�}|j|d%dd&�dS)/N�,��#�R��ȷ�r�tU��rr��6��$��Nr��r��U ��"r��4��\)�nvme0n1Z nvme0n1p1Z nvme0n1p2Z nvme0n1p3r�r�r�i�) r�r�r�r�r�r�r�r�r�) r�rr�r�r�rrrr�) 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�rr�r�r�rrrr�) r�r�r�r�r�r�r�r�r�) r�r�r�r�r�r�r�r�r�)rKr)r@�d�outrDrDrE�test_real_datas zTestWrapNumbers.test_real_datacCsbdtddd�i}t|d�tj�}|j|dd|i�|j|ddii�|j|ddii�dS)Nr�r�r�rrNr)r�r� cache_inforK)r@r��cacherDrDrE�test_cache_first_calls z%TestWrapNumbers.test_cache_first_callc Cs�dtddd�i}t|d�dtddd�i}t|d�tj�}|j|dd|i�|j|ddddd dd dii�|j|ddii�dS)Nr�r�r�r�rrNr)r�r)r�rN)r�r)r�rr�rK)r@r�r�rDrDrE�test_cache_call_twices z%TestWrapNumbers.test_cache_call_twicec s�dtddd�i}t|d�dtddd�i}t|d�tj�}�j|dd|i��j|ddd dddddii��j|dddtdg�ii��fdd �}dtddd�i}t|d�tj�}�j|dd|i�|�dtddd �i}t|d�tj�}�j|dd|i�|�dtddd�i}t|d�tj�}�j|dd|i��j|ddddddddii��j|dddtdg�ii�dS)Nr�r�r�r�rrNrc sJtj�}�j|ddddddd dii��j|dddtd g�ii�dS)NrNr�r�rrr�)r�r)r�rN)r�r)r�r)rr�rKrX)r�)r@rDrE�assert_4s z0TestWrapNumbers.test_cache_wrap..assert_r�r�r�)r�r)r�rN)r�r)r�r)r�r)r�rN)r�r)r�r)r�rr�rKrX)r@r�r�r�rD)r@rE�test_cache_wrap%s: zTestWrapNumbers.test_cache_wrapc Cs�dtddd�i}t|d�tddd�tddd�d�}t|d�tj�}|j|dd|i�|j|ddd dd dddii�|j|ddii�dS)Nr�r�r�r�)r�r�rrNr)r�r)r�rN)r�r)r�rr�rK)r@r�r�rDrDrE�test_cache_changing_keysUs z(TestWrapNumbers.test_cache_changing_keyscCs\dtddd�i}t|d�t|d�tjd�|jtj�iiif�tjd�tjd�dS)Nr�r�r�z?!?)r�rr�rKr�)r@r�rDrDrE�test_cache_clearbs z TestWrapNumbers.test_cache_clearzno disks or NICs availablecCs�tj�tj�tj�}x$|D]}|jd|�|jd|�qWtjj�tj�}x$|D]}|jd|�|jd|�qVWtjj�tj�}|j|iiif�dS)Nzpsutil.disk_io_counterszpsutil.net_io_counters) r4r�r�rr�r6r�r=rK)r@Zcachesr�rDrDrE�test_cache_clear_public_apisks z,TestWrapNumbers.test_cache_clear_public_apisN)r�r�r�r��tearDownr�r�r�r�r�r�r�r�r�r�r�rr,�skipIfr4r�r�rrDrDrDrEr��s" %0 r�zcan't test on TOXzcan't locate scripts directoryc@s�eZdZdZedd��Zed=dd��Zdd�Zej e d �d d��Zdd �Zdd�Z dd�Zdd�Zej epxeoxej�d�dd��Zdd�Zdd�Zdd�Zej ed�dd��Zej ed �d!d"��Zej ed �d#d$��Zd%d&�Zd'd(�Zd)d*�Zd+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.cOs�dtjjt|�}t|g}x|D]}|j|�q Wyt|f|�j�}Wn8tk r~}zdt |�krlt |�S�WYdd}~XnX|S)Nz%sr?) r;r�rtr&rr{r'�strip�RuntimeErrorrG)�exer|r}�cmd�argr��errrDrDrE� assert_stdout�s zTestScripts.assert_stdoutNc CsRtjjt|�}tr"t|ddd�}n t|d�}|�|j�}WdQRXtj|�dS)NZrt�utf8)�encoding) r;r�rtr&r �open�read�ast�parse)rr|r��srcrDrDrE� assert_syntax�s zTestScripts.assert_syntaxcCsZt|�}xLtjt�D]>}|jd�rdtjj|�d|kr|jdtjjt|��qWdS)Nz.pyZtest_rzno test defined for %r script) rfr;�listdirr&�endswithr��splitextrnrt)r@Zmethsr2rDrDrE� test_coverage�s zTestScripts.test_coveragez POSIX onlycCsTxNtjt�D]@}|jd�rtjjt|�}tjtj|�tj@s|j d|�qWdS)Nz.pyz%r is not executable) r;rr&rr�rt�stat�S_IXUSR�ST_MODErn)r@r2r�rDrDrE�test_executable�s zTestScripts.test_executablecCs|jd�dS)Nz disk_usage.py)r)r@rDrDrE�test_disk_usage�szTestScripts.test_disk_usagecCs|jd�dS)Nzfree.py)r)r@rDrDrE� test_free�szTestScripts.test_freecCs|jd�dS)Nz meminfo.py)r)r@rDrDrE�test_meminfo�szTestScripts.test_meminfocCs|jdttj��dS)Nzprocinfo.py)rrGr;r<)r@rDrDrE� test_procinfo�szTestScripts.test_procinfoz unreliable on APPVEYOR or TRAVIScCs|jd�dS)Nzwho.py)r)r@rDrDrE�test_who�szTestScripts.test_whocCs|jd�dS)Nzps.py)r)r@rDrDrE�test_ps�szTestScripts.test_pscCs|jd�dS)Nz pstree.py)r)r@rDrDrE�test_pstree�szTestScripts.test_pstreecCs|jd�dS)Nz netstat.py)r)r@rDrDrE�test_netstat�szTestScripts.test_netstatzunreliable on TRAVIScCs|jd�dS)Nzifconfig.py)r)r@rDrDrE� test_ifconfig�szTestScripts.test_ifconfigz not supportedcCs|jdttj��dS)Nzpmap.py)rrGr;r<)r@rDrDrE� test_pmap�szTestScripts.test_pmapcCs|jdtd�dS)Nzprocsmem.py)�stderr)rr)r@rDrDrE� test_procsmem�szTestScripts.test_procsmemcCs|jd�dS)Nz killall.py)r)r@rDrDrE�test_killall�szTestScripts.test_killallcCs|jd�dS)Nz nettop.py)r)r@rDrDrE�test_nettop�szTestScripts.test_nettopcCs|jd�dS)Nztop.py)r)r@rDrDrE�test_top�szTestScripts.test_topcCs|jd�dS)Nziotop.py)r)r@rDrDrE� test_iotop�szTestScripts.test_iotopcCs,|jdtj�j��}|jttj��|�dS)Nzpidof.py)rr4r5r2r6rGr;r<)r@�outputrDrDrE� test_pidof�szTestScripts.test_pidofzWINDOWS onlycCs|jd�dS)Nzwinservices.py)r)r@rDrDrE�test_winservices�szTestScripts.test_winservicescCs|jd�dS)Nzcpu_distribution.py)r)r@rDrDrE�test_cpu_distribution�sz!TestScripts.test_cpu_distributioncCs|jd�dS)Nztemperatures.py)r)r@rDrDrE�test_temperatures�szTestScripts.test_temperaturescCs|jd�dS)Nzfans.py)r)r@rDrDrE� test_fans�szTestScripts.test_fansz no batterycCs|jd�dS)Nz battery.py)r)r@rDrDrE�test_batteryszTestScripts.test_batterycCs|jd�dS)Nz sensors.py)r)r@rDrDrE�test_sensorsszTestScripts.test_sensors)N)-r�r�r�rl�staticmethodrrrr,rrrrrrrr r+r4r�r r!r"r#r$rr%rr'r(r)r*r+r-rr.r/rr0rr1rrr2r3rDrDrDrEr�s@ rc@sxeZdZejd�dd��Zejd�dd��Zejd�dd��Zejd�dd ��Zejd�d d��Z ejd�dd ��Z dS)�TestRetryDecoratorz time.sleepcsFtdddd��fdd��}ttd��|j|�d�|j|jd�dS)Nr�rN)�retriesr��logfuncsx�r�j�ddqWdS)NrNr)�poprD)�queuerDrErIsz2TestRetryDecorator.test_retry_success..foor�)r"�listr�rK� call_count)r@�sleeprIrD)r9rE�test_retry_successsz%TestRetryDecorator.test_retry_successcsDtdddd��fdd��}ttd��|jt|�|j|jd�dS)Nr�rN)r6r�r7csx�r�j�ddqWdS)NrNr)r8rD)r9rDrErI&sz2TestRetryDecorator.test_retry_failure..foo�)r"r:r�r��ZeroDivisionErrorrKr;)r@r<rIrD)r9rE�test_retry_failure"sz%TestRetryDecorator.test_retry_failurecCs2ttdd�dd��}|jt|�|j|jd�dS)NrN)r�r�cSst�dS)N)� TypeErrorrDrDrDrErI3sz2TestRetryDecorator.test_exception_arg..foor)r"r�r�rArKr;)r@r<rIrDrDrE�test_exception_arg1sz%TestRetryDecorator.test_exception_argcCs4tdddd�dd��}|jt|�|j|jd�dS)Nr�)r6r�r7cSsdddS)NrNrrDrDrDrDrErI>sz4TestRetryDecorator.test_no_interval_arg..foor)r"r�r?rKr;)r@r<rIrDrDrE�test_no_interval_arg:sz'TestRetryDecorator.test_no_interval_argcCs4tdddd�dd��}|jt|�|j|jd�dS)Nr�rN)r6r�r7cSsdddS)NrNrrDrDrDrDrErIHsz0TestRetryDecorator.test_retries_arg..foo)r"r�r?rKr;)r@r<rIrDrDrE�test_retries_argEsz#TestRetryDecorator.test_retries_argcCs|jttddd�dS)Nr�rN)r6Ztimeout)r�r�r")r@r<rDrDrE�test_retries_and_timeout_argsOsz0TestRetryDecorator.test_retries_and_timeout_argsN)r�r�r�rr8r=r@rBrCrDrErDrDrDrEr5s r5c@sDeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dS)�TestSyncTestUtilscCstt�dS)N)r%r))r@rDrDrErVszTestSyncTestUtils.tearDowncCsPttj��ttj��d}tjdtdg�d��|j tj t|�WdQRXdS)Ni��zpsutil.tests.retry.__iter__r)r�)r0r;r<�maxr4Zpidsrr8�iterr�r>)r@ZnopidrDrDrE�test_wait_for_pidYsz#TestSyncTestUtils.test_wait_for_pidc Cs,ttd��}|jd�WdQRXtt�dS)N�wrI)rr)�writer/)r@r�rDrDrE�test_wait_for_file_sz$TestSyncTestUtils.test_wait_for_filecCs&ttd��WdQRXttdd�dS)NrJT)�empty)rr)r/)r@rDrDrE�test_wait_for_file_emptyes z*TestSyncTestUtils.test_wait_for_file_emptycCs2tjdtdg�d��|jttt�WdQRXdS)Nzpsutil.tests.retry.__iter__r)r�)rr8rHr��IOErrorr/r))r@rDrDrE�test_wait_for_file_no_fileksz,TestSyncTestUtils.test_wait_for_file_no_filecCs0ttd��}|jd�WdQRXttdd�dS)NrJrIF)�delete)rr)rKr/)r@r�rDrDrE�test_wait_for_file_no_deleteosz.TestSyncTestUtils.test_wait_for_file_no_deletecCstdd�d�}|j|d�dS)NcSsdS)NrNrDrDrDrDrE�vsz3TestSyncTestUtils.test_call_until..zret == 1rN)r rK)r@r�rDrDrE�test_call_untilusz!TestSyncTestUtils.test_call_untilN) r�r�r�rrIrLrNrPrRrTrDrDrDrErFTsrFc@s(eZdZdd�ZeZdd�Zdd�ZdS)�TestFSTestUtilscCstt�dS)N)r%r))r@rDrDrEr�|szTestFSTestUtils.setUpcCstttd�j�tt�tt�tjt�tt�tjdtt j d�d��$}|jt��tt�WdQRXWdQRXdS)NrJzpsutil.tests.os.statr�)r3)rr)�closer%r;�mkdirrr8r�r�r�r�)r@�mrDrDrE�test_safe_rmpath�s z TestFSTestUtils.test_safe_rmpathcCsTtj�}tjt�tt�� |jtj�tjj|t��WdQRX|jtj�|�dS)N)r;r�rWr)rrKr�rt)r@�baserDrDrE� test_chdir�s $zTestFSTestUtils.test_chdirN)r�r�r�r�rrYr[rDrDrDrErUzsrUc@s2eZdZdd�Zdd�Zejed�dd��ZdS) �TestProcessUtilscCst�}tj|j�}t�dS)N)rr4r5r7r )r@ZsubprBrDrDrE�test_reap_children�sz#TestProcessUtils.test_reap_childrencCs�t�\}}|j|j|j�tj�jdd�}|jt|�d�|j||�|j||�|j|j �t j��|j|j �|j�t�dS)NT)� recursiver) rrTr7r4r5�childrenrKrYr6rOr;r<r )r@rUrVr_rDrDrE�test_create_proc_children_pair�s z/TestProcessUtils.test_create_proc_children_pairz POSIX onlycCs4t�}|jtdd�tj|�}|j|j�tj�dS)NT)r^)rr�r r4r5rKZstatusZ STATUS_ZOMBIE)r@ZzpidrBrDrDrE�test_create_zombie_proc�s z(TestProcessUtils.test_create_zombie_procN) r�r�r�r]r`r,rrrarDrDrDrEr\�s r\c@sPeZdZdd�Zejed�dd��Zdd�Zejed�dd ��Z d d�Z dS) �TestNetUtilsc Cs>t�}tjtd|fd��}|j|j�d|�WdQRXdS)Nr�)�addrrN)r� contextlib�closingrrK�getsockname)r@ZportrZrDrDrEr�szTestNetUtils.bind_socketz POSIX onlycCs�t��T}t|�}tj|��6|j|jtj�|j|jtj �|j|j �|�WdQRXWdQRXt��:}t|tjd�}tj|��|j|jtj�WdQRXWdQRXdS)N)�type)r-rrdrerK�familyr��AF_UNIXrgr�rf� SOCK_DGRAM)r@r2r�rDrDrE�test_bind_unix_socket�sz"TestNetUtils.test_bind_unix_socketcCs|dt�f}ttj|d�\}}tj|��Ltj|��6|j|j�|�|j|j�|�|j |j�|�WdQRXWdQRXdS)Nz 127.0.0.1)rc) rr(r��AF_INETrdrerKrf�getpeernamerT)r@rc�server�clientrDrDrE�tcp_tcp_socketpair�s zTestNetUtils.tcp_tcp_socketpaircCs�tj�}|j�}t��v}t|�\}}zP|j|j�|d�|jt|jdd��d�|j|j�|�|j|j �|�Wd|j �|j �XWdQRXdS)NrZunix)Zkind)r4r5�num_fdsr-r.rKrYZconnectionsrfrmrV)r@rBrqr2rnrorDrDrE�test_unix_socketpair�sz!TestNetUtils.test_unix_socketpaircCs�t��}tjt�}tjt�}x:|D]2}||jd7<||jtjtj�d7<q"W|j |tj d�t�r�|j |tjd�t r�tr�|j |tjd�|j |tjd�|j |tjd�WdQRXdS)NrNr)r�collections�defaultdict�intrhZ getsockoptr�Z SOL_SOCKETZSO_TYPEZassertGreaterEqualrlrr�rrrir�rj)r@ZsocksZfams�typesrZrDrDrE�test_create_sockets�s z TestNetUtils.test_create_socketsN)r�r�r�rr,rrrkrprrrwrDrDrDrErb�s rbc@seZdZdd�ZdS)�TestOtherUtilscCsdS)NrD)r@rDrDrE�test_is_namedtuplesz!TestOtherUtils.test_is_namedtupleN)r�r�r�ryrDrDrDrErxsrx�__main__)Nrlrrsrdr�r�r;r�r�rr4rrrr�rrrrZpsutil._compatr Zpsutil.testsr rrr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0ZTestCaser1r^r�r�rr�r�rr5rFrUr\rbrxr�r�rDrDrDrE� s�Ip C&!&E