OwlCyberSecurity - MANAGER
Edit File: test_linux.cpython-36.pyc
3 ��JZ�3 � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lm Z dd lm Z 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+ ej,j-ej,j.e/��Z0dZ1d Z2d Z3e�r�d Z4d d � Z5d!d"� Z6d#d$� Z7d%d&� Z8d'd(� Z9d)d*� Z:e*j;e d+�G d,d-� d-e*j<��Z=e*j;e d+�G d.d/� d/e*j<��Z>e*j;e d+�G d0d1� d1e*j<��Z?e*j;e d+�G d2d3� d3e*j<��Z@e*j;e d+�G d4d5� d5e*j<��ZAe*j;e d+�G d6d7� d7e*j<��ZBe*j;e d+�G d8d9� d9e*j<��ZCe*j;e d+�e*j;e d:�G d;d<� d<e*j<���ZDe*j;e d+�G d=d>� d>e*j<��ZEe*j;e d+�G d?d@� d@e*j<��ZFe*j;e d+�G dAdB� dBe*j<��ZGe*j;e d+�G dCdD� dDe*j<��ZHe*j;e d+�G dEdF� dFe*j<��ZIeJdGk�r�e#e/� dS )HzLinux specific tests.� )�divisionN)�LINUX)�PY3)�u)� call_until)� HAS_BATTERY)� HAS_CPU_FREQ)� HAS_RLIMIT)�MEMORY_TOLERANCE)�mock)�PYPY)�pyrun)� reap_children)� reload_module)�retry_before_failing)�run_test_module_by_name)� safe_rmpath)�sh)�skip_on_not_implemented)�TESTFN)� ThreadTask)�TRAVIS)�unittest)�whichi� i� i'� i c C 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�closingZ inet_ntoa�ioctl�fileno� SIOCGIFADDR�struct�pack)�ifnamer �s� r- �"/usr/lib64/python3.6/test_linux.py�get_ipv4_address@ s r/ 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 )Nr- )�xr- r- r. �ordW s z get_mac_address.<locals>.ord� c s g | ]}d � |� �qS )z%02x:r- )�.0�char)r1 r- r. � <listcomp>\ s z#get_mac_address.<locals>.<listcomp>� r � ���)r r r r! r" r# r$ r% r& r' � SIOCGIFHWADDRr) r* � __builtin__r1 �join)r+ r r, �infor: r- )r1 r. �get_mac_addressM s r= c C st t d�} | jd�}xJ|D ]B}|jd�r|j� \}}}}tjdd�}|t|�t|�t|��S qW tddj|� ��dS )zQParse 'free' cmd and return swap memory's s total, used and free values. zfree -b� ZSwap�freeztotal used freez&can't find 'Swap' in 'free' output: %sN)r �split� startswith� collections� namedtuple�int� ValueErrorr; )�out�lines�line�_�total�usedr? �ntr- r- r. � free_swap_ s rM c C s~ t d�} | jd�}xT|D ]L}|jd�rdd� |j� dd� D �\}}}}tjdd �}|||||| �S qW td dj|� ��d S ) zSParse 'free' cmd and return physical memory's total, used and free values. zfree -br> ZMemc S s g | ] }t |��qS r- )rD )r3 r0 r- r- r. r5 { s z free_physmem.<locals>.<listcomp>r7 � r? z total used free shared outputz%can't find 'Mem' in 'free' output: %sN)r r@ rA rB rC rE r; )rF rG rH rJ rK r? �sharedrL r- r- r. � free_physmemn s "rP c C sN t d�}x4|jd�D ]&}|j� }| |krt|jd�d �S qW td| ��d S )Nz vmstat -sr> � r z can't find %r in 'vmstat' output)r r@ �striprD rE )�statrF rH r- r- r. �vmstat� s rT c C s( t d�j� } ttt| j� d jd���S )Nzfree -Vr7 �.r8 )r rR �tuple�maprD r@ )rF r- r- r. �get_free_version_info� s rX z LINUX onlyc @ s� e Zd Zdd� Zejeo e� d k d�e� dd� ��Z eje d�e� d d � ��Z e� d d � �Z eje d�e� d d� ��Z eje d�e� dd� ��Ze� dd� �Ze� dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd S )!�TestSystemVirtualMemoryc C s& t d�d }tj� j}| j||� d S )Nz total memoryi )rT �psutil�virtual_memoryrJ �assertAlmostEqual)�self� vmstat_value� psutil_valuer- r- r. � test_total� s z"TestSystemVirtualMemory.test_total� � zold free versionc C s8 t � }|j}tj� j}| j||td|||jf d� d S )Nz %s %s %s)�delta�msg)rP rK rZ r[ r\ r �output)r] r? � free_valuer_ r- r- r. � test_used� s z!TestSystemVirtualMemory.test_usedzunreliable on TRAVISc C s* t d�d }tj� j}| j||td� d S )Nz free memoryi )rc )rT rZ r[ r? r\ r )r] r^ r_ r- r- r. � test_free� s z!TestSystemVirtualMemory.test_freec C s* t d�d }tj� j}| j||td� d S )Nz buffer memoryi )rc )rT rZ r[ �buffersr\ r )r] r^ r_ r- r- r. � test_buffers� s z$TestSystemVirtualMemory.test_buffersc C s* t d�d }tj� j}| j||td� d S )Nz active memoryi )rc )rT rZ r[ �activer\ r )r] r^ r_ r- r- r. � test_active� s z#TestSystemVirtualMemory.test_activec C s* t d�d }tj� j}| j||td� d S )Nzinactive memoryi )rc )rT rZ r[ �inactiver\ r )r] r^ r_ r- r- r. � test_inactive� s z%TestSystemVirtualMemory.test_inactivec C sJ t � }|j}|dkr tjd��tj� j}| j||td|||jf d� d S )Nr z%free does not support 'shared' columnz %s %s %s)rc rd ) rP rO r �SkipTestrZ r[ r\ r re )r] r? rf r_ r- r- r. � test_shared� s z#TestSystemVirtualMemory.test_sharedc C sh t d�}|jd�}d|d kr*tjd��n:t|d j� d �}tj� j}| j||t d|||f d� d S ) Nzfree -br> � availabler z(free does not support 'available' columnr7 z %s %s %s)rc rd r8 ) r r@ r ro rD rZ r[ rq r\ r )r] rF rG rf r_ r- r- r. �test_available� s z&TestSystemVirtualMemory.test_availablec sn � fdd�}t � trdnd}tj|d|d���6}tjdd��� }tjd� tj� }|j s\t �| j t |�d � |d }|j jd �s�t �| jd t|j�� | jd t|j�� | jdt|j�� | jdt|j�� | jdt|j�� | jdt|j�� | jdt|j�� | j |jd � | j |jd � | j |jd � | j |jd � | j |jd � | j |jd � W d Q R X W d Q R X d S )Nc s0 | dkr t jtjd�j� �S � | f|�|�S d S )Nz /proc/meminfoa� 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 )�io�BytesIO�textwrap�dedent�encode)�name�args�kwargs)� orig_openr- r. � open_mock� s zBTestSystemVirtualMemory.test_warnings_on_misses.<locals>.open_mockz builtins.openz__builtin__.openT)�create� side_effect)�record�alwaysr7 r zpsutil/_pslinux.pyz#memory stats couldn't be determined�cachedrO rk rm ri rq )�openr r �patch�warnings�catch_warnings� simplefilterrZ r[ �called�AssertionError� assertEqual�len�filename�endswith�assertIn�str�messager� rk rm rO ri rq )r] r| � patch_point�m�ws�ret�wr- )r{ r. �test_warnings_on_misses� s2 z/TestSystemVirtualMemory.test_warnings_on_missesc C s� ddl m} ddl m} i }|d��4}x,|D ]$}|j� }t|d �d ||d < q,W W d Q R X ||�}d|kr�|d }t|| �| d } | j| d � d S ) Nr )�calculate_avail_vmem)� open_binaryz /proc/meminfor7 i s MemAvailable:�d � )�psutil._pslinuxr� r� r@ rD �absZ assertLess) r] r� r� Zmems�frH �fields�a�bZ diff_percentr- r- r. �test_avail_old_percent s &z.TestSystemVirtualMemory.test_avail_old_percentc s� � fdd�}t � trdnd}tj|d|d��X}tjdd��}tj� }W d Q R X |jsXt �| j |j d � |d }| j d t |j�� W d Q R X d S ) Nc s0 | dkr t jtjd�j� �S � | f|�|�S d S )Nz /proc/meminfoa� 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 )rs rt ru rv rw )rx ry rz )r{ r- r. r| 0 s zKTestSystemVirtualMemory.test_avail_old_comes_from_kernel.<locals>.open_mockz builtins.openz__builtin__.openT)r} r~ )r i�Sd i r z,inactive memory stats couldn't be determinedl �" )r� r r r� r� r� rZ r[ r� r� r� rq r� r� r� )r] r| r� r� r� r� r� r- )r{ r. � test_avail_old_comes_from_kernel- s z8TestSystemVirtualMemory.test_avail_old_comes_from_kernelc s� � fdd�}t � trdnd}tj|d|d��X}tjdd��}tj� }W d Q R X |jsXt �| j |j d� |d }| j d t |j�� W d Q R X d S )Nc s0 | dkr t jtjd�j� �S � | f|�|�S d S )Nz /proc/meminfoa� Active: 9444728 kB Active(anon): 6145416 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB )rs rt ru rv rw )rx ry rz )r{ r- r. r| S s zHTestSystemVirtualMemory.test_avail_old_missing_fields.<locals>.open_mockz builtins.openz__builtin__.openT)r} r~ )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� rZ r[ r� r� r� rq r� r� r� )r] r| r� r� r� r� r� r- )r{ r. � test_avail_old_missing_fieldsO s z5TestSystemVirtualMemory.test_avail_old_missing_fieldsc s� � fdd�}t � trdnd}tj|d|d��X}tjdd��}tj� }W d Q R X |jsXt �| j |j d� |d }| j d t |j�� W d Q R X d S )Nc sF | dkr t jtjd�j� �S | dkr2ttjd��n� | f|�|�S d S )Nz /proc/meminfoaT Active: 9444728 kB Active(anon): 6145416 kB Active(file): 2950064 kB Buffers: 287952 kB Cached: 4818144 kB Inactive(file): 1578132 kB Inactive(anon): 574764 kB Inactive(file): 1567648 kB MemFree: 2057400 kB MemTotal: 16325648 kB Shmem: 577588 kB SReclaimable: 346648 kB z/proc/zoneinfozno such file or directory)rs rt ru rv rw �IOError�errno�ENOENT)rx ry rz )r{ r- r. r| q s zJTestSystemVirtualMemory.test_avail_old_missing_zoneinfo.<locals>.open_mockz builtins.openz__builtin__.openT)r} r~ )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� rZ r[ r� r� r� rq r� r� r� )r] r| r� r� r� r� r� r- )r{ r. � test_avail_old_missing_zoneinfon s z7TestSystemVirtualMemory.test_avail_old_missing_zoneinfoN)ra ra rb )�__name__� __module__� __qualname__r` r �skipIfr rX r rg r rh rj rl rn rp rr r� r� r� r� r� r- r- r- r. rY � s$ ," rY 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:)r� �read)r� �datar- r- r. �meminfo_has_swap_info� s z*TestSystemSwapMemory.meminfo_has_swap_infoc C s" t � j}tj� j}| j||td�S )N)rc )rM rJ rZ � swap_memoryr\ r )r] rf r_ r- r- r. r` � s z TestSystemSwapMemory.test_totalc C s" t � j}tj� j}| j||td�S )N)rc )rM rK rZ r� r\ r )r] rf r_ r- r- r. rg � s z TestSystemSwapMemory.test_usedc C s" t � j}tj� j}| j||td�S )N)rc )rM r? rZ r� r\ r )r] rf r_ r- r- r. rh � s z TestSystemSwapMemory.test_freec C s� t jddd���}tjdd��x}tjd� tj� }|js:t�| j t |�d� |d }|j j d�sbt�| j d t|j�� | j |jd� | j |jd� W d Q R X W d Q R X d S ) Nzpsutil._pslinux.openT)r} )r r� r7 r zpsutil/_pslinux.pyz9'sin' and 'sout' swap memory stats couldn't be determined)r r� r� r� r� rZ r� r� r� r� r� r� r� r� r� r� �sin�sout)r] r� r� r� r� r- r- r. �test_missing_sin_sout� s z*TestSystemSwapMemory.test_missing_sin_soutc s� � fdd�}t � trdnd}tj|d|d���}tjdd��x}tjd� tj� }|j sXt �| j t |�d � |d }|j jd �s�t �| jd t|j�� | j |jd � | j |jd � W d Q R X W d Q R X d S ) Nc s* | dkrt tjd��n� | f|�|�S d S )Nz /proc/vmstatzno such file or directory)r� r� r� )rx ry rz )r{ r- r. r| � s z=TestSystemSwapMemory.test_no_vmstat_mocked.<locals>.open_mockz builtins.openz__builtin__.openT)r} r~ )r r� r7 r zpsutil/_pslinux.pyzK'sin' and 'sout' swap memory stats couldn't be determined and were set to 0)r� r r r� r� r� r� rZ r� r� r� r� r� r� r� r� r� r� r� r� )r] r| r� r� r� r� r� r- )r{ r. �test_no_vmstat_mocked� s z*TestSystemSwapMemory.test_no_vmstat_mockedc C s� | j � stjd�S tjd��}tj� }W d Q R X |j s<t�dd l j }|j � \}}}}}}}||9 }||9 }| j |j |� | j |j|� d S )Nz!/proc/meminfo has no swap metricsz"psutil._pslinux.cext.linux_sysinfor )r� r �skipr r� rZ r� r� r� Zpsutil._psutil_linuxZ _psutil_linuxZ linux_sysinfor� rJ r? )r] r� �swapZcextrI rJ r? Zunit_multiplierr- r- r. � test_meminfo_against_sysinfo� s z1TestSystemSwapMemory.test_meminfo_against_sysinfoc sN � fdd�}t � trdnd}tj|d|d��}tj� |js@t�W d Q R X d S )Nc s& | dkrt jd�S � | f|�|�S d S )Nz /proc/meminfo� )rs rt )rx ry rz )r{ r- r. r| � s zKTestSystemSwapMemory.test_emulate_meminfo_has_no_metrics.<locals>.open_mockz builtins.openz__builtin__.openT)r} r~ )r� r r r� rZ r� r� r� )r] r| r� r� r- )r{ r. �#test_emulate_meminfo_has_no_metrics� s z8TestSystemSwapMemory.test_emulate_meminfo_has_no_metricsN) r� r� r� � staticmethodr� r` r rg rh r� r� r� r� r- r- r- r. r� � s r� c @ s& e Zd Zejed�dd� �Zejejj d� d�dd� �Z ejejj d� d �d d � �Z eje d � d �dd� �Z eje d� d�dd� �Zdd� Zdd� Zeje d�dd� �Zejed�eje d�d d � ��Zeje d�d d � �Zeje d�d d!� �Zejed�eje d�d"d#� ��Zd$S )%� TestSystemCPUzunknown failure on travisc C s� t j� j}tjdtj� d �d }ttt |j d���}|d krL| j d|� n | j d|� |d krn| j d |� n | j d |� |dkr�| j d |� n | j d |� d S )Nz \d+\.\d+\.\d+� r rU � � Zstealr Zguestra Z guest_nice)r� r� r� )r� r� r )ra r� r ) rZ � cpu_times�_fields�re�findall�os�unamerV rW rD r@ r� � assertNotIn)r] r� Z kernel_verZkernel_ver_infor- r- r. �test_cpu_times s z TestSystemCPU.test_cpu_timesz /sys/devices/system/cpu/onlinez-/sys/devices/system/cpu/online does not existc C sV t d��}|j� j� }W d Q R X dt|�krRt|jd�d �d }| jtj� |� d S )Nz /sys/devices/system/cpu/online�-r7 ) r� r� rR r� rD r@ r� rZ � cpu_count)r] r� �valuer- r- r. �*test_cpu_count_logical_w_sysdev_cpu_online s z8TestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_onlinez/sys/devices/system/cpuz&/sys/devices/system/cpu does not existc C s0 t jd�}tdd� |D ��}| jtj� |� d S )Nz/sys/devices/system/cpuc S s g | ]}t jd |�dk r|�qS )zcpu\d+$N)r� �search)r3 r0 r- r- r. r5 % s zITestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_num.<locals>.<listcomp>)r� �listdirr� r� rZ r� )r] Zls�countr- r- r. �'test_cpu_count_logical_w_sysdev_cpu_num! s z5TestSystemCPU.test_cpu_count_logical_w_sysdev_cpu_numZnprocznproc utility not availablec C s$ t td��}| jtjdd�|� d S )Nz nproc --allT)�logical)rD r r� rZ r� )r] �numr- r- r. � test_cpu_count_logical_w_nproc( s z,TestSystemCPU.test_cpu_count_logical_w_nprocZlscpuzlscpu utility not availablec C s8 t d�}tdd� |jd�D ��}| jtjdd�|� d S )Nzlscpu -pc S s g | ]}|j d �s|�qS )�#)rA )r3 r0 r- r- r. r5 0 s z@TestSystemCPU.test_cpu_count_logical_w_lscpu.<locals>.<listcomp>r> T)r� )r r� r@ r� rZ r� )r] rF r� r- r- r. � test_cpu_count_logical_w_lscpu- s z,TestSystemCPU.test_cpu_count_logical_w_lscpuc & s< dd l }|jj� }tjdtd���}| j|jj� |� |js@t�tjddd��:}| j |jj� � | j|j d� | j|j d d d� W d Q R X t d d ��}|j � }W d Q R X tj|�}tjd|dd ��}| j|jj� |� W d Q R X � fd d �}t � tr�dnd}tj||dd�� | j|jj� |� W d Q R X W d Q R X d S )Nr zpsutil._pslinux.os.sysconf)r~ zpsutil._pslinux.openT)r} r� z /proc/statz /proc/cpuinfo�rb)� return_valuer} c s( | j d�rtjd�S � | f|�|�S d S )Nz /proc/cpuinfor� )rA rs rt )rx ry rz )r{ r- r. r| P s z>TestSystemCPU.test_cpu_count_logical_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ r} )r� �_pslinuxZcpu_count_logicalr r� rE r� r� r� � assertIsNoneZ call_count� call_argsr� r� rs rt r ) r] rZ �originalr� r� Z cpuinfo_data� fake_filer| r� r- )r{ r. � test_cpu_count_logical_mocked3 s* z+TestSystemCPU.test_cpu_count_logical_mockedc C s8 t jddd�� }| jtjj� � |js*t�W d Q R X d S )Nzpsutil._pslinux.openT)r} )r r� r� rZ r� Zcpu_count_physicalr� r� )r] r� r- r- r. � test_cpu_count_physical_mocked[ s z,TestSystemCPU.test_cpu_count_physical_mockedz not supportedc C s, t jdg d�� | jtj� � W d Q R X d S )Nzpsutil._pslinux.glob.glob)r� )r r� r� rZ �cpu_freq)r] r- r- r. �test_cpu_freq_no_resultb s z%TestSystemCPU.test_cpu_freq_no_resultzfails on Travisc sT � �fdd�}g � t j �tjd|dd��" tj� s6t�| jt� �d� W d Q R X d S )Nc s. | j d�r� jd � g S � jd � �| �S d S )Nz&/sys/devices/system/cpu/cpufreq/policy)rA �append)�pattern)�flags� orig_globr- r. � glob_mockk s z>TestSystemCPU.test_cpu_freq_use_second_file.<locals>.glob_mockzpsutil._pslinux.glob.globT)r~ r} r� )�globr r� rZ r� r� r� r� )r] r� r- )r� r� r. � test_cpu_freq_use_second_fileg s z+TestSystemCPU.test_cpu_freq_use_second_filec s� � fdd�}t � trdnd}tj||d��T tjddgd��8 tj� }| j|jd � | j|jd � | j|j d � 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/scaling_cur_freqs 500000z/scaling_min_freqs 600000z/scaling_max_freqs 700000)r� rs rt )rx ry rz )r{ r- r. r| | s z;TestSystemCPU.test_cpu_freq_emulate_data.<locals>.open_mockz builtins.openz__builtin__.open)r~ z glob.globz'/sys/devices/system/cpu/cpufreq/policy0)r� g @@g ��@g �@) r� r r r� rZ r� r� �current�min�max)r] r| r� �freqr- )r{ r. �test_cpu_freq_emulate_dataz s z(TestSystemCPU.test_cpu_freq_emulate_datac s� � fdd�}t � trdnd}dddg}tj||d��R tjd |d ��8 tj� }| j|jd � | j|jd � | j|j d � 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/scaling_cur_freqs 100000z/scaling_min_freqs 200000z/scaling_max_freqs 300000)r� rs rt )rx ry rz )r{ r- r. r| � s z@TestSystemCPU.test_cpu_freq_emulate_multi_cpu.<locals>.open_mockz builtins.openz__builtin__.openz'/sys/devices/system/cpu/cpufreq/policy0z'/sys/devices/system/cpu/cpufreq/policy1z'/sys/devices/system/cpu/cpufreq/policy2)r~ z glob.glob)r� g Y@g i@g �r@) r� r r r� rZ r� r� r� r� r� )r] r| r� �policiesr� r- )r{ r. � test_cpu_freq_emulate_multi_cpu� s z-TestSystemCPU.test_cpu_freq_emulate_multi_cpuc s� � fdd�}t � trdnd}dddg}tj||d��6 tjd |d �� tj� }| j|jd � W d Q R X W d Q R X � fd d�}t � tr�dnd}tj||d��. tjd |d �� | jt tj� W d Q R X W d Q R X d S ) Nc s@ | j d�rttjd��n$| j d�r,tjd�S � | f|�|�S d S )Nz/scaling_cur_freqr2 z/cpuinfo_cur_freqs 200000)r� r� r� r� rs rt )rx ry rz )r{ r- r. r| � s zGTestSystemCPU.test_cpu_freq_no_scaling_cur_freq_file.<locals>.open_mockz builtins.openz__builtin__.openz'/sys/devices/system/cpu/cpufreq/policy0z'/sys/devices/system/cpu/cpufreq/policy1z'/sys/devices/system/cpu/cpufreq/policy2)r~ z glob.glob)r� �� c sD | j d�rttjd��n(| j d�r0ttjd��n� | f|�|�S d S )Nz/scaling_cur_freqr2 z/cpuinfo_cur_freq)r� r� r� r� )rx ry rz )r{ r- r. r| � s ) r� r r r� rZ r� r� r� � assertRaises�NotImplementedError)r] r| r� r� r� r- )r{ r. �&test_cpu_freq_no_scaling_cur_freq_file� s " z4TestSystemCPU.test_cpu_freq_no_scaling_cur_freq_fileN)r� r� r� r r� r r� r� �path�existsr� r� r r� r� r� r� r r� r� r� r� r� r- r- r- r. r� s ( r� c @ s4 e Zd Zejed�dd� �Zejed�dd� �ZdS )�TestSystemCPUStatszfails on Travisc C s& t d�}tj� j}| j||dd� d S )Nzcontext switchesi� )rc )rT rZ � cpu_statsZ ctx_switchesr\ )r] r^ r_ r- r- r. �test_ctx_switches� s z$TestSystemCPUStats.test_ctx_switchesc C s& t d�}tj� j}| j||dd� d S )N� interruptsi� )rc )rT rZ r� r� r\ )r] r^ r_ r- r- r. �test_interrupts� s z"TestSystemCPUStats.test_interruptsN)r� r� r� r r� r r� r� r- r- r- r. r� � s r� c @ s| e Zd Zdd� Zdd� Ze� dd� �Zeje d� d�eje d �d d � ��Z e j d ed �e j ddd�dd� ��Zdd� ZdS )�TestSystemNetworkc C sj xdt j� j� D ]T\}}xJ|D ]B}|jt jkr@| j|jt|�� q |jtj kr | j|jt |�� q W qW d S )N) rZ � net_if_addrs�itemsZfamilyZAF_LINKr� Zaddressr= r! r" r/ )r] rx ZaddrsZaddrr- r- r. �test_net_if_addrs_ips� s z'TestSystemNetwork.test_net_if_addrs_ipsc C sb x\t j� j� D ]L\}}ytd| �}W n tk r: Y qX | j|jttj d|�d �� qW d S )Nz ifconfig %sz(?i)MTU[: ](\d+)r ) rZ � net_if_statsr� r � RuntimeErrorr� ZmturD r� r� )r] rx �statsrF r- r- r. �test_net_if_stats� s z#TestSystemNetwork.test_net_if_statsc s � fdd�}t jddd�}x�|j� D ]�\� }y |� �}W n tk rN w$Y nX | j|j|d dd � | j|j|d dd � | j|j|d dd � | j|j|d dd � | j|j |d dd � | j|j |d dd � | j|j |d dd � | j|j |d dd � q$W d S )Nc 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+)�errinr7 �erroutzdropped[: ](\d+)�dropin�dropoutz#RX (?:packets \d+ +)?bytes[: ](\d+)� bytes_recvz#TX (?:packets \d+ +)?bytes[: ](\d+)� bytes_sent)r rD r� r� )Znicr� rF )rx r- r. �ifconfig s z8TestSystemNetwork.test_net_io_counters.<locals>.ifconfigTF)Zpernic�nowrapr i rN )rc r r� r r r� r r r i i ) rZ �net_io_countersr� r� r\ r r r� r r r r r )r] r Znior� Z ifconfig_retr- )rx r. �test_net_io_counters s. z&TestSystemNetwork.test_net_io_countersZipz'ip' utility not availablezskipped on Travisc C s� t d�j� }dd� tj� j� D �}d}xL|jd�D ]>}|j� }tjd|�r2|d7 }|jd�d j� }| j||� q2W | j t |�|d t j |�|f d � d S ) Nzip addrc S s g | ]}d |kr|�qS )�:r- )r3 r0 r- r- r. r5 2 s z7TestSystemNetwork.test_net_if_names.<locals>.<listcomp>r r> z^\d+:r7 r z %s --- %s)rd ) r rR rZ r� �keysr@ r� r� r� r� r� �pprintZpformat)r] rF Znics�foundrH rx r- r- r. �test_net_if_names. s z#TestSystemNetwork.test_net_if_namesz 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 Zinet6)�kind)r r ) r! ZAF_INET6Z SOCK_STREAM� addCleanup�closeZbind�errorrZ �net_connections)r] Z supports_ipv6Z inet_ntopr, r- r- r. �%test_net_connections_ipv6_unsupported= s z7TestSystemNetwork.test_net_connections_ipv6_unsupportedc sP � fdd�}t � trdnd}tj||d�� }tjdd� |jsBt�W d Q R X d S )Nc s, | dkrt jtjd��S � | f|�|�S d S )Nz/proc/net/unixaB 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 )rs �StringIOru rv )rx ry rz )r{ r- r. r| J s z@TestSystemNetwork.test_net_connections_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ Zunix)r )r� r r r� rZ r r� r� )r] r| r� r� r- )r{ r. �test_net_connections_mockedI s z-TestSystemNetwork.test_net_connections_mockedN)r� r� r� r� r� r r r r� r r r r r� rE r r r- r- r- r. r� � s * r� c @ sN e Zd Zejeed� d�e� dd� ��Zdd� Z dd� Z d d � Z d d � Z d S )�TestSystemDisks�statvfszos.statvfs() not availablec C s� dd� }x�t jdd�D ]|}t j|j�}||j�\}}}}| j|j|� t|j| �d krl| jd|j|f � t|j | �d kr| jd|j |f � 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"r> r � Znoner2 )r rR r@ �poprD )r� rF rG rH �devrJ rK r? r- r- r. �dfi s z:TestSystemDisks.test_disk_partitions_and_usage.<locals>.dfF)�allr� i zpsutil=%s, df=%si ( i � i ( i � ) rZ �disk_partitionsZ disk_usageZ mountpointr� rJ r� r? �failrK )r] r �partZusager rJ rK r? r- r- r. � test_disk_partitions_and_usaged s z.TestSystemDisks.test_disk_partitions_and_usagec C s� t dd��}|j� }W d Q R X d|krPx�tj� D ]}|jdkr0P q0W | jd� n|tjtd��}t j d|dd��X}t j d dgd ��<}tj� }|j s�t �|j s�t �|s�t �| j |d jd� W d Q R X W d Q R X d S )Nz/proc/filesystems�r�zfsz couldn't find any ZFS partitionz nodev zfs zpsutil._pslinux.openT)r� r} z$psutil._pslinux.cext.disk_partitions� /dev/sdb3�/�rw)r� r )r& r' r% r( )r� r� rZ r Zfstyper! rs r r r r� r� r� r� )r] r� r� r" r� �m1�m2r� r- r- r. �test_disk_partitions_mocked~ s$ z+TestSystemDisks.test_disk_partitions_mockedc s� � fdd�}t � trdnd}tj||d���}tjdd�}|jsBt�| j|j d� | j|j d � | j|j d t � | j|j d � | j|jd � | j|jd � | j|jdt � | j|jd� | j|jd� W d Q R X d S )Nc sB | dkrt jtjd��S | dkr.t jtd��S � | f|�|�S d S )Nz/proc/partitionszu major minor #blocks name 8 0 488386584 hda z/proc/diskstatsz+ 3 0 1 hda 2 3 4 5 6 7 8 9 10 11 12)rs r ru rv r )rx ry rz )r{ r- r. r| � s zJTestSystemDisks.test_disk_io_counters_kernel_2_4_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ F)r r7 r� ra r rN r� � � r� )r� r r r� rZ �disk_io_countersr� r� r� � read_count�read_merged_count� read_bytes� SECTOR_SIZE� read_time� write_count�write_merged_count� write_bytes� write_time� busy_time)r] r| r� r� r� r- )r{ r. �'test_disk_io_counters_kernel_2_4_mocked� s z7TestSystemDisks.test_disk_io_counters_kernel_2_4_mockedc s� � fdd�}t � trdnd}tj||d���}tjdd�}|jsBt�| j|j d� | j|j d � | j|j d t � | j|j d � | j|jd � | j|jd � | j|jdt � | j|jd� | j|jd� W d Q R X d S )Nc sB | dkrt jtjd��S | dkr.t jtd��S � | f|�|�S d S )Nz/proc/partitionszu major minor #blocks name 8 0 488386584 hda z/proc/diskstatsz' 3 0 hda 1 2 3 4 5 6 7 8 9 10 11)rs r ru rv r )rx ry rz )r{ r- r. r| � s zOTestSystemDisks.test_disk_io_counters_kernel_2_6_full_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ F)r r7 r� ra r rN r� r, r- r� )r� r r r� rZ r. r� r� r� r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 )r] r| r� r� r� r- )r{ r. �,test_disk_io_counters_kernel_2_6_full_mocked� s z<TestSystemDisks.test_disk_io_counters_kernel_2_6_full_mockedc s� � fdd�}t � trdnd}tj||d���}tjdd�}|jsBt�| j|j d� | j|j d t � | j|j d � | j|j d t � | j|jd � | j|jd � | j|jd � | j|jd � | j|jd � W d Q R X d S ) Nc sB | dkrt jtjd��S | dkr.t jtd��S � | f|�|�S d S )Nz/proc/partitionszu major minor #blocks name 8 0 488386584 hda z/proc/diskstatsz 3 1 hda 1 2 3 4)rs r ru rv r )rx ry rz )r{ r- r. r| � s zRTestSystemDisks.test_disk_io_counters_kernel_2_6_limited_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ F)r r7 r� ra r r )r� r r r� rZ r. r� r� r� r/ r1 r2 r4 r6 r0 r3 r5 r7 r8 )r] r| r� r� r� r- )r{ r. �/test_disk_io_counters_kernel_2_6_limited_mocked� s z?TestSystemDisks.test_disk_io_counters_kernel_2_6_limited_mockedN) r� r� r� r r� �hasattrr� r r# r+ r9 r: r; r- r- r- r. r a s r c @ sV e Zd Zdd� Zejd�dd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dS )�TestMiscc C s( t d�}tj� }| jt|�t|�� d S )Nz boot time)rT rZ � boot_timer� rD )r] r^ r_ r- r- r. �test_boot_time s zTestMisc.test_boot_timezpsutil.traceback.print_excc " s t j� }ttjj|d�d��$}|jd� |jd� |jd� W d Q R X �z�t� � fdd�}trbdnd }tj ||d ���x t t � |j s�t �| jtt j� | jtt jd d � | jtt j� | jtt jd d � | jtt j� | jtt jd d � |t _| jt j� d � | jtt j� �d � t jd d �}| jt|�d � t jd d �}| jttt|��d � ttjj|d�d��$}|jd� |jd� |jd� W d Q R X | jt j� d � | jtt jd d ��d � | jtt j� �d � | jtttt jd d ���d � W d Q R X W d tj|� t t � X | jt jd� d S )NrS 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)rA r� r� r� )rx ry rz )r{ r- r. r| s z4TestMisc.test_no_procfs_on_import.<locals>.open_mockz builtins.openz__builtin__.open)r~ T)�percpur zcpu 1 0 0 0 0 0 0 0 0 0 zcpu0 1 0 0 0 0 0 0 0 0 0 zcpu1 1 0 0 0 0 0 0 0 0 0 z/proc)�tempfile�mkdtempr� r� r� r; �writer r r� r rZ r� r� r� r� r� Z cpu_percentZcpu_times_percent� PROCFS_PATHr� �sumrW ZassertNotEqual�shutilZrmtree)r] �tbZ my_procfsr� r| r� Zper_cpu_percentZper_cpu_times_percentr- )r{ r. �test_no_procfs_on_import sN ( z!TestMisc.test_no_procfs_on_importc C s8 t jddd�� }| jttjj� |js*t�W d Q R X d S )Nzpsutil._pslinux.openT)r} ) r r� r� r� rZ r� r> r� r� )r] r� r- r- r. �test_boot_time_mockedD s z TestMisc.test_boot_time_mockedc C s� t jdd gd��&}| jtj� d jd � |js2t�W d Q R X t jddgd��&}| jtj� d jd � |jsnt�W d Q R X t jddgd��&}| jtj� d jd � |js�t�W d Q R X d S )Nzpsutil._pslinux.cext.users� giampaolo�pts/2�:0� �h�ATr� )r� r Z localhost�:0.0�foo)rJ rK rL rM Tr� )rJ rK rN rM Tr� )rJ rK rO rM Tr� )r r� r� rZ Zusers�hostr� r� )r] r� r- r- r. �test_users_mockedK s zTestMisc.test_users_mockedc C s� t j� }z�|t_| jttj� | jttj� | jttjdd� | jttj� | jttj � | jttj � | jttj � | jttj � | jttj � | jtjtj� W d dt_tj|� X d S )NT)r@ z/proc)rA rB rZ rD r� r� r[ r� r> r r r� r. r Z NoSuchProcess�Processr� �rmdir)r] Ztdirr- r- r. �test_procfs_path_ s zTestMisc.test_procfs_pathc sP � �fdd�}g � t �tr dnd}tj||d�� tj� � sBt�W d Q R X d S )Nc sJ t rt| t�r| j� } d| kr6� jd � ttjd��n�| f|�|�S d S )NZhw_sector_sizer2 )r � isinstancer �decoder� r� r� r� )rx ry rz )�flagr{ r- r. r| u s z1TestMisc.test_sector_size_mock.<locals>.open_mockz builtins.openz__builtin__.open)r~ )r� r r r� rZ r. r� )r] r| r� r- )rW r{ r. �test_sector_size_mockr s z TestMisc.test_sector_size_mockc C sn t � }|j� zPtj� }|j� d j}tj|� s:t|��tj|�}|j� | j |tj � � W d |j � X d S )Nr7 ) r �startrZ rR �threads�id� pid_existsr� Zas_dictr� Zpids�stop)r] �t�p�tidZptr- r- r. �test_issue_687� s zTestMisc.test_issue_687c sL � fdd�}t � trdnd}tj||d�� tjtj� �s>t�W d Q R X d S )Nc s2 | dt j� kr tjtd��S � | f|�|�S d S )Nz/proc/%s/statusr2 )r� �getpidrs r r )rx ry rz )r{ r- r. r| � s z:TestMisc.test_pid_exists_no_proc_status.<locals>.open_mockz builtins.openz__builtin__.open)r~ ) r� r r r� rZ r\ r� rb r� )r] r| r� r- )r{ r. � test_pid_exists_no_proc_status� s z'TestMisc.test_pid_exists_no_proc_statusN) r� r� r� r? r r� rH rI rQ rT rX ra rc r- r- r- r. r= � s >r= z no batteryc @ s� e Zd Zejed� d�dd� �Zejed� d�dd� �Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� ZdS ) �TestSensorsBatteryZacpizacpi utility not availablec C sD t d�}t|jd�d j� jdd��}tj� j}| j||dd� d S )Nzacpi -b�,r7 �%r2 )rc ) r rD r@ rR �replacerZ �sensors_battery�percentr\ )r] rF Z acpi_valuer_ r- r- r. � test_percent� s z TestSensorsBattery.test_percentc C sT t d�}d|j� kr tjd�S d|kr,d}nd|jd�d k}| jtj� j|� d S ) Nzacpi -b�unknownzacpi output not reliablezdischarging at zero rateTZChargingr> r ) r �lowerr r� r@ r� rZ rh � power_plugged)r] rF Zpluggedr- r- r. �test_power_plugged� s z%TestSensorsBattery.test_power_pluggedc sj � fdd�}t � trdnd}tj||d��6}| jtj� jd� | jtj� jtj � |j s\t �W d Q R X d S )Nc s2 | j d�s| j d�r tjd�S � | f|�|�S d S )Nz AC0/onlinez AC/online� 1)r� rs rt )rx ry rz )r{ r- r. r| � s z@TestSensorsBattery.test_emulate_power_plugged.<locals>.open_mockz builtins.openz__builtin__.open)r~ T) r� r r r� r� rZ rh rm ZsecsleftZPOWER_TIME_UNLIMITEDr� r� )r] r| r� r� r- )r{ r. �test_emulate_power_plugged� s z-TestSensorsBattery.test_emulate_power_pluggedc sV � fdd�}t � trdnd}tj||d��"}| jtj� jd� |jsHt �W d Q R X d S )Nc sN | j d�s| j d�r"ttjd��n(| j d�r:tjtd��S � | f|�|�S d S )Nz AC0/onlinez AC/onliner2 z/statusZcharging)r� r� r� r� rs r r )rx ry rz )r{ r- r. r| � s zBTestSensorsBattery.test_emulate_power_plugged_2.<locals>.open_mockz builtins.openz__builtin__.open)r~ T) r� r r r� r� rZ rh rm r� r� )r] r| r� r� r- )r{ r. � test_emulate_power_plugged_2� s z/TestSensorsBattery.test_emulate_power_plugged_2c sV � fdd�}t � trdnd}tj||d��"}| jtj� jd� |jsHt �W d Q R X d S )Nc s2 | j d�s| j d�r tjd�S � | f|�|�S d S )Nz AC0/onlinez AC/online� 0)r� rs rt )rx ry rz )r{ r- r. r| � s zDTestSensorsBattery.test_emulate_power_not_plugged.<locals>.open_mockz builtins.openz__builtin__.open)r~ F) r� r r r� r� rZ rh rm r� r� )r] r| r� r� r- )r{ r. � test_emulate_power_not_plugged� s z1TestSensorsBattery.test_emulate_power_not_pluggedc sV � fdd�}t � trdnd}tj||d��"}| jtj� jd� |jsHt �W d Q R X d S )Nc sN | j d�s| j d�r"ttjd��n(| j d�r:tjtd��S � | f|�|�S d S )Nz AC0/onlinez AC/onliner2 z/statusZ discharging)r� r� r� r� rs r r )rx ry rz )r{ r- r. r| � s zFTestSensorsBattery.test_emulate_power_not_plugged_2.<locals>.open_mockz builtins.openz__builtin__.open)r~ F) r� r r r� r� rZ rh rm r� r� )r] r| r� r� r- )r{ r. � test_emulate_power_not_plugged_2� s z3TestSensorsBattery.test_emulate_power_not_plugged_2c sT � fdd�}t � trdnd}tj||d�� }| jtj� j� |jsFt �W d Q R X d S )Nc sJ | j d�s| j d�r"ttjd��n$| j d�r6tjd�S � | f|�|�S d S )Nz"/sys/class/power_supply/AC0/onlinez!/sys/class/power_supply/AC/onliner2 z#/sys/class/power_supply/BAT0/statuss ???)rA r� r� r� rs rt )rx ry rz )r{ r- r. r| s zETestSensorsBattery.test_emulate_power_undetermined.<locals>.open_mockz builtins.openz__builtin__.open)r~ ) r� r r r� r� rZ rh rm r� r� )r] r| r� r� r- )r{ r. � test_emulate_power_undetermined s z2TestSensorsBattery.test_emulate_power_undeterminedc sR � fdd�}t � trdnd}tj||d�� }| jtj� � |jsDt�W d Q R X d S )Nc s6 | j d�s| j d�r"ttjd��n� | f|�|�S d S )Nz'/sys/class/power_supply/BAT0/energy_nowz'/sys/class/power_supply/BAT0/charge_nowr2 )rA r� r� r� )rx ry rz )r{ r- r. r| s z@TestSensorsBattery.test_emulate_no_base_files.<locals>.open_mockz builtins.openz__builtin__.open)r~ ) r� r r r� r� rZ rh r� r� )r] r| r� r� r- )r{ r. �test_emulate_no_base_files s z-TestSensorsBattery.test_emulate_no_base_filesc sV � fdd�}t � trdnd}tj||d��"}| jtj� jd� |jsHt �W d Q R X d S )Nc s( | j d�rtjd�S � | f|�|�S d S )Nz(/sys/class/power_supply/BAT0/energy_fullrr )rA rs rt )rx ry rz )r{ r- r. r| & s z@TestSensorsBattery.test_emulate_energy_full_0.<locals>.open_mockz builtins.openz__builtin__.open)r~ r ) r� r r r� r� rZ rh ri r� r� )r] r| r� r� r- )r{ r. �test_emulate_energy_full_0$ s z-TestSensorsBattery.test_emulate_energy_full_0c sV � fdd�}t � trdnd}tj||d��"}| jtj� jd� |jsHt �W d Q R X d S )Nc sR d}d}| j |�s | j |�r*ttjd��n$| j d�r>tjd�S � | f|�|�S d S )Nz(/sys/class/power_supply/BAT0/energy_fullz(/sys/class/power_supply/BAT0/charge_fullr2 z%/sys/class/power_supply/BAT0/capacitys 88)rA r� r� r� rs rt )rx ry rz Z energy_fullZ charge_full)r{ r- r. r| 5 s zHTestSensorsBattery.test_emulate_energy_full_not_avail.<locals>.open_mockz builtins.openz__builtin__.open)r~ �X ) r� r r r� r� rZ rh ri r� r� )r] r| r� r� r- )r{ r. �"test_emulate_energy_full_not_avail2 s z5TestSensorsBattery.test_emulate_energy_full_not_availc sD � fdd�}t jj� tjd|d��}tj� |js6t�W d Q R X d S )Nc s | j d�rdS � | �S d S )Nz"/sys/class/power_supply/AC0/onlineF)rA )rx )�orig_path_existsr- r. �path_exists_mockG s zGTestSensorsBattery.test_emulate_no_ac0_online.<locals>.path_exists_mockz psutil._pslinux.os.path.exists)r~ ) r� r� r� r r� rZ rh r� r� )r] r{ r� r- )rz r. �test_emulate_no_ac0_onlineE s z-TestSensorsBattery.test_emulate_no_ac0_onlinec sT � fdd�}t � trdnd}tj||d�� }| jtj� j� |jsFt �W d Q R X d S )Nc s@ | j d�s | j d�s | j d�r,ttjd��n� | f|�|�S d S )Nz!/sys/class/power_supply/AC/onlinez"/sys/class/power_supply/AC0/onlinez#/sys/class/power_supply/BAT0/statusr2 )rA r� r� r� )rx ry rz )r{ r- r. r| U s z;TestSensorsBattery.test_emulate_no_power.<locals>.open_mockz builtins.openz__builtin__.open)r~ ) r� r r r� r� rZ rh rm r� r� )r] r| r� r� r- )r{ r. �test_emulate_no_powerS s z(TestSensorsBattery.test_emulate_no_powerN)r� r� r� r r� r rj rn rp rq rs rt ru rv rw ry r| r} r- r- r- r. rd � s rd c @ s( e Zd Zejed�dd� �Zdd� ZdS )�TestSensorsTemperatureszunreliable on TRAVISc s� � fdd�}t � trdnd}tj||d��N}tjdd��6}| jtj� i � |j sTt �| j dt |d j �� W d Q R X W d Q R X d S ) Nc s, | j d�rttjd��n� | f|�|�S d S )NZ_inputr2 )r� �OSErrorr� ZEIO)rx ry rz )r{ r- r. r| i s zATestSensorsTemperatures.test_emulate_eio_error.<locals>.open_mockz builtins.openz__builtin__.open)r~ T)r Zignoringr )r� r r r� r� r� r� rZ �sensors_temperaturesr� r� r� r� r� )r] r| r� r� r� r- )r{ r. �test_emulate_eio_errorg s z.TestSensorsTemperatures.test_emulate_eio_errorc s� � fdd�}t � trdnd}tj||d��j tjddgd��N tj� d d }| j|jd � | j|jd � | j|j d � | j|j d� W d Q R X W d Q R X d S )Nc s� | j d�rtjtd��S | j d�r0tjtd��S | j d�rDtjd�S | j d�rXtjd�S | j d �rltjd �S � | f|�|�S d S ) Nz/namerx z /temp1_label�labelz /temp1_inputs 30000z /temp1_maxs 40000z /temp1_crits 50000)r� rs r r rt )rx ry rz )r{ r- r. r| x s z<TestSensorsTemperatures.test_emulate_data.<locals>.open_mockz builtins.openz__builtin__.open)r~ z glob.globz /sys/class/hwmon/hwmon0/temp1)r� rx r r� g >@g D@g I@) r� r r r� rZ r� r� r� r� ZhighZcritical)r] r| r� Ztempr- )r{ r. �test_emulate_dataw s z)TestSensorsTemperatures.test_emulate_dataN)r� r� r� r r� r r� r� r- r- r- r. r~ d s r~ c @ s e Zd Zdd� ZdS )�TestSensorsFansc s� � fdd�}t � trdnd}tj||d��N tjddgd��2 tj� d d }| j|jd � | j|jd � W d Q R X W d Q R X d S ) Nc s\ | j d�rtjtd��S | j d�r0tjtd��S | j d�rHtjtd��S � | f|�|�S d S )Nz/namerx z /fan1_labelr� z /fan1_inputZ2000)r� rs r r )rx ry rz )r{ r- r. r| � s z4TestSensorsFans.test_emulate_data.<locals>.open_mockz builtins.openz__builtin__.open)r~ z glob.globz /sys/class/hwmon/hwmon2/fan1)r� rx r r� i� ) r� r r r� rZ Z sensors_fansr� r� r� )r] r| r� Zfanr- )r{ r. r� � s z!TestSensorsFans.test_emulate_dataN)r� r� r� r� r- r- r- r. r� � s r� c @ s� e Zd Zdd� ZeZdd� Zejed�dd� �Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zeje d�dd � �Zd d � Zd d � Zd!d"� Zd#S )$� TestProcessc C s t t� d S )N)r r )r] r- r- r. �setUp� s zTestProcess.setUpc C s� t jdt �}t|�}| jt� tdd� dt � tj|j �}t j d� |j � }|j dd�}| j|jtdd � |D ��d d � | j|jtd d � |D ��d d � | j|jtd d � |D ��d d � d S )Nzk import time with open("%s", "w") as f: time.sleep(10) c S s t jd�S )NrU )r� r� r- r- r- r. �<lambda>� s z3TestProcess.test_memory_full_info.<locals>.<lambda>z'%s' not in retg�������?F)Zgroupedc S s g | ]}|j |j �qS r- )Z private_dirtyZ private_clean)r3 r0 r- r- r. r5 � s z5TestProcess.test_memory_full_info.<locals>.<listcomp>i )rc c S s g | ] }|j �qS r- )�pss)r3 r0 r- r- r. r5 � s c S s g | ] }|j �qS r- )r� )r3 r0 r- r- r. r5 � s )ru rv r r r r r rZ rR �pid�timeZsleepZmemory_full_info� memory_mapsr\ ZussrE r� r� )r] �srcZsprocr_ Zmem�mapsr- r- r. �test_memory_full_info� s z!TestProcess.test_memory_full_infozunreliable on PYPYc ; C sR dd� }t td�� | j|� jd� W d Q R X t td�� | j|� jd� W d Q R X t td�� | j|� jd� W d Q R X t td�� | j|� jd� W d Q R X t td�� | j|� jd� W d Q R X t td�� | j|� jd� W d Q R X t�rNtt� t td �� | j|� jd� W d Q R X tt� t td �� | j|� jd� W d Q R X d S ) Nc S s\ t j� } tj� d }x:x4| j� D ](}|jtjjt�kr:|S tj� |kr P q W qW td��d S )Nr� z timeout looking for test file) rZ rR r� � open_filesr� r� �abspathr r� )r_ Z giveup_at�filer- r- r. � get_test_file� s z7TestProcess.test_open_files_mode.<locals>.get_test_filer� r$ r� zr+zw+za+r0 zx+)r� r r� �moder r )r] r� r- r- r. �test_open_files_mode� s( z TestProcess.test_open_files_modec C s� t j� }|j� }tj� �� t|jdt|� � tjdt t j d�d��"}|j� }| sXt �|j sbt �W d Q R X tjdt t jd�d�� }| j|j� g � |j s�t �W d Q R X W d Q R X d S )Nzlen(ret) != %izpsutil._pslinux.os.readlinkr2 )r~ )rZ rR r� rA �NamedTemporaryFiler r� r r� r r� r� r� r� ZEINVALr� )r] r_ �filesr� r- r- r. �test_open_files_file_gone� s z%TestProcess.test_open_files_file_gonec C s� t j� }|j� }tj� �d t|jdt|� � tr6dnd}tj |t t j d�d��"}|j� }| sdt �|jsnt �W d Q R X W d Q R X d S )Nzlen(ret) != %iz builtins.openz__builtin__.openr2 )r~ )rZ rR r� rA r� r r� r r r� r� r� r� r� r� )r] r_ r� r� r� r- r- r. �test_open_files_fd_gone s z#TestProcess.test_open_files_fd_gonec C sB t jdi d��*}| jtjjtj� �j� � |j s4t �W d Q R X d S )Nz)psutil._pslinux._psposix.get_terminal_map)r� ) r r� r� rZ r� rR r� rb Zterminalr� r� )r] r� r- r- r. �test_terminal_mocked s z TestProcess.test_terminal_mockedc C s� t j� }tjtd��}tjd|dd��$}| j|j� ddg� |j sFt �W d Q R X tjtd��}tjd|dd��&}| j|j� dddg� |j s�t �W d Q R X d S ) Nzfoo bar zpsutil._pslinux.openT)r� r} rO �barz foo bar r2 ) rZ rR rs r r r r� r� �cmdliner� r� )r] r_ r� r� r- r- r. �test_cmdline_mocked% s z TestProcess.test_cmdline_mockedc C s� t j� }tjtd��}tjd|dd��$}| j|j� ddg� |j sFt �W d Q R X tjtd��}tjd|dd��&}| j|j� dddg� |j s�t �W d Q R X d S ) Nzfoo bar zpsutil._pslinux.openT)r� r} rO r� z foo bar r2 ) rZ rR rs r r r r� r� r� r� r� )r] r_ r� r� r- r- r. �test_cmdline_spaces_mocked3 s z&TestProcess.test_cmdline_spaces_mockedc C sF t jddd��. | jtj� j� d� | jtj� j� d� W d Q R X d S )Nzpsutil._pslinux.os.readlinkz/home/foo (deleted))r� z /home/foo)r r� r� rZ rR �exe�cwd)r] r- r- r. �!test_readlink_path_deleted_mockedA s z-TestProcess.test_readlink_path_deleted_mockedc s� � fdd�}t � trdnd}tj||d��(}tj� j� }|jsBt�| j |g � W d Q R X � fdd�}tj||d�� | j tj tj� j� W d Q R X d S )Nc s4 | j dtj� �r ttjd��n� | f|�|�S d S )Nz /proc/%s/taskr2 )rA r� rb r� r� r� )rx ry rz )r{ r- r. r| L s z2TestProcess.test_threads_mocked.<locals>.open_mockz builtins.openz__builtin__.open)r~ c s4 | j dtj� �r ttjd��n� | f|�|�S d S )Nz /proc/%s/taskr2 )rA r� rb r� r� ZEPERM)rx ry rz )r{ r- r. r| [ s ) r� r r r� rZ rR rZ r� r� r� r� � AccessDenied)r] r| r� r� r� r- )r{ r. �test_threads_mockedG s z TestProcess.test_threads_mockedc C s� t jdttjd�d���}t jdtjdd�d��`}tj� j� }|j sFt �|j sPt �| j |d� t jddd�� | j tj tj� j� W d Q R X W d Q R X W d Q R X d S ) Nzpsutil._pslinux.readlinkr2 )r~ zpsutil.Process.cmdliner z psutil._pslinux.os.path.lexistsF)r� )r r� r r� r� rZ r� rR r� r� r� r� r� � ZombieProcess)r] r) r* r� r- r- r. �test_exe_mockedd s zTestProcess.test_exe_mockedc s| � fdd�}t � trdnd}tj||d��H}tj� }| jt��}|j� W d Q R X | j |j j t j � |j snt�W d Q R X d S )Nc s4 | j dtj� �r ttjd��n� | f|�|�S d S )Nz/proc/%s/smapsr2 )rA r� rb r� r� r� )rx ry rz )r{ r- r. r| | s z.TestProcess.test_issue_1014.<locals>.open_mockz builtins.openz__builtin__.open)r~ )r� r r r� rZ rR r� r� r� r� � exceptionr� r� r� r� )r] r| r� r� r_ �errr- )r{ r. �test_issue_1014y s zTestProcess.test_issue_1014z not supportedc C s� t jdttjd�d��D}tj� }|j� | jtj ��}|j tj � W d Q R X |j sVt �W d Q R X | j|jj|j� | j|jj|j� � d S )Nz"psutil._pslinux.cext.linux_prlimitr2 )r~ )r r� r r� ZENOSYSrZ rR rx r� r� ZrlimitZ RLIMIT_NOFILEr� r� r� r� r� )r] r� r_ �excr- r- r. �test_rlimit_zombie� s z TestProcess.test_rlimit_zombiec C s� t jdttjd�d��@}tj� }|j� | jtj ��}|j � W d Q R X |j sRt �W d Q R X | j |jj|j� | j |jj|j� � d S )Nzpsutil._pslinux.os.readlinkr2 )r~ )r r� r r� r� rZ rR rx r� r� r� r� r� r� r� r� )r] r� r_ r� r- r- r. �test_cwd_zombie� s zTestProcess.test_cwd_zombiec s� ddl m} � fdd�}t� tr$dnd}tj||d��� tj� }| j|j � d� | j|j � tj � | j|j � d � | j|j � d | tj� � |j� }| j|jd | � | j|jd | � | j|jd | � | j|jd| � | j|j� d � W d Q R X d S )Nr )� CLOCK_TICKSc ' s� | j dtj� �rxddddddddddddddddd ddddd ddddddddddddddddd g'}tjd j|�j� �S � | f|�|�S d S ) Nz /proc/%s/stat�0z(cat)�Z�1�2�3�4�5�6rQ )rA r� rb rs rt r; rw )rx ry rz )r{ r- r. r| � sT z5TestProcess.test_stat_file_parsing.<locals>.open_mockz builtins.openz__builtin__.open)r~ �catr7 r� r� ra r rN )r� r� r� r r r� rZ rR r� rx �statusZ STATUS_ZOMBIE�ppidZ create_timer> r� �user�system� children_user�children_systemZcpu_num)r] r� r| r� r_ Zcpur- )r{ r. �test_stat_file_parsing� s" / z"TestProcess.test_stat_file_parsingc s� � fdd�}t � trdnd}tj||d��� tj� }| j|j� jd� | j|j� j d� | j|j � d� |j � }| j|j d � | j|j d � | j|jd � |j� }| j|j d � | j|j d � | j|jd� | j|jj� ttdd��� W d Q R X d S )Nc s: | j dtj� �r&tjtjd�j� �S � | f|�|�S d S )Nz/proc/%s/statusa+ Uid: 1000 1001 1002 1003 Gid: 1004 1005 1006 1007 Threads: 66 Cpus_allowed: f Cpus_allowed_list: 0-7 voluntary_ctxt_switches: 12 nonvoluntary_ctxt_switches: 13)rA r� rb rs rt ru rv rw )rx ry rz )r{ r- r. r| � s z7TestProcess.test_status_file_parsing.<locals>.open_mockz builtins.openz__builtin__.open)r~ rb � �B i� i� i� i� i� i� r r- )r� r r r� rZ rR r� �num_ctx_switches� voluntary� involuntary� num_threads�uids�realZ effectiveZsaved�gids�_proc�_get_eligible_cpus�list�range)r] r| r� r_ r� r� r- )r{ r. �test_status_file_parsing� s" z$TestProcess.test_status_file_parsingN)r� r� r� r� ZtearDownr� r r� r r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r- r- r- r. r� � s" $ Br� c @ sr e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z e � 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)rZ rR �proc)�clsr- r- r. � 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) rZ � _psplatform� open_textr� r� rR rA � partitionrD rE )r] Z linestartr� rH r� r- r- r. �read_status_file s z)TestProcessAgainstStatus.read_status_filec C s | j d�}| j| jj� |� d S )NzName:)r� r� r� rx )r] r� r- r- r. � test_name! s z"TestProcessAgainstStatus.test_namec C sH | j d�}||jd�d |jd�� }|jdd�}| j| jj� |� d S )NzState:�(r7 �)rQ r� )r� �find�rfindrg r� r� r� )r] r� r- r- r. � test_status% s z$TestProcessAgainstStatus.test_statusc C s | j d�}| j| jj� |� d S )NzPPid:)r� r� r� r� )r] r� r- r- r. � test_ppid+ s z"TestProcessAgainstStatus.test_ppidc C s | j d�}| j| jj� |� d S )NzThreads:)r� r� r� r� )r] r� r- r- r. �test_num_threads/ s z)TestProcessAgainstStatus.test_num_threadsc C s: | j d�}ttt|j� dd� ��}| j| jj� |� d S )NzUid:r7 r )r� rV rW rD r@ r� r� r� )r] r� r- r- r. � test_uids3 s z"TestProcessAgainstStatus.test_uidsc C s: | j d�}ttt|j� dd� ��}| j| jj� |� d S )NzGid:r7 r )r� rV rW rD r@ r� r� r� )r] r� r- r- r. � test_gids8 s z"TestProcessAgainstStatus.test_gidsc C s@ | j d�}| j| jj� j|� | j d�}| j| jj� j|� d S )Nzvoluntary_ctxt_switches:znonvoluntary_ctxt_switches:)r� r� r� r� r� r� )r] r� r- r- r. �test_num_ctx_switches= s z.TestProcessAgainstStatus.test_num_ctx_switchesc C sN | j d�}dt|�krJtt|jd��\}}| j| jj� tt ||d ��� d S )NzCpus_allowed_list:r� r7 ) r� r� rW rD r@ r� r� Z cpu_affinityr� r� )r] r� Zmin_Zmax_r- r- r. �test_cpu_affinityD 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� r� r� r� r� r� r� )r] r� r� r- r- r. � test_cpu_affinity_eligible_cpusK s z8TestProcessAgainstStatus.test_cpu_affinity_eligible_cpusN)r� r� r� �__doc__� classmethodr� r� r� r� r� r� r� r� r r� r� r� r- r- r- r. r� s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestUtilsc C s* t jjt��}| j|jd� W d Q R X d S )NZrt)rZ r� r� �__file__r� r� )r] r� r- r- r. �test_open_text] s zTestUtils.test_open_textc C s* t jjt��}| j|jd� W d Q R X d S )Nr� )rZ r� r� r� r� r� )r] r� r- r- r. �test_open_binarya s zTestUtils.test_open_binaryc C s< t jddd��$}| jtjjd�d� |js.t�W d Q R X d S )Nz os.readlinkz foo (deleted))r� r� rO )r r� r� rZ r� �readlinkr� r� )r] r� r- r- r. � test_readlinke s zTestUtils.test_readlinkc C s| t jjt�}t|d��}|jd� W d Q R X | jtjj tdd�d� | jtjj tdd�d� | jtjj td d d �d � d S ) NZwtzfoo F)ZbinaryrO Ts fooz??r� )Zfallback) r� r� r� r r� rC r� rZ r� r� )r] Zfnamer� r- r- r. �test_catj s zTestUtils.test_catN)r� r� r� r� r� r� r� r- r- r- r. r� Z s r� �__main__)Kr� Z __future__r rB r$ r� r� rs r� r r� rF r! r) rA ru r� r� rZ r Zpsutil._compatr r Z psutil.testsr r r r r r r r r r r r r r r r r r r r r� r� �dirnamer� ZHEREr( Z SIOCGIFCONFr9 r2 r/ r= rM rP rT rX r� ZTestCaserY r� r� r� r� r r= rd r~ r� r� r� r� r� r- r- r- r. �<module> s� k T t . 7 - Y R