Advanced Web Server Manager
Complete File Manager & Terminal - Standalone Version
By Sid Gifari | Gifari Industries
Current path:
/
/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
psutil
/
tests
/
__pycache__
✏️
Editing: test_unicode.cpython-36.pyc
3 �̪e1 � @ s8 d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlm Z ddlm Z ddlm Z ddl m Z ddl mZ dd l mZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" dd lm#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#ej)ed&�ej)e�o�e d'�G d(d)� d)e(����Z*ej)ed*�G d+d,� d,e*��Z+G d-d.� d.e(�Z,e-d/k�r4dd0l.m/Z/ e/e0� 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.connections('unix') * Process.cwd() * Process.environ() * Process.exe() * Process.memory_maps() * Process.name() * 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)�u)�APPVEYOR)�ASCII_FS)� CI_TESTING)�HAS_CONNECTIONS_UNIX)� HAS_ENVIRON)�HAS_MEMORY_MAPS)�INVALID_UNICODE_SUFFIX)�PYPY)� TESTFN_PREFIX)�UNICODE_SUFFIX)�PsutilTestCase)�bind_unix_socket)�chdir)�copyload_shared_lib)� create_py_exe)� get_testfn)� safe_mkdir)� safe_rmpath)� serialrun)�skip_on_access_denied)�spawn_testproc)� terminatec C s6 ddl m} y|| �S tk r0 tj� Y nX d S )Nr )r )� psutil.testsr Z WindowsError� traceback� print_exc)�pathZrm� r# �E/usr/local/lib64/python3.6/site-packages/psutil/tests/test_unicode.pyr s s r c C s� d}t | d�}z^y<t|� t|� t|gd�}tj||d � t|d � W n ttfk rd dS X dS W d|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 �shutil�copyfile�UnicodeEncodeError�IOErrorr )r% �sprocZtestfnr# r# r$ � try_unicode� s r, c s0 e Zd ZdZe� fdd��Z� fdd�Z� ZS )�BaseUnicodeTestNc sN t � j� d| _d | _| jd k rJt| j�s2d| _nt| jd�| _t| j� d S )NFT)r% )r � setUpClass� skip_tests� funky_name� funky_suffixr, r r )�cls)� __class__r# r$ r. � s zBaseUnicodeTest.setUpClassc s t � j� | jr| jd��d S )Nzcan't handle unicode str)r �setUpr/ �skipTest)�self)r3 r# r$ r4 � s zBaseUnicodeTest.setUp)�__name__� __module__� __qualname__r1 � classmethodr. r4 � __classcell__r# r# )r3 r$ r- � s r- zASCII fsztoo much trouble on PYPY2c @ s� e Zd ZdZeZdd� Zdd� Zdd� Zdd � Z d d � Z e j e oBed �d d� �Ze j e d�dd� �Ze j e d�e j e d�e� dd� ���Zdd� Ze j e d�e j e d�e j e d�dd� ���Zd S ) � TestFSAPIsz1Test FS APIs with a funky, valid, UTF8 path name.c C sF t | jt�rdntd�}tj� � tjd� | jtj|�kS Q R X d S )N�.�ignore) � isinstancer0 �strr �warnings�catch_warnings� simplefilter�os�listdir)r6 �herer# r# r$ �expect_exact_path_match� s z"TestFSAPIs.expect_exact_path_matchc C s` | j ddg}| j|�}tj|j�}|j� }| j|t� | j� r\| j t j j |�t j j | j �� d S )Nz-czimport time; time.sleep(10)) r0 r �psutil�Process�pid�exe�assertIsInstancer@ rG � assertEqualrD r" �normcase)r6 r& �subp�prK r# r# r$ � test_proc_exe� s zTestFSAPIs.test_proc_exec C sT | j ddg}| j|�}tj|j�j� }| j|t� | j� rP| j |t j j | j �� d S )Nz-czimport time; time.sleep(10)) r0 r rH rI rJ �namerL r@ rG rM rD r" �basename)r6 r& rO rR r# r# r$ �test_proc_name� s zTestFSAPIs.test_proc_namec C s\ | j ddg}| j|�}tj|j�}|j� }x|D ]}| j|t� q0W | j� rX| j ||� d S )Nz-czimport time; time.sleep(10)) r0 r rH rI rJ �cmdlinerL r@ rG rM )r6 r& rO rP rU �partr# r# r$ �test_proc_cmdline� s z TestFSAPIs.test_proc_cmdlinec C sj | j d }| jt|� t|� t|�� tj� }|j� }W d Q R X | j|j� t � | j � rf| j ||� d S )N�2) r0 � addCleanupr r r rH rI �cwdrL r@ rG rM )r6 �dnamerP rZ r# r# r$ � test_proc_cwd� s zTestFSAPIs.test_proc_cwdzfails on PYPY + WINDOWSc C s� t j� }t|j� �}t| jd�� t|j� �}W d Q R X || j� j}| j|t � t rf| rf| j d�S | j � r�| j tjj|�tjj| j�� d S )N�rbzopen_files on BSD is broken)rH rI �setZ open_files�openr0 �popr" rL r@ r r5 rG rM rD rN )r6 rP �start�newr" r# r# r$ �test_proc_open_files� s z TestFSAPIs.test_proc_open_filesz POSIX onlyc C s� | j | jd�}y t|�}W n& tk r@ tr2� n tjd��Y nX t|��4 tj � j d�d }| j |j t � | j|j |� W d Q R X d S )N)r% z not supported�unixr )r r1 r r) r �unittest�SkipTestr rH rI Z connectionsrL �laddrr@ rM )r6 rR �sock�connr# r# r$ �test_proc_connections s z TestFSAPIs.test_proc_connectionszcan't list UNIX socketsc C s� dd� }| j | jd�}y t|�}W n& tk rH tr:� n tjd��Y nX t|��6 tj dd�}||�}| j |j t � | j |j |� W d Q R X d S )Nc S s2 x$| D ] }t jj|j�jt�r|S qW td��d S )Nzconnection not found)rD r" rS rg � startswithr � ValueError)�consri r# r# r$ � find_sock s z2TestFSAPIs.test_net_connections.<locals>.find_sock)r% z not supportedrd )�kind)r r1 r r) r re rf r rH Znet_connectionsrL rg r@ rM )r6 rn rR rh rm ri r# r# r$ �test_net_connections s z TestFSAPIs.test_net_connectionsc C s, | j d }| jt|� t|� tj|� d S )NrX )r0 rY r r rH � disk_usage)r6 r[ r# r# r$ �test_disk_usage' s zTestFSAPIs.test_disk_usagez not supportedz&ctypes does not support unicode on PY2zunstable on PYPYc sv t | jd��`}dd� � � fdd�tj� j� D �}dd� |D �}| j� |�|� x|D ]}| j|t� qTW W d Q R X d S )N)r% c S s t jjt jj| ��S )N)rD r" �realpathrN )rP r# r# r$ �normpath5 s z-TestFSAPIs.test_memory_maps.<locals>.normpathc s g | ]}� |j ��qS r# )r" )�.0�x)rt r# r$ � <listcomp>9 s z/TestFSAPIs.test_memory_maps.<locals>.<listcomp>c S s g | ]}t |kr|�qS r# )r )ru rv r# r# r$ rw < s )r r1 rH rI Z memory_mapsZassertInrL r@ )r6 Z funky_pathZlibpathsr" r# )rt r$ �test_memory_maps- s zTestFSAPIs.test_memory_mapsN)r7 r8 r9 �__doc__r r1 rG rQ rT rW r\ re �skipIfr r rc r rj r r rp rr r r rx r# r# r# r$ r< � s r< zunreliable on CIc @ s e Zd ZdZeZdd� ZdS )�TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr# )r6 r# r# r$ rG H s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r7 r8 r9 ry r r1 rG r# r# r# r$ r{ B s r{ c @ sB e Zd ZdZerendZeje d�eje o.e d�dd� ��Z dS )� TestNonFSAPISz&Unicode tests for non fs-related APIs.� èz not supportedzsegfaults on PYPY + WINDOWSc C sx t jj� }| j|d< | j|d�}tj|j�}|j� }x,|j� D ] \}}| j |t � | j |t � q>W | j |d | j� d S )NZ FUNNY_ARG)�env) rD �environ�copyr1 r rH rI rJ �itemsrL r@ rM )r6 r~ r+ rP �k�vr# r# r$ �test_proc_environW s z TestNonFSAPIS.test_proc_environN) r7 r8 r9 ry r r r1 re rz r r r r� r# r# r# r$ r| R s r| �__main__)� run_from_name)1ry rD r' r re rA � contextlibr rH 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- rz r< r{ r| r7 Zpsutil.tests.runnerr� �__file__r# r# r# r$ �<module>I sd
💾 Save Changes
❌ Cancel