← Back
Editing: test_misc.cpython-37.pyc
B �=�_�o � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddl Z ddlZ e j'dd � d kZ(G d d � d e �Z)e�*d d!�Z+G d"d#� d#e �Z,e&�-ej.�/e$� d$�G d%d&� d&e ��Z0e1d'k�r dd(l2m3Z3 e3e4� dS ))z Miscellaneous tests. � N)�LINUX)�POSIX)�WINDOWS)�memoize)�memoize_when_activated)� supports_ipv6)� wrap_numbers)�PY3)�APPVEYOR)� CI_TESTING)� HAS_BATTERY)�HAS_MEMORY_MAPS)�HAS_NET_IO_COUNTERS)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�import_module_by_path)�mock)�PsutilTestCase)� PYTHON_EXE)� reload_module)�ROOT_DIR)� SCRIPTS_DIR)�sh)�unittest� )� � c @ s� e Zd Zefdd�Zdd� Zefdd�Zefdd�Zefd d �Zefd d �Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*�TestMiscc C s� t �| �� j�}||�}| �d|� | �d|j |� | �dt|�� � |�dd�� | �d|� | �d|� |� � |� � ||�}| �d|� | �d|� t j j t jd t �t�� �d ��@ t �� }||�}| �d|j |� | �d |� | �d |� W d Q R X t j j t jd t �t�� �d ��@ t �� }||�}| �d|j |� | �d |� | �d |� W d Q R X t j j t jd t �t�� �d ��4 t �� }||�}| �d|j |� | �d |� W d Q R X d S )Nzpsutil.Processzpid=%sz name='%s'zname=u'zname='zstatus=z exitcode=zstatus='terminated'�name)� side_effectzstatus='zombie'zname=Z terminated)�psutil�ProcessZspawn_testproc�pid�assertIn�strr �replace� assertNotIn� terminate�waitr �patch�object� ZombieProcess�os�getpid� NoSuchProcess� AccessDenied)�self�func�p�r� r5 �I/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_misc.py�test_process__repr__= sB z TestMisc.test_process__repr__c C s | j td� d S )N)r2 )r7 r% )r1 r5 r5 r6 �test_process__str__a s z TestMisc.test_process__str__c C sN | � tt�d��d� | � ttjddd��d� | � ttjddd��d� d S )NiA z7psutil.NoSuchProcess process no longer exists (pid=321)�foo)r zCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo'))�msgzpsutil.NoSuchProcess foo)� assertEqual�reprr! r/ )r1 r2 r5 r5 r6 � test_no_such_process__repr__d s z%TestMisc.test_no_such_process__repr__c C sj | � tt�d��d� | � ttjddd��d� | � ttjdddd��d� | � ttjddd ��d � d S ) NiA zEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)r9 )r zQpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo')� )r �ppidzYpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo', ppid=1))r: zpsutil.ZombieProcess foo)r; r< r! r, )r1 r2 r5 r5 r6 �test_zombie_process__repr__p s z$TestMisc.test_zombie_process__repr__c C sN | � tt�d��d� | � ttjddd��d� | � ttjddd��d� d S )NiA z psutil.AccessDenied (pid=321)r9 )r z)psutil.AccessDenied (pid=321, name='foo'))r: zpsutil.AccessDenied foo)r; r< r! r0 )r1 r2 r5 r5 r6 �test_access_denied__repr__� s z#TestMisc.test_access_denied__repr__c C sP | � tt�d��d� | � ttjddd��d� | � ttjdddd��d� d S ) NiA z/psutil.TimeoutExpired timeout after 321 seconds�o )r# z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)r9 )r# r zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))r; r< r! �TimeoutExpired)r1 r2 r5 r5 r6 � test_timeout_expired__repr__� s z%TestMisc.test_timeout_expired__repr__c C s> t �� }t �� }| �||� d|_| �||� | �|d� d S )N)r r r9 )r! r" r; �_identZassertNotEqual)r1 �p1�p2r5 r5 r6 �test_process__eq__� s zTestMisc.test_process__eq__c C s( t t�� t�� g�}| �t|�d� d S )Nr> )�setr! r" r; �len)r1 �sr5 r5 r6 �test_process__hash__� s z TestMisc.test_process__hash__c C s� t t�}x�|D ]~}|dkr q|�d�sy t|� W q tk r� |tjkr�tt|�}|d kr`w|jd k r�d|j�� kr�| � d| � Y qX qW xtjD ]}| � ||� q�W d S )N)�long�tests�test�PermissionError�ProcessLookupError�_Z deprecatedz%r not in psutil.__all__) �dirr! � startswith� __import__� ImportError�__all__�getattr�__doc__�lower�failr$ )r1 Z dir_psutilr �funr5 r5 r6 � test__all__� s" zTestMisc.test__all__c C s$ | � d�dd� tjD ��tj� d S )N�.c S s g | ] }t |��qS r5 )r% )�.0�xr5 r5 r6 � <listcomp>� s z)TestMisc.test_version.<locals>.<listcomp>)r; �joinr! � version_info� __version__)r1 r5 r5 r6 � test_version� s zTestMisc.test_versionc C s" t �� }d|_| �d|�� � d S )N�1r9 )r! r" r9 r' �as_dict)r1 r3 r5 r5 r6 �!test_process_as_dict_no_new_names� s z*TestMisc.test_process_as_dict_no_new_namesc s t � fdd��}g � x:td�D ].}|� }di f}| �||� | �t� �d� q W x<td�D ]0}|d�}di f}| �||� | �t� �d� qZW xDtd�D ]8}|ddd�}dddif}| �||� | �t� �d � q�W |�� |� }di f}| �||� | �t� �d � | �|jd � d S ) Nc s � � d� | |fS )z foo docstringN)�append)�args�kwargs)�callsr5 r6 r9 � s z"TestMisc.test_memoize.<locals>.foor r5 r> )r> )�barrm r � z foo docstring)r �ranger; rJ � cache_clearrY )r1 r9 r` �retZexpectedr5 )rl r6 � test_memoize� s. zTestMisc.test_memoizec s� G � fdd�d�}|� }g � |� � |� � | �t� �d� g � |j �|� |� � |� � | �t� �d� g � |j �|� |� � |� � | �t� �d� d S )Nc s e Zd Ze� fdd��ZdS )z1TestMisc.test_memoize_when_activated.<locals>.Fooc s � � d � d S )N)ri )r1 )rl r5 r6 r9 � s z5TestMisc.test_memoize_when_activated.<locals>.Foo.fooN)�__name__� __module__� __qualname__r r9 r5 )rl r5 r6 �Foo� s rv r r> )r9 r; rJ �cache_activate�cache_deactivate)r1 rv �fr5 )rl r6 �test_memoize_when_activated� s z$TestMisc.test_memoize_when_activatedc C s� ddl m} dd� }| �|d�|d�di� | �|d�|d�d|d �d i� | �|d �|d�d|d �d i� | �|d �|d�d|d �d i� | �|d�|d�di� | �|d�|d�di� | �|d�|d�di� d S )Nr )�parse_environ_blockc S s t r | �� S | S )N)r �upper)rK r5 r5 r6 �k s z,TestMisc.test_parse_environ_block.<locals>.kza=1 �arf 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{ r; )r1 r{ r} r5 r5 r6 �test_parse_environ_block s z!TestMisc.test_parse_environ_blockc C s> | � tj� t� r�t�d�� }d|_t�� t� r6t�W d Q R X t�� tjdtjd��}t� rdt�|j snt�W d Q R X t�� tjdtj d��"}t� r�t�t�� |j s�t�W d Q R X t�� tjdtj d��"}t� r�t�t�� |j s�t�W d Q R X n@| � t ��0 t�tj tj�}z|�d� W d |�� X W d Q R X d S )Nzpsutil._common.socketFz psutil._common.socket.socket)r z!psutil._common.socket.socket.bind)z::1r )Z addCleanupr rp r r* �has_ipv6�AssertionError�socket�error�called�gaierror� assertRaises� Exception�AF_INET6� SOCK_STREAM�bind�close)r1 rK �sockr5 r5 r6 �test_supports_ipv6 s8 zTestMisc.test_supports_ipv6c C s� ddl m} tj�t�}||�s$t�|tj�|��r8t�tj dt t j d�d�� | � t ||� W d Q R X tj dt t jd�d�� | � t ||� W d Q R X tj dt t jd�d�� ||�r�t�W d Q R X tj ddd�� ||�r�t�W d Q R X d S ) Nr )� isfile_strictzpsutil._common.os.statr9 )r zpsutil._common.stat.S_ISREGF)� return_value)r� r� r- �path�abspath�__file__r� �dirnamer r* �OSError�errno�EPERMr� �EACCES�ENOENT)r1 r� Z this_filer5 r5 r6 �test_isfile_strict= s zTestMisc.test_isfile_strictc s� � fdd�}|t �� �� � |t �� � |t �� � |t �� � |t jdd�� |t �� � trnt j � d�snnt s~|t � � � |t �� � |t �t �� �� |t �� � d S )Nc s< t d k rt �t �| �� t�| �}t�|�}� �| |� d S )N)�json�loads�dumps�pickler; )rq r~ r )r1 r5 r6 �checkO s z*TestMisc.test_serialization.<locals>.checkr )�intervalz/proc/diskstats)r! r" rg Zvirtual_memoryZ swap_memoryZ cpu_timesZcpu_times_percent�net_io_countersr r- r� �existsr �disk_io_countersZdisk_partitions� disk_usage�getcwd�users)r1 r� r5 )r1 r6 �test_serializationN s zTestMisc.test_serializationc C sT t j�td�}tr(t j�|�s(| �d�S t|�}| �t |j � | � |� � t j� d S )Nzsetup.pyzcan't find setup.py)r- r� rb r r r� �skipTestr r� � SystemExit�setupr; � get_versionr! rd )r1 Zsetup_py�moduler5 r5 r6 �test_setup_scripte s zTestMisc.test_setup_scriptc C s� t jjtjdtjd��}t�� |js*t�W d Q R X t jjtjdt�d�d��}t�� |jsbt�W d Q R X t jjtjdt d��.}| � t �� t�� W d Q R X |js�t�W d Q R X d S )NZ create_time)r r> ) r r* r+ r! r" r0 r� r� r, � ValueErrorr� )r1 �methr5 r5 r6 �test_ad_on_process_creationm s z$TestMisc.test_ad_on_process_creationc C sR t jddd��: | �t��}tt� W d Q R X | �dt|j�� � � W d Q R X d S )Nz psutil._psplatform.cext.versionz0.0.0)r� zversion conflict) r r* r� rV r r! r$ r% � exceptionrZ )r1 �cmr5 r5 r6 �test_sanity_version_check~ s z"TestMisc.test_sanity_version_checkN)rs rt ru r< r7 r8 r= r@ rA rD rH rL r] re rh rr rz r� r� r� r� r� r� r� r5 r5 r5 r6 r ; s( $ ##r r9 za b cc @ s� e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Ze�e d�d d � �Zd S ) �TestWrapNumbersc C s t �� d S )N)r rp )r1 r5 r5 r6 �setUp� s zTestWrapNumbers.setUpc C s& dt ddd�i}| �t|d�|� d S )N�disk1� �disk_io)�ntr; r )r1 �inputr5 r5 r6 �test_first_call� s z TestWrapNumbers.test_first_callc C s8 dt ddd�i}| �t|d�|� | �t|d�|� d S )Nr� r� r� )r� r; r )r1 r� r5 r5 r6 �test_input_hasnt_changed� s z(TestWrapNumbers.test_input_hasnt_changedc C s� dt ddd�i}| �t|d�|� dt ddd�i}| �t|d�|� dt ddd�i}| �t|d�|� dt ddd�i}| �t|d�|� d S ) Nr� r� r� � � � � � )r� r; r )r1 r� r5 r5 r6 �test_increase_but_no_wrap� s z)TestWrapNumbers.test_increase_but_no_wrapc C s� dt ddd�i}| �t|d�|� dt ddd�i}| �t|d�dt ddd�i� dt ddd�i}| �t|d�dt ddd�i� dt ddd�i}| �t|d�dt ddd�i� dt ddd�i}| �t|d�dt ddd �i� dt ddd�i}| �t|d�dt ddd �i� dt d dd�i}| �t|d�dt d dd �i� dt d dd�i}| �t|d�dt ddd �i� dt d dd�i}| �t|d�dt ddd �i� d S ) Nr� �d r� r� �n �Z � r� �� �2 � �( )r� r; r )r1 r� r5 r5 r6 � test_wrap� s4 zTestWrapNumbers.test_wrapc C st dt ddd�i}| �t|d�|� t ddd�t ddd�d�}| �t|d�|� dt ddd�i}| �t|d�|� d S )Nr� r� r� � )r� �disk2� )r� r; r )r1 r� r5 r5 r6 �test_changing_keys� s z"TestWrapNumbers.test_changing_keysc C s. t ddd�t ddd�d�}| �t|d�|� t ddd�t ddd�d�}| �t|d�t ddd�t ddd�d�� dt ddd�i}| �t|d�|� t ddd�t ddd�d�}| �t|d�|� t ddd�t ddd�d�}| �t|d�|� t ddd�t ddd�d�}| �t|d�t ddd�t ddd�d�� d S )Nr� r� )r� r� r� r� r� r� )r� r; r )r1 r� r5 r5 r6 �test_changing_keys_w_wrap� s* z)TestWrapNumbers.test_changing_keys_w_wrapc C sb ddddd�}| � t|d�|� | � t|d�|� ddddd�}t|d�}| � |d d d � d S ) N) i, i� i� i# iR i� i i�� iȷ ) i� r i tU i i r r r r� ) �6 r� i �$ i �N rn � r� i� � ) iU i� i" r� i� i4 i� i� i\ )�nvme0n1Z nvme0n1p1Z nvme0n1p2Z nvme0n1p3r� ) r� i� i� i# iR i� i i�� iȷ r� r i� )r; r )r1 �d�outr5 r5 r6 �test_real_data� s z TestWrapNumbers.test_real_datac C sb dt ddd�i}t|d� t�� }| �|d d|i� | �|d di i� | �|d di i� d S )Nr� r� r� r r> r )r� r � cache_infor; )r1 r� �cacher5 r5 r6 �test_cache_first_call s z%TestWrapNumbers.test_cache_first_callc C s� dt ddd�i}t|d� dt ddd�i}t|d� t�� }| �|d d|i� | �|d ddddd�i� | �|d di i� d S ) Nr� r� r� r� r r> ))r� r )r� r> )r� r r )r� r r� r; )r1 r� r� r5 r5 r6 �test_cache_call_twice s z%TestWrapNumbers.test_cache_call_twicec s� dt ddd�i}t|d� dt ddd�i}t|d� t�� }� �|d d|i� � �|d ddddd�i� � �|d ddtd g�ii� � fd d �}dt ddd�i}t|d� t�� }� �|d d|i� |� dt ddd �i}t|d� t�� }� �|d d|i� |� dt ddd �i}t|d� t�� }� �|d d|i� � �|d ddddd�i� � �|d ddtd g�ii� d S )Nr� r� r� r� r r> ))r� r )r� r> )r� r r )r� r c sF t �� } � �| d ddddd�i� � �| d ddtdg�ii� d S ) Nr> r� r r� ))r� r )r� r> )r� r r r� )r� r )r r� r; rI )r� )r1 r5 r6 �assert_, s z0TestWrapNumbers.test_cache_wrap.<locals>.assert_r� r� r� )r� r r� r; rI )r1 r� r� r� r5 )r1 r6 �test_cache_wrap s: z TestWrapNumbers.test_cache_wrapc C s� dt ddd�i}t|d� t ddd�t ddd�d�}t|d� t�� }| �|d d|i� | �|d ddddd�i� | �|d di i� d S ) Nr� r� r� r� )r� r� r r> ))r� r )r� r> )r� r r )r� r r� r; )r1 r� r� r5 r5 r6 �test_cache_changing_keysM s z(TestWrapNumbers.test_cache_changing_keysc C s\ dt ddd�i}t|d� t|d� t�d� | �t�� i i i f� t�d� t�d� d S )Nr� r� r� z?!?)r� r rp r; r� )r1 r� r5 r5 r6 �test_cache_clearZ s z TestWrapNumbers.test_cache_clearz not supportedc C s� t �� rt �� s| �d�S t �� t �� t�� }x$|D ] }| �d|� | �d|� q8W t j�� t�� }x$|D ] }| �d|� | �d|� qpW t j�� t�� }| � |i i i f� d S )Nzno disks or NICs availablezpsutil.disk_io_counterszpsutil.net_io_counters) r! r� r� r� r r� r$ rp r' r; )r1 Zcachesr� r5 r5 r6 � test_cache_clear_public_apisc s z,TestWrapNumbers.test_cache_clear_public_apisN)rs rt ru r� ZtearDownr� r� r� r� r� r� r� r� r� r� r� r� r �skipIfr r� r5 r5 r5 r6 r� � s % 0 r� z can't locate scripts directoryc @ s~ e Zd ZdZedd� �Zed<dd��Zdd� Ze� e d �d d � �Z d d � Z dd� Z dd� Zdd� Ze� eote�� d�dd� �Zdd� Zdd� Zdd � Zd d � Ze� e d �d d!� �Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z d,d-� Z e� e d.�d/d0� �Z!d1d2� Z"e� e# d �d3d4� �Z$e� e% d �d5d6� �Z&e� e' d �e� e( d7�d8d9� ��Z)e� e' d �e� e( d7�d:d;� ��Z*dS )=� TestScriptsz-Tests for scripts in the "scripts" directory.c O s� dt j�t| � } t| g}x|D ]}|�|� q W yt|f|��� }W n8 tk r~ } zdt |�krlt |�S � W d d }~X Y nX |s�t |��|S )Nz%sr0 ) r- r� rb r r ri r �strip� RuntimeErrorr% r� )�exerj rk �cmd�argr� �errr5 r5 r6 � assert_stdout� s zTestScripts.assert_stdoutNc C sR t j�t| �} tr"t| ddd�}n t| d�}|� |�� }W d Q R X t�|� d S )N�rt�utf8)�encoding) r- r� rb r r �open�read�ast�parse)r� rj ry �srcr5 r5 r6 � assert_syntax� s zTestScripts.assert_syntaxc C sZ t | �}xLt�t�D ]>}|�d�rdtj�|�d |kr| �dtj�t|� � qW d S )Nz.pyZtest_r z no test defined for %r script) rS r- �listdirr �endswithr� �splitextr[ rb )r1 Zmethsr r5 r5 r6 � test_coverage� s zTestScripts.test_coveragez POSIX onlyc C sT xNt �t�D ]@}|�d�r t j�t|�}tjt �|�tj @ s | � d| � q W d S )Nz.pyz%r is not executable) r- r� r r� r� rb �stat�S_IXUSR�ST_MODEr[ )r1 r r� r5 r5 r6 �test_executable� s zTestScripts.test_executablec C s | � d� d S )Nz disk_usage.py)r� )r1 r5 r5 r6 �test_disk_usage� s zTestScripts.test_disk_usagec C s | � d� d S )Nzfree.py)r� )r1 r5 r5 r6 � test_free� s zTestScripts.test_freec C s | � d� d S )Nz meminfo.py)r� )r1 r5 r5 r6 � test_meminfo� s zTestScripts.test_meminfoc C s | � dtt�� �� d S )Nz procinfo.py)r� r% r- r. )r1 r5 r5 r6 � test_procinfo� s zTestScripts.test_procinfozno usersc C s | � d� d S )Nzwho.py)r� )r1 r5 r5 r6 �test_who� s zTestScripts.test_whoc C s | � d� d S )Nzps.py)r� )r1 r5 r5 r6 �test_ps� s zTestScripts.test_psc C s | � d� d S )Nz pstree.py)r� )r1 r5 r5 r6 � test_pstree� s zTestScripts.test_pstreec C s | � d� d S )Nz netstat.py)r� )r1 r5 r5 r6 � test_netstat� s zTestScripts.test_netstatc C s | � d� d S )Nz ifconfig.py)r� )r1 r5 r5 r6 � test_ifconfig� s zTestScripts.test_ifconfigz not supportedc C s | � dtt�� �� d S )Nzpmap.py)r� r% r- r. )r1 r5 r5 r6 � test_pmap� s zTestScripts.test_pmapc C s* dt �� �� jkr | �d��| �d� d S )NZussz not supportedz procsmem.py)r! r" Zmemory_full_info�_fieldsr� r� )r1 r5 r5 r6 � test_procsmem� s zTestScripts.test_procsmemc C s | � d� d S )Nz killall.py)r� )r1 r5 r5 r6 � test_killall� s zTestScripts.test_killallc C s | � d� d S )Nz nettop.py)r� )r1 r5 r5 r6 � test_nettop� s zTestScripts.test_nettopc C s | � d� d S )Nztop.py)r� )r1 r5 r5 r6 �test_top� s zTestScripts.test_topc C s | � d� d S )Nziotop.py)r� )r1 r5 r5 r6 � test_iotop� s zTestScripts.test_iotopc C s, | � dt�� �� �}| �tt�� �|� d S )Nzpidof.py)r� r! r" r r$ r% r- r. )r1 �outputr5 r5 r6 � test_pidof� s zTestScripts.test_pidofz WINDOWS onlyc C s | � d� d S )Nzwinservices.py)r� )r1 r5 r5 r6 �test_winservices� s z TestScripts.test_winservicesc C s | � d� d S )Nzcpu_distribution.py)r� )r1 r5 r5 r6 �test_cpu_distribution� s z!TestScripts.test_cpu_distributionc C s t �� s| �d� | �d� d S )Nzno temperaturesztemperatures.py)r! Zsensors_temperaturesr� r� )r1 r5 r5 r6 �test_temperatures� s z TestScripts.test_temperaturesc C s t �� s| �d� | �d� d S )Nzno fanszfans.py)r! Z sensors_fansr� r� )r1 r5 r5 r6 � test_fans� s zTestScripts.test_fansz no batteryc C s | � d� d S )Nz battery.py)r� )r1 r5 r5 r6 � test_battery� s zTestScripts.test_batteryc C s | � d� d S )Nz sensors.py)r� )r1 r5 r5 r6 � test_sensors� s zTestScripts.test_sensors)N)+rs rt ru rY � staticmethodr� 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 r5 r5 r5 r6 r� ~ s< r� �__main__)� run_from_name)5rY r� � collectionsr� r� r- r� r� r� �sysr! 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 rc Z PYTHON_39r � namedtupler� r� r� r� r� r� rs Zpsutil.tests.runnerr r� r5 r5 r5 r6 �<module> sd S p
Save File
Cancel