Cs[m� � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l m
Z
d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m! Z! d d l m" Z" d d l m# Z# d d l m$ Z$ d d l m% Z% d d l m& Z& d d l m' Z' d d l m( Z( d d l m) Z) d d l m* Z* d d l m+ Z+ d d l m, Z, d d! l m- Z- Gd" d# � d# e- j. � Z/ e0 d$ k r�e( e1 � d S)%zTests for system APIS.� N)�AIX)�BSD)�FREEBSD)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�long)�APPVEYOR)�ASCII_FS)�check_net_address)�DEVNULL)�enum)�get_test_subprocess)� HAS_BATTERY)� HAS_CPU_FREQ)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�mock)�
reap_children)�retry_before_failing)�run_test_module_by_name)� safe_rmpath)�TESTFN)�TESTFN_UNICODE)�TRAVIS)�unittestc @ sn e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
e j e
d � d d � � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z! d9 d: � Z" d; d< � Z# d= d> � Z$ d? d@ � Z% dA dB � Z& dC dD � Z' dE dF � Z( dG dH � Z) e j e* dI � dJ dK � � Z+ e j e, p�e- p�e. dL � dM dN � � Z/ e j e, o.e0 j1 j2 dO � dP � e j e3 oOe4 j5 � dQ k dR � dS dT � � � Z6 dU dV � Z7 e j e3 p�e* o�e4 j8 � dW � dX dY � � Z9 dZ d[ � Z: e j e; d\ � d] d^ � � Z< d_ d` � Z= e j e> da � db dc � � Z? e j e> da � dd de � � Z@ e j eA da � e j eB df � dg dh � � � ZC e j eD da � di dj � � ZE dQ S)k�TestSystemAPIsz Tests for system-related APIs.c C s t t � d S)N)r r )�self� r# �/test_system.py�setUp? s zTestSystemAPIs.setUpc C s t � d S)N)r )r" r# r# r$ �tearDownB s zTestSystemAPIs.tearDownc C sF | j t j � d d � t j � D� � t � } | j | j d d � t j � D� � t j | j � } | j � | j � | j
| j d d � t j � D� � t j d d t j
t j � � ��! | j t t j � � g � Wd QRXt j d d t j t j � � ��/ | j t j � � t t j � � Wd QRXWd QRXd S)Nc S s g | ] } | j � q Sr# )�pid)�.0�xr# r# r$ �
<listcomp>F s z4TestSystemAPIs.test_process_iter.<locals>.<listcomp>c S s g | ] } | j � q Sr# )r' )r( r) r# r# r$ r* H s c S s g | ] } | j � q Sr# )r' )r( r) r# r# r$ r* L s zpsutil.Process� side_effect)�assertIn�os�getpid�psutil� process_iterr r' �Process�kill�waitZ assertNotInr �patchZ
NoSuchProcess� assertEqual�list� AccessDenied� assertRaises)r" �sproc�pr# r# r$ �test_process_iterE s ) &
& # z TestSystemAPIs.test_process_iterc C s� x? t j d d g � D]( } | j t | j j � � d g � q W| j t � � t t j d d g � � Wd QRXt j d d t j
d d � ��d } xK t j d d d g � D]1 } | j | j d � | j | j d d � q� W| j
s� t � Wd QRXt j d d t j
d d � ��v } t � } xT t j d d d g d | � D]4 } | j | j d | � | j | j d d � qFW| j
s�t � Wd QRXd S)
NZattrsr' Zfooz$psutil._psplatform.Process.cpu_timesr+ r � � cpu_timesZad_value)r/ r0 r5 r6 �info�keysr8 �
ValueErrorr r4 r7 � assertIsNone�assertGreaterEqual�called�AssertionError�object�assertIs)r" r: �m�flagr# r# r$ �test_prcess_iter_w_paramsV s$ & z(TestSystemAPIs.test_prcess_iter_w_paramsc s� � f d d � } g � t � } t � } t � } d d � | | | f D� } � j t t j | d d �� j t t j | d d �t j � } t j | d d d | �\ } } � j t j � | d � � j | g � � j t | � d
� � j � g � x$ | D] } � j
t | d � � q Wt d � � f d
d � � }
| j
� |
| | � \ } } � j | j d d � | D� � t r�� j | j � j t j � n � j | j � j d � � j � | j g � x$ | D] } � j
t | d � � q�Wt d � � f d d � � }
| j
� | j
� |
| | � \ } } � j t � � t | j | j | j g � � x$ | D] } � j t | d � � q�Wd S)Nc s � j | j � d S)N)�appendr' )r: )�pidsr# r$ �callbackk s z0TestSystemAPIs.test_wait_procs.<locals>.callbackc S s" g | ] } t j | j � � q Sr# )r/ r1 r' )r( r) r# r# r$ r* r s z2TestSystemAPIs.test_wait_procs.<locals>.<listcomp>�timeout� rL g{�G�z�?g �?� �
returncode� c sW t j | d d d | �\ } } � j t | � d � � j t | � d � | | f S)NrM g� ��Q��?rL rN � )r/ �
wait_procsr5 �len)�procsrL �gone�alive)r" r# r$ �test s
z,TestSystemAPIs.test_wait_procs.<locals>.testc S s g | ] } | j � q Sr# )r' )r( r) r# r# r$ r* � s c sW t j | d d d | �\ } } � j t | � d � � j t | � d � | | f S)NrM g� ��Q��?rL rO r )r/ rS r5 rT )rU rL rV rW )r" r# r$ rX � s
���)r r8 r@ r/ rS � TypeError�time�
assertLessr5 rT � assertFalse�hasattrr � terminater, r' r �poprP �signal�SIGTERM�set�
assertTrue) r" rL �sproc1�sproc2�sproc3rU �trV rW r: rX r# )rK r" r$ �test_wait_procsj sB !
.
z TestSystemAPIs.test_wait_procsc C sk t � } t � } t � } d d � | | | f D� } x | D] } | j � q> Wt j | � \ } } d S)Nc S s" g | ] } t j | j � � q Sr# )r/ r1 r' )r( r) r# r# r$ r* � s z=TestSystemAPIs.test_wait_procs_no_timeout.<locals>.<listcomp>)r r_ r/ rS )r" re rf rg rU r: rV rW r# r# r$ �test_wait_procs_no_timeout� s
z)TestSystemAPIs.test_wait_procs_no_timeoutc C sF t j � } | j | t � | j | d � | j | t j � � d S)Nr )r/ Z boot_time�assertIsInstance�float�
assertGreaterr\ r[ )r" Zbtr# r# r$ �test_boot_time� s z TestSystemAPIs.test_boot_timez
POSIX onlyc C s/ d d l } | j t j d � | j � � d S)Nr � SC_PAGE_SIZE)�resourcer5 r- �sysconfZ getpagesize)r" rp r# r# r$ �
test_PAGESIZE� s z TestSystemAPIs.test_PAGESIZEc C sJ t j � } | j d k s' t | � � | j d k sB t | � � d | j k o\ d k n sm t | � � | j d k s� t | � � | j d k s� t | � � x� | j D]� } t | | � } | d k r� | j
| t t f � | d k r� | d k s| j
d | | f � | | j k r� | j
d | | j | | f � q� Wd S)Nr �d �percent�totalz %r < 0 (%s)z %r > total (total=%s, %s=%s))r/ Zvirtual_memoryru rD Z availablert �used�free�_fields�getattrrk �intr �fail)r" �mem�name�valuer# r# r$ �test_virtual_memory� s + z"TestSystemAPIs.test_virtual_memoryc C s t j � } | j | j d � | j d k s: t | � � | j d k sU t | � � | j d k r� | j d k s� t | � � n | j d k s� t | � � d | j k o� d k n s� t | � � | j d k s� t | � � | j
d k s� t | � � d S)
Nru rv rw rt �sin�soutr rs )ru rv rw rt r� r� ) r/ Z swap_memoryr5 rx ru rD rv rw rt r� r� )r" r| r# r# r$ �test_swap_memory� s
+z TestSystemAPIs.test_swap_memoryc C s� t � } | j t j | j � � t j | j � } | j � | j � | j t j | j � � | j t j d � � | j t j d � d t j
� k � d S)NrN r rY ) r rd r/ �
pid_existsr' r1 r2 r3 r] r5 rK )r" r9 r: r# r# r$ �test_pid_exists� s
z TestSystemAPIs.test_pid_existsc C s� t � t j � } xh | D]` } y t j | � s8 t � Wq t k
ry t j d � | t j � k ru | j | � Yq Xq Wt t | � d t | � d � } x* | D]" } | j
t j | � d | �q� Wd S)Ng�������?i� ip �msg) r r/ rK r� rD r[ �sleepr{ �range�maxr] )r" rK r' r# r# r$ �test_pid_exists_2� s
#
z TestSystemAPIs.test_pid_exists_2c C sg d d � t j � D� } t j � } | j | j � | j � � | j t | � t t | � � � d S)Nc S s g | ] } | j � q Sr# )r' )r( r) r# r# r$ r* � s z,TestSystemAPIs.test_pids.<locals>.<listcomp>)r/ r0 rK r5 �sortrT rc )r" ZplistZpidlistr# r# r$ � test_pids� s zTestSystemAPIs.test_pidsc C s1 t j } t t _ z t j � Wd | t _ Xd S)N)�sys�stdoutr r/ rX )r" r� r# r# r$ � test_test� s
zTestSystemAPIs.test_testc
C s� t j � } | j | t t j d d � � � | j | d � t j j d � r� t d � � } | j
� } Wd QRXd | k r� t j d � � t j d d � } t
r� t j � d d � d k r� | j | � n | j | d � | j | | � d S) N�percpuTrN z
/proc/cpuinfoz physical idz#cpuinfo doesn't include physical id�logicalFrR � )r� rN )r/ � cpu_countr5 rT r= rB r- �path�exists�open�readr ZSkipTestr r� ZgetwindowsversionrA )r" r� �fdZ cpuinfo_dataZphysicalr# r# r$ �test_cpu_count s " "z TestSystemAPIs.test_cpu_countc C s� x� d D]� } t j d d | ��) } | j t j � � | j sG t � Wd QRXt j d d | ��/ } | j t j d d � � | j s� t � Wd QRXq Wd S)
NrN r z$psutil._psplatform.cpu_count_logical� return_valuez%psutil._psplatform.cpu_count_physicalr� FrY )rY r N)r r4 rA r/ r� rC rD )r" �valrG r# r# r$ �test_cpu_count_none s
z"TestSystemAPIs.test_cpu_count_nonec C s{ d } t j � } t | � x8 | D]0 } | j | t � | j | d � | | 7} q# W| j | t | � � t | � d S)Nr g )r/ r= �sumrk rl rB r5 �str)r" ru �times�cp_timer# r# r$ �test_cpu_times! s
z TestSystemAPIs.test_cpu_timesc C s\ t t j � � } t j d � t t j � � } | | } | d k sX | j d | � d S)Ng�������?g�������?z
difference %s)r� r/ r= r[ r� r{ )r" �t1�t2�
differencer# r# r$ � test_cpu_times_time_increases> s
z,TestSystemAPIs.test_cpu_times_time_increasesc C s� x� t j d d � D]q } d } t | � x8 | D]0 } | j | t � | j | d � | | 7} q0 W| j | t | � � t | � q W| j t t j d d � d � t t j d d � � � d S)Nr� Tr g F) r/ r= r� rk rl rB r5 r� rT )r" r� ru r� r# r# r$ �test_per_cpu_timesG s
z!TestSystemAPIs.test_per_cpu_timesc C s� t j d d � } t j � d } x t j � | k r% Pq% Wt j d d � } xP t | | � D]? \ } } t | � t | � } } | | } | d k r^ d Sq^ W| j � d S)Nr� Tg�������?g�������?)r/ r= r[ �zipr� r{ )r" Ztot1Zstop_atZtot2r� r� r� r# r# r$ �test_per_cpu_times_2h s
z#TestSystemAPIs.test_per_cpu_times_2c C s� t j � } t j d d � } | j d d � t | � D� � } x9 | j D]. } | j t | | � t | | � d d �qJ Wd S)Nr� Tc S s g | ] } t | � � q Sr# )r� )r( �numr# r# r$ r* } s z<TestSystemAPIs.test_cpu_times_comparison.<locals>.<listcomp>�deltarN )r/ r= �_maker� rx �assertAlmostEqualry )r" �baseZper_cpuZ
summed_valuesZfieldr# r# r$ �test_cpu_times_comparisonx s "z(TestSystemAPIs.test_cpu_times_comparisonc C s� yN | j | t � | j | d � | j | d � | j | d t j � � WnP t k
r� } z0 t d | t j | � t j | � f � � WYd d } ~ Xn Xd S)Ng g Y@z
%s
last=%s
new=%sg �)
rk rl rB Z assertIsNot�assertLessEqualr/ r� rD �pprintZpformat)r" rt Zlast_retZnew_ret�errr# r# r$ �_test_cpu_percent� s z TestSystemAPIs._test_cpu_percentc
C s t j d d � } x? t d � D]1 } t j d d � } | j | | | � | } q W| j t � � t j d d � Wd QRXd S)N�intervalg����MbP?rs rN rY )r/ � cpu_percentr� r� r8 r@ )r" �lastr) �newr# r# r$ �test_cpu_percent� s
z TestSystemAPIs.test_cpu_percentc C s� t j d d d d � } | j t | � t j � � xV t d � D]H } t j d d d d � } x! | D] } | j | | | � qf W| } qA W| j t � � t j d d d d � Wd QRXd S)Nr� g����MbP?r� Trs rN rY ) r/ r� r5 rT r� r� r� r8 r@ )r" r� r) r� rt r# r# r$ �test_per_cpu_percent� s
z#TestSystemAPIs.test_per_cpu_percentc C s� t j d d � } xi t d � D][ } t j d d � } x! | D] } | j | | | � q> W| j t | � | | � | } q Wd S)Nr� g����MbP?rs )r/ �cpu_times_percentr� r� r� )r" r� r) r� rt r# r# r$ �test_cpu_times_percent� s
z%TestSystemAPIs.test_cpu_times_percentc C s� t j d d d d � } | j t | � t j � � x� t d � D]r } t j d d d d � } xK | D]C } x! | D] } | j | | | � qs W| j t | � | | � qf W| } qA Wd S)Nr� g����MbP?r� Trs )r/ r� r5 rT r� r� r� r� )r" r� r) r� �cpurt r# r# r$ �test_per_cpu_times_percent� s
z)TestSystemAPIs.test_per_cpu_times_percentc
C s� t j d d � d d � t j d d � D� } t j d d | ��F x> t j d d � D]* } x! | D] } | j | d d � qe WqX WWd QRXd S)Nr� Tc S s; g | ]1 } | j d d � t t | j � � D� � � q S)c S s g | ] } d � q S)r r# )r( r) r# r# r$ r* � s zQTestSystemAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>.<listcomp>)r� r� rT rx )r( r) r# r# r$ r* � s zFTestSystemAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>zpsutil.cpu_timesr� )r/ r� r= r r4 r� )r" Z
zero_timesr� rt r# r# r$ �#test_per_cpu_times_percent_negative� s
z2TestSystemAPIs.test_per_cpu_times_percent_negativec
C s� t j t j � � } | j | j d � | j d k sC t | � � | j d k s^ t | � � | j d k sy t | � � | j | j k s� t | � � | j | j k s� t | � � d | j
k o� d k n s� t | j
� � t t d � r[t j t j � � } d
} | j | j | j � | j
| j | j d
| �| j
| j | j d
| �t j � } | j t � � } t j | � Wd QRX| j | j j t j � d S)Nru rv rw rt r rs �
disk_usage� i r� )ru rv rw rt i i P )r/ r� r- �getcwdr5 rx ru rD rv rw rt r^ �shutilr� �tempfile�mktempr8 �OSError� exception�errno�ENOENT)r" �usageZ shutil_usageZ toleranceZfname�excr# r# r$ �test_disk_usage� s( . z TestSystemAPIs.test_disk_usagec C s. t r* | j t � � t j t � Wd QRXd S)N)r r8 �UnicodeEncodeErrorr/ r� r )r" r# r# r$ �test_disk_usage_unicode� s z&TestSystemAPIs.test_disk_usage_unicodec C s t j d � d S)N� .)r/ r� )r" r# r# r$ �test_disk_usage_bytes� s z$TestSystemAPIs.test_disk_usage_bytesc C s� t j d d � } | j | d | �x | D]} | j | j t � | j | j t � | j | j t � | j | j t � t r� d | j k r� q, t
s� t j j
| j � s� t | � � n | j t s� t r� t j j
| j � st | � � n! t j j | j � st | � � | j s, t | � � q, Wt j d d � } | j | d | �xt j d d � D]� } t sAy t j | j � Wnd t k
r�} zD t r�t r�| j t j k r�wi| j t j t j f k r�� WYd d } ~ XnR Xt s�t r t j j
| j � sAt | � � n! t j j | j � sAt | � � | j | j t � | j | j t � qiWd d � } | t � } d d � t j d d � D� } | j | | � t j | � d S)
N�allFr� ZcdromTc S sG t j j | � } x( t j j | � s< t j j | � } q W| j � S)N)r- r� �abspath�ismount�dirname�lower)r� r# r# r$ �find_mount_point s z=TestSystemAPIs.test_disk_partitions.<locals>.find_mount_pointc S s g | ] } | j j � � q Sr# )�
mountpointr� )r( r) r# r# r$ r* s z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>) r/ Zdisk_partitionsrd rk Zdevicer� r� ZfstypeZoptsr r r- r� r� rD r
r �isdir�statr� r r� ZEIO�EPERMZEACCES�__file__r, r� )r" �lsZdiskr� r� ZmountZmountsr# r# r$ �test_disk_partitions� sL
$ $! $! z#TestSystemAPIs.test_disk_partitionsc s� � f d d � } t j d d � } | | � t j d d � } � j | g � x9 | D]1 } � j | � � j | t � | | | � qW Wd S)Nc s� � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � | j d k s� t | � � | j d k s� t | � � | j d k s t | � � | j d k s$t | � � | j d k s?t | � � | j d k sZt | � � | j d k sut | � � | j d k s�t | � � d S) Nr rN rR rO � r� r� � )
r5 Z
bytes_sentZ
bytes_recvZ packets_sentZ packets_recvZerrinZerroutZdropinZdropoutrD )�nt)r" r# r$ � check_ntuple! s z9TestSystemAPIs.test_net_io_counters.<locals>.check_ntuple�pernicFT)r/ �net_io_counters�assertNotEqualrd rk r� )r" r� �ret�keyr# )r" r$ �test_net_io_counters s
z#TestSystemAPIs.test_net_io_countersc C sg t j d d i ��K } | j t j d d � � | j t j d d � i � | j s\ t � Wd QRXd S)Nz"psutil._psplatform.net_io_countersr� r� FT)r r4 rA r/ r� r5 rC rD )r" rG r# r# r$ � test_net_io_counters_no_nics< s
z+TestSystemAPIs.test_net_io_counters_no_nicsc
C s5 t j � } | s t | � � t j � } t t j t j t j g � } xt| j � D]f\ } } | j
| t � | j t
t | � � t
| � � x%| D] } | j
| j t � | j
| j t � | j
| j t t d � f � | j
| j t t d � f � | j | j | � t j d k r<| j
| j t j � | | j r+| j t j k r�t j | j � } t j | � � | j | j d f � Wd QRXn� | j t j k r+t j | j d t j t j d t j � d } | \ } }
} } }
t j | |
| � } t j | � � | j |
� Wd QRXxT | j | j | j | j f D]4 } | d k rJ| j t j k rJt! | | j � qJW| j r�| j" | j � q� | j r� | j" | j � q� WqU Wt# s�t$ s�t% r�t& t d � r1| j t j t j � n8 t' r| j t j t j( � n t) r1| j t j d � d S)NrO r� r �AF_LINKrN )rO r� rY )*r/ � net_if_addrsrD � net_if_statsrc �socket�AF_INET�AF_INET6r� �itemsrk r� r5 rT �familyrz �addressZnetmask�typeZ broadcastr, r� � version_infor �IntEnum�isup�
contextlib�closing�bind� getaddrinfo� SOCK_STREAM�
AI_PASSIVEZptpr rA r r r
r^ r � AF_PACKETr )r" �nicsZ nic_stats�familiesZnicZaddrs�addr�sr> �af�socktype�proto� canonname�saZipr# r# r$ �test_net_if_addrsE sX "
z TestSystemAPIs.test_net_if_addrsc
C s� t r' d t j d d d d f g } n d g } t j d d | ��Y } t j � d d } | j sk t � t r� | j | j d � n | j | j d � Wd QRXd S) N�em1z06:3d:29rN �06-3d-29z psutil._psplatform.net_if_addrsr� r z06:3d:29:00:00:00z06-3d-29-00-00-00rY )r rY r NNN)
r r/ r� r r4 r� rC rD r5 r� )r" r� rG r� r# r# r$ � test_net_if_addrs_mac_null_bytes� s ! z/TestSystemAPIs.test_net_if_addrs_mac_null_byteszunreliable on TRAVISc C s� t j � } | s t | � � t j t j t j f } x� | j � D]~ \ } } | j | t � | \ } } } } | j | t � | j
| | � | j
| | � | j | d � | j | d � qC Wd S)Nr ) r/ r� rD ZNIC_DUPLEX_FULLZNIC_DUPLEX_HALFZNIC_DUPLEX_UNKNOWNr� rk r� �boolr, rB ) r" r� Z all_duplexesr} Zstatsr� �duplexZspeedZmtur# r# r$ �test_net_if_stats� s z TestSystemAPIs.test_net_if_statsz LINUX or BSD or MACOS specificc
C sZ t j d d t t j d � ��2 } t j � } | j | i � | j sO t � Wd QRXd S)Nz psutil._psutil_posix.net_if_mtur+ r< )
r r4 r� r� ZENODEVr/ r� r5 rC rD )r" rG r� r# r# r$ �test_net_if_stats_enodev� s
z'TestSystemAPIs.test_net_if_stats_enodevz/proc/diskstatsz3/proc/diskstats not available on this linux versionNzunreliable on APPVEYORc s� � f d d � } t j d d � } | d k s< t d � � | | � t j d d � } � j t | � t t | � � � x. | D]& } | s� t | � � | | | � q� Wd S)Nc s: � j | d | j � � j | d | j � � j | d | j � � j | d | j � t pe t s� j | d | j � � j | d | j � t r� � j | d | j
� � j | d | j � � j | d | j � n t
r� j | d | j � x2 | j D]' } t | | � d k s t | � � q Wd S)
Nr rN rR rO r� r� r� r� � )r5 Z
read_countZ write_countZ
read_bytesZ write_bytesr r Z read_timeZ
write_timer Zread_merged_countZwrite_merged_countZ busy_timer rx ry rD )r� r} )r" r# r$ r� � s z:TestSystemAPIs.test_disk_io_counters.<locals>.check_ntuple�perdiskFzno disks on this system?T)r/ �disk_io_countersrD r5 rT rc )r" r� r� r� r# )r" r$ �test_disk_io_counters� s
"
z$TestSystemAPIs.test_disk_io_countersc C sg t j d d i ��K } | j t j d d � � | j t j d d � i � | j s\ t � Wd QRXd S)Nz#psutil._psplatform.disk_io_countersr� r FT)r r4 rA r/ r
r5 rC rD )r" rG r# r# r$ � test_disk_io_counters_no_disks� s
z-TestSystemAPIs.test_disk_io_counters_no_disksz unreliable on APPVEYOR or TRAVISc C s t j � } | j | g � x� | D]� } | j s> t | � � | j | j t � | j | j t t d � f � | j d k r� | j | j t t d � f � | j | j | j
d k s� t | � � t j j | j
� t
s� t r� | j | j � q# t j | j � q# Wd S)Ng )r/ �usersr� r} rD rk r� Zterminalr� �host�started�datetimeZ
fromtimestampr r rA r' r1 )r" r �userr# r# r$ �
test_users� s
zTestSystemAPIs.test_usersc C sy t j � } | j | j d � xS | j D]H } t | | � } | j | d � t r) | d k r) | j | d � q) Wd S)N� ctx_switches�
interrupts�soft_interrupts�syscallsr )r r r r )r r )r/ Z cpu_statsr5 rx ry rB r rm )r" Zinfosr} r~ r# r# r$ �test_cpu_stats� s z TestSystemAPIs.test_cpu_statsz not suportedc s� � f d d � } t j d d � } t r5 | r5 d S| sG t | � � | t j d d � g � t r� � j t | � t j � � d S)Nc s� x� | D]{ } � j | j d � � j | j | j � xI | j D]> } t | | � } � j | t t t f � � j
| d � q@ Wq Wd S)N�current�minr� r )r r r� ) r5 rx r� r r� ry rk rz r rl rB )r� r� r} r~ )r" r# r$ �check_ls� s
z.TestSystemAPIs.test_cpu_freq.<locals>.check_lsr� TF)r/ Zcpu_freqr rD r r5 rT r� )r" r r� r# )r" r$ �
test_cpu_freq� s
z TestSystemAPIs.test_cpu_freqc C s# d d d d d d d d d g } x- | D]% } | j t t | � t d
| �q( Wt j d k r�t j so t � t j s t � | j d � d t
j j � k r� t j
s� t � | j d � q�d
t
j j � k rFt j s� t � | j t j t j t j g j d � d � | j d � | j d � | j d � | j d � q�d t
j j � k spd t
j j � k r�t j st � | j d � q�d t
j j � k r�t j s�t � | j d � n, t j s�t � t j s�t � | j d � x- | D]% } | j t t | � d d
| �q�Wd S)Nr r r r r r r r r
r� �posix�linuxZbsdTrN �sunos�solaris�darwinF)rk ry r/ r r- r} r rD r �remover� �platformr� r r r5 r r r �countr
r rF )r" �namesr} r# r# r$ �test_os_constants s>
#
z TestSystemAPIs.test_os_constantsz
not supportedc C s� t j � } x� | j � D]� \ } } | j | t � x� | D] } | j | j t � | j d k rw | j | j d � | j d k r� | j | j d � | j d k r< | j | j d � q< Wq Wd S)Nr )
r/ �sensors_temperaturesr� rk r� �labelr rB �high�critical)r" �tempsr} �entries�entryr# r# r$ �test_sensors_temperatures1 s
z(TestSystemAPIs.test_sensors_temperaturesc
C s� d d g i } t j d d | ��i } t j d d � d d
} | j sP t � | j | j d � | j | j d � | j | j d
� Wd QRXd S)NZcoretempr* � I@� N@� �Q@z'psutil._psplatform.sensors_temperaturesr� Z
fahrenheitTr g �^@g �a@g �c@)r* r1 r2 r3 )
r r4 r/ r) rC rD r5 r r+ r, )r" �drG r- r# r# r$ �#test_sensors_temperatures_fahreneit? s z2TestSystemAPIs.test_sensors_temperatures_fahreneitz
no batteryc C s� t j � } | j | j d � | j | j d � | j t j t j f k rc | j | j d � n" | j t j k r� | j | j � | j
| j t � d S)Nr rs ) r/ Zsensors_batteryrB rt r� ZsecsleftZPOWER_TIME_UNKNOWNZPOWER_TIME_UNLIMITEDrd Z
power_pluggedrk r )r" r� r# r# r$ �test_sensors_batteryK s z#TestSystemAPIs.test_sensors_batteryc C s� t j � } xz | j � D]l \ } } | j | t � xM | D]E } | j | j t � | j | j t t f � | j | j d � q< Wq Wd S)Nr )
r/ Z sensors_fansr� rk r� r* r rz r rB )r" Zfansr} r. r/ r# r# r$ �test_sensors_fansY s
z TestSystemAPIs.test_sensors_fans)F�__name__�
__module__� __qualname__�__doc__r% r& r; rI ri rj rn r ZskipIfr rr 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 r r r r
r- r� r� r
r/ r
r r r r r r r r( r r0 r5 r r r6 r r7 r# r# r# r$ r! < sl 7
!
> <
% "
r! �__main__)2r; r� r r� r- r� r� ra r� r� r� r[ r/ r r r r r r r r r
r Zpsutil._compatr Z psutil.testsr
r r r r r r r r r r r r r r r r r r r ZTestCaser! r8 r� r# r# r# r$ �<module> sb � � � +