o
    a@l                    @   sh  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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+ erddl,m-Z- ddl,m.Z. ddl,m/Z/ dd l,m0Z0 ej12ej13e4Z5d!Z6d"Z7d#Z8d$Z9d%Z:er	d&Z;ed' Z<d(d) Z=d*d+ Z>d,d- Z?d.d/ Z@d0d1 ZAd2d3 ZBd4d5 ZCd6d7 ZDd8d9 ZEejFd:d; ZGejFd<d= ZHe*Ie d>G d?d@ d@e!ZJe*Ie d>G dAdB dBe!ZKe*Ie d>G dCdD dDe!ZLe*Ie d>G dEdF dFe!ZMe*Ie d>G dGdH dHe!ZNe*Ie d>G dIdJ dJe!ZOe*Ie d>G dKdL dLe!ZPe*Ie d>G dMdN dNe!ZQe*Ie d>G dOdP dPe!ZRe*Ie d>G dQdR dRe!ZSe*Ie d>G dSdT dTe!ZTe*Ie d>G dUdV dVe!ZUe*Ie d>G dWdX dXe!ZVe*Ie d>G dYdZ dZe!ZWe*Ie d>G d[d\ d\e!ZXe*Ie d>G d]d^ d^e!ZYe*Ie d>e*Ie d_G d`da dae!ZZe*Ie d>G dbdc dce!Z[e*Ie d>G ddde dee!Z\e*Ie d>G dfdg dge!Z]e*Ie d>G dhdi die!Z^e*Ie d>G djdk dke!Z_e*Ie d>G dldm dme!Z`eadnkrddolbmcZc ece4 dS dS )pzLinux specific tests.    )divisionN)LINUX)PY3)FileNotFoundError)
basestring)u)GITHUB_ACTIONS)GLOBAL_TIMEOUT)HAS_BATTERY)HAS_CPU_FREQ)HAS_GETLOADAVG)
HAS_RLIMIT)PYPY)TOLERANCE_DISK_USAGE)TOLERANCE_SYS_MEM)PsutilTestCase)
ThreadTask)
call_until)mock)reload_module)retry_on_failure)safe_rmpath)sh)skip_on_not_implemented)unittest)which)CLOCK_TICKS)RootFsDeviceFinder)calculate_avail_vmem)open_binaryi  i  i'  i  i  i   z/sys/class/hwmon/hwmon*c                 C      dd l }| d d } trt| d} ttjtj}t| t|	|
 ttd| dd W  d    S 1 s=w   Y  d S Nr      ascii256s      )fcntlr   bytessocketAF_INET
SOCK_DGRAM
contextlibclosing	inet_ntoaioctlfilenoSIOCGIFADDRstructpackifnamer'   s r7   9/usr/lib/python3/dist-packages/psutil/tests/test_linux.pyget_ipv4_addressL      


$r9   c                 C   r    r!   )r'   r   r(   r)   r*   r+   r,   r-   r.   r/   r0   SIOCGIFNETMASKr2   r3   r4   r7   r7   r8   get_ipv4_netmaskY   r:   r<   c                 C   r    r!   )r'   r   r(   r)   r*   r+   r,   r-   r.   r/   r0   SIOCGIFBRDADDRr2   r3   r4   r7   r7   r8   get_ipv4_broadcastf   r:   r>   c           	      C   s   t dd!}| D ]}| }|d | kr nq
td|  W d    n1 s)w   Y  |d }g }tdt|dD ]}||||d   q<d|}t	tj
|}ttj
|S )Nz/proc/net/if_inet6rtzcould not find interface %rr      :)open	readlinessplit
ValueErrorrangelenappendjoinr)   Z	inet_ptonAF_INET6	inet_ntop)	r5   flinefieldsZunformattedgroupsi	formattedZpackedr7   r7   r8   get_ipv6_addresss   s    
rS   c              	      s   dd l }| d d } trt| d} ttjtj}t|7 ||	 t
td| }tr3dd  ndd l}|j d fdd	|d
d D d d W  d    S 1 sXw   Y  d S )Nr   r"   r#   r$   c                 S   s   | S Nr7   xr7   r7   r8   ord      zget_mac_address.<locals>.ord c                    s   g | ]}d  | qS )z%02x:r7   ).0charrW   r7   r8   
<listcomp>       z#get_mac_address.<locals>.<listcomp>   r&   r@   )r'   r   r(   r)   r*   r+   r,   r-   r/   r0   SIOCGIFHWADDRr2   r3   __builtin__rW   rJ   )r5   r'   r6   infora   r7   r\   r8   get_mac_address   s   

&$rc   c                  C   s|   t ddgddid} | d}|D ]#}|dr4| \}}}}tdd}|t|t|t|  S qtd	d| )
zQParse 'free' cmd and return swap memory's s total, used and free
    values.
    free-bLANGC.UTF-8env
ZSwapztotal used freez&can't find 'Swap' in 'free' output:
%s)r   rE   
startswithcollections
namedtupleintrF   rJ   )outlinesrN   _totalusedrd   ntr7   r7   r8   	free_swap   s   

ru   c                  C   s   t ddgddid} | d}|D ](}|dr9dd	 | d
d D \}}}}tdd}||||||   S qtdd| )zSParse 'free' cmd and return physical memory's total, used
    and free values.
    rd   re   rf   rg   rh   rj   ZMemc                 S   s   g | ]}t |qS r7   )rn   rZ   rV   r7   r7   r8   r]      s    z free_physmem.<locals>.<listcomp>      ztotal used free shared outputz%can't find 'Mem' in 'free' output:
%s)r   rE   rk   rl   rm   rF   rJ   )ro   rp   rN   rr   rs   rd   sharedrt   r7   r7   r8   free_physmem   s   


rz   c                 C   sV   t ddgddid}|dD ]}| }| |v r$t|dd   S qtd	|  )
Nvmstatz-srf   rg   rh   rj    r   z can't find %r in 'vmstat' output)r   rE   striprn   rF   )statro   rN   r7   r7   r8   r{      s   r{   c                  C   s>   t ddg } d| v rtdttt|  d dS )Nrd   z-VZUNKNOWNzcan't determine free versionr@   .)r   r}   r   SkipTesttuplemaprn   rE   )ro   r7   r7   r8   get_free_version_info   s   
r   c                 #   ^     fdd}t trdnd}tj|d|d}|V  W d   dS 1 s(w   Y  dS )zMock open() builtin and forces it to return a certain `content`
    on read() if the path being opened matches `for_path`.
    c                    sJ   | krt rt trt S t S t S | g|R i |S rT   )r   
isinstancer   ioStringIOBytesIOnameargskwargscontentfor_path	orig_openr7   r8   	open_mock   s   



z$mock_open_content.<locals>.open_mockbuiltins.open__builtin__.openTcreateside_effectNrC   r   r   patch)r   r   r   patch_pointmr7   r   r8   mock_open_content   s   "r   c                 #   r   )zZMock open() builtin and raises `exc` if the path being opened
    matches `for_path`.
    c                    s"   | kr | g|R i |S rT   r7   r   excr   r   r7   r8   r      s   z&mock_open_exception.<locals>.open_mockr   r   Tr   Nr   )r   r   r   r   r   r7   r   r8   mock_open_exception   s   "r   z
LINUX onlyc                   @   s   e Zd Zdd Ze dd Ze dd Ze dd Ze d	d
 Ze dd Z	e dd Z
e dd Zdd Ze dd Zdd Zdd Zdd Zdd ZdS )TestSystemVirtualMemoryc                 C   *   t dd }t j}| j||td d S )Nztotal memory   delta)r{   psutilvirtual_memoryrr   assertAlmostEqualr   selfZvmstat_valuepsutil_valuer7   r7   r8   
test_total   s
   

z"TestSystemVirtualMemory.test_totalc                 C   sL   t  dk r
| dt }|j}t j}| j||td|||jf d d S )N)   r      zold free version	%s %s 
%sr   msg)	r   ZskipTestrz   rs   r   r   r   r   outputr   rd   
free_valuer   r7   r7   r8   	test_used  s   



z!TestSystemVirtualMemory.test_usedc                 C   r   )Nzfree memoryr   r   )r{   r   r   rd   r   r   r   r7   r7   r8   	test_free  
   

z!TestSystemVirtualMemory.test_freec                 C   r   )Nzbuffer memoryr   r   )r{   r   r   buffersr   r   r   r7   r7   r8   test_buffers  r   z$TestSystemVirtualMemory.test_buffersc                 C   r   )Nzactive memoryr   r   )r{   r   r   activer   r   r   r7   r7   r8   test_active   r   z#TestSystemVirtualMemory.test_activec                 C   r   )Nzinactive memoryr   r   )r{   r   r   inactiver   r   r   r7   r7   r8   test_inactive'  r   z%TestSystemVirtualMemory.test_inactivec                 C   sJ   t  }|j}|dkrtdt j}| j||td|||jf d d S )Nr   z%free does not support 'shared' columnr   r   )	rz   ry   r   r   r   r   r   r   r   r   r7   r7   r8   test_shared.  s   


z#TestSystemVirtualMemory.test_sharedc                 C   sj   t ddg}|d}d|d vrtdt|d  d }t j}| j||t	d	|||f d
 d S )Nrd   re   rj   	availabler   z(free does not support 'available' columnrw   r@   r   r   )
r   rE   r   r   rn   r   r   r   r   r   )r   ro   rp   r   r   r7   r7   r8   test_available9  s   



z&TestSystemVirtualMemory.test_availablec              	   C   s  t dtd }tjdd}td t }|j	s J | 
t|d |d }|jds4J | d	t|j | d
t|j | dt|j | dt|j | dt|j | dt|j | dt|j | 
|jd | 
|jd | 
|jd | 
|jd | 
|jd | 
|jd | 
|jd W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )N/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
                Trecordalwaysrw   r   psutil/_pslinux.pyz#memory stats couldn't be determinedcachedry   r   r   r   r   )r   textwrapdedentencodewarningscatch_warningssimplefilterr   r   calledassertEqualrH   filenameendswithassertInstrmessager   r   r   ry   r   r   slabr   r   wsretwr7   r7   r8   test_warnings_on_missesH  sB   	


"z/TestSystemVirtualMemory.test_warnings_on_missesc                 C   s   i }t d}|D ]}| }t|d d ||d < q	W d    n1 s&w   Y  t|}d|v rI|d }t|| | d }| |d d S d S )Nr   rw   r   r   s   MemAvailable:d   r"   )r   rE   rn   r   absZ
assertLess)r   ZmemsrM   rN   rO   abZdiff_percentr7   r7   r8   test_avail_old_percento  s   
z.TestSystemVirtualMemory.test_avail_old_percentc              	   C      t dtd <}tjdd}t }W d    n1 s w   Y  |js*J | 	|j
d |d }| dt|j W d    d S 1 sIw   Y  d S )Nr   aL                  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
                Tr   l     " r   ,inactive memory stats couldn't be determinedr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r7   r7   r8    test_avail_old_comes_from_kernel  s    


"z8TestSystemVirtualMemory.test_avail_old_comes_from_kernelc              	   C   r   )Nr   a                      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
                    Tr       `LG r   r   r   r   r7   r7   r8   test_avail_old_missing_fields  s    



"z5TestSystemVirtualMemory.test_avail_old_missing_fieldsc              
   C   s   t dtd X tdttjd6 tj	dd }t
 }| |jd |d }| d	t|j W d    n1 s>w   Y  W d    n1 sMw   Y  W d    d S W d    d S 1 sew   Y  d S )
Nr   aT                      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/zoneinfono such file or directoryTr   r   r   r   )r   r   r   r   r   IOErrorerrnoENOENTr   r   r   r   r   r   r   r   r   )r   r   r   r   r7   r7   r8   test_avail_old_missing_zoneinfo  s2   
"z7TestSystemVirtualMemory.test_avail_old_missing_zoneinfoc                    s    fdd}t  trdnd}tj|d|dQ}t }|js J | |jd | |j	d | |j
d	 | |jd
 | |jd | |jd | |jd | |jd | |jd W d    d S 1 sjw   Y  d S )Nc                    s2   | dkrt td S  | g|R i |S )Nr   a=	                      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   r7   r8   r     s
   13zETestSystemVirtualMemory.test_virtual_memory_mocked.<locals>.open_mockr   r   Tr   i  i      i p  i T  i   i    i X     )rC   r   r   r   r   r   r   r   rr   rd   r   r   ry   r   r   r   r   )r   r   r   r   memr7   r   r8   test_virtual_memory_mocked  s    7
"z2TestSystemVirtualMemory.test_virtual_memory_mockedN)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r7   r7   r7   r8   r      s.    	







'
r   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   s@   t d} |  }W d   n1 sw   Y  d|v od|v S )z3Return True if /proc/meminfo provides swap metrics.r   Nz
SwapTotal:z	SwapFree:)rC   read)rM   datar7   r7   r8   meminfo_has_swap_info%  s   

z*TestSystemSwapMemory.meminfo_has_swap_infoc                 C   "   t  j}t j}| j||tdS Nr   )ru   rr   r   swap_memoryr   r   r   r   r   r7   r7   r8   r   ,  s
   
zTestSystemSwapMemory.test_totalc                 C   r   r   )ru   rs   r   r   r   r   r   r7   r7   r8   r   2  
   
zTestSystemSwapMemory.test_usedc                 C   r   r   )ru   rd   r   r   r   r   r   r7   r7   r8   r   9  r   zTestSystemSwapMemory.test_freec              	   C   s   t jddd`}tjddA}td t }|jsJ | t	|d |d }|j
ds1J | d	t|j | |jd | |jd W d    n1 sRw   Y  W d    d S W d    d S 1 sjw   Y  d S )
Npsutil._common.openTr   r   r   rw   r   r   z9'sin' and 'sout' swap memory stats couldn't be determined)r   r   r   r   r   r   r   r   r   rH   r   r   r   r   r   sinsoutr   r7   r7   r8   test_missing_sin_sout@  s$   

"z*TestSystemSwapMemory.test_missing_sin_soutc              	   C   s   t dttjd`}tjddA}td t }|j	sJ | 
t|d |d }|jds3J | d	t|j | 
|jd | 
|jd W d    n1 sTw   Y  W d    d S W d    d S 1 slw   Y  d S )
Nz/proc/vmstatr   Tr   r   rw   r   r   zK'sin' and 'sout' swap memory stats couldn't be determined and were set to 0)r   r   r   r   r   r   r   r   r   r   r   rH   r   r   r   r   r   r   r   r   r7   r7   r8   test_no_vmstat_mockedO  s,   


"z*TestSystemSwapMemory.test_no_vmstat_mockedc                 C   s   |   s	tdS td}t }W d    n1 sw   Y  |jr'J dd lm	} |
 \}}}}}}}||9 }||9 }| |j| | j|j|td d S )Nz!/proc/meminfo has no swap metricsz"psutil._pslinux.cext.linux_sysinfor   r   )r   r   skipr   r   r   r   r   Zpsutil._psutil_linuxZ_psutil_linuxZlinux_sysinfor   rr   r   rd   r   )r   r   swapZcextrq   rr   rd   Zunit_multiplierr7   r7   r8   test_meminfo_against_sysinfob  s   


z1TestSystemSwapMemory.test_meminfo_against_sysinfoc                 C   sB   t dd}t  |jsJ W d    d S 1 sw   Y  d S )Nr       )r   r   r   r   r   r   r7   r7   r8   #test_emulate_meminfo_has_no_metricsr  s   "z8TestSystemSwapMemory.test_emulate_meminfo_has_no_metricsN)r   r   r   staticmethodr   r   r   r   r   r   r   r  r  r7   r7   r7   r8   r   "  s    


r   c                   @      e Zd Zdd ZdS )TestSystemCPUTimesc                 C   s   t  j}tdt d d }ttt	|
d}|dkr&| d| n| d| |dkr7| d| n| d| |d	krI| d
| d S | d
| d S )Nz\d+\.\d+\.\d+   r   r   )r        steal)r  r	  r&   Zguest)r   r  r   Z
guest_nice)r   	cpu_times_fieldsrefindallosunamer   r   rn   rE   r   assertNotIn)r   rO   Z
kernel_verZkernel_ver_infor7   r7   r8   test_fields  s   
zTestSystemCPUTimes.test_fieldsN)r   r   r   r  r7   r7   r7   r8   r        r  c                   @   s   e Zd Zeejd ddd Zeejd ddd Z	ee
d	 d
dd Zee
d ddd Zdd ZdS )TestSystemCPUCountLogical/sys/devices/system/cpu/onlinez-/sys/devices/system/cpu/online does not existc                 C   sn   t d}|  }W d    n1 sw   Y  dt|v r5t|dd d }| t | d S d S )Nr  -rw   )	rC   r   r}   r   rn   rE   r   r   	cpu_count)r   rM   valuer7   r7   r8   test_against_sysdev_cpu_online  s   
z8TestSystemCPUCountLogical.test_against_sysdev_cpu_online/sys/devices/system/cpuz&/sys/devices/system/cpu does not existc                 C   s0   t d}tdd |D }| t | d S )Nr  c                 S   s    g | ]}t d |dur|qS )zcpu\d+$N)r  searchrv   r7   r7   r8   r]     s     zITestSystemCPUCountLogical.test_against_sysdev_cpu_num.<locals>.<listcomp>)r  listdirrH   r   r   r  )r   Zlscountr7   r7   r8   test_against_sysdev_cpu_num  s   
z5TestSystemCPUCountLogical.test_against_sysdev_cpu_numZnprocznproc utility not availablec                 C   s$   t td}| tjdd| d S )Nznproc --allTZlogical)rn   r   r   r   r  )r   numr7   r7   r8   test_against_nproc  s   z,TestSystemCPUCountLogical.test_against_nproclscpulscpu utility not availablec                 C   s8   t d}tdd |dD }| tjdd| d S )Nlscpu -pc                 S   s   g | ]	}| d s|qS )#rk   rv   r7   r7   r8   r]     s    z@TestSystemCPUCountLogical.test_against_lscpu.<locals>.<listcomp>rj   Tr   )r   rH   rE   r   r   r  )r   ro   r!  r7   r7   r8   test_against_lscpu  s   z,TestSystemCPUCountLogical.test_against_lscpuc              	   C   s  dd l }|j }tjdtd}| |j | |jsJ tjddd"}| |j  | |j	d | |j
d d d W d    n1 sKw   Y  td	d
}| }W d    n1 sdw   Y  t|}tjd|dd}| |j | W d    n1 sw   Y  td	d}| |j | |j W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )Nr   zpsutil._pslinux.os.sysconfr   r   Tr   r  
/proc/stat/proc/cpuinforbreturn_valuer   r  )psutil._pslinux_pslinuxZcpu_count_logicalr   r   rF   r   r   assertIsNoneZ
call_countZ	call_argsrC   r   r   r   r   )r   r   Zoriginalr   rM   Zcpuinfo_data	fake_filer7   r7   r8   test_emulate_fallbacks  s<   



"z0TestSystemCPUCountLogical.test_emulate_fallbacksN)r   r   r   r   skipIfr  pathexistsr  r  r   r"  r(  r3  r7   r7   r7   r8   r    s    



r  c                   @   s6   e Zd Zeed ddd Zdd Zdd Zd	S )
TestSystemCPUCountCoresr#  r$  c                 C   s\   t d}t }|dD ]}|ds|d}||d  q| tjddt| d S )Nr%  rj   r&  ,rw   Fr   )	r   setrE   rk   addr   r   r  rH   )r   ro   Zcore_idsrN   rO   r7   r7   r8   r(    s   

z*TestSystemCPUCountCores.test_against_lscpuc                 C   sh   t j }tjdg d}t j }|jsJ W d    n1 s!w   Y  |d ur2| || d S d S )N	glob.globr.  )r   r0  cpu_count_coresr   r   r   r   )r   Zmeth_1r   Zmeth_2r7   r7   r8   test_method_2  s   

z%TestSystemCPUCountCores.test_method_2c              	   C   s   t jdg d'}t jddd}| tj  W d    n1 s"w   Y  W d    n1 s1w   Y  |js;J |js@J d S )Nr;  r<  r   Tr   )r   r   r1  r   r0  r=  r   )r   m1m2r7   r7   r8   test_emulate_none  s   
z)TestSystemCPUCountCores.test_emulate_noneN)	r   r   r   r   r4  r   r(  r>  rA  r7   r7   r7   r8   r7    s
    
	r7  c                   @   sz   e Zd Zee ddd Zee ddd Zee ddd Zee ddd	 Z	ee dd
d Z
dS )TestSystemCPUFrequencynot supportedc                    sV    fdd}t jj tjd|dd t sJ W d    d S 1 s$w   Y  d S )Nc                       |  drdS  | S )N&/sys/devices/system/cpu/cpufreq/policyFr'  r5  Zorig_existsr7   r8   path_exists_mock     
zMTestSystemCPUFrequency.test_emulate_use_second_file.<locals>.path_exists_mockos.path.existsT)r   r   )r  r5  r6  r   r   r   cpu_freq)r   rH  r7   rG  r8   test_emulate_use_second_file  s   "z3TestSystemCPUFrequency.test_emulate_use_second_filec                    s    fdd}t jj zatjd|d: ttj t }|s J | 	|j
d | 	|jd tjddD ]}| 	|j
d | 	|jd q4W d    n1 sOw   Y  W ttj tt d S W ttj tt d S ttj tt w )Nc                    rD  )Nz/sys/devices/system/cpu/Fr'  rF  Zos_path_existsr7   r8   rH    rI  zITestSystemCPUFrequency.test_emulate_use_cpuinfo.<locals>.path_exists_mockrJ  r)          TZpercpu)r  r5  r6  r   r   r   r   r0  rK  r   maxmin)r   rH  r   freqr7   rM  r8   test_emulate_use_cpuinfo  s*   






z/TestSystemCPUFrequency.test_emulate_use_cpuinfoc              	      s    fdd}t  trdnd}tj||dK tjddd+ t }| |jd	 |jd
kr5| |jd |j	d
krA| |j	d W d    n1 sKw   Y  W d    d S W d    d S 1 scw   Y  d S )Nc                    s   |  dr| drtdS |  dr| drtdS |  dr-| dr-tdS | dkr6td	S  | g|R i |S )
N/scaling_cur_freqrE     500000/scaling_min_freq   600000/scaling_max_freqs   700000r+  s   cpu MHz		: 500r   rk   r   r   r   r   r7   r8   r     s   






z;TestSystemCPUFrequency.test_emulate_data.<locals>.open_mockr   r   r)  rJ  Tr<       @@rN       @g     @
rC   r   r   r   r   rK  r   currentrQ  rP  r   r   r   rR  r7   r   r8   test_emulate_data  s$   

"z(TestSystemCPUFrequency.test_emulate_datac              
      sl   fdd}t  trdnd}tj||d tjdddw tjd	d
d` tjdd}| |d jd |d jdkrE| |d jd |d j	dkrU| |d j	d | |d jd |d jdkrn| |d jd |d j	dkr~| |d j	d W d    n1 sw   Y  W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )Nc                    s   | }| dr|drtdS | dr |dr tdS | dr/|dr/tdS | dr>|dr>td	S | drM|drMtd
S | dr\|dr\tdS | dkretdS  | g|R i |S )NrT  z'/sys/devices/system/cpu/cpufreq/policy0s   100000rV     200000rX  s   300000z'/sys/devices/system/cpu/cpufreq/policy1s   400000rU  rW  r+  s   cpu MHz		: 100
cpu MHz		: 400rY  )r   r   r   nr   r7   r8   r   =  s8   












z@TestSystemCPUFrequency.test_emulate_multi_cpu.<locals>.open_mockr   r   r)  rJ  Tr<  !psutil._pslinux.cpu_count_logicalr  rO  r         Y@rN  g      i@g     r@rw   g      y@rZ  r[  r\  r^  r7   r   r8   test_emulate_multi_cpu;  s4   "z-TestSystemCPUFrequency.test_emulate_multi_cpuc              
      s    fdd}t  trdnd}tj||dJ tjddd* tjd	d
d t }| |jd W d    n1 s;w   Y  W d    n1 sJw   Y  W d    d S W d    d S 1 sbw   Y  d S )Nc                    sR   |  drttjd|  drtdS | dkrtdS  | g|R i |S )NrT  rY   z/cpuinfo_cur_freqr`  r+  s   cpu MHz		: 200)r   r   r   r   r   r   r   r   r7   r8   r   l  s   



zOTestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_file.<locals>.open_mockr   r   r)  rJ  Tr<  rb  rw      )rC   r   r   r   r   rK  r   r]  r^  r7   r   r8   %test_emulate_no_scaling_cur_freq_filei  s    
"z<TestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_fileN)r   r   r   r   r4  r   rL  rS  r_  rd  rf  r7   r7   r7   r8   rB    s    



-rB  c                   @      e Zd Zdd Zdd ZdS )TestSystemCPUStatsc                 C   &   t d}t j}| j||dd d S )Nzcontext switches  r   )r{   r   	cpu_statsZctx_switchesr   r   r7   r7   r8   test_ctx_switches     
z$TestSystemCPUStats.test_ctx_switchesc                 C   ri  )N
interruptsrj  r   )r{   r   rk  rn  r   r   r7   r7   r8   test_interrupts  rm  z"TestSystemCPUStats.test_interruptsN)r   r   r   rl  ro  r7   r7   r7   r8   rh        rh  c                   @   s"   e Zd Zee ddd ZdS )TestLoadAvgrC  c                 C   s   t  }tdd}|  }W d    n1 sw   Y  | jt|d |d dd | jt|d |d dd | jt|d |d dd d S )Nz/proc/loadavgrr   rw   r   r  )r   
getloadavgrC   r   rE   r   float)r   r   rM   Z
proc_valuer7   r7   r8   test_getloadavg  s    zTestLoadAvg.test_getloadavgN)r   r   r   r   r4  r   ru  r7   r7   r7   r8   rq    s    rq  c                   @   r  )TestSystemNetIfAddrsc                 C   s   t   D ]_\}}|D ]X}|jt jkr| |jt| q|jtj	krN| |jt
| | |jt| |jd urE| |jt| q| t|d q|jtjkrd|jdd }| |t| qqd S )Nz0.0.0.0%r   )r   Znet_if_addrsitemsZfamilyZAF_LINKr   addressrc   r)   r*   r9   Znetmaskr<   Z	broadcastr>   rK   rE   rS   )r   r   ZaddrsZaddrry  r7   r7   r8   test_ips  s$   
zTestSystemNetIfAddrs.test_ipsN)r   r   r   rz  r7   r7   r7   r8   rv    r  rv  c                   @   rg  )TestSystemNetIfStatsc              	   C   sr   t   D ]0\}}ztd| }W n	 ty   Y qw | j|jd|v |d | |jtt	
d|d  qd S )Nifconfig %sZRUNNING)r   z(?i)MTU[: ](\d+)r   )r   net_if_statsrx  r   RuntimeErrorr   Zisupmturn   r  r  )r   r   statsro   r7   r7   r8   test_against_ifconfig  s   z*TestSystemNetIfStats.test_against_ifconfigc              	   C   sb   t   D ](\}}td| d}| |jt|   W d    n1 s)w   Y  qd S )Nz/sys/class/net/%s/mtur?   )	r   r}  rx  rC   r   r  rn   r   r}   )r   r   r  rM   r7   r7   r8   test_mtu  s   zTestSystemNetIfStats.test_mtuN)r   r   r   r  r  r7   r7   r7   r8   r{    s    r{  c                   @   s   e Zd Ze dd ZdS )TestSystemNetIOCountersc              	      s    fdd}t jddd}| D ]l\ }z| }W n	 ty$   Y qw | 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 qd S )Nc                    s   i }t d  }ttd|d |d< ttd|d |d< ttd|d |d< ttd|d	 |d
< ttd|d |d< ttd|d	 |d< ttd|d |d< ttd|d |d< |S )Nr|  zRX packets[: ](\d+)r   packets_recvzTX packets[: ](\d+)packets_sentzerrors[: ](\d+)errinrw   erroutzdropped[: ](\d+)dropindropoutz#RX (?:packets \d+ +)?bytes[: ](\d+)
bytes_recvz#TX (?:packets \d+ +)?bytes[: ](\d+)
bytes_sent)r   rn   r  r  )Znicr   ro   r   r7   r8   ifconfig  s&   z?TestSystemNetIOCounters.test_against_ifconfig.<locals>.ifconfigTF)Zpernicnowrapr  i   r   r  r  r   r  r  
   r  r  r  )r   net_io_countersrx  r~  r   r  r  r  r  r  r  r  r  )r   r  Znior  Zifconfig_retr7   r  r8   r    sB   z-TestSystemNetIOCounters.test_against_ifconfigN)r   r   r   r   r  r7   r7   r7   r8   r    s    r  c                   @   s8   e Zd Zejdedejddddd Zdd	 Zd
S )TestSystemNetConnectionsz psutil._pslinux.socket.inet_ntopr)  zpsutil._pslinux.supports_ipv6Fr<  c                 C   sP   zt  t jt j}| |j |d W n
 t jy   Y nw tjdd d S )N)z::1r   Zinet6kind)	r)   rK   ZSOCK_STREAMZ
addCleanupcloseZbinderrorr   net_connections)r   Zsupports_ipv6rL   r6   r7   r7   r8   test_emulate_ipv6_unsupported  s   z6TestSystemNetConnections.test_emulate_ipv6_unsupportedc                 C   sL   t dtd}tjdd |jsJ W d    d S 1 sw   Y  d 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
                Zunixr  )r   r   r   r   r  r   r  r7   r7   r8   test_emulate_unix  s   "z*TestSystemNetConnections.test_emulate_unixN)r   r   r   r   r   rF   r  r  r7   r7   r7   r8   r    s
    
r  c                   @   s>   e Zd Zeeed de dd Zdd Z	dd Z
d	S )
TestSystemDiskPartitionsstatvfszos.statvfs() not availablec                 C   sn   dd }t jddD ]*}t |j}||j\}}}}| |j| | j|j|td | j|j	|td q
d S )Nc                 S   sz   t d|   }|d}|d |d}| d d \}}}}|dkr)d}t|t|t|}}}||||fS )Nzdf -P -B 1 "%s"rj   r   rA   ZnonerY   )r   r}   rE   poprn   )r5  ro   rp   rN   devrr   rs   rd   r7   r7   r8   df6  s   


z4TestSystemDiskPartitions.test_against_df.<locals>.dfF)allr   )
r   disk_partitions
disk_usageZ
mountpointr   rr   r   rd   r   rs   )r   r  partusager  rr   rs   rd   r7   r7   r8   test_against_df1  s   

z(TestSystemDiskPartitions.test_against_dfc              	   C   s   t dd}| }W d    n1 sw   Y  d|v r3t D ]
}|jdkr+ d S q!| d d S ttd}t	j
d|ddD}t	j
d	d
gd#}t }|jsUJ |jsZJ |s^J | |d jd W d    n1 sqw   Y  W d    d S W d    d S 1 sw   Y  d S )Nz/proc/filesystemsrr  zfszcouldn't find any ZFS partitionz
nodev	zfs
r   Tr-  $psutil._pslinux.cext.disk_partitions)z	/dev/sdb3/r  rwr<  r   )rC   r   r   r  ZfstypeZfailr   r   r   r   r   r   r   )r   rM   r   r  r2  r?  r@  r   r7   r7   r8   test_zfs_fsJ  s8   



"z$TestSystemDiskPartitions.test_zfs_fsc              	   C   s   zAt jddd&}| t t  W d    n1 sw   Y  |js'J W d    n1 s1w   Y  W dt_d S W dt_d S dt_w )Nzos.path.realpathz/non/existentr<  /proc)r   r   assertRaisesr   r   r  r   PROCFS_PATHr  r7   r7   r8   test_emulate_realpath_failb  s   

z3TestSystemDiskPartitions.test_emulate_realpath_failN)r   r   r   r   r4  hasattrr  r   r  r  r  r7   r7   r7   r8   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 )TestSystemDiskIoCountersc              	   C      t ddq tjdddQ tjdd}| |jd | |jd	 | |jd
t	  | |j
d | |jd | |jd | |jdt	  | |jd | |jd W d    n1 saw   Y  W d    d S W d    d S 1 syw   Y  d S )N/proc/diskstatsz+   3     0   1 hda 2 3 4 5 6 7 8 9 10 11 12!psutil._pslinux.is_storage_deviceTr<  Fr  rw   r  r   rA   rx   r	        r  r   r   r   r   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r   r   r7   r7   r8   test_emulate_kernel_2_4q  s*   "z0TestSystemDiskIoCounters.test_emulate_kernel_2_4c              	   C   r  )Nr  z'   3    0   hda 1 2 3 4 5 6 7 8 9 10 11r  Tr<  Fr  rw   r  r   rA   rx   r	  r  r  r  r  r  r7   r7   r8   test_emulate_kernel_2_6_full  s*   "z5TestSystemDiskIoCounters.test_emulate_kernel_2_6_fullc              	   C   s   t ddq tjdddQ tjdd}| |jd | |jd	t  | |j	d
 | |j
dt  | |jd | |jd | |jd | |jd | |jd W d    n1 saw   Y  W d    d S W d    d S 1 syw   Y  d S )Nr  z   3    1   hda 1 2 3 4r  Tr<  Fr  rw   r  r   rA   r   )r   r   r   r   r  r   r  r  r  r  r  r  r  r  r  r  r  r7   r7   r8   test_emulate_kernel_2_6_limited  s*   "z8TestSystemDiskIoCounters.test_emulate_kernel_2_6_limitedc              	   C   s   t dtd[ tjddd; tjddd}| t|d | |d	 j	d
 | |d j	d
 | |d	 j
d | |d j
d W d    n1 sNw   Y  W d    d S W d    d S 1 sfw   Y  d S )Nr                      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
                    r  Fr<  Tperdiskr  r  nvme0n1rw   Z	nvme0n1p1rx   )r   r   r   r   r   r   r  r   rH   r  r  r  r7   r7   r8   test_emulate_include_partitions  s"   "z8TestSystemDiskIoCounters.test_emulate_include_partitionsc              	   C   s  t dtd+ tjddd tjddd}| | W d    n1 s'w   Y  W d    n1 s6w   Y  dd }t dtd> tjdd	|d
 tjddd}| |j	d | |j
d W d    n1 spw   Y  W d    d S W d    d S 1 sw   Y  d S )Nr  r  r  Fr<  r  c                 S   s   | dkS )Nr  r7   r  r7   r7   r8   is_storage_device  s   zSTestSystemDiskIoCounters.test_emulate_exclude_partitions.<locals>.is_storage_deviceTr   rw   rx   )r   r   r   r   r   r   r  r1  r   r  r  )r   r   r  r7   r7   r8   test_emulate_exclude_partitions  s6   "z8TestSystemDiskIoCounters.test_emulate_exclude_partitionsc                 C   sh   dd }t jdd}tjdd|d t jdd}W d    n1 s#w   Y  | t|t| d S )Nc                 S   s   | dkrdS dS )Nr  FTr7   rF  r7   r7   r8   r6    s   z?TestSystemDiskIoCounters.test_emulate_use_sysfs.<locals>.existsT)r  psutil._pslinux.os.path.existsr   )r   r  r   r   r   rH   )r   r6  ZwprocfsZwsysfsr7   r7   r8   test_emulate_use_sysfs  s   z/TestSystemDiskIoCounters.test_emulate_use_sysfsc                 C   sL   dd }t jdd|d | ttj W d    d S 1 sw   Y  d S )Nc                 S   s   dS )NFr7   rF  r7   r7   r8   r6    rX   z>TestSystemDiskIoCounters.test_emulate_not_impl.<locals>.existsr  Tr   )r   r   r  NotImplementedErrorr   r  )r   r6  r7   r7   r8   test_emulate_not_impl  s   "z.TestSystemDiskIoCounters.test_emulate_not_implN)
r   r   r   r  r  r  r  r  r  r  r7   r7   r7   r8   r  n  s    r  c                   @   s^   e Zd Zdd Zdd Zeeddd Zee	d d	eedd
d Z
dd ZdS )TestRootFsDeviceFinderc                 C   s(   t dj}t || _t || _d S )Nr  )r  r~   st_devmajorminor)r   r  r7   r7   r8   setUp  s   zTestRootFsDeviceFinder.setUpc                 C   sf   t  }tjdr|  n| t|j tjd| j| jf r&|	  n| t|j	 |
  d S Nz/proc/partitionsz/sys/dev/block/%s:%s/uevent)r   r  r5  r6  ask_proc_partitionsr  r   r  r  ask_sys_dev_blockask_sys_class_block)r   finderr7   r7   r8   test_call_methods  s   

z(TestRootFsDeviceFinder.test_call_methodszunsupported on GITHUB_ACTIONSc                 C   s   t  }| |  d  } }}tjdr| }tjd| j| jf r*|	 }|
 }|p3|p3|}|r>|r>| || |rH|rH| || |rT|rV| || d S d S d S r  )r   assertIsNotNonefindr  r5  r6  r  r  r  r  r  r   )r   r  r   r   cbaser7   r7   r8   test_comparisons  s$   z'TestRootFsDeviceFinder.test_comparisonsZfindmntzfindmnt utility not availablec                 C   s"   t   }td}| || d S )Nzfindmnt -o SOURCE -rn /)r   r  r   r   )r   r   Zfindmnt_valuer7   r7   r8   test_against_findmnt   s   
z+TestRootFsDeviceFinder.test_against_findmntc                 C   s   t jddgd7}t d }|jsJ ts(| |jd | |jt	 
  n| |jd W d    d S W d    d S 1 sBw   Y  d S )Nr  )	/dev/rootr  Zext4r  r<  r   r  )r   r   r   r  r   r   assertNotEqualZdevicer   r   r  )r   r   r  r7   r7   r8   test_disk_partitions_mocked'  s   
"z2TestRootFsDeviceFinder.test_disk_partitions_mockedN)r   r   r   r  r  r   r4  r   r  r   r  r  r7   r7   r7   r8   r    s    


r  c                   @   sR   e Z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S )TestMiscc                 C   s(   t d}t }| t|t| d S )Nz	boot time)r{   r   	boot_timer   rn   r   r7   r7   r8   test_boot_time<  s   zTestMisc.test_boot_timec              	      sj  |   }t| ttj|dd}|d |d |d W d    n1 s-w   Y  zt  fdd}tr?dnd	}tj	||d
 t
t | ttj | jttjdd | ttj | jttjdd | ttj | jttjdd |t_| t d | tt d tjdd}| t|d tjdd}| ttt|d ttj|dd}|d |d |d W d    n1 sw   Y  | t d | ttjddd | tt d | ttttjddd W d    n	1 sw   Y  W t| t
t n
t| t
t w | tjd d S )Nr~   r   zcpu   0 0 0 0 0 0 0 0 0 0
zcpu0  0 0 0 0 0 0 0 0 0 0
zcpu1  0 0 0 0 0 0 0 0 0 0
c                    s,   |  drttjd | g|R i |S )Nr  zrejecting access for test)rk   r   r   r   r   r   r7   r8   r   M  s   
z4TestMisc.test_no_procfs_on_import.<locals>.open_mockr   r   r)  TrO  r   zcpu   1 0 0 0 0 0 0 0 0 0
zcpu0  1 0 0 0 0 0 0 0 0 0
zcpu1  1 0 0 0 0 0 0 0 0 0
r  )
get_testfnr  mkdirrC   r5  rJ   writer   r   r   r   r   r  r   r  cpu_percentcpu_times_percentr  r   sumr   r  shutilrmtree)r   Z	my_procfsrM   r   r   Zper_cpu_percentZper_cpu_times_percentr7   r   r8   test_no_procfs_on_importA  s`   





&


z!TestMisc.test_no_procfs_on_importc                 C   sV  t dtd !}t  |jsJ tjdd t  tjdd W d    n1 s.w   Y  t dtd d}t }|jsGJ tjdd}t }tjdd}| |d | t	|d | t	|d | t	|d | t	t
t	|d | t	t
t	|d | |jd | |jd W d    d S 1 sw   Y  d S )Nr*  z                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
                TrO  z                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   rc  )r   r   r   r   r   r  r   r  r  r  r   r   r  user)r   r   r  Zcpu_percent_percpur  Zcpu_times_percent_percpur7   r7   r8   test_cpu_steal_decrease~  s@   

"z TestMisc.test_cpu_steal_decreasec                 C   sN   t jddd}| ttjj |jsJ W d    d S 1 s w   Y  d S )Nr   Tr   )r   r   r  r~  r   r0  r  r   r  r7   r7   r8   test_boot_time_mocked  s   "zTestMisc.test_boot_time_mockedc                 C   s   t jddgd}| t d jd |jsJ W d    n1 s#w   Y  t jddgd}| t d jd |jsAJ W d    n1 sKw   Y  t jddgd}| t d jd |jsiJ W d    d S 1 stw   Y  d S )	Nzpsutil._pslinux.cext.users)	giampaolopts/2z:0   hATr  r<  r   Z	localhost)r  r  z:0.0r  Tr  )r  r  foor  Tr  r  )r   r   r   r   ZusersZhostr   r  r7   r7   r8   test_users_mocked  s*   "zTestMisc.test_users_mockedc                 C   s   |   }t| zK|t_| ttj | ttj | jttjdd | ttj	 | ttj
 | ttj | ttj | ttj | tjtj W dt_d S dt_w )NTrO  r  )r  r  r  r   r  r  r   r   r  r  r  r  r}  r  ZNoSuchProcessProcess)r   Ztdirr7   r7   r8   test_procfs_path  s   
zTestMisc.test_procfs_pathc                 C   s   t  < t }| }| t|d t|dd dd j}| |j	| t|}|
  | |t  W d    d S 1 sBw   Y  d S )Nr  c                 S   s   | j S rT   )idrU   r7   r7   r8   <lambda>  s    z)TestMisc.test_issue_687.<locals>.<lambda>)keyrw   )r   r   r  threadsr   rH   sortedr  r  pidZas_dictr  Zpids)r   pr  tidptr7   r7   r8   test_issue_687  s   
"zTestMisc.test_issue_687c                 C   sL   t dd}tt sJ |jsJ W d    d S 1 sw   Y  d S )N/proc/%s/statusrY   )r   r   Z
pid_existsr  getpidr   r  r7   r7   r8   test_pid_exists_no_proc_status  s   "z'TestMisc.test_pid_exists_no_proc_statusN)r   r   r   r  r  r  r  r  r  r   r  r  r7   r7   r7   r8   r  9  s    =)
r  z
no batteryc                   @   sf   e Zd Ze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S )TestSensorsBatteryZacpizacpi utility not availablec                 C   sD   t d}t|dd  dd}t j}| j||dd d S )Nzacpi -br8  rw   rw  rY   r   )	r   rn   rE   r}   replacer   sensors_batterypercentr   )r   ro   Z
acpi_valuer   r7   r7   r8   test_percent  s   
zTestSensorsBattery.test_percentc                    s    fdd}t  trdnd}tj||d!}| t jd | t jtj	 |j
s.J W d    d S 1 s9w   Y  d S )Nc                    4   |  ds
|  drtdS  | g|R i |S )N
AC0/online	AC/online   1r   r   r   r   r   r7   r8   r        
z@TestSensorsBattery.test_emulate_power_plugged.<locals>.open_mockr   r   r)  T)rC   r   r   r   r   r   r	  power_pluggedZsecsleftZPOWER_TIME_UNLIMITEDr   r   r   r   r   r7   r   r8   test_emulate_power_plugged  s   "z-TestSensorsBattery.test_emulate_power_pluggedc                    l    fdd}t  trdnd}tj||d}| t jd |js$J W d    d S 1 s/w   Y  d S )Nc                    N   |  ds
|  drttjd|  drttdS  | g|R i |S )Nr  r  rY   /statusZchargingr   r   r   r   r   r   r   r   r   r7   r8   r     
   
zBTestSensorsBattery.test_emulate_power_plugged_2.<locals>.open_mockr   r   r)  T	rC   r   r   r   r   r   r	  r  r   r  r7   r   r8   test_emulate_power_plugged_2     "z/TestSensorsBattery.test_emulate_power_plugged_2c                    r  )Nc                    r  )Nr  r     0r  r   r   r7   r8   r   "  r  zDTestSensorsBattery.test_emulate_power_not_plugged.<locals>.open_mockr   r   r)  Fr  r  r7   r   r8   test_emulate_power_not_plugged   s   "z1TestSensorsBattery.test_emulate_power_not_pluggedc                    r  )Nc                    r  )Nr  r  rY   r  Zdischargingr  r   r   r7   r8   r   1  r  zFTestSensorsBattery.test_emulate_power_not_plugged_2.<locals>.open_mockr   r   r)  Fr  r  r7   r   r8    test_emulate_power_not_plugged_2.  r  z3TestSensorsBattery.test_emulate_power_not_plugged_2c                    sj    fdd}t  trdnd}tj||d}| t j |js#J W d    d S 1 s.w   Y  d S )Nc                    sJ   |  ds
|  drttjd|  drtdS  | g|R i |S )N"/sys/class/power_supply/AC0/online!/sys/class/power_supply/AC/onlinerY   #/sys/class/power_supply/BAT0/statuss   ???)rk   r   r   r   r   r   r   r   r7   r8   r   B  s   


zETestSensorsBattery.test_emulate_power_undetermined.<locals>.open_mockr   r   r)  )	rC   r   r   r   r1  r   r	  r  r   r  r7   r   r8   test_emulate_power_undetermined?  s   	"z2TestSensorsBattery.test_emulate_power_undeterminedc                 C   sL   t dd}| t jd |jsJ W d    d S 1 sw   Y  d S )N(/sys/class/power_supply/BAT0/energy_fullr  r   )r   r   r   r	  r
  r   r  r7   r7   r8   test_emulate_energy_full_0Q  s   "z-TestSensorsBattery.test_emulate_energy_full_0c              
   C   s   t dttjdH t dttjd& tdd | t jd W d    n1 s-w   Y  W d    n1 s<w   Y  W d    d S W d    d S 1 sTw   Y  d S )Nr$  rY   z(/sys/class/power_supply/BAT0/charge_fullz%/sys/class/power_supply/BAT0/capacitys   88X   )	r   r   r   r   r   r   r   r	  r
  r   r7   r7   r8   "test_emulate_energy_full_not_availX  s$   

"z5TestSensorsBattery.test_emulate_energy_full_not_availc              
   C   s   t dttjdK t dttjd) t dttjd | t j W d    n1 s0w   Y  W d    n1 s?w   Y  W d    d S W d    d S 1 sWw   Y  d S )Nr!  rY   r   r"  )r   r   r   r   r1  r   r	  r  r'  r7   r7   r8   test_emulate_no_powere  s&   


"z(TestSensorsBattery.test_emulate_no_powerN)r   r   r   r   r4  r   r  r  r  r  r  r#  r%  r(  r)  r7   r7   r7   r8   r    s    
r  c                   @   r  )TestSensorsBatteryEmulatedc              	      s    fdd}t  trdnd}tjddgd&}tj||d}| t  W d    n1 s0w   Y  W d    n1 s?w   Y  |jsIJ |jsNJ d S )	Nc                    ^   |  drttdS |  drttdS |  dr$ttdS  | g|R i |S )Nz/energy_nowZ60000000z
/power_now0z/energy_fullZ60000001r   r   r   r   r   r   r7   r8   r   w     


z5TestSensorsBatteryEmulated.test_it.<locals>.open_mockr   r   z
os.listdirZBAT0r<  r)  )rC   r   r   r   r  r   r	  r   )r   r   r   ZmlistdirZmopenr7   r   r8   test_itv  s   

z"TestSensorsBatteryEmulated.test_itN)r   r   r   r/  r7   r7   r7   r8   r*  s  r  r*  c                   @   rg  )TestSensorsTemperaturesc              	      s    fdd}t  trdnd}tj||dM tjddgd, t d	 d
 }| |jd | |jd | |j	d | |j
d W d    n1 sMw   Y  W d    d S W d    d S 1 sew   Y  d S )Nc                    s   |  drttdS |  drttdS |  dr"tdS |  dr,tdS |  d	r6td
S  | g|R i |S )N/namer   z/temp1_labellabelz/temp1_input   30000z
/temp1_maxs   40000z/temp1_crit   50000)r   r   r   r   r   r   r   r7   r8   r     s   







zCTestSensorsTemperatures.test_emulate_class_hwmon.<locals>.open_mockr   r   r)  r;  z/sys/class/hwmon/hwmon0/temp1r<  r   r   r2        >@g      D@      I@rC   r   r   r   r   Zsensors_temperaturesr   r2  r]  Zhighcritical)r   r   r   tempr7   r   r8   test_emulate_class_hwmon  s    "z0TestSensorsTemperatures.test_emulate_class_hwmonc              	      s    fdd}dd }t  trdnd}tj||dM tjdd	|d
, t d d }| |jd | |jd | |j	d | |j
d W d    n1 sQw   Y  W d    d S W d    d S 1 siw   Y  d S )Nc                    sn   |  dr
tdS |  drtdS |  dr ttdS |  dr,ttdS  | g|R i |S )	NZ0_tempr4  r9  r3  Z0_typer8  typer   )r   r   r   r   r   r   r   r7   r8   r     s   





zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.open_mockc                 S   s:   | dkrg S | dkrg S | dkrdgS | dkrddgS g S )Nz/sys/class/hwmon/hwmon*/temp*_*z&/sys/class/hwmon/hwmon*/device/temp*_*z /sys/class/thermal/thermal_zone*z /sys/class/thermal/thermal_zone0z,/sys/class/thermal/thermal_zone0/trip_point*z2/sys/class/thermal/thermal_zone1/trip_point_0_typez2/sys/class/thermal/thermal_zone1/trip_point_0_tempr7   rF  r7   r7   r8   	glob_mock  s   zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.glob_mockr   r   r)  r;  Tr   r   r   rY   r5  r6  r7  )r   r   r<  r   r9  r7   r   r8   test_emulate_class_thermal  s   "z2TestSensorsTemperatures.test_emulate_class_thermalN)r   r   r   r:  r=  r7   r7   r7   r8   r0    s    r0  c                   @   r  )TestSensorsFansc              	      s    fdd}t  trdnd}tj||d? tjddgd t d	 d
 }| |jd | |jd W d    n1 s?w   Y  W d    d S W d    d S 1 sWw   Y  d S )Nc                    r+  )Nr1  r   z/fan1_labelr2  z/fan1_inputZ2000r-  r   r   r7   r8   r     r.  z4TestSensorsFans.test_emulate_data.<locals>.open_mockr   r   r)  r;  z/sys/class/hwmon/hwmon2/fan1r<  r   r   r2  i  )	rC   r   r   r   r   Zsensors_fansr   r2  r]  )r   r   r   Zfanr7   r   r8   r_    s   
"z!TestSensorsFans.test_emulate_dataN)r   r   r   r_  r7   r7   r7   r8   r>    r  r>  c                   @   s   e Zd Ze dd Zdd Ze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dd Zdd Zee ddd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)S )*TestProcessc                 C   s   |   }td| }| |}tdd d|  t|j}t	d |
 }|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 dS )Nr   )r  r  r7   r7   r7   r8   r    s   
 z3TestProcess.test_memory_full_info.<locals>.<lambda>z'%s' not in retg?F)Zgroupedc                 S   s   g | ]}|j |j qS r7   )Zprivate_dirtyZprivate_cleanrv   r7   r7   r8   r]     r^   z5TestProcess.test_memory_full_info.<locals>.<listcomp>r   r   c                 S      g | ]}|j qS r7   )pssrv   r7   r7   r8   r]         c                 S   r@  r7   )r   rv   r7   r7   r8   r]      rB  )r  r   r   Zpyrunr   r   r  r  timesleepmemory_full_infomemory_mapsr   ussr  rA  r   )r   testfnsrcZsprocr   r   mapsr7   r7   r8   test_memory_full_info  s(   


z!TestProcess.test_memory_full_infoc                 C   s   t dt  td +}t }| }|j	sJ | 
|jd | 
|jd | 
|jd W d    d S 1 s<w   Y  d S )N/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
                i l  r   i <  )r   r  r  r   r   r   r   r  rE  r   r   rG  rA  r   )r   r   r   r   r7   r7   r8   test_memory_full_info_mocked  s   

"z(TestProcess.test_memory_full_info_mockedzunreliable on PYPYc                 C   s  dd }|   }t|d | ||jd W d    n1 s!w   Y  t|d | ||jd W d    n1 s?w   Y  t|d | ||jd W d    n1 s]w   Y  t|d | ||jd W d    n1 s{w   Y  t|d | ||jd W d    n1 sw   Y  t|d | ||jd W d    n1 sw   Y  trt| t|d	 | ||jd W d    n1 sw   Y  t| t|d
 | ||jd W d    d S 1 sw   Y  d S d S )Nc                 S   sP   t  }t t }	 | D ]}|jtj| kr|  S t |kr& nqqrT   )	r   r  rC  r	   
open_filesr5  r  abspathr~  )fnamer   Z	giveup_atfiler7   r7   r8   get_test_file'  s   z7TestProcess.test_open_files_mode.<locals>.get_test_filer   rr  r   zr+zw+za+rV   zx+)r  rC   r   moder   r   )r   rR  rH  r7   r7   r8   test_open_files_mode%  s<   "z TestProcess.test_open_files_modec              	   C   s  t  }| }t|  dk t|jdt|  tjdt	t
jdd}| }|r.J |js3J W d    n1 s=w   Y  tjdt	t
jdd}| | g  |js[J W d    n1 sew   Y  W d    d S W d    d S 1 s}w   Y  d S Nr   len(ret) != %ipsutil._pslinux.os.readlinkrY   r)  )r   r  rN  rC   r  r   rH   r   r   OSErrorr   r   r   EINVALr   )r   r   filesr   r7   r7   r8   test_open_files_file_goneJ  s,   


"z%TestProcess.test_open_files_file_gonec              	   C   s   t  }| }t|  dI t|jdt|  trdnd}tj	|t
tjdd}| }|r4J |js9J W d    n1 sCw   Y  W d    d S W d    d S 1 s[w   Y  d S )Nr   rV  r   r   rY   r)  )r   r  rN  rC   r  r   rH   r   r   r   r   r   r   r   r   r   rZ  r   r   r7   r7   r8   test_open_files_fd_gone^  s    
"z#TestProcess.test_open_files_fd_gonec              	   C   s   t  }| }t|  dE t|jdt|  d}tj|t	t
jdd}| }|r0J |js5J W d    n1 s?w   Y  W d    d S W d    d S 1 sWw   Y  d S rU  )r   r  rN  rC   r  r   rH   r   r   rX  r   ENAMETOOLONGr   r\  r7   r7   r8   test_open_files_enametoolongn  s    
"z(TestProcess.test_open_files_enametoolongc                 C   sX   t jdi d}| tjt   |j	sJ W d    d S 1 s%w   Y  d S )Nz)psutil._pslinux._psposix.get_terminal_mapr<  )
r   r   r1  r   r0  r  r  r  Zterminalr   r  r7   r7   r8   test_terminal_mocked  s   "z TestProcess.test_terminal_mockedc                 C      t  }ttd}tjd|dd}| | ddg |j	s#J W d    n1 s-w   Y  ttd}tjd|dd}| | g d |j	sQJ W d    d S 1 s\w   Y  d S )	Nzfoo bar r   Tr-  r  barz	foo bar  r  rb  rY   
r   r  r   r   r   r   r   r   Zcmdliner   r   r   r2  r   r7   r7   r8   test_cmdline_mocked  "   "zTestProcess.test_cmdline_mockedc                 C   ra  )	Nzfoo bar r   Tr-  r  rb  z	foo bar  rc  rd  re  r7   r7   r8   test_cmdline_spaces_mocked  rg  z&TestProcess.test_cmdline_spaces_mockedc                 C   sj   t  }ttd}tjd|dd}| | ddg |j	s#J W d    d S 1 s.w   Y  d S )Nzfoo bar r   Tr-  r  rb  rd  re  r7   r7   r8   test_cmdline_mixed_separators  s   "z)TestProcess.test_cmdline_mixed_separatorsc                 C   s\   t jddd | t  d | t  d W d    d S 1 s'w   Y  d S )NrW  z/home/foo (deleted)r<  z	/home/foo)r   r   r   r   r  execwdr'  r7   r7   r8   !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}t  }|js!J | |g  W d    n1 s1w   Y   fdd}tj||d | 	tj
t j W d    d S 1 sYw   Y  d S )Nc                    4   |  dt  rttjd | g|R i |S Nz/proc/%s/taskrY   )rk   r  r  r   r   r   r   r   r7   r8   r        z2TestProcess.test_threads_mocked.<locals>.open_mockr   r   r)  c                    rm  rn  )rk   r  r  r   r   EPERMr   r   r7   r8   r     ro  )rC   r   r   r   r   r  r  r   r   r  AccessDenied)r   r   r   r   r   r7   r   r8   test_threads_mocked  s   
"zTestProcess.test_threads_mockedc              
   C   s   t jdttjddc}t jdtddd?}t  }|j	s#J |j	s(J | 
|d t jddd | tjt j W d    n1 sJw   Y  W d    n1 sYw   Y  W d    d S W d    d S 1 sqw   Y  d S )	Nzpsutil._pslinux.readlinkrY   r)  zpsutil.Process.cmdliner   zpsutil._pslinux.os.path.lexistsFr<  )r   r   rX  r   r   r   rq  r  rj  r   r   r  ZombieProcess)r   r?  r@  r   r7   r7   r8   test_exe_mocked  s0   



"zTestProcess.test_exe_mockedc              	   C   s   t dt  ttjd+}t }| t	 |
  W d    n1 s&w   Y  |js0J W d    d S 1 s;w   Y  d S )NrL  rY   )r   r  r  r   r   r   r   r  r  r   rF  r   r   r   r   r7   r7   r8   test_issue_1014  s   


"zTestProcess.test_issue_1014rC  c              	   C   s   t jdttjdd1}t }|  | tj	}|
tj W d    n1 s+w   Y  |js5J W d    n1 s?w   Y  | |jj|j | |jj|  d S )Nzpsutil._pslinux.prlimitrY   r)  )r   r   rX  r   ZENOSYSr   r  r   r  rs  ZrlimitZRLIMIT_NOFILEr   r   	exceptionr  r   r   r   r   r7   r7   r8   test_rlimit_zombie  s   
zTestProcess.test_rlimit_zombiec              	   C   s   t jdttjdd/}t }|  | tj	}|
  W d    n1 s)w   Y  |js3J W d    n1 s=w   Y  | |jj|j | |jj|  d S NrW  rY   r)  )r   r   rX  r   r   r   r  r   r  rs  rk  r   r   rw  r  rx  r7   r7   r8   test_cwd_zombie  s   

zTestProcess.test_cwd_zombiec                 C   s  g d}d | }tdt  |m t }| | d | |	 tj
 | | d | | dt t   | }| |jdt  | |jdt  | |jd	t  | |jd
t  | |jdt  | | d W d    d S 1 sw   Y  d S )N)*r,  z(cat)Z1r,  r,  r,  r,  r,  r,  r,  r,  r,  2345r,  r,  r,  r,  6r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r,  r  r,  r,  7r|   z/proc/%s/statcatrw   r	  r  r   rA   rx   r  )rJ   r   r   r  r  r   r  r   r   statusZSTATUS_ZOMBIEppidZcreate_timer   r  r  r  systemchildren_userchildren_systemZiowaitZcpu_num)r   r   r   r   Zcpur7   r7   r8   test_stat_file_parsing  s$   ,"z"TestProcess.test_stat_file_parsingc                 C   s   t dt  td g t }| |	 j
d | |	 jd | | d | }| |jd | |jd | |jd | }| |jd	 | |jd
 | |jd | |j ttdd W d    d S 1 sxw   Y  d S )Nr  a                  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:	13r      B   i  i  i  i  i  i  r   r  )r   r  r  r   r   r   r   r  r   num_ctx_switches	voluntaryinvoluntarynum_threadsuidsrealZ	effectiveZsavedgids_proc_get_eligible_cpuslistrG   )r   r   r  r  r7   r7   r8   test_status_file_parsingN  s&   

"z$TestProcess.test_status_file_parsingc                 C   sZ   t jdttjdd}t }| rJ |jsJ W d    d S 1 s&w   Y  d S rz  )	r   r   rX  r   r^  r   r  Zconnectionsr   ru  r7   r7   r8   test_connections_enametoolongg  s   
"z)TestProcess.test_connections_enametoolongN)r   r   r   r   rK  rM  r   r4  r   rT  r[  r]  r_  r`  rf  rh  ri  rl  rr  rt  rv  r   ry  r{  r  r  r  r7   r7   r7   r8   r?    s.    

#
$

=r?  c                   @   sr   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  | _d S rT   )r   r  proc)clsr7   r7   r8   
setUpClass{  s   z#TestProcessAgainstStatus.setUpClassc              
   C   s   t jd| jj ?}|D ]5}| }||rB|dd }zt|W   W  d    S  t	yA   | Y   W  d    S w qt	d| 1 sLw   Y  d S )Nr  	r  zcan't find %r)
r   _psplatformZ	open_textr  r  r}   rk   	partitionrn   rF   )r   Z	linestartrM   rN   r  r7   r7   r8   read_status_file  s$   


z)TestProcessAgainstStatus.read_status_filec                 C       |  d}| | j | d S )NzName:)r  r   r  r   r   r  r7   r7   r8   	test_name     
z"TestProcessAgainstStatus.test_namec                 C   sH   |  d}||dd |d }|dd}| | j | d S )NzState:(rw   )r|   r  )r  r  rfindr  r   r  r  r  r7   r7   r8   test_status  s   
z$TestProcessAgainstStatus.test_statusc                 C   r  )NzPPid:)r  r   r  r  r  r7   r7   r8   	test_ppid  r  z"TestProcessAgainstStatus.test_ppidc                 C   r  )NzThreads:)r  r   r  r  r  r7   r7   r8   test_num_threads  r  z)TestProcessAgainstStatus.test_num_threadsc                 C   :   |  d}ttt| dd }| | j | d S )NzUid:rw   rA   )r  r   r   rn   rE   r   r  r  r  r7   r7   r8   	test_uids     
z"TestProcessAgainstStatus.test_uidsc                 C   r  )NzGid:rw   rA   )r  r   r   rn   rE   r   r  r  r  r7   r7   r8   	test_gids  r  z"TestProcessAgainstStatus.test_gidsc                 C   s@   |  d}| | j j| |  d}| | j j| d S )Nzvoluntary_ctxt_switches:znonvoluntary_ctxt_switches:)r  r   r  r  r  r  r  r7   r7   r8   test_num_ctx_switches  s   

z.TestProcessAgainstStatus.test_num_ctx_switchesc                 C   sR   |  d}dt|v r'tt|d\}}| | j tt	||d  d S d S )NCpus_allowed_list:r  rw   )
r  r   r   rn   rE   r   r  Zcpu_affinityr  rG   )r   r  Zmin_Zmax_r7   r7   r8   test_cpu_affinity  s   
z*TestProcessAgainstStatus.test_cpu_affinityc                 C   sh   |  d}td}| jj  W d    n1 sw   Y  dt|v r-|jr+J d S |js2J d S )Nr  zpsutil._pslinux.per_cpu_timesr  )r  r   r   r  r  r  r   r   )r   r  r   r7   r7   r8   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  r7   r7   r7   r8   r  r  s    

r  c                   @   rg  )	TestUtilsc                 C   sR   t jddd}| tjdd |jsJ W d    d S 1 s"w   Y  d S )Nzos.readlinkzfoo (deleted)r<  rb  r  )r   r   r   r   r  readlinkr   r  r7   r7   r8   test_readlink  s   "zTestUtils.test_readlinkc                 C   s   |   }t|d}|d W d    n1 sw   Y  | tjj|ddd | tjj|ddd | tjj|d d	d
d	 d S )NZwtzfoo F)binaryr  Ts   fooz??rb  )fallback)r  rC   r  r   r   r  r  )r   rH  rM   r7   r7   r8   test_cat  s   zTestUtils.test_catN)r   r   r   r  r  r7   r7   r7   r8   r    rp  r  __main__)run_from_name)dr  Z
__future__r   rl   r,   r   globr   r  r  r  r)   r2   r   rC  r   r   r   Zpsutil._compatr   r   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   r   r   r   r5  rO  dirname__file__ZHEREr1   ZSIOCGIFCONFr`   r;   r=   r  ZEMPTY_TEMPERATURESr9   r<   r>   rS   rc   ru   rz   r{   r   contextmanagerr   r   r4  r   r   r  r  r7  rB  rh  rq  rv  r{  r  r  r  r  r  r  r  r*  r0  r>  r?  r  r  r   Zpsutil.tests.runnerr  r7   r7   r7   r8   <module>   s   	

  ,]? )- ? B ;}A   R
