Cs[�� � @ 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
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, d d" l m- Z- d d# l m. Z. d d$ l m/ Z/ d d% l m0 Z0 d d& l m1 Z1 d d' l m2 Z2 d d( l m3 Z3 d d) l m4 Z4 d d* l m5 Z5 d d+ l m6 Z6 d d, l m7 Z7 d d- l m8 Z8 d d. l m9 Z9 d d/ l m: Z: d d0 l m; Z; d d1 l m< Z< d d2 l m= Z= d d l
Z
d d l Z
Gd3 d4 � d4 e9 j> � Z? e j@ d5 d6 � ZA Gd7 d8 � d8 e9 j> � ZB e9 jC e7 d9 � e9 jC e8 o�e jD jE e3 � d: � Gd; d< � d< e9 j> � � � ZF Gd= d> � d> e9 j> � ZG Gd? d@ � d@ e9 j> � ZH GdA dB � dB e9 j> � ZI GdC dD � dD e9 j> � ZJ GdE dF � dF e9 j> � ZK GdG dH � dH e9 j> � ZL eM dI k r�e0 eN � d S)Jz
Miscellaneous tests.
� N)�LINUX)�POSIX)�WINDOWS)�memoize)�memoize_when_activated)�
supports_ipv6)� wrap_numbers)�PY3)�APPVEYOR)� bind_socket)�bind_unix_socket)�
call_until)�chdir)�create_proc_children_pair)�create_sockets)�create_zombie_proc)�DEVNULL)�
get_free_port)�get_test_subprocess)� HAS_BATTERY)�HAS_CONNECTIONS_UNIX)�HAS_MEMORY_FULL_INFO)�HAS_MEMORY_MAPS)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�import_module_by_path)�
is_namedtuple)�mock)�
PYTHON_EXE)�
reap_children)�
reload_module)�retry)�ROOT_DIR)�run_test_module_by_name)�
safe_mkdir)� safe_rmpath)� SCRIPTS_DIR)�sh)�tcp_socketpair)�TESTFN)�TOX)�TRAVIS)�unittest)�unix_socket_path)�unix_socketpair)�
wait_for_file)� wait_for_pidc @ s e Z d Z e d d � Z d d � Z e d d � Z e d d � Z e d d
� Z e d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) S)*�TestMiscc C s� t j � } | | � } | j d | � | j d | j | � | j d | � | j | j � | � t j j t j d d t j t j
� � ��T t j � } | | � } | j d | j | � | j d | � | j d | � Wd QRXt j j t j d d t j t j
� � ��T t j � } | | � } | j d | j | � | j d | � | j d | � Wd QRXt j j t j d d t j
t j
� � ��D t j � } | | � } | j d | j | � | j d | � Wd QRXd S)Nzpsutil.Processzpid=%szname=�name� side_effectZzombieZ
terminated)�psutil�Process�assertIn�pidr3 r �patch�object�
ZombieProcess�os�getpid� assertNotIn�
NoSuchProcess� AccessDenied)�self�func�p�r� rE �
/test_misc.py�test_process__repr__Q s4 z TestMisc.test_process__repr__c C s | j d t � d S)NrB )rG �str)rA rE rE rF �test_process__str__m s z TestMisc.test_process__str__c C sm | j t t j d � � d � | j t t j d d d �� d � | j t t j d d d �� d � d S)NiA z7psutil.NoSuchProcess process no longer exists (pid=321)r3 �foozCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo')�msgzpsutil.NoSuchProcess foo)� assertEqual�reprr5 r? )rA rB rE rE rF � test_no_such_process__repr__p s z%TestMisc.test_no_such_process__repr__c C s� | j t t j d � � d � | j t t j d d d �� d � | j t t j d d d d d �� d � | j t t j d d d �� d
� d S) NiA zEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)r3 rJ zQpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo')�ppid� zYpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo', ppid=1)rK zpsutil.ZombieProcess foo)rL rM r5 r; )rA rB rE rE rF �test_zombie_process__repr__| s z$TestMisc.test_zombie_process__repr__c C sm | j t t j d � � d � | j t t j d d d �� d � | j t t j d d d �� d � d S)NiA z psutil.AccessDenied (pid=321)r3 rJ z)psutil.AccessDenied (pid=321, name='foo')rK zpsutil.AccessDenied foo)rL rM r5 r@ )rA rB rE rE rF �test_access_denied__repr__� s z#TestMisc.test_access_denied__repr__c C ss | j t t j d � � d � | j t t j d d d �� d � | j t t j d d d d d �� d � d S) NiA z/psutil.TimeoutExpired timeout after 321 secondsr8 �o z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)r3 rJ zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))rL rM r5 �TimeoutExpired)rA rB rE rE rF � test_timeout_expired__repr__� s z%TestMisc.test_timeout_expired__repr__c C sU t j � } t j � } | j | | � d | _ | j | | � | j | d � d S)Nr rJ )r r )r5 r6 rL �_ident�assertNotEqual)rA �p1�p2rE rE rF �test_process__eq__� s zTestMisc.test_process__eq__c C s8 t t j � t j � g � } | j t | � d � d S)NrP )�setr5 r6 rL �len)rA �srE rE rF �test_process__hash__� s z TestMisc.test_process__hash__c C s� t t � } x� | D]� } | d
k r( q | j d
� s y t | � Wq t k
r� | t j k r� t t | � } | d k r� w | j d k r� d | j j � k r� | j d | � Yq Xq Wx! t j D] } | j
| | � q� Wd S)N�callable�error�
namedtuple�tests�long�test�NUM_CPUS� BOOT_TIME� TOTAL_PHYMEM�_Z
deprecatedz%r not in psutil.__all__) r_ r` ra rb rc rd re rf rg ) �dirr5 �
startswith�
__import__� ImportError�__all__�getattr�__doc__�lower�failr7 )rA Z
dir_psutilr3 ZfunrE rE rF � test__all__� s&
zTestMisc.test__all__c C s0 | j d j d d � t j D� � t j � d S)N�.c S s g | ] } t | � � q SrE )rH )�.0�xrE rE rF �
<listcomp>� s z)TestMisc.test_version.<locals>.<listcomp>)rL �joinr5 � version_info� __version__)rA rE rE rF � test_version� s "zTestMisc.test_versionc C s/ t j � } d | _ | j d | j � � d S)N�1rJ )r5 r6 rJ r> �as_dict)rA rC rE rE rF �!test_process_as_dict_no_new_names� s z*TestMisc.test_process_as_dict_no_new_namesc s� t � f d d � � } g � xO t d � D]A } | � } f i f } | j | | � | j t � � d � q+ WxR t d � D]D } | d � } d i f } | j | | � | j t � � d � q} Wx^ t d � D]P } | d d d �} d
d d i f } | j | | � | j t � � d � q� W| j � | � } f i f } | j | | � | j t � � d � | j | j d � d S) Nc s � j d � | | f S)z
foo docstringN)�append)�args�kwargs)�callsrE rF rJ � s
z"TestMisc.test_memoize.<locals>.foo� rP Zbar� � z
foo docstring)rP )rP )r �rangerL r\ � cache_clearro )rA rJ ru �retZexpectedrE )r� rF � test_memoize� s.
zTestMisc.test_memoizec s� G� f d d � d � } | � } g � | j � | j � | j t � � d � g � | j j � | j � | j � | j t � � d � g � | j j � | j � | j � | j t � � d � d S)Nc s( e Z d Z e � f d d � � Z d S)z1TestMisc.test_memoize_when_activated.<locals>.Fooc s � j d � d S)N)r~ )rA )r� rE rF rJ � s z5TestMisc.test_memoize_when_activated.<locals>.Foo.fooN)�__name__�
__module__� __qualname__r rJ rE )r� rE rF �Foo� s r� r� rP )rJ rL r\ Zcache_activateZcache_deactivate)rA r� �frE )r� rF �test_memoize_when_activated� s
z$TestMisc.test_memoize_when_activatedc C s2 d d l m } d d � } | j | d � | d � d i � | j | d � | d � d | d � d
i � | j | d � | d � d | d � d i � | j | d
� | d � d | d � d
i � | j | d � | d � d i � | j | d � | d � d i � | j | d � | d � d i � d S)Nr )�parse_environ_blockc S s t r | j � S| S)N)r �upper)r] rE rE rF �k s z,TestMisc.test_parse_environ_block.<locals>.kza=1 �ar{ z a=1 b=2 �b�2za=1 b= � z
a=1 b=2 c=3 zxxx a=1 z a=1 =b=2 za=1 b=2)�psutil._commonr� rL )rA r� r� rE rE rF �test_parse_environ_block s ""z!TestMisc.test_parse_environ_blockc C s� | j t j � t � rMt j d � �* } d | _ t j � t � sN t � Wd QRXt j � t j d d t j ��& } t � s� t � | j s� t � Wd QRXt j � t j d d t j
��0 } t � s� t � t j � | j s� t � Wd QRXt j � t j d d t j
��0 } t � s*t � t j � | j sCt � Wd QRXn< | j t � �* t j t j
t j � } | j d � Wd QRXd S) Nzpsutil._common.socketFz psutil._common.socket.socketr4 z!psutil._common.socket.socket.bind�::1r )r� r )�
addCleanupr r� r r9 �has_ipv6�AssertionError�socketr` �called�gaierror� assertRaises� Exception�AF_INET6� SOCK_STREAM�bind)rA r] �sockrE rE rF �test_supports_ipv6' s4
zTestMisc.test_supports_ipv6c C s; d d l m } t j j t � } | | � s4 t � | t j j | � � sS t � t j d d t
t j d � �� | j
t
| | � Wd QRXt j d d t
t j d � �� | j
t
| | � Wd QRXt j d d t
t j d � �� | | � s t � Wd QRXt j d d d �� | | � s0t � Wd QRXd S) Nr )�
isfile_strictzpsutil._common.os.statr4 rJ zpsutil._common.stat.S_ISREG� return_valueF)r� r� r< �path�abspath�__file__r� �dirnamer r9 �OSError�errno�EPERMr� ZEACCES�EINVAL)rA r� Z this_filerE rE rF �test_isfile_strictG s zTestMisc.test_isfile_strictc s� � f d d � } | t j � j � � | t j � � | t j � � | t j � � | t j d d � � | t j � � t r� t j
j d � r� n t s� | t j
� � | t j � � | t j t j � � � | t j � � d S)Nc sT t d k r" t j t j | � � t j | � } t j | � } � j | | � d S)N)�json�loads�dumps�picklerL )r� r� r� )rA rE rF �checkY s
z*TestMisc.test_serialization.<locals>.check�intervalr z/proc/diskstats)r5 r6 r| Zvirtual_memoryZ swap_memoryZ cpu_timesZcpu_times_percent�net_io_countersr r< r� �existsr
�disk_io_countersZdisk_partitions�
disk_usage�getcwd�users)rA r� rE )rA rF �test_serializationX s zTestMisc.test_serializationc C sw t j j t d � } t r; t j j | � r; | j d � St | � } | j t | j
� | j | j � t
j � d S)Nzsetup.pyzcan't find setup.py)r< r� rw r# r, r� ZskipTestr r� �
SystemExitZsetuprL Z get_versionr5 ry )rA Zsetup_py�modulerE rE rF �test_setup_scripto s
zTestMisc.test_setup_scriptc C s� t j j t j d d t j �� } t j � | j s= t � Wd QRXt j j t j d d t j d � �� } t j � | j s� t � Wd QRXt j j t j d d t ��7 } | j
t � � t j � Wd QRX| j s� t � Wd QRXd S)NZ create_timer4 rP ) r r9 r: r5 r6 r@ r� r� r; �
ValueErrorr� )rA �methrE rE rF �test_ad_on_process_creationw s
z$TestMisc.test_ad_on_process_creationc C sc t j d d d ��G | j t � � } t t � Wd QRX| j d t | j � j � � Wd QRXd S)Nz psutil._psplatform.cext.versionr� z0.0.0zversion conflict)
r r9 r� rl r! r5 r7 rH � exceptionrp )rA �cmrE rE rF �test_sanity_version_check� s
z"TestMisc.test_sanity_version_checkN)r� r� r� rM rG rI rN rQ rR rU rZ r^ rr rz r} r� r� r� r� r� r� r� r� r� rE rE rE rF r2 O s( # r2 rJ za b cc @ s� e Z d Z d d � Z e 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 e j e j � p� e j � d � d d � � Z d S) �TestWrapNumbersc C s t j � d S)N)r r� )rA rE rE rF �setUp� s zTestWrapNumbers.setUpc C s5 d t d d d � i } | j t | d � | � d S)N�disk1� �disk_io)�ntrL r )rA �inputrE rE rF �test_first_call� s z TestWrapNumbers.test_first_callc C sN d t d d d � i } | j t | d � | � | j t | d � | � d S)Nr� r� r� )r� rL r )rA r� rE rE rF �test_input_hasnt_changed� s z(TestWrapNumbers.test_input_hasnt_changedc C s� d t d d d � i } | j t | d � | � d t d d d � i } | j t | d � | � d t d d d � i } | j t | d � | � d t d d d � i } | j t | d � | � d S) Nr� r� r� �
� � � � )r� rL r )rA r� rE rE rF �test_increase_but_no_wrap� s z)TestWrapNumbers.test_increase_but_no_wrapc C sM d t d d d � i } | j t | d � | � d t d d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d t d
d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d t d d d � i } | j t | d � d t d d d � i � d S)
Nr� �d r� r� �n �Z � r� �� �2 � �( )r� rL r )rA r� rE rE rF � test_wrap� s4 zTestWrapNumbers.test_wrapc C s� d t d d d � i } | j t | d � | � d t d d d � d t d d d � i } | j t | d � | � d t d d d � i } | j t | d � | � d S)Nr� r� r� �disk2� � )r� rL r )rA r� rE rE rF �test_changing_keys� s z"TestWrapNumbers.test_changing_keysc C s� d t d d d � d t d d d � i } | j t | d � | � d t d d d � d t d d d � i } | j t | d � d t d d d � d t d d d � i � d t d d d � i } | j t | d � | � d t d d d � d t d d d � i } | j t | d � | � d t d d d � d t d d d � i } | j t | d � | � d t d d d � d t d d d � i } | j t | d � d t d d d � d t d d d � i � d S)Nr� r� r� r� r� r� r� )r� rL r )rA r� rE rE rF �test_changing_keys_w_wrap� s* z)TestWrapNumbers.test_changing_keys_w_wrapc C s� d d) d d* d d+ d d, i } | j t | d& � | � | j t | d& � | � d d- d d. d d/ d d0 i } t | d& � } | j | d d d( � d S)1NZnvme0n1�, � � �# �R �� � �� �ȷ Z nvme0n1p1� r� � tU � � r r� Z nvme0n1p2�6 � �$ � �N r� � r� � � Z nvme0n1p3�U � �" r� �� �4 �� � �\ r� r� i� ) 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� r� r� r� r� r� r� ) r� r� r� r� r� r� r� r� r� )rL r )rA �d�outrE rE rF �test_real_data s z TestWrapNumbers.test_real_datac C s� d t d d d � i } t | d � t j � } | j | d d | i � | j | d d i i � | j | d d i i � d S)Nr� r� r� r rP r� )r� r �
cache_inforL )rA r� �cacherE rE rF �test_cache_first_call s
z%TestWrapNumbers.test_cache_first_callc C s� d t d d d � i } t | d � d t d d d � i } t | d � t j � } | j | d d | i � | j | d d d d d d d
d i i � | j | d d i i � d S) Nr� r� r� r� r rP r� )r� r )r� rP )r� r� )r� r r rL )rA r� r rE rE rF �test_cache_call_twice s
z%TestWrapNumbers.test_cache_call_twicec s d t d d d � i } t | d � d t d d d � i } t | d � t j � } � j | d d | i � � j | d d d
d d d d d i i � � j | d d d t d g � i i � � f d d � } d t d d d � i } t | d � t j � } � j | d d | i � | � d t d d d
� i } t | d � t j � } � j | d d | i � | � d t d d d � i } t | d � t j � } � j | d d | i � � j | d d d d d d d d i i � � j | d d d t d g � i i � d S)Nr� r� r� r� r rP r� c se t j � } � j | d d d d d d d d i i � � j | d d d t d
g � i i � d S) NrP r� r� r r� r� )r� r )r� rP )r� r� )r� r� )r r rL r[ )r )rA rE rF �assert_6 s
z0TestWrapNumbers.test_cache_wrap.<locals>.assert_r� r� r� )r� r )r� rP )r� r� )r� r� )r� r )r� rP )r� r� )r� r� )r� r r rL r[ )rA r� r r rE )rA rF �test_cache_wrap' s:
)
z TestWrapNumbers.test_cache_wrapc C s� d t d d d � i } t | d � d t d d d � d t d d d � i } t | d � t j � } | j | d d | i � | j | d d d d d
d d d i i � | j | d d i i � d S) Nr� r� r� r� r� r rP r� )r� r )r� rP )r� r� )r� r r rL )rA r� r rE rE rF �test_cache_changing_keysW s
z(TestWrapNumbers.test_cache_changing_keysc C s| d t d d d � i } t | d � t | d � t j d � | j t j � i i i f � t j d � t j d � d S)Nr� r� r� z?!?)r� r r� rL r )rA r� rE rE rF �test_cache_cleard s
z TestWrapNumbers.test_cache_clearzno disks or NICs availablec C s� t j � t j � t j � } x. | D]& } | j d | � | j d | � q' Wt j j � t j � } x. | D]& } | j d | � | j d | � qq Wt j j � t j � } | j | i i i f � d S)Nzpsutil.disk_io_counterszpsutil.net_io_counters) r5 r� r� r r r7 r� r> rL )rA Zcachesr rE rE rF � test_cache_clear_public_apism s
z,TestWrapNumbers.test_cache_clear_public_apisN)r� r� r� r� �tearDownr� r� r� r� r� r� r r r r r r r- �skipIfr5 r� r� r
rE rE rE rF r� � s"
% 0
r� zcan't test on TOXz can't locate scripts directoryc @ sP e Z d Z d Z e d d � � Z e d 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 e j e p� e o� e j � d � d d � � Z d d � Z d d � Z d d � Z e j e d � d d � � Z e j e d � 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! e j e" d/ � d0 d1 � � Z# d2 d3 � Z$ e j e% d � e j e d � d4 d5 � � � Z& e j e' d � e j e d � d6 d7 � � � Z( e j e) d � e j e* d8 � d9 d: � � � Z+ d; d< � Z, d S)=� TestScriptsz-Tests for scripts in the "scripts" directory.c O s� d t j j t | � } t | g } x | D] } | j | � q, Wy t | | � j � } WnD t k
r� } z$ d t | � k r� t | � S� WYd d } ~ Xn X| s� t
| � � | S)Nz%sr@ ) r< r� rw r' r r~ r( �strip� RuntimeErrorrH r� )�exer r� �cmd�argr �errrE rE rF �
assert_stdout� s
zTestScripts.assert_stdoutNc C sm t j j t | � } t r3 t | d d d �} n t | d � } | � | j � } Wd QRXt j | � d S)NZrt�encoding�utf8) r< r� rw r' r �open�read�ast�parse)r r r� �srcrE rE rF �
assert_syntax� s zTestScripts.assert_syntaxc C sy t | � } xf t j t � D]U } | j d � r d t j j | � d | k r | j d t j j t | � � q Wd S)Nz.pyZtest_r z no test defined for %r script) ri r< �listdirr' �endswithr� �splitextrq rw )rA Zmethsr3 rE rE rF �
test_coverage� s zTestScripts.test_coveragez
POSIX onlyc C sp xi t j t � D]X } | j d � r t j j t | � } t j t j | � t j @s | j d | � q Wd S)Nz.pyz%r is not executable)
r< r r' r r� rw �stat�S_IXUSR�ST_MODErq )rA r3 r� rE rE rF �test_executable� s
zTestScripts.test_executablec C s | j d � d S)Nz
disk_usage.py)r )rA rE rE rF �test_disk_usage� s zTestScripts.test_disk_usagec C s | j d � d S)Nzfree.py)r )rA rE rE rF � test_free� s zTestScripts.test_freec C s | j d � d S)Nz
meminfo.py)r )rA rE rE rF � test_meminfo� s zTestScripts.test_meminfoc C s | j d t t j � � � d S)Nz procinfo.py)r rH r< r= )rA rE rE rF �
test_procinfo� s zTestScripts.test_procinfoz unreliable on APPVEYOR or TRAVISc C s | j d � d S)Nzwho.py)r )rA rE rE rF �test_who� s zTestScripts.test_whoc C s | j d � d S)Nzps.py)r )rA rE rE rF �test_ps� s zTestScripts.test_psc C s | j d � d S)Nz pstree.py)r )rA rE rE rF � test_pstree� s zTestScripts.test_pstreec C s | j d � d S)Nz
netstat.py)r )rA rE rE rF � test_netstat� s zTestScripts.test_netstatzunreliable on TRAVISc C s | j d � d S)Nz ifconfig.py)r )rA rE rE rF �
test_ifconfig� s zTestScripts.test_ifconfigz
not supportedc C s | j d t t j � � � d S)Nzpmap.py)r rH r< r= )rA rE rE rF � test_pmap� s zTestScripts.test_pmapc C s | j d d t �d S)Nz procsmem.py�stderr)r r )rA rE rE rF �
test_procsmem� s zTestScripts.test_procsmemc C s | j d � d S)Nz
killall.py)r )rA rE rE rF � test_killall� s zTestScripts.test_killallc C s | j d � d S)Nz nettop.py)r )rA rE rE rF � test_nettop� s zTestScripts.test_nettopc C s | j d � d S)Nztop.py)r )rA rE rE rF �test_top� s zTestScripts.test_topc C s | j d � d S)Nziotop.py)r )rA rE rE rF �
test_iotop� s zTestScripts.test_iotopc C s> | j d t j � j � � } | j t t j � � | � d S)Nzpidof.py)r r5 r6 r3 r7 rH r< r= )rA �outputrE rE rF �
test_pidof� s zTestScripts.test_pidofz WINDOWS onlyc C s | j d � d S)Nzwinservices.py)r )rA rE rE rF �test_winservices� s z TestScripts.test_winservicesc C s | j d � d S)Nzcpu_distribution.py)r )rA rE rE rF �test_cpu_distribution� s z!TestScripts.test_cpu_distributionc C s | j d � d S)Nztemperatures.py)r )rA rE rE rF �test_temperatures� s z TestScripts.test_temperaturesc C s | j d � d S)Nzfans.py)r )rA rE rE rF � test_fans s zTestScripts.test_fansz
no batteryc C s | j d � d S)Nz
battery.py)r )rA rE rE rF � test_battery s zTestScripts.test_batteryc C s | j d � d S)Nz
sensors.py)r )rA rE rE rF � test_sensors
s zTestScripts.test_sensors)-r� r� r� ro � staticmethodr r r r- r r r$ r% r& r' r( r
r, r5 r� r) r* r+ r, r- r r. r r0 r1 r2 r3 r4 r6 r r7 r8 r r9 r r: r r r; r< rE rE rE rF r
� s@
!!"r
c @ s� e Z d Z e j d � d d � � Z e j d � d d � � Z e j d � d d � � Z e j d � d d � � Z e j d � d
d � � Z e j d � d d
� � Z
d S)�TestRetryDecoratorz
time.sleepc si t d d d d d d � � f d d � � } t t d � � � | j | � d � | j | j d � d S) N�retriesr� r� rP �logfunc s# x � r � j � d d q Wd S)NrP r )�poprE )�queuerE rF rJ s
z2TestRetryDecorator.test_retry_success.<locals>.foor� )r" �listr� rL �
call_count)rA �sleeprJ rE )rB rF �test_retry_success s -z%TestRetryDecorator.test_retry_successc sf t d d d d d d � � f d d � � } t t d � � � | j t | � | j | j d � d S) Nr? r� r� rP r@ c s# x � r � j � d d q Wd S)NrP r )rA rE )rB rE rF rJ ( s
z2TestRetryDecorator.test_retry_failure.<locals>.foo� )r" rC r� r� �ZeroDivisionErrorrL rD )rA rE rJ rE )rB rF �test_retry_failure$ s -z%TestRetryDecorator.test_retry_failurec C sH t d t d d � d d � � } | j t | � | j | j d � d S)Nr� r� rP c S s
t � d S)N)� TypeErrorrE rE rE rF rJ 5 s z2TestRetryDecorator.test_exception_arg.<locals>.foor )r" r� r� rJ rL rD )rA rE rJ rE rE rF �test_exception_arg3 s !z%TestRetryDecorator.test_exception_argc C sN t d d d d d d � d d � � } | j t | � | j | j d � d S)Nr? r� r� r@ c S s d d d S)NrP r rE rE rE rE rF rJ @ s z4TestRetryDecorator.test_no_interval_arg.<locals>.foor )r" r� rH rL rD )rA rE rJ rE rE rF �test_no_interval_arg< s 'z'TestRetryDecorator.test_no_interval_argc C sN t d d d d d d � d d � � } | j t | � | j | j d � d S)Nr? r� r� rP r@ c S s d d d S)NrP r rE rE rE rE rF rJ J s z0TestRetryDecorator.test_retries_arg.<locals>.foo)r" r� rH rL rD )rA rE rJ rE rE rF �test_retries_argG s 'z#TestRetryDecorator.test_retries_argc C s | j t t d d d d �d S)Nr? r� �timeoutrP )r� r� r" )rA rE rE rE rF � test_retries_and_timeout_argsQ s z0TestRetryDecorator.test_retries_and_timeout_argsN) r� r� r� r r9 rF rI rK rL rM rO rE rE rE rF r> s
r> c @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)�TestSyncTestUtilsc C s t t � d S)N)r& r* )rA rE rE rF r X s zTestSyncTestUtils.tearDownc C sf t t j � � t t j � � d } t j d d t d g � �� | j t j
t | � Wd QRXd S)Ni�� zpsutil.tests.retry.__iter__r� r ) r1 r< r= �maxr5 Zpidsr r9 �iterr� r? )rA ZnopidrE rE rF �test_wait_for_pid[ s z#TestSyncTestUtils.test_wait_for_pidc C sM t t d � � } | j d � Wd QRXt t � t j j t � sI t � d S)N�wrJ )r r* �writer0 r< r� r� r� )rA r� rE rE rF �test_wait_for_filea s
z$TestSyncTestUtils.test_wait_for_filec C sD t t d � � Wd QRXt t d d �t j j t � s@ t � d S)NrT �emptyT)r r* r0 r< r� r� r� )rA rE rE rF �test_wait_for_file_emptyg s z*TestSyncTestUtils.test_wait_for_file_emptyc C s= t j d d t d g � �� | j t t t � Wd QRXd S)Nzpsutil.tests.retry.__iter__r� r )r r9 rR r� �IOErrorr0 r* )rA rE rE rF �test_wait_for_file_no_filem s z,TestSyncTestUtils.test_wait_for_file_no_filec C sR t t d � � } | j d � Wd QRXt t d d �t j j t � sN t � d S)NrT rJ �deleteF)r r* rU r0 r< r� r� r� )rA r� rE rE rF � test_wait_for_file_no_deleteq s z.TestSyncTestUtils.test_wait_for_file_no_deletec C s) t d d � d � } | j | d � d S)Nc S s d S)NrP rE rE rE rE rF �<lambda>x s z3TestSyncTestUtils.test_call_until.<locals>.<lambda>zret == 1rP )r
rL )rA r� rE rE rF �test_call_untilw s z!TestSyncTestUtils.test_call_untilN)
r� r� r� r rS rV rX rZ r\ r^ rE rE rE rF rP V s rP c @ sF e Z d Z d d � Z e Z d d � Z d d � Z d d � Z d S)
�TestFSTestUtilsc C s t t � d S)N)r& r* )rA rE rE rF r� ~ s zTestFSTestUtils.setUpc C sH t t � t j j t � s" t � t t � t j j t � sD t � d S)N)r% r* r< r� �isdirr� )rA rE rE rF �test_safe_mkdir� s
z TestFSTestUtils.test_safe_mkdirc C s� t t d � j � t t � t j j t � s6 t � t t � t j t � t t � t j j t � sp t � t j
d d t t j
d � ��7 } | j t � � t t � Wd QRX| j s� t � Wd QRXd S)NrT zpsutil.tests.os.statr4 r� )r r* �closer& r< r� r� r� �mkdirr r9 r� r� r� r� r� )rA �mrE rE rF �test_safe_rmpath� s
z TestFSTestUtils.test_safe_rmpathc C sl t j � } t j t � t t � �* | j t j � t j j | t � � Wd QRX| j t j � | � d S)N)r< r� rc r* r rL r� rw )rA �baserE rE rF �
test_chdir� s
,zTestFSTestUtils.test_chdirN)r� r� r� r� r ra re rg rE rE rE rF r_ | s
r_ c @ sG e Z d Z d d � Z d d � Z e j e d � d d � � Z d S) �TestProcessUtilsc C sq t � } t j | j � } | j � s- t � t � | j � sG t � t j j sZ t � t j j sm t � d S)N)
r r5 r6 r8 �
is_runningr� r rb �
_pids_started�_subprocesses_started)rA ZsubprC rE rE rF �test_reap_children� s z#TestProcessUtils.test_reap_childrenc C s# t � \ } } | j | j | j � | j � s7 t � | j � sI t � t j � j d d � } | j t | � d � | j
| | � | j
| | � | j | j � t j
� � | j | j � | j � t � | j � s� t � | j � s� t � t j j s t � t j j s t � d S)N� recursiveTr� )r rW r8 ri r� r5 r6 �childrenrL r\ r7 rO r< r= r rb rj rk )rA rX rY rn rE rE rF � test_create_proc_children_pair� s z/TestProcessUtils.test_create_proc_children_pairz
POSIX onlyc C sH t � } | j t d d �t j | � } | j | j � t j � d S)Nrm T)r r� r r5 r6 rL �statusZ
STATUS_ZOMBIE)rA ZzpidrC rE rE rF �test_create_zombie_proc� s z(TestProcessUtils.test_create_zombie_procN) r� r� r� rl ro r- r r rq rE rE rE rF rh � s rh c @ sr e Z d Z d d � Z e j e d � d d � � Z d d � Z e j e d � d d � � Z d
d � Z
d S)
� TestNetUtilsc
C sO t � } t j t d d | f � � �! } | j | j � d | � Wd QRXd S)N�addrr� rP )r �
contextlib�closingr rL � getsockname)rA �portr] rE rE rF r � s !zTestNetUtils.bind_socketz
POSIX onlyc C s
t � �� } t | � } t j | � �� | j | j t j � | j | j t j � | j | j
� | � t j j
| � s� t � t j t j | � j � s� t � Wd QRXWd QRXt � �I } t | d t j �} t j | � � | j | j t j � Wd QRXWd QRXd S)N�type)r. r rt ru rL �familyr� �AF_UNIXrx r� rv r< r� r� r� r! �S_ISSOCK�st_mode�
SOCK_DGRAM)rA r3 r� rE rE rF �test_bind_unix_socket� s / z"TestNetUtils.test_bind_unix_socketc C s� d t � f } t t j d | �\ } } t j | � �^ t j | � �G | j | j � | � | j | j � | � | j | j � | � Wd QRXWd QRXd S)Nz 127.0.0.1rs )
r r) r� �AF_INETrt ru rL rv � getpeernamerW )rA rs �server�clientrE rE rF �tcp_tcp_socketpair� s z TestNetUtils.tcp_tcp_socketpairc C s t j � } | j � } | j d d � s1 t � t � �� } t | � \ } } z� t j j | � sj t � t
j t j
| � j � s� t � | j
| j � | d � | j
t | j d d � � d � | j
| j � | � | j
| j � | � Wd | j � | j � XWd QRXd S)N�kindZunixr� )r5 r6 �num_fdsZ connectionsr� r. r/ r< r� r� r! r{ r| rL r\ rv r� rb )rA rC r� r3 r� r� rE rE rF �test_unix_socketpair� s !"
z!TestNetUtils.test_unix_socketpairc C s� t � �� } t j t � } t j t � } x? | D]7 } | | j d 7<| | j t j t j � d 7<q1 W| j | t j
d � t � r� | j | t j d � t
r� t r� | j | t j d � | j | t j d � | j | t j d � Wd QRXd S)NrP r� )r � collections� defaultdict�intry �
getsockoptr� �
SOL_SOCKET�SO_TYPEZassertGreaterEqualr r r� r r rz r� r} )rA ZsocksZfams�typesr] rE rE rF �test_create_sockets� s
$ z TestNetUtils.test_create_socketsN) r� r� r� r r- r r r~ r� r� r� rE rE rE rF rr � s
rr c @ s e Z d Z d d � Z d S)�TestOtherUtilsc C sD t t j d d � d d d � � s* t � t t � � s@ t � d S)NrJ za b crP r� r� )r r� ra r� �tuple)rA rE rE rF �test_is_namedtuple s *z!TestOtherUtils.test_is_namedtupleN)r� r� r� r� rE rE rE rF r� s r� �__main__)Oro r r� rt r� r� r< r� r� r! r5 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 r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 ZTestCaser2 ra r� r� r r� r� r
r> rP r_ rh rr r� r� r� rE rE rE rF �<module>
s� � H� �C&'&E