OwlCyberSecurity - MANAGER
Edit File: test_system.cpython-36.pyc
3 ��JZ/� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl m Z ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd lmZ ddlmZ ddlmZ 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* dd lm+Z+ dd lm,Z, dd!lm-Z- G d"d#� d#e-j.�Z/e0d$k�r�e(e1� dS )%zTests for system APIS.� N)�AIX)�BSD)�FREEBSD)�LINUX)�NETBSD)�OPENBSD)�OSX)�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 @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z e j e d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGdH� Z)e j e*dI�dJdK� �Z+e j e,�o\e-j.j/dL� dM�e j e0�ote1j2� dNkdO�dPdQ� ��Z3dRdS� Z4e j e0�p�e*�o�e1j5� dT�dUdV� �Z6dWdX� Z7e j e8 dY�dZd[� �Z9d\d]� Z:e j e; d^�d_d`� �Z<e j e; d^�dadb� �Z=e j e> d^�e j e? dc�ddde� ��Z@e j eA d^�dfdg� �ZBdNS )h�TestSystemAPIsz Tests for system-related APIs.c C s t t� d S )N)r r )�self� r# �#/usr/lib64/python3.6/test_system.py�setUp> s zTestSystemAPIs.setUpc C s t � d S )N)r )r" r# r# r$ �tearDownA s zTestSystemAPIs.tearDownc C s� | j tj� dd� tj� D �� t� }| j |jdd� tj� D �� tj|j�}|j� |j � | j |jdd� tj� D �� t j dtj tj� �d�� | jttj� �g � W d Q R X t j dtjtj� �d��* | jtj�� ttj� � W d Q R X W d Q R X d S )Nc S s g | ] }|j �qS r# )�pid)�.0�xr# r# r$ � <listcomp>E s z4TestSystemAPIs.test_process_iter.<locals>.<listcomp>c S s g | ] }|j �qS r# )r' )r( r) r# r# r$ r* G s c S s g | ] }|j �qS r# )r' )r( r) r# r# r$ r* K s zpsutil.Process)� side_effect)�assertIn�os�getpid�psutil� process_iterr r' �Process�kill�wait� assertNotInr �patchZ NoSuchProcess� assertEqual�list� AccessDenied� assertRaises)r" �sproc�pr# r# r$ �test_process_iterD s z TestSystemAPIs.test_process_iterc C s8 x.t jdgd�D ] }| jt|jj� �dg� qW | jt�� tt jdgd�� W d Q R X tj dt j dd�d��L}x:t jddgd�D ]&}| j |jd � | j |jd d� q�W |j s�t�W d Q R X tj dt j dd�d��X}t� }x>t jddg|d �D ](}| j|jd |� | j |jd d� q�W |j �s*t�W d Q R X d S ) Nr' )�attrsZfooz$psutil._psplatform.Process.cpu_timesr � )r+ � cpu_times)r= Zad_value)r/ r0 r6 r7 �info�keysr9 � ValueErrorr r5 r8 � assertIsNone�assertGreaterEqual�called�AssertionError�object�assertIs)r" r; �m�flagr# r# r$ �test_prcess_iter_w_paramsU s$ z(TestSystemAPIs.test_prcess_iter_w_paramsc s� � fdd�}g � t � }t � }t � }dd� |||fD �}�jttj|dd� �jttj|dd� tj� }tj|d|d �\}}�jtj� | d � �j|g � �jt |�d � �j� g � x |D ]} �j t | d �� q�W t d ��fdd��} |j � | ||�\}}�j|jdd� |D �� t�r4�j|j� jtj � n�j|j� jd� �j� |jg� x |D ]} �j t | d �� �q\W t d ��fdd��} |j � |j � | ||�\}}�jt� �t|j|j|jg�� x |D ]} �jt | d �� �q�W d S )Nc s � j | j� d S )N)�appendr' )r; )�pidsr# r$ �callbackj s z0TestSystemAPIs.test_wait_procs.<locals>.callbackc S s g | ]}t j|j��qS r# )r/ r1 r' )r( r) r# r# r$ r* q s z2TestSystemAPIs.test_wait_procs.<locals>.<listcomp>� )�timeout)rN g{�G�z�?)rP rN g �?� � returncode� c s< t j| d|d�\}}� jt|�d� � jt|�d� ||fS )Ng� ��Q��?)rP rN rO � )r/ � wait_procsr6 �len)�procsrN �gone�alive)r" r# r$ �test~ s z,TestSystemAPIs.test_wait_procs.<locals>.testc S s g | ] }|j �qS r# )r' )r( r) r# r# r$ r* � s c s< t j| d|d�\}}� jt|�d� � jt|�d� ||fS )Ng� ��Q��?)rP rN rQ r )r/ rU r6 rV )rW rN rX rY )r" r# r$ rZ � s ���)r r9 rB r/ rU � TypeError�time� assertLessr6 rV � assertFalse�hasattrr � terminater, r' r �poprR �signal�SIGTERM�set� assertTrue) r" rN �sproc1�sproc2�sproc3rW �trX rY r; rZ r# )rM r" r$ �test_wait_procsi sB z TestSystemAPIs.test_wait_procsc C sN t � }t � }t � }dd� |||fD �}x|D ] }|j� q,W tj|�\}}d S )Nc S s g | ]}t j|j��qS r# )r/ r1 r' )r( r) r# r# r$ r* � s z=TestSystemAPIs.test_wait_procs_no_timeout.<locals>.<listcomp>)r ra r/ rU )r" rg rh ri rW r; rX rY r# r# r$ �test_wait_procs_no_timeout� s z)TestSystemAPIs.test_wait_procs_no_timeoutc C s4 t j� }| j|t� | j|d� | j|tj� � 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" dd l }| jtjd�|j� � d S )Nr � SC_PAGE_SIZE)�resourcer6 r- �sysconfZ getpagesize)r" rr r# r# r$ � test_PAGESIZE� s z TestSystemAPIs.test_PAGESIZEc C s� t j� }|jdkst|��|jdks,t|��d|j ko@dkn sNt|��|jdks`t|��|jdksrt|��xt|jD ]j}t ||�}|dkr�| j |t t f� |dkrz|dks�| j d||f � ||jkrz| j d||j||f � qzW d S )Nr �d �percent�totalz %r < 0 (%s)z %r > total (total=%s, %s=%s))r/ Zvirtual_memoryrw rF Z availablerv �used�free�_fields�getattrrm �intr �fail)r" �mem�name�valuer# r# r$ �test_virtual_memory� s " z"TestSystemAPIs.test_virtual_memoryc C s� t j� }| j|jd � |jdks(t|��|jdks:t|��|jdkrX|jdksjt|��n|jdksjt|��d|j ko~dkn s�t|��|j dks�t|��|j dks�t|��d S ) Nrw rx ry rv �sin�soutr ru )rw rx ry rv r� r� ) r/ Z swap_memoryr6 rz rw rF rx ry rv r� r� )r" r~ r# r# r$ �test_swap_memory� s "z TestSystemAPIs.test_swap_memoryc C st t � }| jtj|j�� tj|j�}|j� |j� | jtj|j�� | jtjd�� | j tjd�dtj � k� d S )NrO r r[ ) r rf r/ � pid_existsr' r1 r2 r3 r_ r6 rM )r" r: r; r# r# r$ �test_pid_exists� s z TestSystemAPIs.test_pid_existsc C s� t � tj� }xT|D ]L}ytj|�s(t�W q tk r^ tjd� |tj� krZ| j|� Y qX qW tt |�d t |�d �}x |D ]}| j tj|�|d� q�W d S )Ng�������?i� ip )�msg) r r/ rM r� rF r] �sleepr} �range�maxr_ )r" rM r' r# r# r$ �test_pid_exists_2� s z TestSystemAPIs.test_pid_exists_2c C sJ dd� t j� D �}t j� }| j|j� |j� � | jt|�tt|��� d S )Nc S s g | ] }|j �qS r# )r' )r( r) r# r# r$ r* � s z,TestSystemAPIs.test_pids.<locals>.<listcomp>)r/ r0 rM r6 �sortrV re )r" ZplistZpidlistr# r# r$ � test_pids� s zTestSystemAPIs.test_pidsc C s& t j}tt _z tj� W d |t _X d S )N)�sys�stdoutr r/ rZ )r" r� r# r# r$ � test_test� s zTestSystemAPIs.test_testc C s� t j� }| j|tt jdd��� | j|d� tjjd�rft d��}|j � }W d Q R X d|krft j d��t jdd�}| j|d� | j||� d S ) NT)�percpurO z /proc/cpuinfoz physical idz#cpuinfo doesn't include physical idF)�logical) r/ � cpu_countr6 rV r? rD r- �path�exists�open�readr ZSkipTest)r" r� �fdZ cpuinfo_dataZphysicalr# r# r$ �test_cpu_count s z TestSystemAPIs.test_cpu_countc C sz xtd D ]l}t jd|d�� }| jtj� � |js2t�W d Q R X t jd|d��"}| jtjdd�� |jsht�W d Q R X qW d S ) NrO r z$psutil._psplatform.cpu_count_logical)� return_valuez%psutil._psplatform.cpu_count_physicalF)r� r[ )r[ r N)r r5 rC r/ r� rE rF )r" �valrI r# r# r$ �test_cpu_count_none s z"TestSystemAPIs.test_cpu_count_nonec C s^ d}t j� }t|� x,|D ]$}| j|t� | j|d� ||7 }qW | j|t|�� t|� d S )Nr g )r/ r? �sumrm rn rD r6 �str)r" rw �times�cp_timer# r# r$ �test_cpu_times s z TestSystemAPIs.test_cpu_timesc C sD t tj� �}tjd� t tj� �}|| }|dks@| jd| � 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� xft jdd�D ]V}d}t|� x,|D ]$}| j|t� | j|d� ||7 }q$W | j|t|�� t|� qW | jtt jdd�d �tt jdd��� d S )NT)r� r g F) r/ r? r� rm rn rD r6 r� rV )r" r� rw r� r# r# r$ �test_per_cpu_timesC s z!TestSystemAPIs.test_per_cpu_timesc C s� t jdd�}tj� d }xtj� |krP qW t jdd�}x<t||�D ].\}}t|�t|� }}|| }|dkrDd S qDW | j� d S )NT)r� g�������?g�������?)r/ r? r] �zipr� r} )r" Ztot1Zstop_atZtot2r� r� r� r# r# r$ �test_per_cpu_times_2d s z#TestSystemAPIs.test_per_cpu_times_2c C s\ t j� }t jdd�}|jdd� t|� D ��}x*|jD ] }| jt||�t||�dd� q4W d S )NT)r� c S s g | ] }t |��qS r# )r� )r( Znumr# r# r$ r* y s z<TestSystemAPIs.test_cpu_times_comparison.<locals>.<listcomp>rO )�delta)r/ r? �_maker� rz �assertAlmostEqualr{ )r" �baseZper_cpuZ summed_valuesZfieldr# r# r$ �test_cpu_times_comparisont s z(TestSystemAPIs.test_cpu_times_comparisonc C s� y<| j |t� | j|d� | j|d� | j|dtj� � W n@ tk r| } z$td|tj |�tj |�f ��W Y d d }~X nX d S )Ng g Y@z %s last=%s new=%sg �) rm rn rD Z assertIsNot�assertLessEqualr/ r� rF �pprintZpformat)r" rv Zlast_retZnew_ret�errr# r# r$ �_test_cpu_percent~ s z TestSystemAPIs._test_cpu_percentc C sb t jdd�}x.td�D ]"}t jd d�}| j|||� |}qW | jt�� t jdd� W d Q R X d S )Ng����MbP?)�intervalru rO r[ )r/ � cpu_percentr� r� r9 rB )r" �lastr) �newr# r# r$ �test_cpu_percent� s z TestSystemAPIs.test_cpu_percentc C s� t jddd�}| jt|�t j� � x>td�D ]2}t jd dd�}x|D ]}| j|||� qDW |}q,W | jt�� t jddd� W d Q R X d S )Ng����MbP?T)r� r� ru rO r[ ) r/ r� r6 rV r� r� r� r9 rB )r" r� r) r� rv r# r# r$ �test_per_cpu_percent� s z#TestSystemAPIs.test_per_cpu_percentc C s` t jdd�}xNtd�D ]B}t jd d�}x|D ]}| j|||� q,W | jt|�||� |}qW d S )Ng����MbP?)r� ru )r/ �cpu_times_percentr� r� r� )r" r� r) r� rv r# r# r$ �test_cpu_times_percent� s z%TestSystemAPIs.test_cpu_times_percentc C s� t jddd�}| jt|�t j� � x^td�D ]R}t jd dd�}x:|D ]2}x|D ]}| j|||� qNW | jt|�||� qDW |}q,W d S )Ng����MbP?T)r� r� ru )r/ r� r6 rV r� r� r� r� )r" r� r) r� �cpurv r# r# r$ �test_per_cpu_times_percent� s z)TestSystemAPIs.test_per_cpu_times_percentc C sr t jdd� dd� t jdd�D �}tjd|d��8 x0t jdd�D ] }x|D ]}| j|d d � qJW q@W W d Q R X d S )NT)r� c S s* g | ]"}|j d d� tt|j��D ���qS )c S s g | ]}d �qS )r r# )r( r) r# r# r$ r* � s zQTestSystemAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>.<listcomp>)r� r� rV rz )r( r) r# r# r$ r* � s zFTestSystemAPIs.test_per_cpu_times_percent_negative.<locals>.<listcomp>zpsutil.cpu_times)r� )r/ r� r? r r5 r� )r" Z zero_timesr� rv r# r# r$ �#test_per_cpu_times_percent_negative� s z2TestSystemAPIs.test_per_cpu_times_percent_negativec C s2 t jtj� �}| j|jd � |jdks.t|��|jdks@t|��|j dksRt|��|j|jksft|��|j|j kszt|��d|j ko�dkn s�t|j ��t t d�� r�t jtj� �}d }| j|j|j� | j |j |j |d � | j |j|j|d � tj� }| jt��}t j|� W d Q R X | j|jjtj� d S )Nrw rx ry rv r ru � disk_usage� i )r� )rw rx ry rv i i P )r/ r� r- �getcwdr6 rz rw rF rx ry rv r` �shutilr� �tempfileZmktempr9 �OSErrorZ exception�errno�ENOENT)r" ZusageZ shutil_usageZ toleranceZfname�excr# r# r$ �test_disk_usage� s( $ z TestSystemAPIs.test_disk_usagec C s( t r$| jt�� tjt� W d Q R X d S )N)r r9 �UnicodeEncodeErrorr/ r� r )r" r# r# r$ �test_disk_usage_unicode� s z&TestSystemAPIs.test_disk_usage_unicodec C s t jd� d S )N� .)r/ r� )r" r# r# r$ �test_disk_usage_bytes� s z$TestSystemAPIs.test_disk_usage_bytesc C s t jdd�}| j||d� x�|D ]�}| j|jt� | j|jt� | j|jt� | j|jt� t rld|jkrlq t s�t j j |j�s�t|��n|j ts�tr�t j j |j�s�t|��nt j j|j�s�t|��|js t|��q W t jdd�}| j||d� x�t jdd�D ]�}t �s�yt j|j� W nX tk �rp } z:t�rJt�rJ|jtjk�rJ� w�|jtjtjfk�r`� W Y d d }~X n@X t�s~t�r�t j j |j��s�t|��nt j j|j��s�t|��| j|jt� | j|jt� � q�W dd� }|t�}dd � t jdd�D �}| j||� t j|� d S ) NF)�all)r� ZcdromTc S s2 t jj| �} x t jj| �s(t jj| �} qW | 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� �qS r# )� mountpointr� )r( r) r# r# r$ r* s z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>) r/ Zdisk_partitionsrf rm Zdevicer� r� ZfstypeZoptsr r r- r� r� rF r r �isdir�statr� r r� ZEIOZEPERMZEACCES�__file__r, r� )r" �lsZdiskr� r� ZmountZmountsr# r# r$ �test_disk_partitions� sL z#TestSystemAPIs.test_disk_partitionsc sl � fdd�}t jdd�}||� t jdd�}� j|g � x.|D ]&}� j|� � j|t� ||| � q>W d 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� | jdks�t | ��| jdks�t | ��| jdks�t | ��| jdks�t | ��| jdks�t | ��| jdks�t | ��| jdk�st | ��| jdk�s$t | ��d S ) Nr rO rT rQ � r� � � ) r6 Z bytes_sentZ bytes_recvZ packets_sentZ packets_recvZerrinZerroutZdropinZdropoutrF )�nt)r" r# r$ � check_ntuple s z9TestSystemAPIs.test_net_io_counters.<locals>.check_ntupleF)�pernicT)r/ �net_io_counters�assertNotEqualrf rm r� )r" r� �ret�keyr# )r" r$ �test_net_io_counters s z#TestSystemAPIs.test_net_io_countersc C sN t jdi d��6}| jtjdd�� | jtjdd�i � |js@t�W d Q R X d S )Nz"psutil._psplatform.net_io_counters)r� F)r� T)r r5 rC r/ r� r6 rE rF )r" rI r# r# r$ � test_net_io_counters_no_nics8 s z+TestSystemAPIs.test_net_io_counters_no_nicsc C st t j� }|st|��t j� }ttjtjt jg�}�x�|j � D �]�\}}| j |t � | j t t|��t |�� �x�|D �]�}| j |jt� | j |jt � | j |jt td �f� | j |jt td �f� | j|j|� tjdkr�| j |jtj� || j�r�|jtjk�r6tj|j�}tj|�� |j |jdf� W d Q R X nj|jtjk�r�tj |jdtjtj dtj �d }|\} } } } } tj| | | �}tj|�� |j | � W d Q R X xB|j|j|j|j fD ]*}|d k �r�|jtjk�r�t!||j� �q�W |j�r�| j"|j � qr|j rr| j"|j� qrW q<W t#�s&t$�s&t%�rDt&td��rp| j t jtj� n,t'�r\| j t jtj(� nt)�rp| j t jd� d S )NrQ r� r �AF_LINKrO )rQ r� r[ )*r/ � net_if_addrsrF � net_if_statsre �socketZAF_INETZAF_INET6r� �itemsrm r� r6 rV Zfamilyr| �addressZnetmask�typeZ broadcastr, r� � version_infor �IntEnum�isup� contextlib�closingZbindZ getaddrinfoZ SOCK_STREAMZ AI_PASSIVEZptpr rC r r r r` r Z AF_PACKETr )r" �nicsZ nic_statsZfamiliesZnicZaddrs�addr�sr@ ZafZsocktype�protoZ canonnameZsaZipr# r# r$ �test_net_if_addrsA sX z TestSystemAPIs.test_net_if_addrsc C sz t rdtjdd d d fg}nd g}tjd|d��B}tj� d d }|jsJt�t r^| j|j d� n| j|j d � W d Q R X d S ) N�em1z06:3d:29rO �06-3d-29z psutil._psplatform.net_if_addrs)r� r z06:3d:29:00:00:00z06-3d-29-00-00-00r[ )r� r[ r� NNN) r r/ r� r r5 r� rE rF r6 r� )r" r� rI 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� }|st|��t jt jt jf}xh|j� D ]\\}}| j|t� |\}}}}| j|t � | j ||� | j ||� | j |d� | j |d� q.W d S )Nr ) r/ r� rF ZNIC_DUPLEX_FULLZNIC_DUPLEX_HALFZNIC_DUPLEX_UNKNOWNr� rm r� �boolr, rD ) r" r� Z all_duplexesr Zstatsr� ZduplexZspeedZmtur# r# r$ �test_net_if_stats� s z TestSystemAPIs.test_net_if_statsz/proc/diskstatsz3/proc/diskstats not available on this linux versionNzunreliable on APPVEYORc s� � fdd�}t jdd�}|d k s(td��||� t jdd�}� jt|�tt|��� xb|D ]Z}|sjt|��||| � trZ|d j� rZx |d j� r�|d d � }q�W � j||j � � qZW d S ) Nc s� � j | d | j� � j | d | j� � j | d | j� � j | d | j� tpNtsƈ j | d | j� � j | d | j� t r�� j | d | j � � j | d | j � � j | d | j � nt rƈ j | d | j � x$| jD ]}t| |�dks�t| ��q�W d S ) Nr rO rT rQ r� r� r� r� � )r6 Z read_countZ write_countZ read_bytesZ write_bytesr r Z read_timeZ write_timer Zread_merged_countZwrite_merged_countZ busy_timer rz r{ rF )r� r )r" r# r$ r� � s z:TestSystemAPIs.test_disk_io_counters.<locals>.check_ntupleF)�perdiskzno disks on this system?TrO r[ r[ r[ ) r/ �disk_io_countersrF r6 rV re r �isdigitr4 rA )r" r� r� r� r# )r" r$ �test_disk_io_counters� s z$TestSystemAPIs.test_disk_io_countersc C sN t jdi d��6}| jtjdd�� | jtjdd�i � |js@t�W d Q R X d S )Nz#psutil._psplatform.disk_io_counters)r� F)r� T)r r5 rC r/ r� r6 rE rF )r" rI 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 ]�}|js,t|��| j|jt� | j|jttd �f� |j d k rp| j|j ttd �f� |j |j |j dks�t|��t j j |j � t s�tr�| j|j� qt j|j� qW d S )Ng )r/ �usersr� r rF rm r� Zterminalr� �hostZstarted�datetimeZ fromtimestampr r rC r' r1 )r" r� �userr# r# r$ � test_users� s zTestSystemAPIs.test_usersc C sZ t j� }| j|jd� x>|jD ]4}t||�}| j|d� t r |dkr | j|d� q W d S )N� ctx_switches� interrupts�soft_interrupts�syscallsr )r r r r )r r )r/ Z cpu_statsr6 rz r{ rD r ro )r" Zinfosr r� r# r# r$ �test_cpu_stats� s z TestSystemAPIs.test_cpu_statsz not suportedc s` � fdd�}t jdd�}tr&| r&d S |s2t|��|t jdd�g� tr\� jt|�t j� � d S )Nc sh xb| D ]Z}� j |jd� � j|j|j� x6|jD ],}t||�}� j|ttt f� � j |d� q0W qW d S )N�current�minr� r )r r r� ) r6 rz r� r r� r{ rm r| r rn rD )r� r� r r� )r" r# r$ �check_ls� s z.TestSystemAPIs.test_cpu_freq.<locals>.check_lsT)r� F)r/ Zcpu_freqr rF r r6 rV r� )r" r r� r# )r" r$ � test_cpu_freq� s z TestSystemAPIs.test_cpu_freqc C s� ddddddddd g }x"|D ]}| j tt|�t|d � q W tjd k�rJtjsPt�tj s\t�|j d� d t j j � kr�tj s~t�|j d� n�d t j j � kr�tjs�t�| jtjtjtjgjd�d� |j d� |j d� |j d� |j d� n^dt j j � k�s dt j j � k�r"tj�st�|j d � n&dt j j � k�rntj�s>t�|j d� n$tj�sVt�tj �sdt�|j d� x$|D ] }| jtt|�d|d � �qtW d S )Nr r r r r r r r r )r� �posix�linuxZbsdTrO �sunos�solaris�darwinF)rm r{ r/ r� r- r r rF r �remover� �platformr� r r r6 r r r �countr r rH )r" �namesr r# r# r$ �test_os_constants s> z TestSystemAPIs.test_os_constantsz not supportedc C s� t j� }x�|j� D ]x\}}| j|t� xb|D ]Z}| j|jt� |jd k rV| j|jd� |jd k rn| j|jd� |j d k r,| j|j d� q,W qW d S )Nr ) r/ �sensors_temperaturesr� rm r� �labelr rD �high�critical)r" �tempsr �entries�entryr# r# r$ �test_sensors_temperatures) s z(TestSystemAPIs.test_sensors_temperaturesc C sp ddgi}t jd|d��N}tjdd �d d }|js8t�| j|jd � | j|jd � | j|j d � W d Q R X d S )NZcoretempr � I@� N@� �Q@z'psutil._psplatform.sensors_temperatures)r� T)Z fahrenheitr g �^@g �a@g �c@)r r r r! ) r r5 r/ r rE rF r6 r r r )r" �drI r r# r# r$ �#test_sensors_temperatures_fahreneit7 s z2TestSystemAPIs.test_sensors_temperatures_fahreneitz no batteryc C sp t j� }| j|jd� | j|jd� |jt jt jfkrF| j|jd� n|jt jkr^| j|j � | j |j t � d S )Nr ru ) r/ Zsensors_batteryrD rv r� ZsecsleftZPOWER_TIME_UNKNOWNZPOWER_TIME_UNLIMITEDrf Z power_pluggedrm r� )r" r� r# r# r$ �test_sensors_batteryC s z#TestSystemAPIs.test_sensors_batteryc C sj t j� }x\|j� D ]P\}}| j|t� x:|D ]2}| j|jt� | j|jttf� | j |jd� q,W qW d S )Nr ) r/ Z sensors_fansr� rm r� r r r| r rD )r" Zfansr r r r# r# r$ �test_sensors_fansQ s z TestSystemAPIs.test_sensors_fans)C�__name__� __module__� __qualname__�__doc__r% r& r< rK rk rl rp r ZskipIfr rt 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 r r$ r r% r# r# r# r$ r! ; sh 7 ! > <# % r! �__main__)2r) r� r r� r- r� r� rc 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! r& r� r# r# r# r$ �<module> sh '