Cs[ @ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z d d l
m Z d d l
m
Z
d d l m Z d d l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m! Z! d d l m" Z" d d l m# Z# d d l m$ Z$ d d l m% Z% d d l m& Z& d d l m' Z' d d l m( Z( d d l m) Z) d d l m* Z* d d l m+ Z+ d d! l m, Z, d d" l m- Z- d d# l m. Z. d d$ l m/ Z/ d d% l m0 Z0 d d& l m1 Z1 d d' l m2 Z2 d d( l m3 Z3 d d) l m4 Z4 d d* l m5 Z5 d d+ l m6 Z6 d d, l m7 Z7 d d- l m8 Z8 d d. l m9 Z9 d d/ l m: Z: d d0 l m; Z; d d1 l m< Z< d d2 l m= Z= d d l
Z
d d l Z
Gd3 d4 d4 e9 j> Z? e j@ d5 d6 ZA Gd7 d8 d8 e9 j> ZB e9 jC e7 d9 e9 jC e8 oe jD jE e3 d: Gd; d< d< e9 j> ZF Gd= d> d> e9 j> ZG Gd? d@ d@ e9 j> ZH GdA dB dB e9 j> ZI GdC dD dD e9 j> ZJ GdE dF dF e9 j> ZK GdG dH dH e9 j> ZL eM dI k re0 eN d S)Jz
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_mkdir) safe_rmpath) SCRIPTS_DIR)sh)tcp_socketpair)TESTFN)TOX)TRAVIS)unittest)unix_socket_path)unix_socketpair)
wait_for_file) wait_for_pidc @ s e Z d Z e d d Z d d Z e d d Z e d d Z e d d
Z e d d Z d
d Z
d d Z d d Z d d Z
d d Z d d Z d d Z d d Z d d Z d d Z d! d" Z d# d$ Z d% d& Z d' d( Z d) S)*TestMiscc C s t j } | | } | j d | | j d | j | | j d | | j | j | t j j t j d d t j t j
T t j } | | } | j d | j | | j d | | j d | Wd QRXt j j t j d d t j t j
T t j } | | } | j d | j | | j d | | j d | Wd QRXt j j t j d d t j
t j
D t j } | | } | j d | j | | j d | Wd QRXd S)Nzpsutil.Processzpid=%szname=name side_effectZzombieZ
terminated)psutilProcessassertInpidr3 r patchobject
ZombieProcessosgetpid assertNotIn
NoSuchProcess AccessDenied)selffuncpr rE
/test_misc.pytest_process__repr__Q s4 z TestMisc.test_process__repr__c C s | j d t d S)NrB )rG str)rA rE rE rF test_process__str__m s z TestMisc.test_process__str__c C sm | j t t j d d | j t t j d d d d | j t t j d d d d d S)NiA z7psutil.NoSuchProcess process no longer exists (pid=321)r3 foozCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo')msgzpsutil.NoSuchProcess foo) assertEqualreprr5 r? )rA rB rE rE rF test_no_such_process__repr__p s z%TestMisc.test_no_such_process__repr__c C s | j t t j d d | j t t j d d d d | j t t j d d d d d d | j t t j d d d d
d S) NiA zEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)r3 rJ zQpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo')ppid zYpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo', ppid=1)rK zpsutil.ZombieProcess foo)rL rM r5 r; )rA rB rE rE rF test_zombie_process__repr__| s z$TestMisc.test_zombie_process__repr__c C sm | j t t j d d | j t t j d d d d | j t t j d d d d d S)NiA z psutil.AccessDenied (pid=321)r3 rJ z)psutil.AccessDenied (pid=321, name='foo')rK zpsutil.AccessDenied foo)rL rM r5 r@ )rA rB rE rE rF test_access_denied__repr__ s z#TestMisc.test_access_denied__repr__c C ss | j t t j d d | j t t j d d d d | j t t j d d d d d d d S) NiA z/psutil.TimeoutExpired timeout after 321 secondsr8 o z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)r3 rJ zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))rL rM r5 TimeoutExpired)rA rB rE rE rF test_timeout_expired__repr__ s z%TestMisc.test_timeout_expired__repr__c C sU t j } t j } | j | | d | _ | j | | | j | d d S)Nr rJ )r r )r5 r6 rL _identassertNotEqual)rA p1p2rE rE rF test_process__eq__ s zTestMisc.test_process__eq__c C s8 t t j t j g } | j t | d d S)NrP )setr5 r6 rL len)rA srE rE rF test_process__hash__ s z TestMisc.test_process__hash__c C s t t } x | D] } | d
k r( q | j d
s y t | Wq t k
r | t j k r t t | } | d k r w | j d k r d | j j k r | j d | Yq Xq Wx! t j D] } | j
| | q Wd S)Ncallableerror
namedtupletestslongtestNUM_CPUS BOOT_TIME TOTAL_PHYMEM_Z
deprecatedz%r not in psutil.__all__) r_ r` ra rb rc rd re rf rg ) dirr5
startswith
__import__ ImportError__all__getattr__doc__lowerfailr7 )rA Z
dir_psutilr3 ZfunrE rE rF test__all__ s&
zTestMisc.test__all__c C s0 | j d j d d t j D t j d S)N.c S s g | ] } t | q SrE )rH ).0xrE rE rF
<listcomp> s z)TestMisc.test_version.<locals>.<listcomp>)rL joinr5 version_info __version__)rA rE rE rF test_version s "zTestMisc.test_versionc C s/ t j } d | _ | j d | j d S)N1rJ )r5 r6 rJ r> as_dict)rA rC rE rE rF !test_process_as_dict_no_new_names s z*TestMisc.test_process_as_dict_no_new_namesc s t f d d } g xO t d D]A } | } f i f } | j | | | j t d q+ WxR t d D]D } | d } d i f } | j | | | j t d q} Wx^ t d D]P } | d d d } d
d d i f } | j | | | j t d q W| j | } f i f } | j | | | j t d | j | j d d S) Nc s j d | | f S)z
foo docstringN)append)argskwargs)callsrE rF rJ s
z"TestMisc.test_memoize.<locals>.foo rP Zbar z
foo docstring)rP )rP )r rangerL r\ cache_clearro )rA rJ ru retZexpectedrE )r rF test_memoize s.
zTestMisc.test_memoizec s G f d d d } | } g | j | j | j t d g | j j | j | j | j t d g | j j | j | j | j t d d S)Nc s( e Z d Z e f d d Z d S)z1TestMisc.test_memoize_when_activated.<locals>.Fooc s j d d S)N)r~ )rA )r rE rF rJ s z5TestMisc.test_memoize_when_activated.<locals>.Foo.fooN)__name__
__module__ __qualname__r rJ rE )r rE rF Foo s r r rP )rJ rL r\ Zcache_activateZcache_deactivate)rA r frE )r rF test_memoize_when_activated s
z$TestMisc.test_memoize_when_activatedc C s2 d d l m } d d } | j | d | d d i | j | d | d d | d d
i | j | d | d d | d d i | j | d
| d d | d d
i | j | d | d d i | j | d | d d i | j | d | d d i d S)Nr )parse_environ_blockc S s t r | j S| S)N)r upper)r] rE rE rF k s z,TestMisc.test_parse_environ_block.<locals>.kza=1 ar{ z a=1 b=2 b2za=1 b= z
a=1 b=2 c=3 zxxx a=1 z a=1 =b=2 za=1 b=2)psutil._commonr rL )rA r r rE rE rF test_parse_environ_block s ""z!TestMisc.test_parse_environ_blockc C s | j t j t rMt j d * } d | _ t j t sN t Wd QRXt j t j d d t j & } t s t | j s t Wd QRXt j t j d d t j
0 } t s t t j | j s t Wd QRXt j t j d d t j
0 } t s*t t j | j sCt Wd QRXn< | j t * t j t j
t j } | j d Wd QRXd S) Nzpsutil._common.socketFz psutil._common.socket.socketr4 z!psutil._common.socket.socket.bind::1r )r r )
addCleanupr r r r9 has_ipv6AssertionErrorsocketr` calledgaierror assertRaises ExceptionAF_INET6 SOCK_STREAMbind)rA r] sockrE rE rF test_supports_ipv6' s4
zTestMisc.test_supports_ipv6c C s; d d l m } t j j t } | | s4 t | t j j | sS t t j d d t
t j d | j
t
| | Wd QRXt j d d t
t j d | j
t
| | Wd QRXt j d d t
t j d | | s t Wd QRXt j d d d | | s0t Wd QRXd S) Nr )
isfile_strictzpsutil._common.os.statr4 rJ zpsutil._common.stat.S_ISREG return_valueF)r r r< pathabspath__file__r dirnamer r9 OSErrorerrnoEPERMr ZEACCESEINVAL)rA r Z this_filerE rE rF test_isfile_strictG s zTestMisc.test_isfile_strictc s f d d } | t j j | t j | t j | t j | t j d d | t j t r t j
j d r n t s | t j
| t j | t j t j | t j d S)Nc sT t d k r" t j t j | t j | } t j | } j | | d S)N)jsonloadsdumpspicklerL )r r r )rA rE rF checkY s
z*TestMisc.test_serialization.<locals>.checkintervalr z/proc/diskstats)r5 r6 r| Zvirtual_memoryZ swap_memoryZ cpu_timesZcpu_times_percentnet_io_countersr r< r existsr
disk_io_countersZdisk_partitions
disk_usagegetcwdusers)rA r rE )rA rF test_serializationX s zTestMisc.test_serializationc C sw t j j t d } t r; t j j | r; | j d St | } | j t | j
| j | j t
j d S)Nzsetup.pyzcan't find setup.py)r< r rw r# r, r ZskipTestr r
SystemExitZsetuprL Z get_versionr5 ry )rA Zsetup_pymodulerE rE rF test_setup_scripto s
zTestMisc.test_setup_scriptc C s t j j t j d d t j } t j | j s= t Wd QRXt j j t j d d t j d } t j | j s t Wd QRXt j j t j d d t 7 } | j
t t j Wd QRX| j s t Wd QRXd S)NZ create_timer4 rP ) r r9 r: r5 r6 r@ r r r;
ValueErrorr )rA methrE rE rF test_ad_on_process_creationw s
z$TestMisc.test_ad_on_process_creationc C sc t j d d d G | j t } t t Wd QRX| j d t | j j Wd QRXd S)Nz psutil._psplatform.cext.versionr z0.0.0zversion conflict)
r r9 r rl r! r5 r7 rH exceptionrp )rA cmrE rE rF test_sanity_version_check s
z"TestMisc.test_sanity_version_checkN)r r r rM rG rI rN rQ rR rU rZ r^ rr rz r} r r r r r r r r r rE rE rE rF r2 O s( # r2 rJ za b cc @ s e Z d Z d d Z e Z d d Z d d Z d d Z d d
Z d d Z d
d Z
d d Z d d Z d d Z
d d Z d d Z d d Z e j e j p e j d d d Z d S) TestWrapNumbersc C s t j d S)N)r r )rA rE rE rF setUp s zTestWrapNumbers.setUpc C s5 d t d d d i } | j t | d | d S)Ndisk1 disk_io)ntrL r )rA inputrE rE rF test_first_call s z TestWrapNumbers.test_first_callc C sN d t d d d i } | j t | d | | j t | d | d S)Nr r r )r rL r )rA r rE rE rF test_input_hasnt_changed s z(TestWrapNumbers.test_input_hasnt_changedc C s d t d d d i } | j t | d | d t d d d i } | j t | d | d t d d d i } | j t | d | d t d d d i } | j t | d | d S) Nr r r
)r rL r )rA r rE rE rF test_increase_but_no_wrap s z)TestWrapNumbers.test_increase_but_no_wrapc C sM d t d d d i } | j t | d | d t d d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d t d
d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d t d d d i } | j t | d d t d d d i d S)
Nr d r r n Z r 2 ( )r rL r )rA r rE rE rF test_wrap s4 zTestWrapNumbers.test_wrapc C s d t d d d i } | j t | d | d t d d d d t d d d i } | j t | d | d t d d d i } | j t | d | d S)Nr r r disk2 )r rL r )rA r rE rE rF test_changing_keys s z"TestWrapNumbers.test_changing_keysc C s d t d d d d t d d d i } | j t | d | d t d d d d t d d d i } | j t | d d t d d d d t d d d i d t d d d i } | j t | d | d t d d d d t d d d i } | j t | d | d t d d d d t d d d i } | j t | d | d t d d d d t d d d i } | j t | d d t d d d d t d d d i d S)Nr r r r r r r )r rL r )rA r rE rE rF test_changing_keys_w_wrap s* z)TestWrapNumbers.test_changing_keys_w_wrapc C s d d) d d* d d+ d d, i } | j t | d& | | j t | d& | d d- d d. d d/ d d0 i } t | d& } | j | d d d( d S)1NZnvme0n1, # R ȷ Z nvme0n1p1 r tU r r Z nvme0n1p26 $ N r r Z nvme0n1p3U " r 4 \ 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 )rL r )rA doutrE rE rF test_real_data s z TestWrapNumbers.test_real_datac C s d t d d d i } t | d t j } | j | d d | i | j | d d i i | j | d d i i d S)Nr r r r rP r )r r
cache_inforL )rA r cacherE rE rF test_cache_first_call s
z%TestWrapNumbers.test_cache_first_callc C s d t d d d i } t | d d t d d d i } t | d t j } | j | d d | i | j | d d d d d d d
d i i | j | d d i i d S) Nr r r r r rP r )r r )r rP )r r )r r r rL )rA r r rE rE rF test_cache_call_twice s
z%TestWrapNumbers.test_cache_call_twicec s d t d d d i } t | d d t d d d i } t | d t j } j | d d | i j | d d d
d d d d d i i j | d d d t d g i i f d d } d t d d d i } t | d t j } j | d d | i | d t d d d
i } t | d t j } j | d d | i | d t d d d i } t | d t j } j | d d | i j | d d d d d d d d i i j | d d d t d g i i d S)Nr r r r r rP r c se t j } j | d d d d d d d d i i j | d d d t d
g i i d S) NrP r r r r r )r r )r rP )r r )r r )r r rL r[ )r )rA rE rF assert_6 s
z0TestWrapNumbers.test_cache_wrap.<locals>.assert_r r r )r r )r rP )r r )r r )r r )r rP )r r )r r )r r r rL r[ )rA r r r rE )rA rF test_cache_wrap' s:
)
z TestWrapNumbers.test_cache_wrapc C s d t d d d i } t | d d t d d d d t d d d i } t | d t j } | j | d d | i | j | d d d d d
d d d i i | j | d d i i d S) Nr r r r r r rP r )r r )r rP )r r )r r r rL )rA r r rE rE rF test_cache_changing_keysW s
z(TestWrapNumbers.test_cache_changing_keysc C s| d t d d d i } t | d t | d t j d | j t j i i i f t j d t j d d S)Nr r r z?!?)r r r rL r )rA r rE rE rF test_cache_cleard s
z TestWrapNumbers.test_cache_clearzno disks or NICs availablec C s t j t j t j } x. | D]& } | j d | | j d | q' Wt j j t j } x. | D]& } | j d | | j d | qq Wt j j t j } | j | i i i f d S)Nzpsutil.disk_io_counterszpsutil.net_io_counters) r5 r r r r r7 r r> rL )rA Zcachesr rE rE rF test_cache_clear_public_apism 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- skipIfr5 r r r
rE rE rE rF r s"
% 0
r zcan't test on TOXz can't locate scripts directoryc @ sP e Z d Z d Z e d d Z e d d d Z d d Z e j e
d d
d Z d d
Z d d Z
d d Z d d Z e j e p e o e j d d d Z d d Z d d Z d d Z e j e d d d Z e j e d d! d" Z e j e d d# d$ Z d% d& Z d' d( Z d) d* Z d+ d, Z d- d. Z! e j e" d/ d0 d1 Z# d2 d3 Z$ e j e% d e j e d d4 d5 Z& e j e' d e j e d d6 d7 Z( e j e) d e j e* d8 d9 d: Z+ d; d< Z, d S)= TestScriptsz-Tests for scripts in the "scripts" directory.c O s d t j j t | } t | g } x | D] } | j | q, Wy t | | j } WnD t k
r } z$ d t | k r t | S WYd d } ~ Xn X| s t
| | S)Nz%sr@ ) r< r rw r' r r~ r( strip RuntimeErrorrH r )exer r cmdargr errrE rE rF
assert_stdout s
zTestScripts.assert_stdoutNc C sm t j j t | } t r3 t | d d d } n t | d } | | j } Wd QRXt j | d S)NZrtencodingutf8) r< r rw r' r openreadastparse)r r r srcrE rE rF
assert_syntax s zTestScripts.assert_syntaxc C sy t | } xf t j t D]U } | j d r d t j j | d | k r | j d t j j t | q Wd S)Nz.pyZtest_r z no test defined for %r script) ri r< listdirr' endswithr splitextrq rw )rA Zmethsr3 rE rE rF
test_coverage s zTestScripts.test_coveragez
POSIX onlyc C sp xi t j t D]X } | j d r t j j t | } t j t j | t j @s | j d | q Wd S)Nz.pyz%r is not executable)
r< r r' r r rw statS_IXUSRST_MODErq )rA r3 r rE rE rF test_executable s
zTestScripts.test_executablec C s | j d d S)Nz
disk_usage.py)r )rA rE rE rF test_disk_usage s zTestScripts.test_disk_usagec C s | j d d S)Nzfree.py)r )rA rE rE rF test_free s zTestScripts.test_freec C s | j d d S)Nz
meminfo.py)r )rA rE rE rF test_meminfo s zTestScripts.test_meminfoc C s | j d t t j d S)Nz procinfo.py)r rH r< r= )rA rE rE rF
test_procinfo s zTestScripts.test_procinfoz unreliable on APPVEYOR or TRAVISc C s | j d d S)Nzwho.py)r )rA rE rE rF test_who s zTestScripts.test_whoc C s | j d d S)Nzps.py)r )rA rE rE rF test_ps s zTestScripts.test_psc C s | j d d S)Nz pstree.py)r )rA rE rE rF test_pstree s zTestScripts.test_pstreec C s | j d d S)Nz
netstat.py)r )rA rE rE rF test_netstat s zTestScripts.test_netstatzunreliable on TRAVISc C s | j d d S)Nz ifconfig.py)r )rA rE rE rF
test_ifconfig s zTestScripts.test_ifconfigz
not supportedc C s | j d t t j d S)Nzpmap.py)r rH r< r= )rA rE rE rF test_pmap s zTestScripts.test_pmapc C s | j d d t d S)Nz procsmem.pystderr)r r )rA rE rE rF
test_procsmem s zTestScripts.test_procsmemc C s | j d d S)Nz
killall.py)r )rA rE rE rF test_killall s zTestScripts.test_killallc C s | j d d S)Nz nettop.py)r )rA rE rE rF test_nettop s zTestScripts.test_nettopc C s | j d d S)Nztop.py)r )rA rE rE rF test_top s zTestScripts.test_topc C s | j d d S)Nziotop.py)r )rA rE rE rF
test_iotop s zTestScripts.test_iotopc C s> | j d t j j } | j t t j | d S)Nzpidof.py)r r5 r6 r3 r7 rH r< r= )rA outputrE rE rF
test_pidof s zTestScripts.test_pidofz WINDOWS onlyc C s | j d d S)Nzwinservices.py)r )rA rE rE rF test_winservices s z TestScripts.test_winservicesc C s | j d d S)Nzcpu_distribution.py)r )rA rE rE rF test_cpu_distribution s z!TestScripts.test_cpu_distributionc C s | j d d S)Nztemperatures.py)r )rA rE rE rF test_temperatures s z TestScripts.test_temperaturesc C s | j d d S)Nzfans.py)r )rA rE rE rF test_fans s zTestScripts.test_fansz
no batteryc C s | j d d S)Nz
battery.py)r )rA rE rE rF test_battery s zTestScripts.test_batteryc C s | j d d S)Nz
sensors.py)r )rA rE rE rF test_sensors
s zTestScripts.test_sensors)-r r r ro staticmethodr r r r- r r r$ r% r& r' r( r
r, r5 r r) r* r+ r, r- r r. r r0 r1 r2 r3 r4 r6 r r7 r8 r r9 r r: r r r; r< rE rE rE rF r
s@
!!"r
c @ s e Z d Z e j d d d Z e j d d d Z e j d d d Z e j d d d Z e j d d
d Z e j d d d
Z
d S)TestRetryDecoratorz
time.sleepc si t d d d d d d f d d } t t d | j | d | j | j d d S) Nretriesr r rP logfunc s# x r j d d q Wd S)NrP r )poprE )queuerE rF rJ s
z2TestRetryDecorator.test_retry_success.<locals>.foor )r" listr rL
call_count)rA sleeprJ rE )rB rF test_retry_success s -z%TestRetryDecorator.test_retry_successc sf t d d d d d d f d d } t t d | j t | | j | j d d S) Nr? r r rP r@ c s# x r j d d q Wd S)NrP r )rA rE )rB rE rF rJ ( s
z2TestRetryDecorator.test_retry_failure.<locals>.foo )r" rC r r ZeroDivisionErrorrL rD )rA rE rJ rE )rB rF test_retry_failure$ s -z%TestRetryDecorator.test_retry_failurec C sH t d t d d d d } | j t | | j | j d d S)Nr r rP c S s
t d S)N) TypeErrorrE rE rE rF rJ 5 s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r" r r rJ rL rD )rA rE rJ rE rE rF test_exception_arg3 s !z%TestRetryDecorator.test_exception_argc C sN t d d d d d d d d } | j t | | j | j d d S)Nr? r r r@ c S s d d d S)NrP r rE rE rE rE rF rJ @ s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r" r rH rL rD )rA rE rJ rE rE rF test_no_interval_arg< s 'z'TestRetryDecorator.test_no_interval_argc C sN t d d d d d d d d } | j t | | j | j d d S)Nr? r r rP r@ c S s d d d S)NrP r rE rE rE rE rF rJ J s z0TestRetryDecorator.test_retries_arg.<locals>.foo)r" r rH rL rD )rA rE rJ rE rE rF test_retries_argG s 'z#TestRetryDecorator.test_retries_argc C s | j t t d d d d d S)Nr? r timeoutrP )r r r" )rA rE rE rE rF test_retries_and_timeout_argsQ s z0TestRetryDecorator.test_retries_and_timeout_argsN) r r r r r9 rF rI rK rL rM rO rE rE rE rF r> s
r> c @ sd e Z d Z d d Z d d Z d d Z d d Z d d
Z d d Z d
d Z d S)TestSyncTestUtilsc C s t t d S)N)r& r* )rA rE rE rF r X s zTestSyncTestUtils.tearDownc C sf t t j t t j d } t j d d t d g | j t j
t | Wd QRXd S)Ni zpsutil.tests.retry.__iter__r r ) r1 r< r= maxr5 Zpidsr r9 iterr r? )rA ZnopidrE rE rF test_wait_for_pid[ s z#TestSyncTestUtils.test_wait_for_pidc C sM t t d } | j d Wd QRXt t t j j t sI t d S)NwrJ )r r* writer0 r< r r r )rA r rE rE rF test_wait_for_filea s
z$TestSyncTestUtils.test_wait_for_filec C sD t t d Wd QRXt t d d t j j t s@ t d S)NrT emptyT)r r* r0 r< r r r )rA rE rE rF test_wait_for_file_emptyg s z*TestSyncTestUtils.test_wait_for_file_emptyc C s= t j d d t d g | j t t t Wd QRXd S)Nzpsutil.tests.retry.__iter__r r )r r9 rR r IOErrorr0 r* )rA rE rE rF test_wait_for_file_no_filem s z,TestSyncTestUtils.test_wait_for_file_no_filec C sR t t d } | j d Wd QRXt t d d t j j t sN t d S)NrT rJ deleteF)r r* rU r0 r< r r r )rA r rE rE rF test_wait_for_file_no_deleteq s z.TestSyncTestUtils.test_wait_for_file_no_deletec C s) t d d d } | j | d d S)Nc S s d S)NrP rE rE rE rE rF <lambda>x s z3TestSyncTestUtils.test_call_until.<locals>.<lambda>zret == 1rP )r
rL )rA r rE rE rF test_call_untilw s z!TestSyncTestUtils.test_call_untilN)
r r r r rS rV rX rZ r\ r^ rE rE rE rF rP V s rP c @ sF e Z d Z d d Z e Z d d Z d d Z d d Z d S)
TestFSTestUtilsc C s t t d S)N)r& r* )rA rE rE rF r ~ s zTestFSTestUtils.setUpc C sH t t t j j t s" t t t t j j t sD t d S)N)r% r* r< r isdirr )rA rE rE rF test_safe_mkdir s
z TestFSTestUtils.test_safe_mkdirc C s t t d j t t t j j t s6 t t t t j t t t t j j t sp t t j
d d t t j
d 7 } | j t t t Wd QRX| j s t Wd QRXd S)NrT zpsutil.tests.os.statr4 r )r r* closer& r< r r r mkdirr r9 r r r r r )rA mrE rE rF test_safe_rmpath s
z TestFSTestUtils.test_safe_rmpathc C sl t j } t j t t t * | j t j t j j | t Wd QRX| j t j | d S)N)r< r rc r* r rL r rw )rA baserE rE rF
test_chdir s
,zTestFSTestUtils.test_chdirN)r r r r r ra re rg rE rE rE rF r_ | s
r_ c @ sG e Z d Z d d Z d d Z e j e d d d Z d S) TestProcessUtilsc C sq t } t j | j } | j s- t t | j sG t t j j sZ t t j j sm t d S)N)
r r5 r6 r8
is_runningr r rb
_pids_started_subprocesses_started)rA ZsubprC rE rE rF test_reap_children s z#TestProcessUtils.test_reap_childrenc C s# t \ } } | j | j | j | j s7 t | j sI t t j j d d } | j t | d | j
| | | j
| | | j | j t j
| j | j | j t | j s t | j s t t j j s t t j j s t d S)N recursiveTr )r rW r8 ri r r5 r6 childrenrL r\ r7 rO r< r= r rb rj rk )rA rX rY rn rE rE rF test_create_proc_children_pair s z/TestProcessUtils.test_create_proc_children_pairz
POSIX onlyc C sH t } | j t d d t j | } | j | j t j d S)Nrm T)r r r r5 r6 rL statusZ
STATUS_ZOMBIE)rA ZzpidrC rE rE rF test_create_zombie_proc s z(TestProcessUtils.test_create_zombie_procN) r r r rl ro r- r r rq rE rE rE rF rh s rh c @ sr e Z d Z d d Z e j e d d d Z d d Z e j e d d d Z d
d Z
d S)
TestNetUtilsc
C sO t } t j t d d | f ! } | j | j d | Wd QRXd S)Naddrr rP )r
contextlibclosingr rL getsockname)rA portr] rE rE rF r s !zTestNetUtils.bind_socketz
POSIX onlyc C s
t } t | } t j | | j | j t j | j | j t j | j | j
| t j j
| s t t j t j | j s t Wd QRXWd QRXt I } t | d t j } t j | | j | j t j Wd QRXWd QRXd S)Ntype)r. r rt ru rL familyr AF_UNIXrx r rv r< r r r r! S_ISSOCKst_mode
SOCK_DGRAM)rA r3 r rE rE rF test_bind_unix_socket s / z"TestNetUtils.test_bind_unix_socketc C s d t f } t t j d | \ } } t j | ^ t j | G | j | j | | j | j | | j | j | Wd QRXWd QRXd S)Nz 127.0.0.1rs )
r r) r AF_INETrt ru rL rv getpeernamerW )rA rs serverclientrE rE rF tcp_tcp_socketpair s z TestNetUtils.tcp_tcp_socketpairc C s t j } | j } | j d d s1 t t } t | \ } } z t j j | sj t t
j t j
| j s t | j
| j | d | j
t | j d d d | j
| j | | j
| j | Wd | j | j XWd QRXd S)NkindZunixr )r5 r6 num_fdsZ connectionsr r. r/ r< r r r! r{ r| rL r\ rv r rb )rA rC r r3 r r rE rE rF test_unix_socketpair s !"
z!TestNetUtils.test_unix_socketpairc C s t } t j t } t j t } x? | D]7 } | | j d 7<| | j t j t j d 7<q1 W| j | t j
d t r | j | t j d t
r t r | j | t j d | j | t j d | j | t j d Wd QRXd S)NrP r )r collections defaultdictintry
getsockoptr
SOL_SOCKETSO_TYPEZassertGreaterEqualr r r r r rz r r} )rA ZsocksZfamstypesr] rE rE rF test_create_sockets s
$ z TestNetUtils.test_create_socketsN) r r r r r- r r r~ r r r rE rE rE rF rr s
rr c @ s e Z d Z d d Z d S)TestOtherUtilsc C sD t t j d d d d d s* t t t s@ t d S)NrJ za b crP r r )r r ra r tuple)rA rE rE rF test_is_namedtuple s *z!TestOtherUtils.test_is_namedtupleN)r r r r rE rE rE rF r s r __main__)Oro r r rt r r r< r r r! r5 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 r1 ZTestCaser2 ra r r r r r r
r> rP r_ rh rr r r r rE rE rE rF <module>
s H C&'&E