File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_unicode.cpython-39.pyc
a
��?h;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 s6 ddl m} z
|| �W S ty0 t�� Y n0 d S )Nr r )� psutil.testsr Z WindowsError� traceback� print_exc)�pathZrm� r$ �E/usr/local/lib64/python3.9/site-packages/psutil/tests/test_unicode.pyr q s
r c C s� d}t | d�}z�z<t|� t|� t|gd�}t�||d � t|d � W n2 ttfy~ Y W |durrt|� t|� dS 0 W |dur�t|� t|� dS W |dur�t|� t|� n|dur�t|� t|� 0 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 �shutil�copyfile�UnicodeEncodeError�IOErrorr )r' �sprocZtestfnr$ r$ r% � try_unicode� s2
��
�r. c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sN t � �� d| _d | _| jd urJt| 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 sZ t | jt�rdnd}t�� �* t�d� | jt�|�v W d � S 1 sL0 Y 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�s8J �| �� r^t j
� |�t j
� | j �ks^J �d S �Nz-cz2import time; [time.sleep(0.1) for x in range(100)]) r2 r �psutil�Process�pid�exerG rH rO 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�s4J �| �� rR|t j
� | j �ksRJ �d S rP ) r2 r rQ rR rS rA rG rH rO rL r# �basename)r: r( rV rA r$ r$ r% �test_proc_name� s �
zTestFSAPIs.test_proc_namec C sZ | j ddg}| �|�}t�|j�}|�� }|D ]}t|t�s.J �q.| �� rV||ksVJ �d S rP ) r2 r rQ rR rS �cmdlinerG rH rO )r: r( rV rW r[ �partr$ r$ r% �test_proc_cmdline� s �
z TestFSAPIs.test_proc_cmdlinec C s� | j d }| �t|� t|� t|�� t�� }|�� }W d � n1 sL0 Y t|�� t �shJ �| �
� r|||ks|J �d S �N�2) r2 �
addCleanupr r r rQ rR �cwdrG rH rO )r: �dnamerW ra r$ r$ r% �
test_proc_cwd� s
&zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSrB c C s� t �� }t|�� �}t| jd�� t|�� �}W d � n1 sB0 Y || �� j}t|t �shJ �t
rz|szt � d��| �
� r�tj�|�tj�| j�ks�J �d S )N�rbzopen_files on BSD is broken)rQ rR �setZ
open_files�openr2 �popr# rG rH r r r8 rO rL rU )r: rW �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$ ty> tr0� n
t�d��Y n0 t|��@ t� � �
d�d }t |j t
�slJ �|j |kszJ �W d � n1 s�0 Y d S )Nr& �
not supported�unixr )r r3 r r+ r r r8 r rQ rR �net_connectionsrG �laddrrH )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$ tyF tr8� n
t�d��Y n0 t|��B tj dd�}||�}t
|j t �svJ �|j |ks�J �W d � n1 s�0 Y d S )Nc S s2 | D ] }t j�|j��t�r| S qtd��d S )Nzconnection not found)rL r# rY rn �
startswithr �
ValueError)�consrp r$ r$ r% � find_sock s
z2TestFSAPIs.test_net_connections.<locals>.find_sockr& rk rl )�kind)
r r3 r r+ r r r8 r rQ rm rG rn rH )r: ru rA ro rt rp r$ r$ r% �test_net_connections s
z TestFSAPIs.test_net_connectionsc C s, | j d }| �t|� t|� t�|� d S r^ )r2 r` r r rQ �
disk_usage)r: rb r$ r$ r% �test_disk_usage/ s
zTestFSAPIs.test_disk_usagerk z&ctypes does not support unicode on PY2zunstable on PYPYc s� t | jd��h}dd� � � fdd�t�� �� D �}dd� |D �}� |�|v sNJ �|D ]}t|t�sRJ �qRW d � n1 sz0 Y d S )Nr& c S s t j�t j�| ��S )N)rL r# �realpathrU )rW 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 |v r|�qS r$ r r| r$ r$ r% r� F � )r r3 rQ rR Z memory_mapsrG rH )r: Z
funky_pathZlibpathsr# r$ r r% �test_memory_maps5 s
�zTestFSAPIs.test_memory_mapsN)r; r< r= �__doc__r r3 rO rX rZ r] rc r �mark�skipifr r rj r rq r
r rw ry r r r� r$ r$ r$ r% rD � s0
��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.� èrk rB zsegfaults on PYPY + WINDOWSc C sx t j�� }| j|d< | j|d�}t�|j�}|�� }|�� D ]$\}}t |t
�sRJ �t |t
�s<J �q<|d | jkstJ �d S )NZ FUNNY_ARG)�env) rL �environ�copyr3 r rQ rR rS �itemsrG rH )r: r� r- rW �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