관리-도구
편집 파일: test_unicode.cpython-38.pyc
U �Mg;1 � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm Z ddl m Z ddl m Z dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ e�r�d d � Z!d!d"� Z%G d#d$� d$e�Z&e j'j(d%d&�e j'j)ed'd(�e j'j)e�o�e d)d(�G d*d+� d+e&����Z*e j'j)ed,d(�G d-d.� d.e*��Z+G d/d0� d0e&�Z,dS )1a> Notes about unicode handling in psutil ======================================. Starting from version 5.3.0 psutil adds unicode support, see: https://github.com/giampaolo/psutil/issues/1040 The notes below apply to *any* API returning a string such as process exe(), cwd() or username(): * all strings are encoded by using the OS filesystem encoding (sys.getfilesystemencoding()) which varies depending on the platform (e.g. "UTF-8" on macOS, "mbcs" on Win) * no API call is supposed to crash with UnicodeDecodeError * instead, in case of badly encoded data returned by the OS, the following error handlers are used to replace the corrupted characters in the string: * Python 3: sys.getfilesystemencodeerrors() (PY 3.6+) or "surrogatescape" on POSIX and "replace" on Windows * Python 2: "replace" * on Python 2 all APIs return bytes (str type), never unicode * on Python 2, you can go back to unicode by doing: >>> unicode(p.exe(), sys.getdefaultencoding(), errors="replace") For a detailed explanation of how psutil handles unicode see #1040. Tests ===== List of APIs returning or dealing with a string: ('not tested' means they are not tested to deal with non-ASCII strings): * Process.cmdline() * Process.cwd() * Process.environ() * Process.exe() * Process.memory_maps() * Process.name() * Process.net_connections('unix') * Process.open_files() * Process.username() (not tested) * disk_io_counters() (not tested) * disk_partitions() (not tested) * disk_usage(str) * net_connections('unix') * net_if_addrs() (not tested) * net_if_stats() (not tested) * net_io_counters() (not tested) * sensors_fans() (not tested) * sensors_temperatures() (not tested) * users() (not tested) * WindowsService.binpath() (not tested) * WindowsService.description() (not tested) * WindowsService.display_name() (not tested) * WindowsService.name() (not tested) * WindowsService.status() (not tested) * WindowsService.username() (not tested) In here we create a unicode path with a funky non-ASCII name and (where possible) make psutil return it back (e.g. on name(), exe(), open_files(), etc.) and make sure that: * psutil never crashes with UnicodeDecodeError * the returned path matches � N)�closing)�BSD)�POSIX)�WINDOWS)�PY3)�super)�APPVEYOR)�ASCII_FS)� CI_TESTING)� HAS_ENVIRON)�HAS_MEMORY_MAPS)�HAS_NET_CONNECTIONS_UNIX)�INVALID_UNICODE_SUFFIX)�PYPY�� TESTFN_PREFIX)�UNICODE_SUFFIX)�PsutilTestCase)�bind_unix_socket)�chdir)�copyload_shared_lib)� create_py_exe)� get_testfn)�pytest)� safe_mkdir�� safe_rmpath)�skip_on_access_denied)�spawn_testproc)� terminatec C s8 ddl m} z || �W S tk r2 t�� Y nX d S )Nr r )� psutil.testsr Z WindowsError� traceback� print_exc)�pathZrm� r$ �I/opt/hc_python/lib64/python3.8/site-packages/psutil/tests/test_unicode.pyr q s r c C s� d}t | d�}zhz<t|� t|� t|gd�}t�||d � t|d � W n ttfk rj Y W �dS X W �dS W 5 |dk r�t|� t|� X dS )z`Return True if both the fs and the subprocess module can deal with a unicode file name. N��suffix)�cmdz-2FT) r r r r r �shutil�copyfile�UnicodeEncodeError�IOError)r' �sprocZtestfnr$ r$ r% � try_unicode� s r. c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sN t � �� d| _d | _| jd k rJt| j�s2d| _nt| jd�| _t| j� d S )NFTr& )r � setUpClass� skip_tests� funky_name� funky_suffixr. r r )�cls�� __class__r$ r% r0 � s zBaseUnicodeTest.setUpClassc s t � �� | jrt�d��d S )Nzcan't handle unicode str)r �setUpr1 r �skip��selfr5 r$ r% r7 � s zBaseUnicodeTest.setUp)�__name__� __module__� __qualname__r3 � classmethodr0 r7 � __classcell__r$ r$ r5 r% r/ � s r/ �serial)�namezASCII fs��reasonztoo much trouble on PYPY2c @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z e j j eoDed d �dd� �Ze j j e dd �dd� �Ze j j e dd �e j j e dd �e� dd� ���Zdd� Ze j j e dd �e j j e dd �e j j edd �dd � ���Zd S ) � TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c C sN t | jt�rdnd}t�� �( t�d� | jt�|�kW 5 Q R � S Q R X d S )N�.�ignore)� isinstancer2 �str�warnings�catch_warnings� simplefilter�os�listdir)r: �herer$ r$ r% �expect_exact_path_match� s z"TestFSAPIs.expect_exact_path_matchc C sb | j ddg}| �|�}t�|j�}|�� }t|t�s8t�| � � r^t j � |�t j � | j �ks^t�d S �Nz-cz2import time; [time.sleep(0.1) for x in range(100)]) r2 r �psutil�Process�pid�exerG rH �AssertionErrorrO rL r# �normcase)r: r( �subp�prT r$ r$ r% � test_proc_exe� s � zTestFSAPIs.test_proc_exec C sV | j ddg}| �|�}t�|j��� }t|t�s4t�| � � rR|t j � | j �ksRt�d S rP ) r2 r rQ rR rS rA rG rH rU rO rL r# �basename)r: r( rW rA r$ r$ r% �test_proc_name� s � zTestFSAPIs.test_proc_namec C sZ | j ddg}| �|�}t�|j�}|�� }|D ]}t|t�s.t�q.| � � rV||ksVt�d S rP ) r2 r rQ rR rS �cmdlinerG rH rU rO )r: r( rW rX r\ �partr$ r$ r% �test_proc_cmdline� s � z TestFSAPIs.test_proc_cmdlinec C sl | j d }| �t|� t|� t|�� t�� }|�� }W 5 Q R X t|�� t �sTt �| � � rh||ksht �d S �N�2) r2 � addCleanupr r r rQ rR �cwdrG rH rU rO )r: �dnamerX rb r$ r$ r% � test_proc_cwd� s zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSrB c C s� t �� }t|�� �}t| jd�� t|�� �}W 5 Q R X || �� j}t|t �sTt �t rf|sft � d��| �� r�tj�|�tj�| j�ks�t �d S )N�rbzopen_files on BSD is broken)rQ rR �setZ open_files�openr2 �popr# rG rH rU r r r8 rO rL rV )r: rX �start�newr# r$ r$ r% �test_proc_open_files� s z TestFSAPIs.test_proc_open_filesz POSIX onlyc C s� | j | jd�}z t|�}W n& tk r@ tr2� n t�d��Y nX t|��6 t� � � d�d }t |j t �snt�|j |ks|t�W 5 Q R X d S )Nr& � not supported�unixr )r r3 r r+ r r r8 r rQ rR �net_connectionsrG �laddrrH rU )r: rA �sock�connr$ r$ r% �test_proc_net_connections s z$TestFSAPIs.test_proc_net_connectionszcan't list UNIX socketsc C s� dd� }| j | jd�}z t|�}W n& tk rH tr:� n t�d��Y nX t|��8 tj dd�}||�}t |j t �sxt �|j |ks�t �W 5 Q R X d S )Nc S s2 | D ] }t j�|j��t�r| S qtd��d S )Nzconnection not found)rL r# rZ ro � startswithr � ValueError)�consrq r$ r$ r% � find_sock s z2TestFSAPIs.test_net_connections.<locals>.find_sockr& rl rm )�kind)r r3 r r+ r r r8 r rQ rn rG ro rH rU )r: rv rA rp ru rq r$ r$ r% �test_net_connections s z TestFSAPIs.test_net_connectionsc C s, | j d }| �t|� t|� t�|� d S r_ )r2 ra r r rQ � disk_usage)r: rc r$ r$ r% �test_disk_usage/ s zTestFSAPIs.test_disk_usagerl z&ctypes does not support unicode on PY2zunstable on PYPYc st t | jd��^}dd� � � fdd�t�� �� D �}dd� |D �}� |�|ksNt�|D ]}t|t�sRt�qRW 5 Q R X d S )Nr& c S s t j�t j�| ��S )N)rL r# �realpathrV )rX r$ r$ r% �normpath? s z-TestFSAPIs.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r$ )r# ��.0�x�r| r$ r% � <listcomp>B s z/TestFSAPIs.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |kr|�qS r$ r r} r$ r$ r% r� F s )r r3 rQ rR Z memory_mapsrU rG rH )r: Z funky_pathZlibpathsr# r$ r� r% �test_memory_maps5 s �zTestFSAPIs.test_memory_mapsN)r; r< r= �__doc__r r3 rO rY r[ r^ rd r �mark�skipifr r rk r rr r r rx rz r r r� r$ r$ r$ r% rD � s4 � �rD zunreliable on CIc @ s e Zd ZdZeZdd� ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr$ r9 r$ r$ r% rO R s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r; r< r= r� r r3 rO r$ r$ r$ r% r� L s r� c @ sJ e Zd ZdZerendZejj e dd�ejj e o4e dd�dd� ��Z dS ) � TestNonFSAPISz&Unicode tests for non fs-related APIs.� èrl rB zsegfaults on PYPY + WINDOWSc C sx t j�� }| j|d< | j|d�}t�|j�}|�� }|�� D ]$\}}t |t �sRt �t |t �s<t �q<|d | jkstt �d S )NZ FUNNY_ARG)�env) rL �environ�copyr3 r rQ rR rS �itemsrG rH rU )r: r� r- rX �k�vr$ r$ r% �test_proc_environa s z TestNonFSAPIS.test_proc_environN)r; r< r= r� r r r3 r r� r� r r r r� r$ r$ r$ r% r� \ s r� )-r� rL r) r! rI � contextlibr rQ r r r Zpsutil._compatr 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� Z xdist_groupr� rD r� r� r$ r$ r$ r% �<module> sZ C