3
�[g[d � @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddlm'Z' ddlm(Z( ddlm)Z) ddlm*Z* dd lm+Z+ dd lm,Z, dd lm-Z- e�r�dd l.m/Z/ dd l.m0Z0 dd!l.m1Z1 dd"l.m2Z2 ej3j4ej3j5e6��Z7d#Z8d$Z9d%Z:d&Z;d'Z<e�r,d(Z=ejd)� Z>d*d+� Z?d,d-� Z@d.d/� ZAd0d1� ZBd2d3� ZCd4d5� ZDd6d7� ZEd8d9� ZFd:d;� ZGejHd<d=� �ZIejHd>d?� �ZJe'jKjLe d@dA�G dBdC� dCe#��ZMe'jKjLe d@dA�G dDdE� dEe#��ZNe'jKjLe d@dA�G dFdG� dGe#��ZOe'jKjLe d@dA�G dHdI� dIe#��ZPe'jKjLe d@dA�G dJdK� dKe#��ZQe'jKjLe d@dA�G dLdM� dMe#��ZRe'jKjLe d@dA�G dNdO� dOe#��ZSe'jKjLe d@dA�G dPdQ� dQe#��ZTe'jKjLe d@dA�G dRdS� dSe#��ZUe'jKjLe d@dA�G dTdU� dUe#��ZVe'jKjLe d@dA�G dVdW� dWe#��ZWe'jKjLe d@dA�e'jKjLe dXdA�G dYdZ� dZe#���ZXe'jKjLe d@dA�G d[d\� d\e#��ZYe'jKjLe d@dA�G d]d^� d^e#��ZZe'jKjLe d@dA�G d_d`� d`e#��Z[e'jKjLe d@dA�G dadb� dbe#��Z\e'jKjLe d@dA�G dcdd� dde#��Z]e'jKjLe d@dA�G dedf� dfe#��Z^e'jKjLe d@dA�e'jKjLe dgdA�G dhdi� die#���Z_e'jKjLe d@dA�G djdk� dke#��Z`e'jKjLe d@dA�G dldm� dme#��Zae'jKjLe d@dA�G dndo� doe#��Zbe'jKjLe d@dA�G dpdq� dqe#��Zce'jKjLe d@dA�G drds� dse#��Zde'jKjLe d@dA�G dtdu� due#��ZedS )vzLinux specific tests.� )�divisionN)�LINUX)�PY3)�FileNotFoundError)�
basestring)�AARCH64)�GITHUB_ACTIONS)�GLOBAL_TIMEOUT)� HAS_BATTERY)� HAS_CPU_FREQ)�HAS_GETLOADAVG)�
HAS_RLIMIT)�PYPY)�PYTEST_PARALLEL)� QEMU_USER)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�
ThreadTask)�
call_until)�mock)�pytest)�
reload_module)�retry_on_failure)� safe_rmpath)�sh)�skip_on_not_implemented)�which)� CLOCK_TICKS)�RootFsDeviceFinder)�calculate_avail_vmem)� open_binaryi� i� i'� i� i� i z/sys/class/hwmon/hwmon*c C sp dd l }| d d� } tr"t| d�} tjtjtj�}tj|��* tj|j |j
� t t j
d| ��dd� �S Q R X d S )Nr � �ascii�256s� � )�fcntlr �bytes�socket�AF_INET�
SOCK_DGRAM�
contextlib�closing� inet_ntoa�ioctl�fileno� SIOCGIFADDR�struct�pack)�ifnamer' �s� r6 �C/usr/local/lib64/python3.6/site-packages/psutil/tests/test_linux.py�get_ipv4_addressN s
r8 c C sp dd l }| d d� } tr"t| d�} tjtjtj�}tj|��* tj|j |j
� t t j
d| ��dd� �S Q R X d S )Nr r" r# r$ r% r& )r' r r( r) r* r+ r, r- r. r/ r0 �SIOCGIFNETMASKr2 r3 )r4 r' r5 r6 r6 r7 �get_ipv4_netmask] s
r: c C sp dd l }| d d� } tr"t| d�} tjtjtj�}tj|��* tj|j |j
� t t j
d| ��dd� �S Q R X d S )Nr r" r# r$ r% r& )r' r r( r) r* r+ r, r- r. r/ r0 �SIOCGIFBRDADDRr2 r3 )r4 r' r5 r6 r6 r7 �get_ipv4_broadcastl s
r< c C s� t d��N}g }x*|D ]"}|j� }|d | kr|j|� qW t|�dkrRtd| ��W d Q R X x~tt|��D ]n}|| d }g }x.tdt|�d�D ]}|j|||d � � q�W dj|�} tjtj | �}
tj
tj |
�||< qjW |S )Nz/proc/net/if_inet6� r zcould not find interface %r� �:���) �open�split�append�len�
ValueError�range�joinr) � inet_pton�AF_INET6� inet_ntop) r4 �fZ
all_fields�line�fields�iZ unformatted�groups�j� formatted�packedr6 r6 r7 �get_ipv6_addresses{ s"
rS c
s� dd l }| d d� } tr"t| d�} tjtjtj�}tj|��` |j|j � t
t j d| ��}trfdd� � ndd l
}|j� dj� fdd �|d
d � D ��d d
� S Q R X d S )Nr r" r# r$ c S s | S )Nr6 )�xr6 r6 r7 �ord� s z get_mac_address.<locals>.ord� c s g | ]}d � |� �qS )z%02x:r6 )�.0�char)rU r6 r7 �
<listcomp>� s z#get_mac_address.<locals>.<listcomp>� r& r= r@ )r' r r( r) r* r+ r, r- r/ r0 �
SIOCGIFHWADDRr2 r3 � __builtin__rU rG )r4 r' r5 �infor\ r6 )rU r7 �get_mac_address� s
r^ c C s� t ddgddid�} | jd�}xJ|D ]B}|jd�r$|j� \}}}}tjdd�}|t|�t|�t|��S q$W td dj|� ��d
S ) zQParse 'free' cmd and return swap memory's s total, used and free
values.
�freez-b�LANGzC.UTF-8)�env�
ZSwapztotal used freez&can't find 'Swap' in 'free' output:
%sN)r rB �
startswith� collections�
namedtuple�intrE rG )�out�linesrL �_�total�usedr_ �ntr6 r6 r7 � free_swap� s
rm c C s� t ddgddid�} | jd�}xT|D ]L}|jd�r$dd � |j� d
d � D �\}}}}tjdd �}|||||| �S q$W td
dj|� ��dS )zSParse 'free' cmd and return physical memory's total, used
and free values.
r_ z-br` zC.UTF-8)ra rb ZMemc s s | ]}t |�V qd S )N)rf )rW rT r6 r6 r7 � <genexpr>� s z free_physmem.<locals>.<genexpr>r= � z total used free shared outputz%can't find 'Mem' in 'free' output:
%sN)r rB rc rd re rE rG )rg rh rL rj rk r_ �sharedrl r6 r6 r7 � free_physmem� s
"rq c C sZ t ddgddid�}x4|jd�D ]&}|j� }| |kr t|jd�d �S q W td | ��d S )
N�vmstatz-sr` zC.UTF-8)ra rb � r z can't find %r in 'vmstat' output)r rB �striprf rE )�statrg rL r6 r6 r7 rr � s rr c C s@ t ddg�j� } d| kr"tjd��ttttjd| j � d ���S )Nr_ z-V�UNKNOWNz can't determine free versionz\d+r= r@ )
r rt r �skip�tuple�maprf �re�findallrB )rg r6 r6 r7 �get_free_version_info� s
r| c # sD � �fdd�}t � trdnd}tj|d|d�� }|V W dQ R X dS )z�Mock open() builtin and forces it to return a certain content
for a given path. `pairs` is a {"path": "content", ...} dict.
c sT | �kr@�| }t r4t|t�r(tj|�S tj|�S qPtj|�S n� | f|�|�S d S )N)r �
isinstancer �io�StringIO�BytesIO)�name�args�kwargs�content)� orig_open�pairsr6 r7 � open_mock� s
z$mock_open_content.<locals>.open_mockz
builtins.openz__builtin__.openT)�create� side_effectN)rA r r �patch)r� r� � patch_point�mr6 )r� r� r7 �mock_open_content� s
r� c # sF � ��fdd�}t �tr dnd}tj|d|d�� }|V W dQ R X dS )zZMock open() builtin and raises `exc` if the path being opened
matches `for_path`.
c s" | �kr� �n�| f|�|�S d S )Nr6 )r� r� r� )�exc�for_pathr� r6 r7 r� � s z&mock_open_exception.<locals>.open_mockz
builtins.openz__builtin__.openT)r� r� N)rA r r r� )r� r� r� r� r� r6 )r� r� r� r7 �mock_open_exception� s
r� z
LINUX only)�reasonc @ sL e Zd Zdd� Ze� dd� �Ze� dd� �Ze� dd� �Ze� d d
� �Zd S ) �"TestSystemVirtualMemoryAgainstFreec C s" t � j}tj� j}||ks t�d S )N)rq rj �psutil�virtual_memory�AssertionError)�self� cli_value� psutil_valuer6 r6 r7 �
test_total s
z-TestSystemVirtualMemoryAgainstFree.test_totalc C sR t � dk rtjd��t � dkr(tjd��t� j}tj� j}t|| �tk sNt �d S ) N� � zfree version too oldr> r zfree version too recent)r� r� r� )r> r r )
r| r rw rq rk r� r� �absr r� )r� r� r� r6 r6 r7 � test_used s
z,TestSystemVirtualMemoryAgainstFree.test_usedc C s* t � j}tj� j}t|| �tk s&t�d S )N)rq r_ r� r� r� r r� )r� r� r� r6 r6 r7 � test_free' s
z,TestSystemVirtualMemoryAgainstFree.test_freec C sP t � }|j}|dkr tjd��tj� j}t|| �tk sLtd|||j f ��d S )Nr z%free does not support 'shared' columnz %s %s
%s)
rq rp r rw r� r� r� r r� �output)r� r_ �
free_valuer� r6 r6 r7 � test_shared- s
z.TestSystemVirtualMemoryAgainstFree.test_sharedc C sr t ddg�}|jd�}d|d kr.tjd��n@t|d j� d �}tj� j}t|| �t k snt
d|||f ��d S )
Nr_ z-brb � availabler z(free does not support 'available' columnr= z %s %s
%sr@ ) r rB r rw rf r� r� r� r� r r� )r� rg rh r� r� r6 r6 r7 �test_available8 s
z1TestSystemVirtualMemoryAgainstFree.test_availableN) �__name__�
__module__� __qualname__r� r r� r� r� r� r6 r6 r6 r7 r� s
r� c @ sZ e Zd Zdd� Ze� dd� �Ze� dd� �Ze� dd� �Ze� d d
� �Ze� d d � �Z d
S )�$TestSystemVirtualMemoryAgainstVmstatc C s. t d�d }tj� j}t|| �tk s*t�d S )Nz total memoryi )rr r� r� rj r� r r� )r� � vmstat_valuer� r6 r6 r7 r� J s
z/TestSystemVirtualMemoryAgainstVmstat.test_totalc C sV t � d k rtjd��t � d
kr(tjd��td�d }tj� j}t|| �tk sRt �d S ) Nr� r� zfree version too oldr> r zfree version too recentz used memoryi )r� r� r� )r> r r )
r| r rw rr r� r� rk r� r r� )r� r� r� r6 r6 r7 r� O s
z.TestSystemVirtualMemoryAgainstVmstat.test_usedc C s. t d�d }tj� j}t|| �tk s*t�d S )Nz free memoryi )rr r� r� r_ r� r r� )r� r� r� r6 r6 r7 r� a s
z.TestSystemVirtualMemoryAgainstVmstat.test_freec C s. t d�d }tj� j}t|| �tk s*t�d S )Nz
buffer memoryi )rr r� r� �buffersr� r r� )r� r� r� r6 r6 r7 � test_buffersg s
z1TestSystemVirtualMemoryAgainstVmstat.test_buffersc C s. t d�d }tj� j}t|| �tk s*t�d S )Nz
active memoryi )rr r� r� �activer� r r� )r� r� r� r6 r6 r7 � test_activem s
z0TestSystemVirtualMemoryAgainstVmstat.test_activec C s. t d�d }tj� j}t|| �tk s*t�d S )Nzinactive memoryi )rr r� r� �inactiver� r r� )r� r� r� r6 r6 r7 �
test_inactives s
z2TestSystemVirtualMemoryAgainstVmstat.test_inactiveN)
r� r� r� r� r r� r� r� r� r� r6 r6 r6 r7 r� H s r� c @ sB e Zd Zdd� Ze� dd� �Zdd� Zdd� Zd d
� Zd d � Z d
S )� TestSystemVirtualMemoryMocksc C sd t jd�j� }td|i���>}tjdd���$}tjd� tj� }|j sJt
�t |�dksZt
�|d }dt |j
�kstt
�d t |j
�ks�t
�d
t |j
�ks�t
�d t |j
�ks�t
�d t |j
�ks�t
�d
t |j
�ks�t
�dt |j
�ks�t
�|jdks�t
�|jdks�t
�|jdk�s t
�|jdk�s t
�|jdk�s,t
�|jdk�s<t
�|jdk�sLt
�W d Q R X W d Q R X d S )NaL 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
z
/proc/meminfoT)�record�alwaysr= r z#memory stats couldn't be determined�cachedrp r� r� r� r� )�textwrap�dedent�encoder� �warnings�catch_warnings� simplefilterr� r� �calledr� rD �str�messager� r� r� rp r� r� �slab)r� r� r� �ws�ret�wr6 r6 r7 �test_warnings_on_misses| s.
z4TestSystemVirtualMemoryMocks.test_warnings_on_missesc C s� i }t d��4}x,|D ]$}|j� }t|d �d ||d < qW W d Q R X t|�}d|kr~|d }t|| �| d }|dk s~t�d S )Nz
/proc/meminfor= i r s
MemAvailable:�d r" )r! rB rf r r� r� )r� ZmemsrK rL rM �a�bZ diff_percentr6 r6 r7 �test_avail_old_percent� s
&z3TestSystemVirtualMemoryMocks.test_avail_old_percentc C s| t jd�j� }td|i��X}tjdd��}tj� }W d Q R X |jsFt �|j
d ksTt �|d }dt |j �ksnt �W d Q R X d S )
Na 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
z
/proc/meminfoT)r� i�Sd i r z,inactive memory stats couldn't be determinedl �" )
r� r� r� r� r� r� r� r� r� r� r� r� r� )r� r� r� r� r� r� r6 r6 r7 � test_avail_old_comes_from_kernel� s
z=TestSystemVirtualMemoryMocks.test_avail_old_comes_from_kernelc C s| t jd�j� }td|i��X}tjdd��}tj� }W d Q R X |jsFt �|j
d ksTt �|d }d t |j �ksnt �W d Q R X d S )
Nat 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
z
/proc/meminfoT)r� i�d i i�I r z,inactive memory stats couldn't be determinedi �}l 'L l `LG )
r� r� r� r� r� r� r� r� r� r� r� r� r� )r� r� r� r� r� r� r6 r6 r7 � test_avail_old_missing_fields� s
z:TestSystemVirtualMemoryMocks.test_avail_old_missing_fieldsc C s� t jd�j� }td|i��l tdttjd���N tj dd��6}t
j � }|j dksTt
�|d
}d t|j�ksnt
�W d Q R X W d Q R X W d Q R X d S )Na� 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/meminfoz/proc/zoneinfozno such file or directoryT)r� i�d i i�I r z,inactive memory stats couldn't be determinedi �}l 'L l `LG )r� r� r� r� r� �IOError�errno�ENOENTr� r� r� r� r� r� r� r� )r� r� r� r� r� r6 r6 r7 � test_avail_old_missing_zoneinfo� s
z<TestSystemVirtualMemoryMocks.test_avail_old_missing_zoneinfoc
C s� t jd�j� }td|i���}tj� }|js.t�|jdks<t�|j dksJt�|j
dksXt�|j dksft�|j dkstt�|j
dks�t�|jdks�t�|jdks�t�|jdks�t�W d Q R X d S )Na� 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
z
/proc/meminfor� i � r> ro � � � � � r� i � i i � i p i T i i i X i )r� r� r� r� r� r� r� r� rj r_ r� r� rp r� r� r� r� )r� r� r� Zmemr6 r6 r7 �test_virtual_memory_mocked s 1
z7TestSystemVirtualMemoryMocks.test_virtual_memory_mockedN)
r� r� r� r� r r� r� r� r� r� r6 r6 r6 r7 r� z s $ r� c @ s\ e Zd Zedd� �Zdd� Ze� dd� �Ze� dd� �Zd d
� Z d d � Z
d
d� Z dd� Z dS )�TestSystemSwapMemoryc C s, t d��} | j� }W dQ R X d|ko*d|kS )z3Return True if /proc/meminfo provides swap metrics.z
/proc/meminfoNz
SwapTotal:z SwapFree:)rA �read)rK �datar6 r6 r7 �meminfo_has_swap_infoN s
z*TestSystemSwapMemory.meminfo_has_swap_infoc C s* t � j}tj� j}t|| �tk s&t�d S )N)rm rj r� � swap_memoryr� r r� )r� r� r� r6 r6 r7 r� U s
z TestSystemSwapMemory.test_totalc C s* t � j}tj� j}t|| �tk s&t�d S )N)rm rk r� r� r� r r� )r� r� r� r6 r6 r7 r� Z s
z TestSystemSwapMemory.test_usedc C s* t � j}tj� j}t|| �tk s&t�d S )N)rm r_ r� r� r� r r� )r� r� r� r6 r6 r7 r� ` s
z TestSystemSwapMemory.test_freec C s� t jddd���}tjdd��h}tjd� tj� }|js:t�t |�dksJt�|d }dt
|j �ksdt�|j dksrt�|j
dks�t�W d Q R X W d Q R X d S ) Nzpsutil._common.openT)r� )r� r� r= r z9'sin' and 'sout' swap memory stats couldn't be determined)r r� r� r� r� r� r� r� r� rD r� r� �sin�sout)r� r� r� r� r� r6 r6 r7 �test_missing_sin_soutf s
z*TestSystemSwapMemory.test_missing_sin_soutc C s� t dttjd����}tjdd��h}tjd� tj� }|j s>t
�t |�dksNt
�|d }dt |j
�ksht
�|jdksvt
�|jdks�t
�W d Q R X W d Q R X d S ) Nz /proc/vmstatzno such file or directoryT)r� r� 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� rD r� r� r� r� )r� r� r� r� r� r6 r6 r7 �test_no_vmstat_mockedu s
z*TestSystemSwapMemory.test_no_vmstat_mockedc C s� | j � stjd��tjd��}tj� }W d Q R X |j s<t�dd l j
}|j � \}}}}}}}||9 }||9 }|j |kszt�t
|j| �tk s�t�d S )Nz!/proc/meminfo has no swap metricsz"psutil._pslinux.cext.linux_sysinfor )r� r rw r r� r� r� r� r� Zpsutil._psutil_linuxZ
_psutil_linuxZ
linux_sysinforj r� r_ r )r� r� �swapZcextri rj r_ Zunit_multiplierr6 r6 r7 � test_meminfo_against_sysinfo� s
z1TestSystemSwapMemory.test_meminfo_against_sysinfoc C s. t ddi��}tj� |js t�W d Q R X d S )Nz
/proc/meminfo� )r� r� r� r� r� )r� r� r6 r6 r7 �#test_emulate_meminfo_has_no_metrics� s z8TestSystemSwapMemory.test_emulate_meminfo_has_no_metricsN)
r� r� r� � staticmethodr� r� r r� r� r� r� r� r� r6 r6 r6 r7 r� L s r� c @ s e Zd Zdd� ZdS )�TestSystemCPUTimesc C s� t j� j}tjdtj� d �d }ttt |j
d���}|d krLd|ksXt �n d|ksXt �|d
krnd |kszt �n d |kszt �|dkr�d |ks�t �n d |ks�t �d S )Nz
\d+\.\d+\.\d+r� r �.� � �stealr& Zguestr� Z
guest_nice)r� r� r� )r� r� r& )r� r� r ) r� � cpu_times�_fieldsrz r{ �os�unamerx ry rf rB r� )r� rM Z
kernel_verZkernel_ver_infor6 r6 r7 � test_fields� s
z TestSystemCPUTimes.test_fieldsN)r� r� r� r� r6 r6 r6 r7 r� � s r� c @ s� e Zd Zejjejjd� dd�dd� �Z ejjejjd� dd�dd � �Z
ejje d
� d d�d d
� �Z ejje d� dd�dd� �Z
dd� ZdS )�TestSystemCPUCountLogicalz /sys/devices/system/cpu/onlinez-/sys/devices/system/cpu/online does not exist)r� c
C sV t d��}|j� j� }W d Q R X dt|�krRt|jd�d �d }tj� |ksRt�d S )Nz /sys/devices/system/cpu/online�-r= ) rA r� rt r� rf rB r� � cpu_countr� )r� rK �valuer6 r6 r7 � test_against_sysdev_cpu_online� s
z8TestSystemCPUCountLogical.test_against_sysdev_cpu_onlinez/sys/devices/system/cpuz&/sys/devices/system/cpu does not existc C s0 t jd�}tdd� |D ��}tj� |ks,t�d S )Nz/sys/devices/system/cpuc S s g | ]}t jd |�dk r|�qS )zcpu\d+$N)rz �search)rW rT r6 r6 r7 rY � s zITestSystemCPUCountLogical.test_against_sysdev_cpu_num.<locals>.<listcomp>)r� �listdirrD r� r� r� )r� Zls�countr6 r6 r7 �test_against_sysdev_cpu_num� s
z5TestSystemCPUCountLogical.test_against_sysdev_cpu_numZnprocznproc utility not availablec C s$ t td��}tjdd�|ks t�d S )Nz nproc --allT)�logical)rf r r� r� r� )r� �numr6 r6 r7 �test_against_nproc� s z,TestSystemCPUCountLogical.test_against_nproc�lscpuzlscpu utility not availablec C s8 t d�}tdd� |jd�D ��}tjdd�|ks4t�d S )Nzlscpu -pc S s g | ]}|j d �s|�qS )�#)rc )rW rT r6 r6 r7 rY � s z@TestSystemCPUCountLogical.test_against_lscpu.<locals>.<listcomp>rb T)r� )r rD rB r� r� r� )r� rg r� r6 r6 r7 �test_against_lscpu� s z,TestSystemCPUCountLogical.test_against_lscpuc % C s, dd l }|jj� }tjdtd��� }|jj� |ks6t�|js@t�tjddd��<}|jj� d ksbt�|jdkspt�|j d d dks�t�W d Q R X t
d d
��}|j � }W d Q R X t j
|�}tjd|dd ��}|jj� |ks�t�W d Q R X td d i��&}|jj� |k�st�|j�st�W d Q R X W d Q R X d S )
Nr zpsutil._pslinux.os.sysconf)r� zpsutil._common.openT)r� r� z
/proc/statz
/proc/cpuinfo�rb)� return_valuer� r� )�psutil._pslinux�_pslinuxZcpu_count_logicalr r� rE r� r� Z
call_count� call_argsrA r� r~ r� r� )r� r� �originalr� rK Z cpuinfo_data� fake_filer6 r6 r7 �test_emulate_fallbacks� s*
z0TestSystemCPUCountLogical.test_emulate_fallbacksN)r� r� r� r �mark�skipifr� �path�existsr� r� r r� r� r r6 r6 r6 r7 r� � s r� c @ s: e Zd Zejjed� dd�dd� �Zdd� Zdd � Z d
S ) �TestSystemCPUCountCoresr� zlscpu utility not available)r� c C s` t d�}t� }x4|jd�D ]&}|jd�s|jd�}|j|d � qW tjdd�t|�ks\t�d S )Nzlscpu -prb r� �,r= F)r� ) r �setrB rc �addr� r� rD r� )r� rg Zcore_idsrL rM r6 r6 r7 r� s
z*TestSystemCPUCountCores.test_against_lscpuc C sP t jj� }tjdg d��}t jj� }|js.t�W d Q R X |d k rL||ksLt�d S )Nz glob.glob)r� )r� r� �cpu_count_coresr r� r� r� )r� Zmeth_1r� Zmeth_2r6 r6 r7 �
test_method_2 s
z%TestSystemCPUCountCores.test_method_2c C s^ t jdg d��2}t jddd��}tjj� d ks2t�W d Q R X W d Q R X |jsPt�|jsZt�d S )Nz glob.glob)r� zpsutil._common.openT)r� )r r� r� r� r
r� r� )r� �m1�m2r6 r6 r7 �test_emulate_none s
&
z)TestSystemCPUCountCores.test_emulate_noneN)
r� r� r� r r r r r� r r r6 r6 r6 r7 r s r c @ s� e Zd Zejje dd�dd� �Zejje dd�ejjedd�dd� ��Z ejje dd�dd � �Z
ejje dd�d
d � �Z ejje dd�d d
� �Z dS )�TestSystemCPUFrequencyz
not supported)r� c s@ � fdd�}t jj� tjd|dd�� tj� s2t�W d Q R X d S )Nc s | j d�rdS � | �S d S )Nz&/sys/devices/system/cpu/cpufreq/policyF)rc )r )� orig_existsr6 r7 �path_exists_mock( s
zMTestSystemCPUFrequency.test_emulate_use_second_file.<locals>.path_exists_mockzos.path.existsT)r� r� )r� r r r r� r� �cpu_freqr� )r� r r6 )r r7 � test_emulate_use_second_file% s
z3TestSystemCPUFrequency.test_emulate_use_second_filez,aarch64 does not report mhz in /proc/cpuinfoc s� � fdd�}t jj� z�tjd|d��r ttj� tj� }|sDt |��|j
dksRt �|j dks`t �x0tjdd�D ] }|j
dks�t �|j dksnt �qnW W d Q R X W d ttj� tt� X d S )Nc s | j d�rdS � | �S d S )Nz/sys/devices/system/cpu/F)rc )r )�os_path_existsr6 r7 r ; s
zITestSystemCPUFrequency.test_emulate_use_cpuinfo.<locals>.path_exists_mockzos.path.exists)r� g T)�percpu) r� r r r r� r r� r� r r� �max�min)r� r r� �freqr6 )r r7 �test_emulate_use_cpuinfo4 s
z/TestSystemCPUFrequency.test_emulate_use_cpuinfoc s� � fdd�}t � trdnd}tj||d��f tjddd��L tj� }|jd ksRt�|jd
krj|jd ksjt�|j d
kr�|j d ks�t�W d Q R X W d Q R X d S )
Nc s� | j d�r | jd�r tjd�S | j d�r<| jd�r<tjd�S | j d�rZ| jd�rZtjd�S | dkrltjd �S � | f|�|�S d S )
Nz/scaling_cur_freqz&/sys/devices/system/cpu/cpufreq/policys 500000z/scaling_min_freqs 600000z/scaling_max_freqs 700000z
/proc/cpuinfos cpu MHz : 500)�endswithrc r~ r� )r� r� r� )r� r6 r7 r� R s
z;TestSystemCPUFrequency.test_emulate_data.<locals>.open_mockz
builtins.openz__builtin__.open)r� zos.path.existsT)r� g @@g g ��@g �@)
rA r r r� r� r �currentr� r r )r� r� r� r r6 )r� r7 �test_emulate_dataP s
z(TestSystemCPUFrequency.test_emulate_datac s � fdd�}t � trdnd}tj||d��� tjddd��� tjd d
d��� tjdd �}|d jd
ksjt�|d jdkr�|d jdks�t�|d j dkr�|d j dks�t�|d jdks�t�|d jdkr�|d jdks�t�|d j dkr�|d j dks�t�W d Q R X W d Q R X W d Q R X d S )Nc s� | }|j d�r"|jd�r"tjd�S |j d�r@|jd�r@tjd�S |j d�r^|jd�r^tjd�S |j d�r||jd�r|tjd �S |j d�r�|jd�r�tjd
�S |j d�r�|jd�r�tjd �S | d kr�tjd
�S � | f|�|�S d S )Nz/scaling_cur_freqz'/sys/devices/system/cpu/cpufreq/policy0s 100000z/scaling_min_freqs 200000z/scaling_max_freqs 300000z'/sys/devices/system/cpu/cpufreq/policy1s 400000s 500000s 600000z
/proc/cpuinfos# cpu MHz : 100
cpu MHz : 400)r rc r~ r� )r� r� r� �n)r� r6 r7 r� s s,
z@TestSystemCPUFrequency.test_emulate_multi_cpu.<locals>.open_mockz
builtins.openz__builtin__.open)r� zos.path.existsT)r� z!psutil._pslinux.cpu_count_logicalr� )r r g Y@g g i@g �r@r= g y@g @@g ��@)
rA r r r� r� r r r� r r )r� r� r� r r6 )r� r7 �test_emulate_multi_cpuq s$ z-TestSystemCPUFrequency.test_emulate_multi_cpuc s� � fdd�}t � trdnd}tj||d��P tjddd��6 tjd d
d�� tj� }|jd ksbt�W d Q R X W d Q R X W d Q R X d S ) Nc sR | j d�rttjd��n6| j d�r,tjd�S | dkr>tjd�S � | f|�|�S d S )Nz/scaling_cur_freqrV z/cpuinfo_cur_freqs 200000z
/proc/cpuinfos cpu MHz : 200)r r� r� r� r~ r� )r� r� r� )r� r6 r7 r� � s
zOTestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_file.<locals>.open_mockz
builtins.openz__builtin__.open)r� zos.path.existsT)r� z!psutil._pslinux.cpu_count_logicalr= �� )rA r r r� r� r r r� )r� r� r� r r6 )r� r7 �%test_emulate_no_scaling_cur_freq_file� s
z<TestSystemCPUFrequency.test_emulate_no_scaling_cur_freq_fileN)
r� r� r� r r r r r r r r r r r6 r6 r6 r7 r # s !4r c @ s e Zd Zdd� ZdS )�TestSystemCPUStatsc C s* t d�}tj� j}t|| �dk s&t�d S )N�
interruptsi� )rr r� Z cpu_statsr" r� r� )r� r� r� r6 r6 r7 �test_interrupts� s
z"TestSystemCPUStats.test_interruptsN)r� r� r� r# r6 r6 r6 r7 r! � s r! c @ s& e Zd Zejje dd�dd� �ZdS )� TestLoadAvgz
not supported)r� c C s� t j� }td��}|j� j� }W d Q R X tt|d �|d �dk sHt�tt|d �|d �dk sht�tt|d �|d �dk s�t�d S )Nz
/proc/loadavgr r= r� )r� �
getloadavgrA r� rB r� �floatr� )r� r� rK Z
proc_valuer6 r6 r7 �test_getloadavg� s
zTestLoadAvg.test_getloadavgN)r� r� r� r r r r r' r6 r6 r6 r7 r$ � s r$ c @ s e Zd Zdd� ZdS )�TestSystemNetIfAddrsc C s� x�t j� j� D ]�\}}x�|D ]�}|jt jkr@|jt|�ks�t�q |jtj kr�|jt
|�ks^t�|j t |�kspt�|j
d k r�|j
t|�ks�t�q�t|�dks�t�q |jtjkr |jjd�d }|t|�ks t�q W qW d S )Nz0.0.0.0�%r )r� Z net_if_addrs�items�familyZAF_LINK�addressr^ r� r) r* r8 �netmaskr: � broadcastr< rI rB rS )r� r� �addrs�addrr, r6 r6 r7 �test_ips� s
z TestSystemNetIfAddrs.test_ipsN)r� r� r� r1 r6 r6 r6 r7 r( � s r( zQEMU user not supportedc @ sP e Zd Zejjed� dd�dd� �Zdd� Zejjed� dd�dd � �Z d
S ) �TestSystemNetIfStats�ifconfigz ifconfig utility not available)r� c C sx xrt j� j� D ]b\}}ytd| �}W n tk
r: Y qX |jd|kksRt|��|jtt j
d|�d �kst�qW d S )Nz ifconfig %sZRUNNINGz(?i)MTU[: ](\d+)r ) r� � net_if_statsr* r � RuntimeError�isupr� �mturf rz r{ )r� r� �statsrg r6 r6 r7 �test_against_ifconfig s z*TestSystemNetIfStats.test_against_ifconfigc C sP xJt j� j� D ]:\}}td| �� }|jt|j� j� �ks>t�W d Q R X qW d S )Nz/sys/class/net/%s/mtu) r� r4 r* rA r7 rf r� rt r� )r� r� r8 rK r6 r6 r7 �test_mtu s z TestSystemNetIfStats.test_mtuc C s d}x�t j� j� D ]�\}}ytd| �}W n tk
r> Y qX tjd|�}|r�t|j� �dkr�|d7 }t |j
d�j � j d��}t |j
j d��}||ks�t�qtjd|�}|rt|j� �dkr|d7 }t |j
d�j � j � �}t |j
j d��}||kst�qW |�s | jd ��d S )
Nr z ifconfig %szflags=(\d+)?<(.*?)>r� r= r z (.*) MTU:(\d+) Metric:(\d+)r� zno matches were found)r� r4 r* r r5 rz r� rD rO r �group�lowerrB �flagsr� �fail)r� Z
matches_foundr� r8 rg �matchZifconfig_flagsZ psutil_flagsr6 r6 r7 �
test_flags s( z TestSystemNetIfStats.test_flagsN)
r� r� r� r r r r r9 r: r@ r6 r6 r6 r7 r2 s
r2 c @ s0 e Zd Zejjed� dd�e� dd� ��ZdS )�TestSystemNetIOCountersr3 z ifconfig utility not available)r� c
C s* dd� }t jddd�}�x |j� D � ]�\}}y ||�}W n tk
rN w"Y nX t|j|d �dk sjt�t|j|d �dk s�t�t|j|d
�dk s�t�t|j |d �dk s�t�t|j
|d �dk s�t�t|j |d
�dk s�t�t|j |d �dk �st�t|j
|d �dk s"t�q"W d S )Nc S s� i }t d| �}ttjd|�d �|d< ttjd|�d �|d< ttjd|�d �|d< ttjd|�d �|d
< ttjd |�d �|d < ttjd |�d �|d
< ttjd|�d �|d< ttjd|�d �|d< |S )Nz ifconfig %szRX packets[: ](\d+)r � packets_recvzTX packets[: ](\d+)� packets_sentzerrors[: ](\d+)�errinr= �erroutzdropped[: ](\d+)�dropin�dropoutz#RX (?:packets \d+ +)?bytes[: ](\d+)�
bytes_recvz#TX (?:packets \d+ +)?bytes[: ](\d+)�
bytes_sent)r rf rz r{ )Znicr� rg r6 r6 r7 r3 E s z?TestSystemNetIOCounters.test_against_ifconfig.<locals>.ifconfigTF)Zpernic�nowraprH i �
rI rB rC rD rE rF rG i ( i ( )r� �net_io_countersr* r5 r� rH r� rI rB rC rD rE rF rG )r� r3 Znior� r8 Z ifconfig_retr6 r6 r7 r9 @ s z-TestSystemNetIOCounters.test_against_ifconfigN) r� r� r� r r r r r r9 r6 r6 r6 r7 rA > s rA c @ s8 e Zd Zejded�ejddd�dd� ��Zdd � Zd
S ) �TestSystemNetConnectionsz psutil._pslinux.socket.inet_ntop)r� z psutil._pslinux.supports_ipv6F)r� c C sR y*t j t jt j�}| j|j� |jd� W n t jk
r@ Y nX tjdd� d S )N�::1r �inet6)�kind)rN r ) r) rI � SOCK_STREAMZ
addCleanup�close�bind�errorr� �net_connections)r� �
supports_ipv6rJ r5 r6 r6 r7 � test_emulate_ipv6_unsupportedt s z6TestSystemNetConnections.test_emulate_ipv6_unsupportedc
C s< t jd�}td|i�� }tjdd� |js.t�W d Q R X d S )Na 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/proc/net/unix�unix)rP )r� r� r� r� rU r� r� )r� r� r� r6 r6 r7 �test_emulate_unix� s
z*TestSystemNetConnections.test_emulate_unixN)r� r� r� r r� rE rW rY r6 r6 r6 r7 rM r s rM c @ sB e Zd Zejjeed� dd�e� dd� ��Z dd� Z
dd � Z d
S ) �TestSystemDiskPartitions�statvfszos.statvfs() not available)r� c C sz dd� }xlt jdd�D ]\}t j|j�}||j�\}}}}|j|ksFt�t|j| �tk s\t�t|j | �tk st�qW d S )Nc S sz t d| �j� }|jd�}|jd� |jd�}|j� d d� \}}}}|dkrRd}t|�t|�t|� }}}||||fS )Nzdf -P -B 1 "%s"rb r r> ZnonerV )r rt rB �poprf )r rg rh rL �devrj rk r_ r6 r6 r7 �df� s
z4TestSystemDiskPartitions.test_against_df.<locals>.dfF)�all)
r� �disk_partitions�
disk_usage�
mountpointrj r� r� r_ r rk )r� r^ �part�usageri rj rk r_ r6 r6 r7 �test_against_df� s z(TestSystemDiskPartitions.test_against_dfc C s� t d��}|j� }W d Q R X d|krNx�tj� D ]}|jdkr.P q.W | jd��nxtjd�}tj d|dd��X}tj ddgd ��<}tj� }|j
s�t �|j
s�t �|s�t �|d
jdks�t �W d Q R X W d Q R X d S )Nz/proc/filesystems�zfsz couldn't find any ZFS partitionz
nodev zfs
zpsutil._common.openT)r� r� z$psutil._pslinux.cext.disk_partitions� /dev/sdb3�/�rw)r� r )rg rh rf ri ) rA r� r� r` �fstyper> r~ r r r� r� r� )r� rK r� rc r r r
r� r6 r6 r7 � test_zfs_fs� s(
z$TestSystemDiskPartitions.test_zfs_fsc C sT zFt jddd��.}tjt�� tj� W d Q R X |js:t�W d Q R X W d dt_ X d S )Nzos.path.realpathz
/non/existent)r� z/proc)
r r� r �raisesr r� r` r� r� � PROCFS_PATH)r� r� r6 r6 r7 �test_emulate_realpath_fail� s
z3TestSystemDiskPartitions.test_emulate_realpath_failN) r� r� r� r r r �hasattrr� r re rk rn r6 r6 r6 r7 rZ � s
rZ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TestSystemDiskIoCountersc C s� d}t d|i��� tjddd��� tjdd�}|jdks<t�|jd ksJt�|jd
t ks\t�|j
d ksjt�|j d ksxt�|j d
ks�t�|j
dt ks�t�|jdks�t�|jdks�t�W d Q R X W d Q R X d S )Nz+ 3 0 1 hda 2 3 4 5 6 7 8 9 10 11 12z/proc/diskstatsz!psutil._pslinux.is_storage_deviceT)r� F)rJ r= r� r� r> ro r� r� r� rK )r� r r� r� �disk_io_counters�
read_countr� �read_merged_count�
read_bytes� SECTOR_SIZE� read_time� write_count�write_merged_count� write_bytes�
write_time� busy_time)r� r� r� r6 r6 r7 �test_emulate_kernel_2_4� s z0TestSystemDiskIoCounters.test_emulate_kernel_2_4c C s� d}t d|i��� tjddd��� tjdd�}|jdks<t�|jd ksJt�|jd
t ks\t�|j
d ksjt�|j d ksxt�|j d
ks�t�|j
dt ks�t�|jdks�t�|jdks�t�W d Q R X W d Q R X d S )Nz' 3 0 hda 1 2 3 4 5 6 7 8 9 10 11z/proc/diskstatsz!psutil._pslinux.is_storage_deviceT)r� F)rJ r= r� r� r> ro r� r� r� rK )r� r r� r� rq rr r� rs rt ru rv rw rx ry rz r{ )r� r� r� r6 r6 r7 � test_emulate_kernel_2_6_full� s z5TestSystemDiskIoCounters.test_emulate_kernel_2_6_fullc C s� t ddi��� tjddd��� tjdd�}|jdks8t�|jd t ksJt�|j d
ksXt�|j
d t ksjt�|j d ksxt�|j d ks�t�|j
d ks�t�|jd ks�t�|jd ks�t�W d Q R X W d Q R X d S )
Nz/proc/diskstatsz 3 1 hda 1 2 3 4z!psutil._pslinux.is_storage_deviceT)r� F)rJ r= r� r� r> r )r� r r� r� rq rr r� rt ru rw ry rs rv rx rz r{ )r� r� r6 r6 r7 � test_emulate_kernel_2_6_limited� s z8TestSystemDiskIoCounters.test_emulate_kernel_2_6_limitedc C s� t jd�}td|i��� tjddd��l tjddd�}t|�dksFt�|d j d
ksXt�|d j d
ksjt�|d j
d ks|t�|d j
d ks�t�W d Q R X W d Q R X d S )
Nzx 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/proc/diskstatsz!psutil._pslinux.is_storage_deviceF)r� T)�perdiskrJ r� �nvme0n1r= Z nvme0n1p1ro ) r� r� r� r r� r� rq rD r� rr rw )r� r� r� r6 r6 r7 � test_emulate_include_partitions s z8TestSystemDiskIoCounters.test_emulate_include_partitionsc C s� t jd�}td|i��: tjddd�� tjddd�}|d ksBt�W d Q R X W d Q R X dd� }t jd�}td|i��L tjdd |d
��0 tjddd�}|jd ks�t�|j d ks�t�W d Q R X W d Q R X d S )
Nzx 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/proc/diskstatsz!psutil._pslinux.is_storage_deviceF)r� )r rJ c S s | dkS )Nr� r6 )r� r6 r6 r7 �is_storage_device4 s zSTestSystemDiskIoCounters.test_emulate_exclude_partitions.<locals>.is_storage_deviceT)r� r� r= ro )
r� r� r� r r� r� rq r� rr rw )r� r� r� r� r6 r6 r7 � test_emulate_exclude_partitions% s$
z8TestSystemDiskIoCounters.test_emulate_exclude_partitionsc
C sT dd� }t jdd�}tjdd|d�� t jdd�}W d Q R X t|�t|�ksPt�d S )Nc S s | dkS )Nz/proc/diskstatsr6 )r r6 r6 r7 r F s z?TestSystemDiskIoCounters.test_emulate_use_sysfs.<locals>.existsT)r z psutil._pslinux.os.path.exists)r� r� )r� rq r r� rD r� )r� r ZwprocfsZwsysfsr6 r6 r7 �test_emulate_use_sysfsE s z/TestSystemDiskIoCounters.test_emulate_use_sysfsc C sF dd� }t jdd|d��$ tjt�� tj� W d Q R X W d Q R X d S )Nc S s dS )NFr6 )r r6 r6 r7 r Q s z>TestSystemDiskIoCounters.test_emulate_not_impl.<locals>.existsz psutil._pslinux.os.path.existsT)r� r� )r r� r rl �NotImplementedErrorr� rq )r� r r6 r6 r7 �test_emulate_not_implP s
z.TestSystemDiskIoCounters.test_emulate_not_implN)
r� r� r� r| r} r~ r� r� r� r� r6 r6 r6 r7 rp � s rp c @ sj e Zd Zdd� Zdd� Zejjedd�dd� �Z ejje
d � d
d�ejjedd�d d � ��Z d
d� Z dS )�TestRootFsDeviceFinderc C s( t jd�j}t j|�| _t j|�| _d S )Nrh )r� ru �st_dev�major�minor)r� r] r6 r6 r7 �setUp] s z TestRootFsDeviceFinder.setUpc C s� t � }tjjd�r |j� n tjt�� |j� W d Q R X tjjd| j| j f �r\|j
� n tjt�� |j
� W d Q R X |j � d S )Nz/proc/partitionsz/sys/dev/block/%s:%s/uevent) r r� r r �ask_proc_partitionsr rl r r� r� �ask_sys_dev_block�ask_sys_class_block)r� �finderr6 r6 r7 �test_call_methodsb s
z(TestRootFsDeviceFinder.test_call_methodsz unsupported on GITHUB_ACTIONS)r� c C s� t � }|j� d k st�d } }}tjjd�r6|j� }tjjd| j| jf �rV|j � }|j
� }|ph|ph|}|r~|r~||ks~t�|r�|r�||ks�t�|r�|r�||ks�t�d S )Nz/proc/partitionsz/sys/dev/block/%s:%s/uevent) r �findr� r� r r r� r� r� r� r� )r� r� r� r� �c�baser6 r6 r7 �test_comparisonsr s z'TestRootFsDeviceFinder.test_comparisonsZfindmntz findmnt utility not availablec C s" t � j� }td�}||ks t�d S )Nzfindmnt -o SOURCE -rn /)r r� r r� )r� r� Z
findmnt_valuer6 r6 r7 �test_against_findmnt� s
z+TestRootFsDeviceFinder.test_against_findmntc C sl t jddgd��R}tj� d }|js(t�tsP|jdks:t�|jt� j � ks^t�n|jdks^t�W d Q R X d S ) Nz$psutil._pslinux.cext.disk_partitions� /dev/rootrh �ext4ri )r� r )r� rh r� ri )
r r� r� r` r� r� r �devicer r� )r� r� rc r6 r6 r7 �test_disk_partitions_mocked� s
z2TestRootFsDeviceFinder.test_disk_partitions_mockedN)
r� r� r� r� r� r r r r r� r r� r� r6 r6 r6 r7 r� [ s r� c @ sb e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze � e
j j e
d
d�dd� ��Zdd� ZdS )�TestMiscc C s( t d�}tj� }t|�t|�ks$t�d S )Nz boot time)rr r� � boot_timerf r� )r� r� r� r6 r6 r7 �test_boot_time� s zTestMisc.test_boot_timec K s� | j � }tj|� ttjj|d�d��$}|jd� |jd� |jd� W d Q R X �zt� � fdd�}trldnd }tj ||d
���� t
t � t j
t�� t j� W d Q R X t j
t�� t jd d � W d Q R X t j
t�� t j� W d Q R X t j
t�� t jd d � W d Q R X t j
t�� t j� W d Q R X t j
t�� t jd d � W d Q R X |t _t j� d
k�sbt�tt j� �d
k�sxt�t jd d �}t|�d
k�s�t�t jd d �}ttt|��d
k�s�t�ttjj|d�d��$}|jd� |jd� |jd� W d Q R X t j� d
k�s
t�tt jd d ��d
k�s$t�tt j� �d
k�s:t�tttt jd d ���d
k�sZt�W d Q R X W d tj|� t
t � X t jdk�s�t�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 s& | j d�rttjd��� | f|�|�S )Nz/proczrejecting access for test)rc r� r� r� )r� r� r� )r� r6 r7 r� � s
z4TestMisc.test_no_procfs_on_import.<locals>.open_mockz
builtins.openz__builtin__.open)r� T)r r 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)�
get_testfnr� �mkdirrA r rG �writer r r� r r� r rl r� r� � cpu_percent�cpu_times_percentrm r� �sumry �shutil�rmtree)r� Z my_procfsrK r� r� Zper_cpu_percentZper_cpu_times_percentr6 )r� r7 �test_no_procfs_on_import� sT
.
z!TestMisc.test_no_procfs_on_importc C s> t jd�j� }td|i��8}tj� |js.t�tjdd� tj� tjdd� W d Q R X t jd�j� }td|i��� tj� }|js�t�tjdd�}tj� }tjdd�}|dks�t�t |�dks�t�t |�dks�t�t |�dks�t�t t
t |��dks�t�t t
t |��dk�st�|j dk�s t�|j dk�s0t�W d Q R X d S )Nz~ 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
z
/proc/statT)r 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 g Y@)
r� r� r� r� r� r� r� r� r� r� ry r� �user)r� r� r� r� Zcpu_percent_percpur� Zcpu_times_percent_percpur6 r6 r7 �test_cpu_steal_decrease� s0
z TestMisc.test_cpu_steal_decreasec C sH t jddd��0}tjt�� tjj� W d Q R X |js:t �W d Q R X d S )Nzpsutil._common.openT)r� )
r r� r rl r5 r� r� r� r� r� )r� r� r6 r6 r7 �test_boot_time_mocked s z TestMisc.test_boot_time_mockedc C s$ x t j� D ]}|jdks
t�q
W d S )N�:0�:0.0)r� r� )r� Zusers�hostr� )r� r� r6 r6 r7 �
test_users s zTestMisc.test_usersc H C s@ | j � }tj|� �z |t_tjt�� tj� W d Q R X tjt�� tj � W d Q R X tjt�� tj dd� W d Q R X tjt�� tj
� W d Q R X tjt�� tj � W d Q R X tjt�� tj � W d Q R X tjt�� tj
� W d Q R X tjt�� tj� W d Q R X tjtj�� tj� W d Q R X W d dt_X d S )NT)r z/proc)r� r� r� r� rm r rl r� r� r� r� rU rL r4 r` �
NoSuchProcess�Process)r� Ztdirr6 r6 r7 �test_procfs_path s.
zTestMisc.test_procfs_pathzskip if pytest-parallel)r� c C s� t � �t tj� }|j� }t|�tr&dndks0t�t|dd� d�d j}|j |ksTt�tj|�}|j
� |tj � ksvt�W d Q R X d S )Nr� r� c S s | j S )N)�id)rT r6 r6 r7 �<lambda>H s z)TestMisc.test_issue_687.<locals>.<lambda>)�keyr= ) r r� r� �threadsrD r r� �sortedr� �pidZas_dictZpids)r� �pr� �tid�ptr6 r6 r7 �test_issue_687< s
zTestMisc.test_issue_687c C s8 t ddi��"}tjtj� �s t�|js*t�W d Q R X d S )Nz/proc/%s/statusrV )r� r� Z
pid_existsr� �getpidr� r� )r� r� r6 r6 r7 � test_pid_exists_no_proc_statusN s z'TestMisc.test_pid_exists_no_proc_statusN)r� r� r� r� r� r� r� r� r� r r r r r r� r� r6 r6 r6 r7 r� � s B' r� z
no batteryc @ sj e Zd Zejjed� dd�dd� �Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� ZdS )�TestSensorsBatteryZacpizacpi utility not available)r� c C sH t d�}t|jd�d j� jdd��}tj� j}t|| �dk sDt �d S )Nzacpi -br r= r) rV )
r rf rB rt �replacer� �sensors_battery�percentr� r� )r� rg Z
acpi_valuer� r6 r6 r7 � test_percent_ s
z TestSensorsBattery.test_percentc sj � fdd�}t � trdnd}tj||d��6}tj� jdks>t�tj� jtj ksRt�|j
s\t�W d Q R X d S )Nc s( | j d�rtjd�S � | f|�|�S d S )N�
AC0/online� AC/online� 1)r� r� )r r~ r� )r� r� r� )r� r6 r7 r� h s
z@TestSensorsBattery.test_emulate_power_plugged.<locals>.open_mockz
builtins.openz__builtin__.open)r� T) rA r r r� r� r� �
power_pluggedr� �secsleft�POWER_TIME_UNLIMITEDr� )r� r� r� r� r6 )r� r7 �test_emulate_power_pluggedf s z-TestSensorsBattery.test_emulate_power_pluggedc sV � fdd�}t � trdnd}tj||d��"}tj� jdks>t�|jsHt�W d Q R X d S )Nc s@ | j d�rttjd��n$| j d�r,tjd�S � | f|�|�S d S )N�
AC0/online� AC/onlinerV z/statusZcharging)r� r� )r r� r� r� r~ r )r� r� r� )r� r6 r7 r� { s
zBTestSensorsBattery.test_emulate_power_plugged_2.<locals>.open_mockz
builtins.openz__builtin__.open)r� T) rA r r r� r� r� r� r� r� )r� r� r� r� r6 )r� r7 � test_emulate_power_plugged_2x s z/TestSensorsBattery.test_emulate_power_plugged_2c sV � fdd�}t � trdnd}tj||d��"}tj� jdks>t�|jsHt�W d Q R X d S )Nc s( | j d�rtjd�S � | f|�|�S d S )N�
AC0/online� AC/online� 0)r� r� )r r~ r� )r� r� r� )r� r6 r7 r� � s
zDTestSensorsBattery.test_emulate_power_not_plugged.<locals>.open_mockz
builtins.openz__builtin__.open)r� F) rA r r r� r� r� r� r� r� )r� r� r� r� r6 )r� r7 � test_emulate_power_not_plugged� s z1TestSensorsBattery.test_emulate_power_not_pluggedc sV � fdd�}t � trdnd}tj||d��"}tj� jdks>t�|jsHt�W d Q R X d S )Nc s@ | j d�rttjd��n$| j d�r,tjd�S � | f|�|�S d S )N�
AC0/online� AC/onlinerV z/statusZ discharging)r� r� )r r� r� r� r~ r )r� r� r� )r� r6 r7 r� � s
zFTestSensorsBattery.test_emulate_power_not_plugged_2.<locals>.open_mockz
builtins.openz__builtin__.open)r� F) rA r r r� r� r� r� r� r� )r� r� r� r� r6 )r� r7 � test_emulate_power_not_plugged_2� s z3TestSensorsBattery.test_emulate_power_not_plugged_2c sV � fdd�}t � trdnd}tj||d��"}tj� jd ks>t�|jsHt�W d Q R X d S )Nc s@ | j d�rttjd��n$| j d�r,tjd�S � | f|�|�S d S )N�"/sys/class/power_supply/AC0/online�!/sys/class/power_supply/AC/onlinerV z#/sys/class/power_supply/BAT0/statuss ???)r� r� )rc r� r� r� r~ r� )r� r� r� )r� r6 r7 r� � s
zETestSensorsBattery.test_emulate_power_undetermined.<locals>.open_mockz
builtins.openz__builtin__.open)r� ) rA r r r� r� r� r� r� r� )r� r� r� r� r6 )r� r7 � test_emulate_power_undetermined� s z2TestSensorsBattery.test_emulate_power_undeterminedc C s8 t ddi��"}tj� jdks t�|js*t�W d Q R X d S )Nz(/sys/class/power_supply/BAT0/energy_fullr� r )r� r� r� r� r� r� )r� r� r6 r6 r7 �test_emulate_energy_full_0� s
z-TestSensorsBattery.test_emulate_energy_full_0c C sj t dttjd���N t dttjd���0 tddi�� tj� jdksHt�W d Q R X W d Q R X W d Q R X d S )Nz(/sys/class/power_supply/BAT0/energy_fullrV z(/sys/class/power_supply/BAT0/charge_fullz%/sys/class/power_supply/BAT0/capacitys 88�X ) r� r� r� r� r� r� r� r� r� )r� r6 r6 r7 �"test_emulate_energy_full_not_avail� s z5TestSensorsBattery.test_emulate_energy_full_not_availc C sp t dttjd���T t dttjd���6 t dttjd��� tj� jd ksNt�W d Q R X W d Q R X W d Q R X d S )Nz!/sys/class/power_supply/AC/onlinerV z"/sys/class/power_supply/AC0/onlinez#/sys/class/power_supply/BAT0/status)r� r� r� r� r� r� r� r� )r� r6 r6 r7 �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� r6 r6 r6 r7 r� \ s r� c @ s e Zd Zdd� ZdS )�TestSensorsBatteryEmulatedc sz � fdd�}t � trdnd}tjddgd��0}tj||d��}tj� d k sNt�W d Q R X W d Q R X |jslt�|jsvt�d S ) Nc sP | j d�rtjd�S | j d�r(tjd�S | j d�r<tjd�S � | f|�|�S d S )Nz /energy_nowZ60000000z
/power_now�0z /energy_fullZ60000001)r r~ r )r� r� r� )r� r6 r7 r� � s
z5TestSensorsBatteryEmulated.test_it.<locals>.open_mockz
builtins.openz__builtin__.openz
os.listdirZBAT0)r� )r� )rA r r r� r� r� r� r� )r� r� r� ZmlistdirZmopenr6 )r� r7 �test_it� s
$
z"TestSensorsBatteryEmulated.test_itN)r� r� r� r� r6 r6 r6 r7 r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestSensorsTemperaturesc s� � fdd�}t � trdnd}tj||d��j tjddgd��N tj� d d
}|jd ks\t�|jd ksjt�|j d
ksxt�|j
dks�t�W d Q R X W d Q R X d S )Nc sx | j d�rtjd�S | j d�r(tjd�S | j d�r<tjd�S | j d�rPtjd�S | j d �rdtjd
�S � | f|�|�S d S ) Nz/namer� z /temp1_label�labelz /temp1_inputs 30000z
/temp1_maxs 40000z /temp1_crits 50000)r r~ r r� )r� r� r� )r� r6 r7 r� � s
zCTestSensorsTemperatures.test_emulate_class_hwmon.<locals>.open_mockz
builtins.openz__builtin__.open)r� z glob.globz /sys/class/hwmon/hwmon0/temp1)r� r� r r� g >@g D@g I@) rA r r r� r� �sensors_temperaturesr� r� r �high�critical)r� r� r� �tempr6 )r� r7 �test_emulate_class_hwmon� s z0TestSensorsTemperatures.test_emulate_class_hwmonc s� � fdd�}dd� }t � tr dnd}tj||d��j tjdd |d
��N tj� d d }|jd
ksdt�|jdksrt�|j dks�t�|j
dks�t�W d Q R X W d Q R X d S )Nc sd | j d�rtjd�S | j d�r(tjd�S | j d�r<tjd�S | j d�rPtjd�S � | f|�|�S d S ) NZ0_temps 50000r� s 30000Z0_typer� �typer� )r r~ r� r )r� r� r� )r� r6 r7 r� s
zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.open_mockc S s: | dkr g S | dkrg S | dkr&dgS | dkr6ddgS 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_tempr6 )r r6 r6 r7 � glob_mock$ s zETestSensorsTemperatures.test_emulate_class_thermal.<locals>.glob_mockz
builtins.openz__builtin__.open)r� z glob.globT)r� r� r� r rV g >@g I@) rA r r r� r� r� r� r� r r� r� )r� r� r� r� r� r6 )r� r7 �test_emulate_class_thermal s z2TestSensorsTemperatures.test_emulate_class_thermalN)r� r� r� r� r� r6 r6 r6 r7 r� � s r� c @ s e Zd Zdd� ZdS )�TestSensorsFansc s� � fdd�}t � trdnd}tj||d��N tjddgd��2 tj� d d
}|jd ks\t�|jd ksjt�W d Q R X W d Q R X d S )
Nc sP | j d�rtjd�S | j d�r(tjd�S | j d�r<tjd�S � | f|�|�S d S )Nz/namer� z /fan1_labelr� z /fan1_input�2000)r r~ r )r� r� r� )r� r6 r7 r� @ s
z4TestSensorsFans.test_emulate_data.<locals>.open_mockz
builtins.openz__builtin__.open)r� z glob.globz /sys/class/hwmon/hwmon2/fan1)r� r� r r� i� ) rA r r r� r� Z sensors_fansr� r� r )r� r� r� Zfanr6 )r� r7 r ? s
z!TestSensorsFans.test_emulate_dataN)r� r� r� r r6 r6 r6 r7 r� = s r� c @ s� e Zd Ze� dd� �Zdd� Zejje dd�dd� �Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zejje d d�d d!� �Zd"d#� Zd$d%� Zd&d'� Zd(S ))� TestProcessc C s� | j � }tjj|j�j� \}}}tj|j�jdd�}t|tdd� |D �� �dk sVt �t|tdd� |D �� �dk sxt �t|tdd� |D �� �dk s�t �d S )NF)Zgroupedc S s g | ]}|j |j �qS r6 )Z
private_dirtyZ
private_clean)rW rT r6 r6 r7 rY b s z?TestProcess.test_parse_smaps_vs_memory_maps.<locals>.<listcomp>i c S s g | ]
}|j �qS r6 )�pss)rW rT r6 r6 r7 rY e s c S s g | ]
}|j �qS r6 )r� )rW rT r6 r6 r7 rY f s )
Zspawn_testprocr� r� r� r� � _parse_smaps� memory_mapsr� r� r� )r� Zsproc�ussr� r� �mapsr6 r6 r7 � test_parse_smaps_vs_memory_maps\ s
"z+TestProcess.test_parse_smaps_vs_memory_mapsc
C s~ t jd�j� }tdtj� |i��R}tjjtj� �}|j � \}}}|j
sLt �|d ksXt �|d ksdt �|d
kspt �W d Q R X d S )Nan 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
z/proc/%s/smapsr� r� � i r� r" �
� i l i i < ) r� r� r� r� r� r� r� r� r� r� r� r� )r� r� r� r� r� r� r� r6 r6 r7 �test_parse_smaps_mockedh s
z#TestProcess.test_parse_smaps_mockedzunreliable on PYPY)r� c : C sl dd� }| j � }t|d�� ||�jdks.t�W d Q R X t|�� ||�jdksTt�W d Q R X t|d�� ||�jdks|t�W d Q R X t|d�� ||�jdks�t�W d Q R X t|d�� ||�jdks�t�W d Q R X t|d�� ||�jdks�t�W d Q R X t�rht|� t|d �� ||�jdk�s,t�W d Q R X t|� t|d
�� ||�jdk�s^t�W d Q R X d S ) Nc S s\ t j� }tj� t }x:x4|j� D ](}|jtjj| �kr:|S tj� |kr P q W qW td��d S )Nz timeout looking for test file) r� r� �timer �
open_filesr r� �abspathr5 )�fnamer� Z giveup_at�filer6 r6 r7 �
get_test_file� s
z7TestProcess.test_open_files_mode.<locals>.get_test_filer� �rr� zr+zw+za+rT zx+)r� rA �moder� r r )r� r� Ztestfnr6 r6 r7 �test_open_files_mode� s*
z TestProcess.test_open_files_modec s� t j� ��j� � t| j� d��� t� �fdd�� tjdtt j
d�d�� }�j� g ksZt �|j sdt �W d Q R X tjdtt j
d�d�� }�j� g ks�t �|j s�t �W d Q R X W d Q R X d S )Nr� c s t �j� �t � �kS )N)rD r� r6 )�filesr� r6 r7 r� � s z7TestProcess.test_open_files_file_gone.<locals>.<lambda>zpsutil._pslinux.os.readlinkrV )r� )r� r� r� rA r� r r r� �OSErrorr� r� r� r� �EINVAL)r� r� r6 )r� r� r7 �test_open_files_file_gone� s z%TestProcess.test_open_files_file_gonec s� t j� ��j� � t| j� d��` t� �fdd�� tr:dnd}tj|t t
j d�d�� }�j� g ksft �|j
spt �W d Q R X W d Q R X d S )Nr� c s t �j� �t � �kS )N)rD r� r6 )r� r� r6 r7 r� � s z5TestProcess.test_open_files_fd_gone.<locals>.<lambda>z
builtins.openz__builtin__.openrV )r� )r� r� r� rA r� r r r r� r� r� r� r� r� )r� r� r� r6 )r� r� r7 �test_open_files_fd_gone� s z#TestProcess.test_open_files_fd_gonec s� t j� ��j� � t| j� d��n t� �fdd�� d}tj|tt j
d�d��6}tjd�� �j� g ksjt �|j stt �W d Q R X W d Q R X W d Q R X d S )Nr� c s t �j� �t � �kS )N)rD r� r6 )r� r� r6 r7 r� � s z:TestProcess.test_open_files_enametoolong.<locals>.<lambda>zpsutil._pslinux.os.readlinkrV )r� zpsutil._pslinux.debug)
r� r� r� rA r� r r r� r� r� � ENAMETOOLONGr� r� )r� r� r� r6 )r� r� r7 � test_open_files_enametoolong� s z(TestProcess.test_open_files_enametoolongc C sD t jdi d��,}tjjtj� �j� d ks,t�|j s6t�W d Q R X d S )Nz)psutil._pslinux._psposix.get_terminal_map)r� )
r r� r� r� r� r� r� �terminalr� r� )r� r� r6 r6 r7 �test_terminal_mocked� s
z TestProcess.test_terminal_mockedc C s� t j� }tjd�}tjd|dd��$}|j� ddgks8t�|jsBt�W d Q R X tjd�}tjd|dd��&}|j� dddgks~t�|js�t�W d Q R X d S ) Nzfoo bar zpsutil._common.openT)r� r� �foo�barz foo bar rV ) r� r� r~ r r r� �cmdliner� r� )r� r� r r� r6 r6 r7 �test_cmdline_mocked� s
z TestProcess.test_cmdline_mockedc C s� t j� }tjd�}tjd|dd��$}|j� ddgks8t�|jsBt�W d Q R X tjd�}tjd|dd��&}|j� dddgks~t�|js�t�W d Q R X d S ) Nzfoo bar zpsutil._common.openT)r� r� r r z foo bar rV ) r� r� r~ r r r� r r� r� )r� r� r r� r6 r6 r7 �test_cmdline_spaces_mocked s
z&TestProcess.test_cmdline_spaces_mockedc
C sP t j� }tjd�}tjd|dd��$}|j� ddgks8t�|jsBt�W d Q R X d S )Nzfoo bar zpsutil._common.openT)r� r� r r ) r� r� r~ r r r� r r� r� )r� r� r r� r6 r6 r7 � test_cmdline_mixed_separators s
z)TestProcess.test_cmdline_mixed_separatorsc C sF t jddd��. tj� j� dks$t�tj� j� dks8t�W d Q R X d S )Nzpsutil._pslinux.os.readlinkz/home/foo (deleted))r� z /home/foo)r r� r� r� �exer� �cwd)r� r6 r6 r7 �!test_readlink_path_deleted_mocked! s z-TestProcess.test_readlink_path_deleted_mockedc s� � fdd�}t � trdnd}tj||d��(}tj� j� }|jsBt�|g ksNt�W d Q R X � fdd�}tj||d��* t j
tj �� tj� j� W d Q R X W d Q R X d S )Nc s4 | j dtj� �r ttjd��n� | f|�|�S d S )Nz
/proc/%s/taskrV )rc r� r� r� r� r� )r� r� r� )r� r6 r7 � open_mock_1- s z4TestProcess.test_threads_mocked.<locals>.open_mock_1z
builtins.openz__builtin__.open)r� c s4 | j dtj� �r ttjd��n� | f|�|�S d S )Nz
/proc/%s/taskrV )rc r� r� r� r� �EPERM)r� r� r� )r� r6 r7 � open_mock_2< s z4TestProcess.test_threads_mocked.<locals>.open_mock_2) rA r r r� r� r� r� r� r� r rl � AccessDenied)r� r r� r� r� r r6 )r� r7 �test_threads_mocked( s
z TestProcess.test_threads_mockedc C sb t jdttjd�d��B}t jdg d��( tj� j� }|js>t �|dksJt �W d Q R X W d Q R X d S )Nzpsutil._pslinux.readlinkrV )r� z psutil._pslinux.Process.cmdline)r� )
r r� r� r� r� r� r� r r� r� )r� r� r� r6 r6 r7 �test_exe_mockedF s
zTestProcess.test_exe_mockedc C sZ t dtj� ttjd���6}tj� }tj t
�� |j � W d Q R X |j sLt
�W d Q R X d S )Nz/proc/%s/smapsrV )r� r� r� r� r� r� r� r� r rl r r� r� r� )r� r� r� r6 r6 r7 �test_issue_1014R s zTestProcess.test_issue_1014z
not supportedc C s� t jdttjd�d��T}t jddd��:}tj� }|j� tj tj
��}|j tj � W d Q R X W d Q R X W d Q R X |j
szt�|j
s�t�|jj|jks�t�|jj|j� ks�t�d S )Nzpsutil._pslinux.prlimitrV )r� z"psutil._pslinux.Process._is_zombieT)r� )r r� r� r� ZENOSYSr� r� r� r rl �
ZombieProcessZrlimitZ
RLIMIT_NOFILEr� r� r� r� )r� r r
r� �cmr6 r6 r7 �test_rlimit_zombie] s
*
z TestProcess.test_rlimit_zombiec * C sZ dddddddddddddddddddddd ddddddddddddddddd ddd
g*}d j |�j� }td tj� |i��� tj� }|j� d
ks�t�|j � tj
ks�t�|j � dks�t�|j � dt
tj� ks�t�|j� }|jdt
ks�t�|jdt
ks�t�|jdt
k�st�|jdt
k�s&t�|jdt
k�s:t�|j� dk�sLt�W d Q R X d S )Nr� z(cat)�Z�1�2�3�4�5�6�7rs z
/proc/%s/stat�catr= r� r� r� r> ro r� )rG r� r� r� r� r� r� r� r� �status�
STATUS_ZOMBIE�ppidZ create_timer r� r� r� �system�
children_user�children_systemZiowaitZcpu_num)r� r� r� r� Zcpur6 r6 r7 �test_stat_file_parsingq sp z"TestProcess.test_stat_file_parsingc C s� t jd�j� }tdtj� |i��� tj� }|j� j dks>t
�|j� j dksPt
�|j � dks`t
�|j
� }|jdksvt
�|jdks�t
�|jdks�t
�|j� }|jd ks�t
�|jd
ks�t
�|jd ks�t
�|jj� ttd ��ks�t
�W d Q R X d S )
Nz� 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: 13z/proc/%s/statusr� r� �B i� i� i� i� i� i� r� )r� r� r� r� r� r� r� r� �num_ctx_switches� voluntaryr� � involuntary� num_threads�uids�real� effective�saved�gids�_proc�_get_eligible_cpus�listrF )r� r� r� r- r1 r6 r6 r7 �test_status_file_parsing� s
z$TestProcess.test_status_file_parsingc C s^ t jdttjd�d��>}tj� }t jd�� |j� g ks<t�|j sFt�W d Q R X W d Q R X d S )Nzpsutil._pslinux.os.readlinkrV )r� zpsutil._pslinux.debug)
r r� r� r� r r� r� rU r� r� )r� r� r� r6 r6 r7 �!test_net_connections_enametoolong� s z-TestProcess.test_net_connections_enametoolongN)r� r� r� r r� r� r r r r r� r� r r r r r r
r
r r r r
r r' r5 r6 r6 r6 r6 r7 r� Z s$ "# <r� c @ s� e Zd ZdZedd� �Zdd� Zdd� Zej j
e dd �d
d � �Z d d
� Z
dd� Zdd� Zdd� Ze� dd� �Zdd� Zdd� ZdS )�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)r� r� �proc)�clsr6 r6 r7 �
setUpClass� s z#TestProcessAgainstStatus.setUpClassc C s| t jjd| jj ��^}xJ|D ]B}|j� }|j|�r |jd�d }yt|�S t k
r\ |S X q W t d| ��W d Q R X d S )Nz/proc/%s/status� r� z
can't find %r)
r� � _psplatform� open_textr8 r� rt rc � partitionrf rE )r� Z linestartrK rL r� r6 r6 r7 �read_status_file� s
z)TestProcessAgainstStatus.read_status_filec C s | j d�}| jj� |ks t�d S )NzName:)r? r8 r� r� )r� r� r6 r6 r7 � test_name� s
z"TestProcessAgainstStatus.test_namezQEMU user not supported)r� c C sH | j d�}||jd�d |jd�� }|jdd�}| jj� |ksDt�d S )NzState:�(r= �)rs r� )r? r� �rfindr� r8 r! r� )r� r� r6 r6 r7 � test_status� s
z$TestProcessAgainstStatus.test_statusc C s | j d�}| jj� |ks t�d S )NzPPid:)r? r8 r# r� )r� r� r6 r6 r7 � test_ppid� s
z"TestProcessAgainstStatus.test_ppidc C s | j d�}| jj� |ks t�d S )NzThreads:)r? r8 r, r� )r� r� r6 r6 r7 �test_num_threads� s
z)TestProcessAgainstStatus.test_num_threadsc C s: | j d�}ttt|j� dd� ��}| jj� |ks6t�d S )NzUid:r= r> )r? rx ry rf rB r8 r- r� )r� r� r6 r6 r7 � test_uids s
z"TestProcessAgainstStatus.test_uidsc C s: | j d�}ttt|j� dd� ��}| jj� |ks6t�d S )NzGid:r= r> )r? rx ry rf rB r8 r1 r� )r� r� r6 r6 r7 � test_gids s
z"TestProcessAgainstStatus.test_gidsc C s@ | j d�}| jj� j|ks t�| j d�}| jj� j|ks<t�d S )Nzvoluntary_ctxt_switches:znonvoluntary_ctxt_switches:)r? r8 r) r* r� r+ )r� r� r6 r6 r7 �test_num_ctx_switches s
z.TestProcessAgainstStatus.test_num_ctx_switchesc C sN | j d�}dt|�krJtt|jd��\}}| jj� tt||d ��ksJt �d S )NzCpus_allowed_list:r� r= )
r? r� ry rf rB r8 Z cpu_affinityr4 rF r� )r� r� Zmin_Zmax_r6 r6 r7 �test_cpu_affinity s
z*TestProcessAgainstStatus.test_cpu_affinityc
C sT | j d�}tjd��}| jjj� W d Q R X dt|�krF|j sPt�n
|jsPt�d S )NzCpus_allowed_list:z psutil._pslinux.per_cpu_timesr� ) r? r r� r8 r2 r3 r� r� r� )r� r� r� r6 r6 r7 � test_cpu_affinity_eligible_cpus s
z8TestProcessAgainstStatus.test_cpu_affinity_eligible_cpusN)r� r� r� �__doc__� classmethodr: r? r@ r r r r rD rE rF rG rH r rI rJ rK r6 r6 r6 r7 r7 � s r7 c @ s e Zd Zdd� ZdS )� TestUtilsc C s< t jddd��$}tjjd�dks$t�|js.t�W d Q R X d S )Nz os.readlinkz
foo (deleted))r� r r )r r� r� r< �readlinkr� r� )r� r� r6 r6 r7 �
test_readlink) s zTestUtils.test_readlinkN)r� r� r� rP r6 r6 r6 r7 rN ' s rN )frL �
__future__r rd r, r� �globr~ r� rz r� r) r2 r� r� r� r� 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 r r� r r r r! r r� �dirname�__file__�HEREr1 Z SIOCGIFCONFr[ r9 r; ru ZEMPTY_TEMPERATURESr8 r: r< rS r^ rm rq rr r| �contextmanagerr� r� r r r� r� r� r� r� r� r r r! r$ r( r2 rA rM rZ rp r� r� r� r� r� r� r� r7 rN r6 r6 r6 r7 �<module> s� 91 RZI '
63 A H 8
C zS |