Your IP : 216.73.216.213
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 r�d 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 � �ascii�256s� � )�fcntlr �bytes�socket�AF_INET�
SOCK_DGRAM�
contextlib�closing� inet_ntoa�ioctl�fileno� SIOCGIFADDR�struct�pack)�ifnamer �s� r/ �J/opt/alt/python35/lib64/python3.5/site-packages/psutil/tests/test_linux.py�get_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/ )�.0�char)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�
ZSwap�freeztotal used freez&can't find 'Swap' in 'free' output:
%sN)r �split�
startswith� collections�
namedtuple�int�
ValueErrorr= )�out�lines�line�_�total�usedrA �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 �tuple�maprF 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 �io�StringIO�BytesIO)�name�args�kwargs)�content�for_path� orig_openr/ r0 � open_mock� s
z$mock_open_content.<locals>.open_mockz
builtins.openz__builtin__.open�createT� side_effectN)�openr r �patch)rc rb re � patch_point�mr/ )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 �psutil�virtual_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)N�delta�msgz %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
�recordT�alwaysr9 r zpsutil/_pslinux.pyz#memory stats couldn't be determined�cachedrQ r� r� r r� )rl �textwrap�dedent�encode�warnings�catch_warnings� simplefilterrp rq �called�AssertionError� assertEqual�len�filename�endswith�assertIn�str�messager� r� r� rQ r r� �slab)rs rk �ws�ret�wr/ 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� Zmems�frJ �fields�a�bZ 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� Ti�Sd 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� Ti�d i i�I 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� Ti�d i i�I r z,inactive memory stats couldn't be determinedi �}l 'L l `LG )rl r� r� r� rn �IOError�errno�ENOENTr� 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� �sin�sout)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�_fields�re�findall�os�unamerX 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 Zls�countr/ 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 --all�logicalT)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/cpuinfo�rb� return_valuer� )r� �_pslinuxZcpu_count_logicalr ri rG r� r� r� � assertIsNoneZ
call_count� call_argsrh r� r\ r^ rl )rs rp �originalrk 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� �current�min�max)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� � assertRaises�NotImplementedError)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� �path�existsr� 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_stats� 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_addrs�items�familyZAF_LINKr� �addressr? r" r# r1 )rs r_ �addrs�addrr/ 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� �mturF 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+)�dropin�dropoutz#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>.ifconfigZpernicT�nowrapFr 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 �kind�inet6)r$ r ) r" �AF_INET6� SOCK_STREAM�
addCleanup�close�bind�errorrp �net_connections)rs �
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% �unix)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 r0 r2 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)�TestSystemDisks�statvfszos.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>.df�allFr� i zpsutil=%s, df=%si ( i � i ( i � )
rp �disk_partitions�
disk_usage�
mountpointr� rL r� rA �failrM )rs r7 �part�usager6 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/filesystems�r�zfsz couldn't find any ZFS partitionz
nodev zfs
zpsutil._pslinux.openr� rf Tz$psutil._pslinux.cext.disk_partitions� /dev/sdb3�/�rwr )rB rC rA rD )rh r� rp r9 �fstyper< r\ r] r r ri r� r� r� )rs r� r� r= r� �m1�m2r� 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_count�read_merged_count�
read_bytes� SECTOR_SIZE� read_time� write_count�write_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 rI r� rJ rK rL rM rN rO rP rQ rR rS )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 rI r� rJ rL rM rO rQ rK rN rP rR rS )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� F�perdiskTr! r� �nvme0n1r9 Z nvme0n1p1rP ) rl r� r� r ri rp rI r� r� rJ rO )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� FrW r! c S s
| d k S)NrX 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 rI r� r� rJ rO )rs r� rZ 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>.existsrW Tz psutil._pslinux.os.path.existsrf rg )rp rI 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 rI )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 r? rH rT rU rV rY r[ r\ r] r/ r/ r/ r0 r3 � s r3 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)�tempfile�mkdtemprh r� r r= �writer r ri r rp r� r� r r� r� � cpu_percent�cpu_times_percent� PROCFS_PATHr� �sumrY �assertNotEqual�shutil�rmtree)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
rb 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 rf r� r� rg rj ri rY r� r� �user)rs rk rf Zcpu_percent_percpurg 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� r` 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� � giampaolo�pts/2�:0� �h�ATr� r Z localhost�:0.0�foo)rr rs rt ru Tr� )rr rs rv ru Tr� )rr rs rw ru Tr� )r ri r� rp Zusers�hostr� 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)Nrb Tz/proc)rc rd rp rh r r� rq r� r` r- r" r r9 Z
NoSuchProcess�Processr� �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 rz �threads�id�
pid_existsr� Zas_dictr� Zpids�stop)rs �t�p�tid�ptr/ 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 r� 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� ra r ri rn rp rq ry r| r� r� r/ r/ r/ r0 r_ c s > ) r_ 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_battery�percentrr )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 -b�unknownzacpi 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� �secsleft�POWER_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 88�X ) 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_label�labelz /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� �high�critical)rs re rj �tempr/ )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_input�2000)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 rz �pid�time�sleep�memory_full_info� memory_mapsrr �ussri 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 rz 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 r�t 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 rz r� �
open_filesr r� �abspathr r )r� Z giveup_at�filer/ r/ r0 �
get_test_fileB s z7TestProcess.test_open_files_mode.<locals>.get_test_filer� r@ 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 rz r� rc �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 rz r� rc 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� rz r� r� �terminalr� 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 Trw �barz foo bar r4 ) rp rz 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 Trw r� z foo bar r4 ) rp rz 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 rz �exe�cwd)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 rz r~ 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� rz r� r� r� r� r �
ZombieProcess)rs rF rG 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 rz 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 rz 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 rz 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_TICKS�0z(cat)�Z�1�2�3�4�5�6rS z
/proc/%s/stat�catr9 r� r� rw r� rP )r� r� r= r� rl r� r� rp rz r� r_ �status�
STATUS_ZOMBIE�ppidZ create_timer` r� ro �system�
children_user�children_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 rz r� �num_ctx_switches� voluntary� involuntary� num_threads�uids�real� effective�saved�gids�_proc�_get_eligible_cpus�list�range)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 rz �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 �find�rfindr� 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� rw )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) N�wtzfoo �binaryFrw Ts fooz??�fallbackr� )
r� r r� r rh re r� rp r r� )rs �fnamer� 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 �
__future__r rD r% r� r� r\ r� r� rk r" r+ rc 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 �HEREr* Z SIOCGIFCONFr; rM r1 r? rO rR rV rZ �contextmanagerrl rn r� ZTestCasero r� r� r r r3 r_ r� r� r� r� r� r r� r/ r/ r/ r0 �<module> s�
� 5 ] � o � � � - � g R