Cs[D @ sa d Z d d l m 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 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+ e j, j- e j, j. e/ Z0 d Z1 d Z2 d Z3 e rd Z4 d d! Z5 d" d# Z6 d$ d% Z7 d& d' Z8 d( d) Z9 d* d+ Z: e j; d, d- Z< e j; d. d/ Z= e* j> e d0 Gd1 d2 d2 e* j? Z@ e* j> e d0 Gd3 d4 d4 e* j? ZA e* j> e d0 Gd5 d6 d6 e* j? ZB e* j> e d0 Gd7 d8 d8 e* j? ZC e* j> e d0 Gd9 d: d: e* j? ZD e* j> e d0 Gd; d< d< e* j? ZE e* j> e d0 Gd= d> d> e* j? ZF e* j> e d0 e* j> e d? Gd@ dA dA e* j? ZG e* j> e d0 GdB dC dC e* j? ZH e* j> e d0 GdD dE dE e* j? ZI e* j> e d0 GdF dG dG e* j? ZJ e* j> e d0 GdH dI dI e* j? ZK e* j> e d0 GdJ dK dK e* j? ZL eM dL k r]e# e/ d S)MzLinux specific tests. )divisionN)LINUX)
basestring)PY3)u)
call_until) HAS_BATTERY) HAS_CPU_FREQ)
HAS_RLIMIT)MEMORY_TOLERANCE)mock)PYPY)pyrun)
reap_children)
reload_module)retry_before_failing)run_test_module_by_name) safe_rmpath)sh)skip_on_not_implemented)TESTFN)
ThreadTask)TRAVIS)unittest)whichi i i' i c C s d d l } | d d } t r1 t | d } t j t j t j } t j | = t j | j | j
t t j
d | d d SWd QRXd S)Nr ascii256s )fcntlr bytessocketAF_INET
SOCK_DGRAM
contextlibclosing inet_ntoaioctlfileno SIOCGIFADDRstructpack)ifnamer s r/ /test_linux.pyget_ipv4_addressA s r1 c
s d d l } | d d } t r1 t | d } t j t j t j } t j | | j | j t
t j d | } t r d d n d d l
} | j d j f d d | d
d D d d
SWd QRXd S)Nr r r r c S s | S)Nr/ )xr/ r/ r0 ordX s z get_mac_address.<locals>.ord c s g | ] } d | q S)z%02x:r/ ).0char)r3 r/ r0
<listcomp>] s z#get_mac_address.<locals>.<listcomp> r )r r r! r" r# r$ r% r& r( r)
SIOCGIFHWADDRr+ r, __builtin__r3 join)r- r r. infor< r/ )r3 r0 get_mac_addressN s ! r? c C s t d } | j d } xi | D]a } | j d r" | j \ } } } } t j d d } | t | t | t | Sq" Wt d d j | d S)zQParse 'free' cmd and return swap memory's s total, used and free
values.
zfree -b
ZSwapfreeztotal used freez&can't find 'Swap' in 'free' output:
%sN)r split
startswith collections
namedtupleint
ValueErrorr= )outlinesline_totalusedrA ntr/ r/ r0 free_swap` s
&rO c C s t d } | j d } xt | D]l } | j d r" d d | j d d D \ } } } } t j d d } | | | | | | Sq" Wt d
d j | d S) zSParse 'free' cmd and return physical memory's total, used
and free values.
zfree -br@ ZMemc S s g | ] } t | q Sr/ )rF )r5 r2 r/ r/ r0 r7 | s z free_physmem.<locals>.<listcomp>r9 rA z total used free shared outputz%can't find 'Mem' in 'free' output:
%sN)r rB rC rD rE rG r= )rH rI rJ rL rM rA sharedrN r/ r/ r0 free_physmemo s
/ rR c C si t d } xF | j d D]5 } | j } | | k r t | j d d Sq Wt d | d S)Nz vmstat -sr@ r z can't find %r in 'vmstat' output)r rB striprF rG )statrH rJ r/ r/ r0 vmstat s rV c C s8 t d j } t t t | j d j d S)Nzfree -Vr9 .r: )r rT tuplemaprF rB )rH r/ r/ r0 get_free_version_info s rZ c # s^ f d d } t t r* d n d } t j | d d d | } | VWd QRXd S) zMock open() builtin and forces it to return a certain `content`
on read() if the path being opened matches `for_path`.
c sb | k rN t r> t t r. t j St j Sq^ t j Sn | | | Sd S)N)r
isinstancer ioStringIOBytesIO)nameargskwargs)contentfor_path orig_openr/ r0 open_mock s
z$mock_open_content.<locals>.open_mockz
builtins.openz__builtin__.opencreateT side_effectN)openr r patch)rc rb re patch_pointmr/ )rb rc rd r0 mock_open_content s
rl c # s^ f d d } t t r* d n d } t j | d d d | } | VWd QRXd S) zZMock open() builtin and raises `exc` if the path being opened
matches `for_path`.
c s) | k r n | | | Sd S)Nr/ )r_ r` ra )excrc rd r/ r0 re s z&mock_open_exception.<locals>.open_mockz
builtins.openz__builtin__.openrf Trg N)rh r r ri )rc rm re rj rk r/ )rm rc rd r0 mock_open_exception s
rn z
LINUX onlyc @ sN e Z d Z d d Z e j e o0 e d" k d e d d Z e j e
d e d d
Z e d d Z e j e
d e d
d Z
e j e
d 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! S)#TestSystemVirtualMemoryc C s3 t d d } t j j } | j | | d S)Nz total memoryi )rV psutilvirtual_memoryrL assertAlmostEqual)self vmstat_value psutil_valuer/ r/ r0
test_total s z"TestSystemVirtualMemory.test_total zold free versionc
C sQ t } | j } t j j } | j | | d t d d | | | j f d S)Ndeltamsgz %s %s
%s)rR rM rp rq rr r output)rs rA
free_valueru r/ r/ r0 test_used s z!TestSystemVirtualMemory.test_usedzunreliable on TRAVISc C s9 t d d } t j j } | j | | d t d S)Nz free memoryi ry )rV rp rq rA rr r )rs rt ru r/ r/ r0 test_free s z!TestSystemVirtualMemory.test_freec C s9 t d d } t j j } | j | | d t d S)Nz
buffer memoryi ry )rV rp rq buffersrr r )rs rt ru r/ r/ r0 test_buffers s z$TestSystemVirtualMemory.test_buffersc C s9 t d d } t j j } | j | | d t d S)Nz
active memoryi ry )rV rp rq activerr r )rs rt ru r/ r/ r0 test_active s z#TestSystemVirtualMemory.test_activec C s9 t d d } t j j } | j | | d t d S)Nzinactive memoryi ry )rV rp rq inactiverr r )rs rt ru r/ r/ r0
test_inactive s z%TestSystemVirtualMemory.test_inactivec
C sl t } | j } | d k r- t j d t j j } | j | | d t d d | | | j f d S)Nr z%free does not support 'shared' columnry rz z %s %s
%s) rR rQ r SkipTestrp rq rr r r{ )rs rA r| ru r/ r/ r0 test_shared s z#TestSystemVirtualMemory.test_sharedc
C s t d } | j d } d | d k r= t j d nR t | d j d
} t j j } | j | | d t d d | | | f d S) Nzfree -br@ availabler z(free does not support 'available' columnr9 ry rz z %s %s
%sr: )
r rB r r rF rp rq r rr r )rs rH rI r| ru r/ r/ r0 test_available s z&TestSystemVirtualMemory.test_availablec C s t d t j d j } t j d d } t j d t j } | j s^ t
| j t | d | d } | j
j d s t
| j d t | j | j d
t | j | j d t | j | j d t | j | j d
t | j | j d t | j | j d t | j | j | j d | j | j d | j | j d | j | j d | j | j d | j | j d | j | j d Wd QRXWd QRXd S)Nz
/proc/meminfoap Active(anon): 6145416 kB
Active(file): 2950064 kB
Inactive(anon): 574764 kB
Inactive(file): 1567648 kB
MemAvailable: -1 kB
MemFree: 2057400 kB
MemTotal: 16325648 kB
SReclaimable: 346648 kB
recordTalwaysr9 r zpsutil/_pslinux.pyz#memory stats couldn't be determinedcachedrQ r r r r )rl textwrapdedentencodewarningscatch_warnings simplefilterrp rq calledAssertionError assertEquallenfilenameendswithassertInstrmessager r r rQ r r slab)rs rk wsretwr/ r/ r0 test_warnings_on_misses s4
z/TestSystemVirtualMemory.test_warnings_on_missesc
C s d d l m } d d l m } i } | d @ } x6 | D]. } | j } t | d d | | d <q< WWd QRX| | } d | k r | d } t | | | d } | j | d d S)
Nr )calculate_avail_vmem) open_binaryz
/proc/meminfor9 i s
MemAvailable:d
)psutil._pslinuxr r rB rF absZ
assertLess)
rs r r ZmemsfrJ fieldsabZ diff_percentr/ r/ r0 test_avail_old_percent? s
'
z.TestSystemVirtualMemory.test_avail_old_percentc C s t d t j d j t } t j d d } t j } Wd QRX| j sX t | j
| j d | d } | j d t
| j Wd QRXd S)
Nz
/proc/meminfoaL Active: 9444728 kB
Active(anon): 6145416 kB
Active(file): 2950064 kB
Buffers: 287952 kB
Cached: 4818144 kB
Inactive(file): 1578132 kB
Inactive(anon): 574764 kB
Inactive(file): 1567648 kB
MemAvailable: 6574984 kB
MemFree: 2057400 kB
MemTotal: 16325648 kB
Shmem: 577588 kB
SReclaimable: 346648 kB
r TiSd i r z,inactive memory stats couldn't be determinedl " )rl r r r r r rp rq r r r r r r r )rs rk r r r r/ r/ r0 test_avail_old_comes_from_kernelQ s
z8TestSystemVirtualMemory.test_avail_old_comes_from_kernelc C s t d t j d j t } t j d d } t j } Wd QRX| j sX t | j
| j d | d } | j d t
| j Wd QRXd S)
Nz
/proc/meminfoa Active: 9444728 kB
Active(anon): 6145416 kB
Buffers: 287952 kB
Cached: 4818144 kB
Inactive(file): 1578132 kB
Inactive(anon): 574764 kB
MemFree: 2057400 kB
MemTotal: 16325648 kB
Shmem: 577588 kB
r Tid i iI r z,inactive memory stats couldn't be determinedi }l 'L l `LG )rl r r r r r rp rq r r r r r r r )rs rk r r r r/ r/ r0 test_avail_old_missing_fieldsm s
z5TestSystemVirtualMemory.test_avail_old_missing_fieldsc C s t d t j d j t d t t j d c t j d d I } t
j } | j | j
d | d
} | j d t | j Wd QRXWd QRXWd QRXd S)Nz
/proc/meminfoaT Active: 9444728 kB
Active(anon): 6145416 kB
Active(file): 2950064 kB
Buffers: 287952 kB
Cached: 4818144 kB
Inactive(file): 1578132 kB
Inactive(anon): 574764 kB
Inactive(file): 1567648 kB
MemFree: 2057400 kB
MemTotal: 16325648 kB
Shmem: 577588 kB
SReclaimable: 346648 kB
z/proc/zoneinfozno such file or directoryr Tid i iI r z,inactive memory stats couldn't be determinedi }l 'L l `LG )rl r r r rn IOErrorerrnoENOENTr r rp rq r r r r r )rs r r r r/ r/ r0 test_avail_old_missing_zoneinfo s
z7TestSystemVirtualMemory.test_avail_old_missing_zoneinfoc s f d d } t t r$ d n d } t j | d d d | } t j } | j sc t | j | j d | j | j
d | j | j d | j | j d | j | j
d | j | j d | j | j d | j | j d | j | j d Wd QRXd S) Nc s< | d k r( t j t j d j S | | | Sd S)Nz
/proc/meminfoa= MemTotal: 100 kB
MemFree: 2 kB
MemAvailable: 3 kB
Buffers: 4 kB
Cached: 5 kB
SwapCached: 6 kB
Active: 7 kB
Inactive: 8 kB
Active(anon): 9 kB
Inactive(anon): 10 kB
Active(file): 11 kB
Inactive(file): 12 kB
Unevictable: 13 kB
Mlocked: 14 kB
SwapTotal: 15 kB
SwapFree: 16 kB
Dirty: 17 kB
Writeback: 18 kB
AnonPages: 19 kB
Mapped: 20 kB
Shmem: 21 kB
Slab: 22 kB
SReclaimable: 23 kB
SUnreclaim: 24 kB
KernelStack: 25 kB
PageTables: 26 kB
NFS_Unstable: 27 kB
Bounce: 28 kB
WritebackTmp: 29 kB
CommitLimit: 30 kB
Committed_AS: 31 kB
VmallocTotal: 32 kB
VmallocUsed: 33 kB
VmallocChunk: 34 kB
HardwareCorrupted: 35 kB
AnonHugePages: 36 kB
ShmemHugePages: 37 kB
ShmemPmdMapped: 38 kB
CmaTotal: 39 kB
CmaFree: 40 kB
HugePages_Total: 41 kB
HugePages_Free: 42 kB
HugePages_Rsvd: 43 kB
HugePages_Surp: 44 kB
Hugepagesize: 45 kB
DirectMap46k: 46 kB
DirectMap47M: 47 kB
DirectMap48G: 48 kB
)r\ r^ r r r )r_ r` ra )rd r/ r0 re s 1zETestSystemVirtualMemory.test_virtual_memory_mocked.<locals>.open_mockz
builtins.openz__builtin__.openrf Trg r i rP rw i i i i p i T i i i X i )rh r r ri rp rq r r r rL rA r r rQ r r r r )rs re rj rk memr/ )rd r0 test_virtual_memory_mocked s 7 z2TestSystemVirtualMemory.test_virtual_memory_mockedN)rw rw rx )__name__
__module__ __qualname__rv r skipIfr rZ r r} r r~ r r r r r r r r r r r r/ r/ r/ r0 ro s& ' ro c @ s e Z d Z e d d Z d d Z e d d Z e d d Z d d
Z d d Z
d
d Z d d Z d S)TestSystemSwapMemoryc C s8 t d } | j } Wd QRXd | k o7 d | k S)z3Return True if /proc/meminfo provides swap metrics.z
/proc/meminfoNz
SwapTotal:z SwapFree:)rh read)r datar/ r/ r0 meminfo_has_swap_info s z*TestSystemSwapMemory.meminfo_has_swap_infoc C s1 t j } t j j } | j | | d t S)Nry )rO rL rp swap_memoryrr r )rs r| ru r/ r/ r0 rv s z TestSystemSwapMemory.test_totalc C s1 t j } t j j } | j | | d t S)Nry )rO rM rp r rr r )rs r| ru r/ r/ r0 r} s z TestSystemSwapMemory.test_usedc C s1 t j } t j j } | j | | d t S)Nry )rO rA rp r rr r )rs r| ru r/ r/ r0 r~ s z TestSystemSwapMemory.test_freec C s t j d d d } t j d d } t j d t j } | j sU t | j t
| d | d } | j j d s t | j
d t | j | j | j d | j | j d Wd QRXWd QRXd S)
Nzpsutil._pslinux.openrf Tr r r9 r zpsutil/_pslinux.pyz9'sin' and 'sout' swap memory stats couldn't be determined)r ri r r r rp r r r r r r r r r r sinsout)rs rk r r r r/ r/ r0 test_missing_sin_sout s
z*TestSystemSwapMemory.test_missing_sin_soutc C s t d t t j d } t j d d } t j d t j } | j s[ t
| j t | d | d } | j
j d s t
| j d t | j | j | j d | j | j d Wd QRXWd QRXd S)
Nz /proc/vmstatzno such file or directoryr Tr r9 r zpsutil/_pslinux.pyzK'sin' and 'sout' swap memory stats couldn't be determined and were set to 0)rn r r r r r r rp r r r r r r r r r r r r )rs rk r r r r/ r/ r0 test_no_vmstat_mocked! s
z*TestSystemSwapMemory.test_no_vmstat_mockedc C s | j s t j d St j d } t j } Wd QRX| j sN t d d l j
} | j \ } } } } } } } | | 9} | | 9} | j | j
| | j | j | d S)Nz!/proc/meminfo has no swap metricsz"psutil._pslinux.cext.linux_sysinfor )r r skipr ri rp r r r Zpsutil._psutil_linuxZ
_psutil_linuxZ
linux_sysinfor rL rA )rs rk swapZcextrK rL rA Zunit_multiplierr/ r/ r0 test_meminfo_against_sysinfo4 s
!
z1TestSystemSwapMemory.test_meminfo_against_sysinfoc C s6 t d d } t j | j s+ t Wd QRXd S)Nz
/proc/meminfo )rl rp r r r )rs rk r/ r/ r0 #test_emulate_meminfo_has_no_metricsD s
z8TestSystemSwapMemory.test_emulate_meminfo_has_no_metricsN)
r r r staticmethodr rv r r} r~ r r r r r/ r/ r/ r0 r s r c @ s e Z d Z e j e d d d Z e j e j j d d d d Z
e j e j j d d d
d Z e j e d d
d d Z
e j e d d d d Z d d Z d d Z e j e d d d Z e j e d e j e d d d Z e j e d d d Z e j e d d d! Z e j e d e j e d d" d# Z d$ S)%
TestSystemCPUzunknown failure on travisc C s t j j } t j d t j d d } t t t | j
d } | d k rl | j d | n | j d | | d
k r | j d | n | j d | | d k r | j d | n | j d | d S)Nz
\d+\.\d+\.\d+r r rW stealr Zguestrw Z
guest_nice)r r r )r r r )rw r r )
rp cpu_times_fieldsrefindallosunamerX rY rF rB r assertNotIn)rs r Z
kernel_verZkernel_ver_infor/ r/ r0 test_cpu_timesU s z TestSystemCPU.test_cpu_timesz /sys/devices/system/cpu/onlinez-/sys/devices/system/cpu/online does not existc
C sq t d } | j j } Wd QRXd t | k rm t | j d d d } | j t j | d S)Nz /sys/devices/system/cpu/online-r9 ) rh r rT r rF rB r rp cpu_count)rs r valuer/ r/ r0 *test_cpu_count_logical_w_sysdev_cpu_onlineg s
z8TestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_onlinez/sys/devices/system/cpuz&/sys/devices/system/cpu does not existc C sB t j d } t d d | D } | j t j | d S)Nz/sys/devices/system/cpuc S s. g | ]$ } t j d | d k r | q S)zcpu\d+$N)r search)r5 r2 r/ r/ r0 r7 t s zITestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_num.<locals>.<listcomp>)r listdirr r rp r )rs Zlscountr/ r/ r0 'test_cpu_count_logical_w_sysdev_cpu_nump s z5TestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_numZnprocznproc utility not availablec C s2 t t d } | j t j d d | d S)Nz nproc --alllogicalT)rF r r rp r )rs numr/ r/ r0 test_cpu_count_logical_w_nprocw s z,TestSystemCPU.test_cpu_count_logical_w_nprocZlscpuzlscpu utility not availablec C sN t d } t d d | j d D } | j t j d d | d S)Nzlscpu -pc S s% g | ] } | j d s | q S)#)rC )r5 r2 r/ r/ r0 r7 s z@TestSystemCPU.test_cpu_count_logical_w_lscpu.<locals>.<listcomp>r@ r T)r r rB r rp r )rs rH r r/ r/ r0 test_cpu_count_logical_w_lscpu| s "z,TestSystemCPU.test_cpu_count_logical_w_lscpuc & C st d d l } | j j } t j d d t =} | j | j j | | j s[ t t j d d d K } | j | j j | j | j
d | j | j d d d Wd QRXt d d
} | j
} Wd QRXt j | } t j d d | d d } | j | j j | Wd QRXt d d ' } | j | j j | | j Wd QRXWd QRXd S)
Nr zpsutil._pslinux.os.sysconfrg zpsutil._pslinux.openrf Tr z
/proc/statz
/proc/cpuinforb return_valuer )r _pslinuxZcpu_count_logicalr ri rG r r r assertIsNoneZ
call_count call_argsrh r r\ r^ rl )rs rp Zoriginalrk r Z cpuinfo_data fake_filer/ r/ r0 test_cpu_count_logical_mocked s& " z+TestSystemCPU.test_cpu_count_logical_mockedc C sH t j d d d , } | j t j j | j s= t Wd QRXd S)Nzpsutil._pslinux.openrf T)r ri r rp r Zcpu_count_physicalr r )rs rk r/ r/ r0 test_cpu_count_physical_mocked s z,TestSystemCPU.test_cpu_count_physical_mockedz
not supportedc C s4 t j d d g | j t j Wd QRXd S)Nzpsutil._pslinux.glob.globr )r ri r rp cpu_freq)rs r/ r/ r0 test_cpu_freq_no_result s z%TestSystemCPU.test_cpu_freq_no_resultzfails on Travisc
ss f d d } g t j t j d d | d d - t j sR t | j t d Wd QRXd S)Nc s; | j d r j d g S j d | Sd S)Nz&/sys/devices/system/cpu/cpufreq/policy)rC append)pattern)flags orig_globr/ r0 glob_mock s
z>TestSystemCPU.test_cpu_freq_use_second_file.<locals>.glob_mockzpsutil._pslinux.glob.globrg rf Tr )globr ri rp r r r r )rs r r/ )r r r0 test_cpu_freq_use_second_file s
z+TestSystemCPU.test_cpu_freq_use_second_filec s f d d } t t r$ d n d } t j | d | j t j d d d g J t j } | j | j d | j | j d
| j | j d Wd QRXWd QRXd S) Nc sh | j d r t j d S| j d r8 t j d S| j d rT t j d S | | | Sd S)Nz/scaling_cur_freqs 500000z/scaling_min_freqs 600000z/scaling_max_freqs 700000)r r\ r^ )r_ r` ra )rd r/ r0 re s
z;TestSystemCPU.test_cpu_freq_emulate_data.<locals>.open_mockz
builtins.openz__builtin__.openrg z glob.globr z'/sys/devices/system/cpu/cpufreq/policy0g @@g @g @)
rh r r ri rp r r currentminmax)rs re rj freqr/ )rd r0 test_cpu_freq_emulate_data s
z(TestSystemCPU.test_cpu_freq_emulate_datac s f d d } t t r$ d n d } d d d g } t j | d | g t j d d
| J t j } | j | j d | j | j d | j | j d
Wd QRXWd QRXd S)Nc sh | j d r t j d S| j d r8 t j d S| j d rT t j d S | | | Sd S)Nz/scaling_cur_freqs 100000z/scaling_min_freqs 200000z/scaling_max_freqs 300000)r r\ r^ )r_ r` ra )rd r/ r0 re s
z@TestSystemCPU.test_cpu_freq_emulate_multi_cpu.<locals>.open_mockz
builtins.openz__builtin__.openz'/sys/devices/system/cpu/cpufreq/policy0z'/sys/devices/system/cpu/cpufreq/policy1z'/sys/devices/system/cpu/cpufreq/policy2rg z glob.globr g Y@g i@g r@)
rh r r ri rp r r r r r )rs re rj policiesr r/ )rd r0 test_cpu_freq_emulate_multi_cpu s
z-TestSystemCPU.test_cpu_freq_emulate_multi_cpuc s
f d d } t t r$ d n d } d d d g } t j | d | A t j d d
| $ t j } | j | j d Wd QRXWd QRX f d d } t t r d n d } t j | d | 5 t j d d
| | j t t j Wd QRXWd QRXd S)
Nc sT | j d r$ t t j d n, | j d r@ t j d S | | | Sd S)Nz/scaling_cur_freqr4 z/cpuinfo_cur_freqs 200000)r r r r r\ r^ )r_ r` ra )rd r/ r0 re s
zGTestSystemCPU.test_cpu_freq_no_scaling_cur_freq_file.<locals>.open_mockz
builtins.openz__builtin__.openz'/sys/devices/system/cpu/cpufreq/policy0z'/sys/devices/system/cpu/cpufreq/policy1z'/sys/devices/system/cpu/cpufreq/policy2rg z glob.globr c s\ | j d r$ t t j d n4 | j d rH t t j d n | | | Sd S)Nz/scaling_cur_freqr4 z/cpuinfo_cur_freq)r r r r )r_ r` ra )rd r/ r0 re s
)
rh r r ri rp r r r assertRaisesNotImplementedError)rs re rj r r r/ )rd r0 &test_cpu_freq_no_scaling_cur_freq_file s !z4TestSystemCPU.test_cpu_freq_no_scaling_cur_freq_fileN)r r r r r r r r pathexistsr r r r r r r r r r r r r r/ r/ r/ r0 r R s %% ! " r c @ sL e Z d Z e j e d d d Z e j e d d d Z d S)TestSystemCPUStatszfails on Travisc C s5 t d } t j j } | j | | d d d S)Nzcontext switchesry i )rV rp cpu_statsZ ctx_switchesrr )rs rt ru r/ r/ r0 test_ctx_switches" s z$TestSystemCPUStats.test_ctx_switchesc C s5 t d } t j j } | j | | d d d S)N
interruptsry i )rV rp r r rr )rs rt ru r/ r/ r0 test_interrupts( s z"TestSystemCPUStats.test_interruptsN)r r r r r r r r r/ r/ r/ r0 r s r c @ s e Z d Z d d Z d d Z e d d Z e j d d e e j d d
d d d
Z
d d Z d S)TestSystemNetworkc C s x t j j D]v \ } } xg | D]_ } | j t j k rZ | j | j t | q& | j t j k r& | j | j t
| q& Wq Wd S)N) rp Z net_if_addrsitemsfamilyZAF_LINKr addressr? r" r# r1 )rs r_ Zaddrsaddrr/ r/ r0 test_net_if_addrs_ips7 s
z'TestSystemNetwork.test_net_if_addrs_ipsc C sy xr t j j D]^ \ } } y t d | } Wn t k
rG Yq X| j | j t t j d | d q Wd S)Nz ifconfig %sz(?i)MTU[: ](\d+)r )
rp net_if_statsr r RuntimeErrorr ZmturF r r )rs r_ statsrH r/ r/ r0 test_net_if_stats@ s
z#TestSystemNetwork.test_net_if_statsc s[ f d d } t j d d d d } x*| j D] \ } y | } Wn t k
rj w7 Yn X| j | j | d d d | j | j | d d d | j | j | d d d | j | j | d
d d | j | j | d d d | j | j
| d d d | j | j | d d d | j | j | d d d q7 Wd S)Nc s i } t d } t t j d | d | d <t t j d | d | d <t t j d | d | d <t t j d | d | d
<t t j d | d | d <t t j d | d | d
<t t j d | d | d <t t j d | d | d <| S)Nz ifconfig %szRX packets[: ](\d+)r packets_recvzTX packets[: ](\d+) packets_sentzerrors[: ](\d+)errinr9 erroutzdropped[: ](\d+)dropindropoutz#RX (?:packets \d+ +)?bytes[: ](\d+)
bytes_recvz#TX (?:packets \d+ +)?bytes[: ](\d+)
bytes_sent)r rF r r )Znicr rH )r_ r/ r0 ifconfigN s z8TestSystemNetwork.test_net_io_counters.<locals>.ifconfigZpernicTnowrapFr ry i rP r r r r r r r r i i )
rp net_io_countersr r rr r r r r r r r r )rs r Znior Z ifconfig_retr/ )r_ r0 test_net_io_countersL s.
z&TestSystemNetwork.test_net_io_countersz psutil._pslinux.socket.inet_ntoprg z psutil._pslinux.supports_ipv6r Fc C se y9 t j t j t j } | j | j | j d Wn t j k
rP Yn Xt j d d d S)N::1r kindZinet6)r r ) r" AF_INET6 SOCK_STREAM
addCleanupclosebinderrorrp net_connections)rs Z
supports_ipv6 inet_ntopr. r/ r/ r0 %test_net_connections_ipv6_unsupported s z7TestSystemNetwork.test_net_connections_ipv6_unsupportedc
C sE t d t j d & } t j d d | j s: t Wd QRXd S)Nz/proc/net/unixa. 0: 00000003 000 000 0001 03 462170 @/tmp/dbus-Qw2hMPIU3n
0: 00000003 000 000 0001 03 35010 @/tmp/dbus-tB2X8h69BQ
0: 00000003 000 000 0001 03 34424 @/tmp/dbus-cHy80Y8O
000000000000000000000000000000000000000000000000000000
r! Zunix)rl r r rp r( r r )rs rk r/ r/ r0 test_net_connections_mocked s z-TestSystemNetwork.test_net_connections_mockedN) r r r r r r r r ri rG r* r+ r/ r/ r/ r0 r
4 s :$ r
c @ s e Z d Z e j e e d d 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 S)TestSystemDisksstatvfszos.statvfs() not availablec C s d d } x t j d d D] } t j | j } | | j \ } } } } | j | j | t | j | d k r | j d | j | f t | j | d k r | j d | j | f q Wd S) Nc S s t d | j } | j d } | j d | j d } | j d d \ } } } } | d k ru d } t | t | t | } } } | | | | f S)Nzdf -P -B 1 "%s"r@ r r Znoner4 )r rT rB poprF )r rH rI rJ devrL rM rA r/ r/ r0 df s
" &z:TestSystemDisks.test_disk_partitions_and_usage.<locals>.dfallFr i zpsutil=%s, df=%si ( i i ( i )
rp disk_partitions
disk_usageZ
mountpointr rL r rA failrM )rs r0 partusager/ rL rM rA r/ r/ r0 test_disk_partitions_and_usage s z.TestSystemDisks.test_disk_partitions_and_usagec C s t d d } | j } Wd QRXd | k rh x t j D] } | j d k r> Pq> W| j d n t j t d } t j
d d | d d v } t j
d
d d g T } t j } | j s t | j s t | s t | j
| d j d Wd QRXWd QRXd S)Nz/proc/filesystemsrzfsz couldn't find any ZFS partitionz
nodev zfs
zpsutil._pslinux.openr rf Tz$psutil._pslinux.cext.disk_partitions /dev/sdb3/rwr )r: r; r9 r< )rh r rp r2 Zfstyper4 r\ r] r r ri r r r )rs r r r5 r m1m2r r/ r/ r0 test_disk_partitions_mocked s$ z+TestSystemDisks.test_disk_partitions_mockedc C s t d d t j d d d t j d d } | j | j d | j | j d | j | j d
t | j | j
d | j | j d | j | j d
| j | j
d t | j | j d | j | j d Wd QRXWd QRXd S)Nz/proc/diskstatsz+ 3 0 1 hda 2 3 4 5 6 7 8 9 10 11 12z!psutil._pslinux.is_storage_devicer Tr Fr9 r rw r rP r r r r )rl r ri rp disk_io_countersr
read_countread_merged_count
read_bytes SECTOR_SIZE read_time write_countwrite_merged_count write_bytes
write_time busy_time)rs r r/ r/ r0 'test_disk_io_counters_kernel_2_4_mocked s
z7TestSystemDisks.test_disk_io_counters_kernel_2_4_mockedc C s t d d t j d d d t j d d } | j | j d | j | j d | j | j d
t | j | j
d | j | j d | j | j d
| j | j
d t | j | j d | j | j d Wd QRXWd QRXd S)Nz/proc/diskstatsz' 3 0 hda 1 2 3 4 5 6 7 8 9 10 11z!psutil._pslinux.is_storage_devicer Tr Fr9 r rw r rP r r r r )rl r ri rp r@ r rA rB rC rD rE rF rG rH rI rJ )rs r r/ r/ r0 ,test_disk_io_counters_kernel_2_6_full_mocked s
z<TestSystemDisks.test_disk_io_counters_kernel_2_6_full_mockedc C s t d d t j d d d t j d d } | j | j d | j | j d t | j | j d
| j | j
d t | j | j d | j | j d | j | j
d | j | j d | j | j d Wd QRXWd QRXd S)
Nz/proc/diskstatsz 3 1 hda 1 2 3 4z!psutil._pslinux.is_storage_devicer Tr Fr9 r rw r r )rl r ri rp r@ r rA rC rD rF rH rB rE rG rI rJ )rs r r/ r/ r0 /test_disk_io_counters_kernel_2_6_limited_mocked s
z?TestSystemDisks.test_disk_io_counters_kernel_2_6_limited_mockedc C s t d t j d t j d d d t j d d d d } | j t | d | j | d
j d | j | d j d | j | d
j
d
| j | d j
d
Wd QRXWd QRXd S)Nz/proc/diskstatsz 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11
3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11
z!psutil._pslinux.is_storage_devicer FperdiskTr r nvme0n1r9 Z nvme0n1p1rP ) rl r r r ri rp r@ r r rA rF )rs r r/ r/ r0 (test_disk_io_counters_include_partitions s
z8TestSystemDisks.test_disk_io_counters_include_partitionsc ! C s t d t j d G t j d d d * t j d d d d } | j | Wd QRXWd QRXd d } t d t j d f t j d d
d d | C t j d d d d } | j | j d
| j | j
d Wd QRXWd QRXd S)Nz/proc/diskstatsz 3 0 nvme0n1 1 2 3 4 5 6 7 8 9 10 11
3 0 nvme0n1p1 1 2 3 4 5 6 7 8 9 10 11
z!psutil._pslinux.is_storage_devicer FrN r c S s
| d k S)NrO r/ )r_ r/ r/ r0 is_storage_device: s zSTestSystemDisks.test_disk_io_counters_exclude_partitions.<locals>.is_storage_devicerf Trg r9 rP ) rl r r r ri rp r@ r r rA rF )rs r rQ r/ r/ r0 (test_disk_io_counters_exclude_partitions* s$
z8TestSystemDisks.test_disk_io_counters_exclude_partitionsc C ss d d } t j d d } t j d d d d | t j d d } Wd QRX| j t | t | d S)Nc S s | d k r d Sd S)Nz/proc/diskstatsFTr/ )r r/ r/ r0 r J s z;TestSystemDisks.test_disk_io_counters_sysfs.<locals>.existsrN Tz psutil._pslinux.os.path.existsrf rg )rp r@ r ri r r )rs r ZwprocfsZwsysfsr/ r/ r0 test_disk_io_counters_sysfsI s z+TestSystemDisks.test_disk_io_counters_sysfsc
C sF d d } t j d d d d | | j t t j Wd QRXd S)Nc S s d S)NFr/ )r r/ r/ r0 r V s z>TestSystemDisks.test_disk_io_counters_not_impl.<locals>.existsz psutil._pslinux.os.path.existsrf Trg )r ri r r rp r@ )rs r r/ r/ r0 test_disk_io_counters_not_implU s z.TestSystemDisks.test_disk_io_counters_not_implN)r r r r r hasattrr r r7 r? rK rL rM rP rR rS rT r/ r/ r/ r0 r, s r, c @ s e Z d Z d d Z e j d 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 s8 t d } t j } | j t | t | d S)Nz boot time)rV rp boot_timer rF )rs rt ru r/ r/ r0 test_boot_timef s zTestMisc.test_boot_timezpsutil.traceback.print_excc " s t j } t t j j | d d . } | j d | j d | j d Wd QRXzJt f d d } t r d n d } t j | d
| t
t | j s t
| j t t j | j t t j d d | j t t j | j t t j d d | j t t j | j t t j d d | t _ | j t j d
| j t t j d
t j d d } | j t | d
t j d d } | j t t t | d
t t j j | d d . } | j d | j d | j d Wd QRX| j t j d
| j t t j d d d
| j t t j d
| j t t t t j d d d
Wd QRXWd t j | t
t X| j t j d d S)NrU r zcpu 0 0 0 0 0 0 0 0 0 0
zcpu0 0 0 0 0 0 0 0 0 0 0
zcpu1 0 0 0 0 0 0 0 0 0 0
c s1 | j d r! t t j d | | | S)Nz/proczrejecting access for test)rC r r r )r_ r` ra )rd r/ r0 re w s z4TestMisc.test_no_procfs_on_import.<locals>.open_mockz
builtins.openz__builtin__.openrg percpuTr zcpu 1 0 0 0 0 0 0 0 0 0
zcpu0 1 0 0 0 0 0 0 0 0 0
zcpu1 1 0 0 0 0 0 0 0 0 0
z/proc)tempfilemkdtemprh r r r= writer r ri r rp r r r r r cpu_percentcpu_times_percent PROCFS_PATHr sumrY assertNotEqualshutilrmtree)rs tbZ my_procfsr re rj Zper_cpu_percentZper_cpu_times_percentr/ )rd r0 test_no_procfs_on_importk sN !
!
0
z!TestMisc.test_no_procfs_on_importc C s t d t j d j J } t j | j s: t t j d d t j t j d d Wd QRXt d t j d j } t j } | j s t t j d d } t j } t j d d } | j | d | j t
| d | j t
| d | j t
| d | j t
t t
| d | j t
t t
| d | j | j
d | j | j d Wd QRXd S)Nz
/proc/statz cpu 0 0 0 0 0 0 0 1 0 0
cpu0 0 0 0 0 0 0 0 1 0 0
cpu1 0 0 0 0 0 0 0 1 0 0
rY Tz cpu 1 0 0 0 0 0 0 0 0 0
cpu0 1 0 0 0 0 0 0 0 0 0
cpu1 1 0 0 0 0 0 0 0 0 0
r g Y@)rl r r r rp r] r r r^ ra r` rY r r user)rs rk r] Zcpu_percent_percpur^ Zcpu_times_percent_percpur/ r/ r0 test_cpu_steal_decrease s4
z TestMisc.test_cpu_steal_decreasec
C sH t j d d d , } | j t t j j | j s= t Wd QRXd S)Nzpsutil._pslinux.openrf T) r ri r r rp r rW r r )rs rk r/ r/ r0 test_boot_time_mocked s
z TestMisc.test_boot_time_mockedc C s t j d d d
g 3 } | j t j d j d
| j sG t Wd QRXt j d d d g 3 } | j t j d j d
| j s t Wd QRXt j d d d g 3 } | j t j d j d | j s t Wd QRXd S)Nzpsutil._pslinux.cext.usersr giampaolopts/2:0 hATr r localhost:0.0foo)ri rj rk rl Tr )ri rj rn rl Tr )ri rj ro rl Tr )r ri r rp Zusershostr r )rs rk r/ r/ r0 test_users_mocked s zTestMisc.test_users_mockedc C s t j } z | t _ | j t t j | j t t j | j t t j d d | j t t j | j t t j | j t t j
| j t t j | j t t j | j t j
t j Wd d t _ t j | Xd S)NrY Tz/proc)rZ r[ rp r_ r r rq r rW r( r r r2 Z
NoSuchProcessProcessr rmdir)rs Ztdirr/ r/ r0 test_procfs_path s zTestMisc.test_procfs_pathc
C s t } | j zn t j } | j d j } t j | sQ t | t j | } | j | j | t j
Wd | j Xd S)Nr9 ) r startrp rr threadsid
pid_existsr Zas_dictr Zpidsstop)rs tptidZptr/ r/ r0 test_issue_687 s
zTestMisc.test_issue_687c C sG t d d 1 } t j t j s- t | j s< t Wd QRXd S)Nz/proc/%s/statusr4 )rl rp rx r getpidr r )rs rk r/ r/ r0 test_pid_exists_no_proc_status s z'TestMisc.test_pid_exists_no_proc_statusN)
r r r rX r ri re rg rh rq rt r} r r/ r/ r/ r0 rV c s > ) rV z
no batteryc @ s e Z d Z e j e d d d d Z e j e d d 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)TestSensorsBatteryZacpizacpi utility not availablec C s` t d } t | j d d j j d d } t j j } | j | | d d d S)Nzacpi -b,r9 %r4 ry ) r rF rB rT replacerp sensors_batterypercentrr )rs rH Z
acpi_valueru r/ r/ r0 test_percent$ s +z TestSensorsBattery.test_percentc C sv t d } d | j k r+ t j d Sd | k r@ d } n d | j d d k } | j t j j | d S) Nzacpi -bunknownzacpi output not reliablezdischarging at zero rateTZChargingr@ r ) r lowerr r rB r rp r
power_plugged)rs rH Zpluggedr/ r/ r0 test_power_plugged+ s
z%TestSensorsBattery.test_power_pluggedc
s f d d } t t r$ d n d } t j | d | K } | j t j j d | j t j j t j | j
s t Wd QRXd S)Nc s? | j d s | j d r+ t j d S | | | Sd S)Nz
AC0/onlinez AC/online 1)r r\ r^ )r_ r` ra )rd r/ r0 re 8 s
z@TestSensorsBattery.test_emulate_power_plugged.<locals>.open_mockz
builtins.openz__builtin__.openrg T) rh r r ri r rp r r ZsecsleftZPOWER_TIME_UNLIMITEDr r )rs re rj rk r/ )rd r0 test_emulate_power_plugged6 s z-TestSensorsBattery.test_emulate_power_pluggedc
su f d d } t t r$ d n d } t j | d | / } | j t j j d | j sj t Wd QRXd S)Nc si | j d s | j d r3 t t j d n2 | j d rU t j t d S | | | Sd S)Nz
AC0/onlinez AC/onliner4 z/statusZcharging)r r r r r\ r] r )r_ r` ra )rd r/ r0 re I s
zBTestSensorsBattery.test_emulate_power_plugged_2.<locals>.open_mockz
builtins.openz__builtin__.openrg T)
rh r r ri r rp r r r r )rs re rj rk r/ )rd r0 test_emulate_power_plugged_2F s z/TestSensorsBattery.test_emulate_power_plugged_2c
su f d d } t t r$ d n d } t j | d | / } | j t j j d | j sj t Wd QRXd S)Nc s? | j d s | j d r+ t j d S | | | Sd S)Nz
AC0/onlinez AC/online 0)r r\ r^ )r_ r` ra )rd r/ r0 re Y s
zDTestSensorsBattery.test_emulate_power_not_plugged.<locals>.open_mockz
builtins.openz__builtin__.openrg F)
rh r r ri r rp r r r r )rs re rj rk r/ )rd r0 test_emulate_power_not_pluggedW s z1TestSensorsBattery.test_emulate_power_not_pluggedc
su f d d } t t r$ d n d } t j | d | / } | j t j j d | j sj t Wd QRXd S)Nc si | j d s | j d r3 t t j d n2 | j d rU t j t d S | | | Sd S)Nz
AC0/onlinez AC/onliner4 z/statusZ discharging)r r r r r\ r] r )r_ r` ra )rd r/ r0 re h s
zFTestSensorsBattery.test_emulate_power_not_plugged_2.<locals>.open_mockz
builtins.openz__builtin__.openrg F)
rh r r ri r rp r r r r )rs re rj rk r/ )rd r0 test_emulate_power_not_plugged_2e s z3TestSensorsBattery.test_emulate_power_not_plugged_2c sr f d d } t t r$ d n d } t j | d | , } | j t j j | j sg t Wd QRXd S)Nc sc | j d s | j d r3 t t j d n, | j d rO t j d S | | | Sd S)Nz"/sys/class/power_supply/AC0/onlinez!/sys/class/power_supply/AC/onliner4 z#/sys/class/power_supply/BAT0/statuss ???)rC r r r r\ r^ )r_ r` ra )rd r/ r0 re y s
zETestSensorsBattery.test_emulate_power_undetermined.<locals>.open_mockz
builtins.openz__builtin__.openrg )
rh r r ri r rp r r r r )rs re rj rk r/ )rd r0 test_emulate_power_undeterminedv s z2TestSensorsBattery.test_emulate_power_undeterminedc C s] t d t t j d ; t d t t j d | j t j Wd QRXWd QRXd S)Nz'/sys/class/power_supply/BAT0/energy_nowr4 z'/sys/class/power_supply/BAT0/charge_now)rn r r r r rp r )rs r/ r/ r0 test_emulate_no_base_files s z-TestSensorsBattery.test_emulate_no_base_filesc
C sE t d d / } | j t j j d | j s: t Wd QRXd S)Nz(/sys/class/power_supply/BAT0/energy_fullr r )rl r rp r r r r )rs rk r/ r/ r0 test_emulate_energy_full_0 s z-TestSensorsBattery.test_emulate_energy_full_0c C sz t d t t j d X t d t t j d 5 t d d | j t j j d Wd QRXWd QRXWd QRXd S)Nz(/sys/class/power_supply/BAT0/energy_fullr4 z(/sys/class/power_supply/BAT0/charge_fullz%/sys/class/power_supply/BAT0/capacitys 88X ) rn r r r rl r rp r r )rs r/ r/ r0 "test_emulate_energy_full_not_avail s
z5TestSensorsBattery.test_emulate_energy_full_not_availc C s t d t t j d a t d t t j d > t d t t j d | j t j j Wd QRXWd QRXWd QRXd S)Nz!/sys/class/power_supply/AC/onliner4 z"/sys/class/power_supply/AC0/onlinez#/sys/class/power_supply/BAT0/status)rn r r r r rp r r )rs r/ r/ r0 test_emulate_no_power s z(TestSensorsBattery.test_emulate_no_powerN)r r r r r r r r r r r r r r r r r r/ r/ r/ r0 r s %%
r c @ s: e Z d Z e j e d d d Z d d Z d S)TestSensorsTemperatureszunreliable on TRAVISc s f d d } t t r$ d n d } t j | d | e } t j d d I } | j t j i | j s| t
| j d t | d j
Wd QRXWd QRXd S)
Nc s8 | j d r$ t t j d n | | | Sd S)N_inputr4 )r OSErrorr ZEIO)r_ r` ra )rd r/ r0 re s zATestSensorsTemperatures.test_emulate_eio_error.<locals>.open_mockz
builtins.openz__builtin__.openrg r TZignoringr )rh r r ri r r r rp sensors_temperaturesr r r r r )rs re rj rk r r/ )rd r0 test_emulate_eio_error s z.TestSensorsTemperatures.test_emulate_eio_errorc s f d d } t t r$ d n d } t j | d | t j d d d g e t j d d
} | j | j d | j | j d | j | j d
| j | j
d Wd QRXWd QRXd S)Nc s | j d r" t j t d S| j d rD t j t d S| j d r` t j d S| j d r| t j d S| j d r t j d
S | | | Sd S) Nz/namer_ z /temp1_labellabelz /temp1_inputs 30000z
/temp1_maxs 40000z /temp1_crits 50000)r r\ r] r r^ )r_ r` ra )rd r/ r0 re s
z<TestSensorsTemperatures.test_emulate_data.<locals>.open_mockz
builtins.openz__builtin__.openrg z glob.globr z /sys/class/hwmon/hwmon0/temp1r_ r r g >@g D@g I@) rh r r ri rp r r r r highcritical)rs re rj Ztempr/ )rd r0 test_emulate_data s
z)TestSensorsTemperatures.test_emulate_dataN)r r r r r r r r r/ r/ r/ r0 r s r c @ s e Z d Z d d Z d S)TestSensorsFansc s f d d } t t r$ d n d } t j | d | _ t j d d d g ? t j d d
} | j | j d | j | j d Wd QRXWd QRXd S)
Nc sz | j d r" t j t d S| j d rD t j t d S| j d rf t j t d S | | | Sd S)Nz/namer_ z /fan1_labelr z /fan1_inputZ2000)r r\ r] r )r_ r` ra )rd r/ r0 re s z4TestSensorsFans.test_emulate_data.<locals>.open_mockz
builtins.openz__builtin__.openrg z glob.globr z /sys/class/hwmon/hwmon2/fan1r_ r r i ) rh r r ri rp Z sensors_fansr r r )rs re rj Zfanr/ )rd r0 r s
z!TestSensorsFans.test_emulate_dataN)r r r r r/ r/ r/ r0 r s r c @ s e Z d Z d d Z e Z 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 d d Z d d Z d d Z d d Z d d Z e j e d d d Z d d Z d! d" Z d# d$ Z d% S)& TestProcessc C s t t d S)N)r r )rs r/ r/ r0 setUp s zTestProcess.setUpc C s t j d t } t | } | j t t d d d t t j | j } t
j d | j } | j
d d } | j | j t d d | D d
d | j | j t d d | D d
d | j | j t d
d | D d
d d S)Nzk
import time
with open("%s", "w") as f:
time.sleep(10)
c S s
t j d S)NrW )r r r/ r/ r/ r0 <lambda> s z3TestProcess.test_memory_full_info.<locals>.<lambda>z'%s' not in retg?ZgroupedFc S s g | ] } | j | j q Sr/ )Z
private_dirtyZ
private_clean)r5 r2 r/ r/ r0 r7 s z5TestProcess.test_memory_full_info.<locals>.<listcomp>ry i c S s g | ] } | j q Sr/ )pss)r5 r2 r/ r/ r0 r7 s c S s g | ] } | j q Sr/ )r )r5 r2 r/ r/ r0 r7 s )r r r r r$ r r rp rr pidtimesleepmemory_full_info memory_mapsrr ussr` r r )rs srcZsprocr{ r mapsr/ r/ r0 test_memory_full_info s
&z!TestProcess.test_memory_full_infoc C s t d t j t j d j g } t j } | j } | j sR t
| j | j d | j | j
d | j | j d
Wd QRXd S)Nz/proc/%s/smapsa fffff0 r-xp 00000000 00:00 0 [vsyscall]
Size: 1 kB
Rss: 2 kB
Pss: 3 kB
Shared_Clean: 4 kB
Shared_Dirty: 5 kB
Private_Clean: 6 kB
Private_Dirty: 7 kB
Referenced: 8 kB
Anonymous: 9 kB
LazyFree: 10 kB
AnonHugePages: 11 kB
ShmemPmdMapped: 12 kB
Shared_Hugetlb: 13 kB
Private_Hugetlb: 14 kB
Swap: 15 kB
SwapPss: 16 kB
KernelPageSize: 17 kB
MMUPageSize: 18 kB
Locked: 19 kB
VmFlags: rd ex
r r i rw r
i l i i < )rl r r~ r r r rp rr r r r r r r r )rs rk r{ r r/ r/ r0 test_memory_full_info_mocked s
z(TestProcess.test_memory_full_info_mockedzunreliable on PYPYc ; C s d d } t t d | j | j d Wd QRXt t d | j | j d Wd QRXt t d | j | j d Wd QRXt t d | j | j d Wd QRXt t d | j | j d Wd QRXt t d | j | j d Wd QRXt rt t t t d | j | j d Wd QRXt t t t d
| j | j d Wd QRXd S) Nc S s| t j } t j d } xM xF | j D]8 } | j t j j t k rQ | St j | k r, Pq, Wq Wt d d S)Nr z timeout looking for test file) rp rr r
open_filesr r abspathr r )r{ Z giveup_atfiler/ r/ r0
get_test_fileB s z7TestProcess.test_open_files_mode.<locals>.get_test_filer r8 r zr+zw+za+r2 zx+)rh r r moder r )rs r r/ r/ r0 test_open_files_mode@ s(
z TestProcess.test_open_files_modec C s t j } | j } t j t | j d t | t j d d t t
j d / } | j } | s| t | j
s t Wd QRXt j d d t t
j d , } | j | j g | j
s t Wd QRXWd QRXd S)Nzlen(ret) != %izpsutil._pslinux.os.readlinkrg r4 )rp rr r rZ NamedTemporaryFiler r r ri r r r r r EINVALr )rs r{ filesrk r/ r/ r0 test_open_files_file_goned s
z%TestProcess.test_open_files_file_gonec C s t j } | j } t j t | j d t | t rK d n d } t j | d t
t j d / } | j } | s t
| j s t
Wd QRXWd QRXd S)Nzlen(ret) != %iz
builtins.openz__builtin__.openrg r4 )rp rr r rZ r r r r r ri r r r r r )rs r{ r rj rk r/ r/ r0 test_open_files_fd_gonex s
z#TestProcess.test_open_files_fd_gonec
C sW t j d d i ; } | j t j j t j j | j sL t
Wd QRXd S)Nz)psutil._pslinux._psposix.get_terminal_mapr ) r ri r rp r rr r r~ Zterminalr r )rs rk r/ r/ r0 test_terminal_mocked s %z TestProcess.test_terminal_mockedc C s t j } t j t d } t j d d | d d 2 } | j | j d d g | j sj t
Wd QRXt j t d } t j d d | d d 5 } | j | j d d d g | j s t
Wd QRXd S)
Nzfoo bar zpsutil._pslinux.openr rf Tro barz foo bar r4 ) rp rr r\ r] r r ri r cmdliner r )rs r{ r rk r/ r/ r0 test_cmdline_mocked s z TestProcess.test_cmdline_mockedc C s t j } t j t d } t j d d | d d 2 } | j | j d d g | j sj t
Wd QRXt j t d } t j d d | d d 5 } | j | j d d d g | j s t
Wd QRXd S)
Nzfoo bar zpsutil._pslinux.openr rf Tro r z foo bar r4 ) rp rr r\ r] r r ri r r r r )rs r{ r rk r/ r/ r0 test_cmdline_spaces_mocked s z&TestProcess.test_cmdline_spaces_mockedc
C sY t j d d d = | j t j j d | j t j j d Wd QRXd S)Nzpsutil._pslinux.os.readlinkr z/home/foo (deleted)z /home/foo)r ri r rp rr execwd)rs r/ r/ r0 !test_readlink_path_deleted_mocked s
z-TestProcess.test_readlink_path_deleted_mockedc s f d d } t t r$ d n d } t j | d | 8 } t j j } | j sc t | j | g Wd QRX f d d } t j | d | ! | j
t j t j j Wd QRXd S)Nc sB | j d t j r. t t j d n | | | Sd S)Nz
/proc/%s/taskr4 )rC r r~ r r r )r_ r` ra )rd r/ r0 re s z2TestProcess.test_threads_mocked.<locals>.open_mockz
builtins.openz__builtin__.openrg c sB | j d t j r. t t j d n | | | Sd S)Nz
/proc/%s/taskr4 )rC r r~ r r EPERM)r_ r` ra )rd r/ r0 re s ) rh r r ri rp rr rv r r r r AccessDenied)rs re rj rk r r/ )rd r0 test_threads_mocked s z TestProcess.test_threads_mockedc C s t j d d t t j d } t j d d t j d d } t j j } | j si t
| j sx t
| j | d t j d d d ! | j t j
t j j Wd QRXWd QRXWd QRXd S) Nzpsutil._pslinux.readlinkrg r4 zpsutil.Process.cmdliner z psutil._pslinux.os.path.lexistsr F)r ri r r r rp r rr r r r r r
ZombieProcess)rs r= r> r r/ r/ r0 test_exe_mocked s
zTestProcess.test_exe_mockedc C s t d t j t t j d ^ } t j } | j t } | j Wd QRX| j
| j j t j | j s t
Wd QRXd S)Nz/proc/%s/smapsr4 )rn r r~ r r r rp rr r r r exceptionr r )rs rk r{ errr/ r/ r0 test_issue_1014 s
zTestProcess.test_issue_1014z
not supportedc C s t j d d t t j d X } t j } | j | j t j } | j
t j Wd QRX| j su t
Wd QRX| j | j j | j | j | j j | j d S)Nz"psutil._pslinux.cext.linux_prlimitrg r4 )r ri r r ZENOSYSrp rr r_ r r ZrlimitZ
RLIMIT_NOFILEr r r r r )rs rk r{ rm r/ r/ r0 test_rlimit_zombie s
z TestProcess.test_rlimit_zombiec C s t j d d t t j d R } t j } | j | j t j } | j
Wd QRX| j so t Wd QRX| j
| j j | j | j
| j j | j d S)Nzpsutil._pslinux.os.readlinkrg r4 )r ri r r r rp rr r_ r r r r r r r r )rs rk r{ rm r/ r/ r0 test_cwd_zombie s
zTestProcess.test_cwd_zombiec ' C s d d l m } d d d d d d d d d d d d d d d d d
d d d d d d d d d d d d d d d d d d d d d d g' } d j | j } t d
t j | t j } | j | j
d | j | j t j | j | j
d | j | j d | t j | j } | j | j d | | j | j d | | j | j d | | j | j d | | j | j d Wd QRXd S)Nr ) CLOCK_TICKS0z(cat)Z123456rS z
/proc/%s/statcatr9 r r rw r rP )r r r= r rl r r~ rp rr r r_ statusZ
STATUS_ZOMBIEppidZ create_timerW r rf system
children_userchildren_systemZcpu_num)rs r r` rb r{ Zcpur/ r/ r0 test_stat_file_parsing sl z"TestProcess.test_stat_file_parsingc
C s: t d t j t j d j t j } | j | j j
d | j | j j d | j | j d | j
} | j | j d | j | j d | j | j d | j } | j | j d | j | j d
| j | j d | j | j j t t d d
Wd QRXd S)Nz/proc/%s/statusa Uid: 1000 1001 1002 1003
Gid: 1004 1005 1006 1007
Threads: 66
Cpus_allowed: f
Cpus_allowed_list: 0-7
voluntary_ctxt_switches: 12
nonvoluntary_ctxt_switches: 13rx r B i i i i i i r r )rl r r~ r r r rp rr r num_ctx_switches voluntary involuntary num_threadsuidsrealZ effectiveZsavedgids_proc_get_eligible_cpuslistrange)rs r{ r r r/ r/ r0 test_status_file_parsingM s"
z$TestProcess.test_status_file_parsingN)r r r r ZtearDownr r r r r
r r r r r r r r r r r
r r r r r/ r/ r/ r0 r s$ # $ ;r c @ s e Z d Z 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 e
d d Z d d Z d d Z d S)TestProcessAgainstStatusa /proc/pid/stat and /proc/pid/status have many values in common.
Whenever possible, psutil uses /proc/pid/stat (it's faster).
For all those cases we check that the value found in
/proc/pid/stat (by psutil) matches the one found in
/proc/pid/status.
c C s t j | _ d S)N)rp rr proc)clsr/ r/ r0
setUpClassp s z#TestProcessAgainstStatus.setUpClassc C s t j j d | j j } } xc | D][ } | j } | j | r& | j d d } y t | SWq& t k
r | SYq& Xq& Wt d | Wd QRXd S)Nz/proc/%s/status r z
can't find %r)
rp _psplatform open_textr r rT rC partitionrF rG )rs Z linestartr rJ r r/ r/ r0 read_status_filet s
z)TestProcessAgainstStatus.read_status_filec C s, | j d } | j | j j | d S)NzName:)r r r r_ )rs r r/ r/ r0 test_name s z"TestProcessAgainstStatus.test_namec C sd | j d } | | j d d | j d } | j d d } | j | j j | d S)NzState:(r9 )rS r )r findrfindr r r r )rs r r/ r/ r0 test_status s &z$TestProcessAgainstStatus.test_statusc C s, | j d } | j | j j | d S)NzPPid:)r r r r )rs r r/ r/ r0 test_ppid s z"TestProcessAgainstStatus.test_ppidc C s, | j d } | j | j j | d S)NzThreads:)r r r r )rs r r/ r/ r0 test_num_threads s z)TestProcessAgainstStatus.test_num_threadsc C sQ | j d } t t t | j d d } | j | j j | d S)NzUid:r9 r )r rX rY rF rB r r r )rs r r/ r/ r0 test_uids s %z"TestProcessAgainstStatus.test_uidsc C sQ | j d } t t t | j d d } | j | j j | d S)NzGid:r9 r )r rX rY rF rB r r r )rs r r/ r/ r0 test_gids s %z"TestProcessAgainstStatus.test_gidsc C sZ | j d } | j | j j j | | j d } | j | j j j | d S)Nzvoluntary_ctxt_switches:znonvoluntary_ctxt_switches:)r r r r r r )rs r r/ r/ r0 test_num_ctx_switches s z.TestProcessAgainstStatus.test_num_ctx_switchesc C so | j d } d t | k rk t t | j d \ } } | j | j j t t | | d d S)NzCpus_allowed_list:r r9 )
r r rY rF rB r r Z cpu_affinityr r )rs r Zmin_Zmax_r/ r/ r0 test_cpu_affinity s
z*TestProcessAgainstStatus.test_cpu_affinityc
C sp | j d } t j d } | j j j Wd QRXd t | k r] | j sl t n | j sl t d S)NzCpus_allowed_list:z psutil._pslinux.per_cpu_timesr ) r r ri r r r r r r )rs r rk r/ r/ r0 test_cpu_affinity_eligible_cpus s z8TestProcessAgainstStatus.test_cpu_affinity_eligible_cpusN)r r r __doc__ classmethodr r r r r r r r r r r r r/ r/ r/ r0 r g s
r c @ s@ e Z d Z d d Z d d Z d d Z d d Z d S)
TestUtilsc
C s3 t j j t } | j | j d Wd QRXd S)NZrt)rp r r __file__r r )rs r r/ r/ r0 test_open_text s zTestUtils.test_open_textc
C s3 t j j t } | j | j d Wd QRXd S)Nr )rp r r r r r )rs r r/ r/ r0 test_open_binary s zTestUtils.test_open_binaryc
C sN t j d d d 2 } | j t j j d d | j sC t Wd QRXd S)Nz os.readlinkr z
foo (deleted)r ro )r ri r rp r readlinkr r )rs rk r/ r/ r0
test_readlink s zTestUtils.test_readlinkc C s t j j t } t | d } | j d Wd QRX| j t j j t d d d | j t j j t d d d | j t j j t d d d
d
d S) NZwtzfoo binaryFro Ts fooz??fallbackr )
r r r r rh r\ r rp r r )rs Zfnamer r/ r/ r0 test_cat s ""zTestUtils.test_catN)r r r r r r r r/ r/ r/ r0 r s r __main__)Nr Z
__future__r rD r% r r r\ r r rb r" r+ rZ r r r rp r Zpsutil._compatr r r Z psutil.testsr r r r
r r r
r r r r r r r r r r r r r r r dirnamer ZHEREr* Z SIOCGIFCONFr; rD r1 r? rO rR rV rZ contextmanagerrl rn r ZTestCasero r r r r
r, rV r r r r r r r r/ r/ r/ r0 <module> s
5 ] o - g R