File: //opt/osm/venv/lib/python3.10/site-packages/psutil/tests/__pycache__/test_unicode.cpython-310.pyc
o
=i�) � @ s� d Z ddlZddlZddlZddlmZ ddlZddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z dd lm Z dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
m Z ddl
m Z ddl
m Z ddl
m Z dd l
m Z dd l
m!Z! dd l
m"Z" d d � Z#G d!d"� d"e�Z$e j%j&d#d$�e j%j'ed%d&�G d'd(� d(e$���Z(e j%j'ed)d&�G d*d+� d+e(��Z)G d,d-� d-e$�Z*dS ).aY 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:
* sys.getfilesystemencodeerrors() or "surrogatescape" on POSIX and
"replace" on Windows.
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��MACOS)�NETBSD)�OPENBSD)�POSIX)�WINDOWS)�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_subproc)� terminatec C s� d}t | d�}zGz t|� t|� t|gd�}t�||d � t|d � W n ttfy? Y W |dur9t|� t|� dS w W |durIt|� t|� dS |durWt|� t|� w )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�OSErrorr )r! �sprocZtestfn� r( �G/opt/osm/venv/lib/python3.10/site-packages/psutil/tests/test_unicode.py� try_unicodeh s*
� �
r* c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sT t � �� d| _d | _| jd ur(t| j�sd| _d S t| jd�| _t| j� d S d S )NFTr )�super�
setUpClass�
skip_tests�
funky_name� funky_suffixr* r r )�cls�� __class__r( r) r- � s
�zBaseUnicodeTest.setUpClassc s t � �� | jr
t�d�S d S )Nzcan't handle unicode str)r, �setUpr. r �skip��selfr2 r( r) r4 � s
�zBaseUnicodeTest.setUp)�__name__�
__module__� __qualname__r0 � classmethodr- r4 �
__classcell__r( r( r2 r) r+ � s
r+ �serial)�namezASCII fs��reasonc @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d
d � Z
e j j
eo"ed d
�e j j
ep+edd
�dd� ��Ze j j
e dd
�e j j
e dd
�dd� ��Ze j j
e dd
�e j j
e dd
�e� dd� ���Zdd� Ze j j
e dd
�dd� �Zd S ) �
TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c C sF t �� � t �d� | jt�d�v W d � S 1 s w Y d S )N�ignore�.)�warnings�catch_warnings� simplefilterr/ �os�listdirr6 r( r( r) �expect_exact_path_match� s
$�z"TestFSAPIs.expect_exact_path_matchc C sf | j ddg}| �|�}t�|j�}|�� }t|t�s J �| �� r/t j
� |�t j
� | j �ks1J �d S d S �Nz-cz2import time; [time.sleep(0.1) for x in range(100)]) r/ r �psutil�Process�pid�exe�
isinstance�strrI rG �path�normcase)r7 r" �subp�prN r( r( r) �
test_proc_exe� s �
�zTestFSAPIs.test_proc_exec C sZ | j ddg}| �|�}t�|j��� }t|t�sJ �| �� r)|t j
� | j �ks+J �d S d S rJ ) r/ r rK rL rM r> rO rP rI rG rQ �basename)r7 r" rS r> r( r( r) �test_proc_name� s �
�zTestFSAPIs.test_proc_namec C s^ | j ddg}| �|�}t�|j�}|�� }|D ] }t|t�s J �q| �� r+||ks-J �d S d S rJ ) r/ r rK rL rM �cmdlinerO rP rI )r7 r" rS rT rX �partr( r( r) �test_proc_cmdline� s �
�z TestFSAPIs.test_proc_cmdlinec C s� | j d }| �t|� t|� t|�� t�� }|�� }W d � n1 s&w Y t|�� t �s4J �| �
� r>||ks@J �d S d S �N�2) r/ �
addCleanupr r r rK rL �cwdrO rP rI )r7 �dnamerT r^ r( r( r) �
test_proc_cwd� s
� �zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSr? zbroken on NETBSD or OPENBSDc C s� t �� }t|�� �}t| jd�� t|�� �}W d � n1 s!w Y || �� j}t|t �s4J �t
r=|s=t � d�S | �
� rPtj�|�tj�| j�ksRJ �d S d S )N�rbzopen_files on BSD is broken)rK rL �setZ
open_files�openr/ �poprQ rO rP r r r5 rI rG rR )r7 rT �start�newrQ r( r( r) �test_proc_open_files� s �
�z TestFSAPIs.test_proc_open_filesz
POSIX onlyzcan't list UNIX socketsc C s� | j | jd�}t|�}t|��4 t�� �d�d }t|jt �s!J �|js4t
r4t r4t �
d�W d � S |j|ks;J �W d � d S 1 sFw Y d S )Nr �unixr zunreliable on OSX)r r0 r r rK rL �net_connectionsrO �laddrrP r r r r5 )r7 r> �sock�connr( r( r) �test_proc_net_connections� s
�"�z$TestFSAPIs.test_proc_net_connectionsc C s~ dd� }| j | jd�}t|�}t|��" tjdd�}||�}t|jt�s&J �|j|ks-J �W d � d S 1 s8w Y d S )Nc S s. | D ]}t j�|j��t�r| S qtd��)Nzconnection not found)rG rQ rV rj �
startswithr �
ValueError)�consrl r( r( r) � find_sock� s
�z2TestFSAPIs.test_net_connections.<locals>.find_sockr rh )�kind) r r0 r r rK ri rO rj rP )r7 rq r> rk rp rl r( r( r) �test_net_connections� s
"�z TestFSAPIs.test_net_connectionsc C s, | j d }| �t|� t|� t�|� d S r[ )r/ r] r r rK �
disk_usage)r7 r_ r( r( r) �test_disk_usage s
zTestFSAPIs.test_disk_usage�
not supportedc s� t | jd��5}dd� � � fdd�t�� �� D �}dd� |D �}� |�|v s'J �|D ] }t|t�s2J �q)W d � d S 1 s>w Y d S )Nr c S s t j�t j�| ��S �N)rG rQ �realpathrR )rT r( r( r) �normpath s z-TestFSAPIs.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r( )rQ ��.0�x�ry r( r) �
<listcomp> s
�z/TestFSAPIs.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |v r|�qS r( r rz r( r( r) r~ s )r r0 rK rL Z memory_mapsrO rP )r7 Z
funky_pathZlibpathsrQ r( r} r) �test_memory_maps s
��"�zTestFSAPIs.test_memory_mapsN)r8 r9 r: �__doc__r r0 rI rU rW rZ r` r �mark�skipifr r r r rg r r rm r rs ru r
r r( r( r( r) rA � s6
�
�
�rA zunreliable on CIc @ s e Zd ZdZeZdd� ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s t S rw r r6 r( r( r) rI ( s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r8 r9 r: r� r r0 rI r( r( r( r) r� " s r� c @ sB e Zd ZdZeZejje dd�ejje
oe dd�dd� ��Z dS )�
TestNonFSAPISz&Unicode tests for non fs-related APIs.rv r? zsegfaults on PYPY + WINDOWSc C sx t j�� }| j|d< | j|d�}t�|j�}|�� }|�� D ]\}}t |t
�s)J �t |t
�s0J �q |d | jks:J �d S )NZ FUNNY_ARG)�env) rG �environ�copyr0 r rK rL rM �itemsrO rP )r7 r� r' rT �k�vr( r( r) �test_proc_environ6 s
z TestNonFSAPIS.test_proc_environN)
r8 r9 r: r� r r0 r r� r� r r r r� r( r( r( r) r� 1 s r� )+r� rG r# rD �
contextlibr rK r r r r r r 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� Z xdist_groupr� rA r� r� r( r( r( r) �<module> sT >