File: //opt/osm/venv/lib/python3.10/site-packages/psutil/tests/__pycache__/test_windows.cpython-310.pyc
o
=iu� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlm Z ddlm Z er�es�e � � � e � d� ddl Z ddl!Z!ddl"Z"ddl#Z#W d � n1 s�w Y er�ddl$m%Z% ej&j'Z'ej(j)e dd�ej(j)edd�G dd� de���Z*dd� Z+e,fdd�Z-G d d � d e*�Z.G d d � d e*�Z/G d d!� d!e*�Z0G d"d#� d#e*�Z1G d$d%� d%e*�Z2ej(j)e dd�G d&d'� d'e��Z3ej(j)e dd�G d(d)� d)e��Z4ej(j)e dd�G d*d+� d+e��Z5dS ),zWindows specific tests.� N)�mock)�WINDOWS)�GITHUB_ACTIONS)� HAS_BATTERY)�IS_64BIT)�PYPY)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�
spawn_subproc)� terminate�ignore)�convert_oserrorz WINDOWS only��reasonz pywin32 not available on PYPYc @ s e Zd ZdS )�WindowsTestCaseN)�__name__�
__module__� __qualname__� r r �G/opt/osm/venv/lib/python3.10/site-packages/psutil/tests/test_windows.pyr 5 s r c C s( t �d�s
t�d�S d| � d�}t|�S )z�Currently not used, but available just in case. Usage:
>>> powershell(
"Get-CIMInstance Win32_PageFileUsage | Select AllocatedBaseSize")
zpowershell.exez powershell.exe not availablezhpowershell.exe -ExecutionPolicy Bypass -NoLogo -NonInteractive -NoProfile -WindowStyle Hidden -Command "�")�shutil�whichr �skipr
)�cmd�cmdliner r r �
powershell; s
��r c sh t d| � d|� ���� }d�|�� dd� ��� }� dur2d|v r.t� fdd�|�� D ��S � |�S |S ) z�Currently not used, but available just in case. Usage:
>>> wmic("Win32_OperatingSystem", "FreePhysicalMemory")
2134124534
z
wmic path z get � � N�,c 3 s � | ]}� |�V qd S �Nr ��.0�x�� converterr r � <genexpr>T s � zwmic.<locals>.<genexpr>)r
�strip�join�
splitlines�tuple�split)�path�whatr) �out�datar r( r �wmicJ s r4 c @ sR e Zd Zejjdejvdd�dd� �Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dS )� TestCpuAPIs�NUMBER_OF_PROCESSORSz-NUMBER_OF_PROCESSORS env var is not availabler c C s" t tjd �}|t�� ksJ �d S )Nr6 )�int�os�environ�psutil� cpu_count)�selfZnum_cpusr r r �&test_cpu_count_vs_NUMBER_OF_PROCESSORSa s z2TestCpuAPIs.test_cpu_count_vs_NUMBER_OF_PROCESSORSc C s$ t �� d }t�� }||ksJ �d S )N� )�win32apiZ
GetSystemInfor: r; �r< � sys_value� psutil_valuer r r � test_cpu_count_vs_GetSystemInfok s z+TestCpuAPIs.test_cpu_count_vs_GetSystemInfoc C s2 t �� }tdd� |�� D ��}t�� |ksJ �d S )Nc s � � | ]}|j V qd S r$ )ZNumberOfLogicalProcessors�r& �procr r r r* t s �
�z<TestCpuAPIs.test_cpu_count_logical_vs_wmi.<locals>.<genexpr>��wmi�WMI�sum�Win32_Processorr: r; )r< �wZprocsr r r � test_cpu_count_logical_vs_wmir s
�z)TestCpuAPIs.test_cpu_count_logical_vs_wmic C s6 t �� }tdd� |�� D ��}tjdd�|ksJ �d S )Nc s rD r$ )Z
NumberOfCoresrE r r r r* { s � z:TestCpuAPIs.test_cpu_count_cores_vs_wmi.<locals>.<genexpr>F)ZlogicalrG )r< rL Zcoresr r r �test_cpu_count_cores_vs_wmiy s z'TestCpuAPIs.test_cpu_count_cores_vs_wmic C s t �� tt jdd��ksJ �d S )NT)Zpercpu)r: r; �len� cpu_times�r< r r r �test_cpu_count_vs_cpu_times~ s z'TestCpuAPIs.test_cpu_count_vs_cpu_timesc C s@ t �� }|�� d }|jt�� jksJ �|jt�� jks J �d S �Nr ) rH rI rK ZCurrentClockSpeedr: Zcpu_freq�currentZ
MaxClockSpeed�max)r< rL rF r r r �
test_cpu_freq� s zTestCpuAPIs.test_cpu_freqN)r r r r �mark�skipifr8 r9 r= rC rM rN rR rV r r r r r5 ` s �
r5 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Ze� d d � �Z e� d
d� �Z
e� dd� �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd S ) �TestSystemAPIsc C sX t d�}tjdd��� }|D ]}d|�dd��� v rq||vr)t�|�d�� S qd S )Nz
ipconfig /allT)Zperniczpseudo-interface� �-z+ nic wasn't found in 'ipconfig /all' output)r
r: Znet_io_counters�keys�replace�lowerr �fail)r< r2 ZnicsZnicr r r �test_nic_names� s ���z TestSystemAPIs.test_nic_namesc C s, t �� �� d }t|j�t�� jksJ �d S rS )rH rI ZWin32_ComputerSystemr7 ZTotalPhysicalMemoryr: �virtual_memory�total�r< rL r r r �test_total_phymem� s z TestSystemAPIs.test_total_phymemc C s4 t �� �� d }tt|j�t�� j �t k sJ �d S rS )
rH rI � Win32_PerfRawData_PerfOS_Memory�absr7 ZAvailableBytesr: ra �freer rc r r r �test_free_phymem� s �z TestSystemAPIs.test_free_phymemc C sl t �� �� d }t|j�t�� j t�� jksJ �t�� jdkr2t�� j dks)J �t�� j
dks4J �d S d S rS ) rH rI re r7 Z CommitLimitr: ra rb � swap_memoryrg �usedrc r r r �test_total_swapmem� s ��z!TestSystemAPIs.test_total_swapmemc C s� t �� jdkr<t�� jdd�d }t|j�d t|j� }t �� j dks&J �t
t �� j | �dk s3J �t �� j dks>J �d S d S )Nr Z_Total)�Name�d r> ) r: ri rb rH rI Z#Win32_PerfRawData_PerfOS_PagingFiler7 Z PercentUsageZPercentUsage_Base�percentrf )r< rL Z percentSwapr r r �test_percent_swapmem� s �z#TestSystemAPIs.test_percent_swapmemc C s6 t �� �� }dd� |D �}tt�� �}||ksJ �d S )Nc S s h | ]}|j �qS r �Z ProcessIdr% r r r � <setcomp>� s z+TestSystemAPIs.test_pids.<locals>.<setcomp>)rH rI �
Win32_Process�setr: �pids)r< rL Zwmi_pidsZ psutil_pidsr r r � test_pids� s zTestSystemAPIs.test_pidsc
C s t jdd�}t�� �� }|D ]s}|D ]d}|j�dd�|jkrv|js# n^d|j v r* nW|j�
d�r2 nOzt � |j�}W n
t yD Y n=w |j
t|j�ksOJ �t|j�}|j|ks[J �t|j| �dkrtt�d|j� d |� �� S n qt�d
|��� S qd S ) NT��all�\r! �cdrom�A:i � zpsutil=z, wmi=zcan't find partition )r: �disk_partitionsrH rI ZWin32_LogicalDiskZdevicer] ZDeviceID�
mountpoint�opts�
startswith�
disk_usage�FileNotFoundErrorrb r7 �SizeZ FreeSpacerg rf r r_ )r< Zps_partsZ wmi_partsZps_partZwmi_part�usageZwmi_freer r r �
test_disks� s8
�
����zTestSystemAPIs.test_disksc C s� t �� D ]9}d|jv r qt�|j�}t �|j�}t|d |j �t k s%J �t|d |j
�t k s2J �|j |j
|j ks=J �qd S )Nry r r" ) r: r{ r} r? ZGetDiskFreeSpaceExr| r rf rg r rb rj )r< ZdiskrA rB r r r �test_disk_usage� s
�z TestSystemAPIs.test_disk_usagec C s> dd� t �� �d�D �}dd� tjdd�D �}||ks J �d S )Nc S s" g | ]
}|r|� d �s|d �qS )rz rx )r~ r% r r r �
<listcomp>� s ���z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>z\ c S s g | ] }|j �d �s|j �qS )rz )r| r~ r% r r r r� � s
��Trv )r? ZGetLogicalDriveStringsr/ r: r{ r@ r r r �test_disk_partitions� s �
�z#TestSystemAPIs.test_disk_partitionsc C s� d}d}d}d}d}||i}t jd|jd��"}tj�|�|ks!J �tj�|�|ks+J �|js0J �W d � d S 1 s;w Y d S )NzC:\Windows\TempzC:z\Device\HarddiskVolume1z\??\C:\Windows\Tempz$\Device\HarddiskVolume1\Windows\Tempz%psutil._pswindows.cext.QueryDosDevice�Z side_effect)r �patch�getr: �
_pswindows�convert_dos_path�called)r< �winpathZ driveletterZ
devicepath�ntpath1�ntpath2Zdevices�mr r r �test_convert_dos_path_drive s � "�z*TestSystemAPIs.test_convert_dos_path_drivec C sL d}d}d}t j�|�|ksJ �t j�|�|ksJ �t j�|�|ks$J �d S )Nz\\localhost\C$\Windows\Tempz!\??\UNC\localhost\C$\Windows\Tempz%\Device\Mup\localhost\C$\Windows\Temp)r: r� r� )r< r� r� r� r r r �test_convert_dos_path_unc s z(TestSystemAPIs.test_convert_dos_path_uncc C s` t t�� �}t�� �� }t � }|D ]}|�|j� |�|j� q||@ s.J d|� d|� ���d S )Nzno common entries in z, ) rs �cextZ net_if_statsrH rI ZWin32_NetworkAdapter�addrl ZNetConnectionID)r< Zps_namesZ wmi_adaptersZ wmi_namesZ wmi_adapterr r r �test_net_if_stats" s ��z TestSystemAPIs.test_net_if_statsc C sf t �� �� }|d j�d�d }tj�|d�}tj�t� � �}t
|| � � �}|dks1J ||f��d S )Nr �.� %Y%m%d%H%M%Sr> ) rH rI ZWin32_OperatingSystemZLastBootUpTimer/ �datetime�strptime�
fromtimestampr: Z boot_timerf �
total_seconds)r< Zwmi_osZ
wmi_btime_strZ wmi_btime_dtZ psutil_dt�diffr r r �test_boot_time- s �z TestSystemAPIs.test_boot_timec C s0 t jj�� }|d }tt�� | �dk sJ �d S )Ng @�@g �?)�ctypes�windll�kernel32ZGetTickCount64rf r� Zuptime)r< �msZsecsr r r � test_uptime7 s zTestSystemAPIs.test_uptimeN)r r r r` rd rh rk ro r ru r� r� r� r� r� r� r� r� r r r r rY � s$
rY c @ sh e Zd Zdd� Zejje dd�dd� �Zejje dd�dd� �Z d d
� Z
d d � Z d
d� Z dd� Z
dS )�TestSensorsBatteryc C s4 t �� d rt�� d usJ �d S t�� d u sJ �d S )NZSystemBatteriesPresent)r? ZGetPwrCapabilitiesr: �sensors_batteryrQ r r r �test_has_batteryE s z#TestSensorsBattery.test_has_batteryz
no batteryr c C s: t �� }|�d�d }t�� }t|j|j �dk sJ �d S )N�select * from Win32_Batteryr r" )rH rI �queryr: r� rf rn ZEstimatedChargeRemaining�r< rL Z battery_wmiZbattery_psutilr r r � test_percentK s �z TestSensorsBattery.test_percentc C s6 t �� }|�d�d }t�� }|j|jdkksJ �d S )Nr� r � )rH rI r� r: r� Z
power_pluggedZ
BatteryStatusr� r r r �test_power_pluggedU s z%TestSensorsBattery.test_power_pluggedc C sN t jddd��}t�� d u sJ �|jsJ �W d � d S 1 s w Y d S )N�&psutil._pswindows.cext.sensors_battery)r � r r �� return_value)r r� r: r� r� �r< r� r r r �test_emulate_no_battery^ s � "�z*TestSensorsBattery.test_emulate_no_batteryc C �R t jddd��}t�� jtjksJ �|jsJ �W d � d S 1 s"w Y d S )Nr� )r" r r r r� �r r� r: r� �secsleftZPOWER_TIME_UNLIMITEDr� r� r r r � test_emulate_power_connectedf � �� "�z/TestSensorsBattery.test_emulate_power_connectedc C r� )Nr� )r � r r r� r� r� r r r �test_emulate_power_chargingp r� z.TestSensorsBattery.test_emulate_power_chargingc C r� )Nr� )r r r ���r� )r r� r: r� r� ZPOWER_TIME_UNKNOWNr� r� r r r � test_emulate_secs_left_unknownz s � "�z1TestSensorsBattery.test_emulate_secs_left_unknownN)r r r r� r rW rX r r� r� r� r� r� r� r r r r r� D s
r� c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$� TestProcessc C � t � j| _d S r$ �r �pid��clsr r r �
setUpClass� � zTestProcess.setUpClassc C � t | j� d S r$ �r r� r� r r r �
tearDownClass� r� zTestProcess.tearDownClassc C sD t �d�}t�t j��
|�� W d � d S 1 sw Y d S rS )r: �Processr �raises� AccessDenied�kill�r< �pr r r �
test_issue_24� s
"�zTestProcess.test_issue_24c C s� t �d�}|�� dks
J �t|� |�� |�� dks J �z |�� d d� \}}W n t jy= t� � d dvr:� Y d S w |dksDJ �d S )N� �Systemg r� r" > ZvistaZwin7zwin-7r )
r: r� �name�str�username� create_time� memory_infor� �platform�uname)r< r� �rssZ_vmsr r r �test_special_pid� s
��z TestProcess.test_special_pidc C sH t �| j�}t�t�� |�tj� W d � d S 1 s w Y d S r$ ) r: r� r� r r� �
ValueError� send_signal�signal�SIGINTr� r r r �test_send_signal� s "�z TestProcess.test_send_signalc C sb t �t�� �}|�� }t�tjtj t�� �}|�� }||d ks"J �t�
|� |�� |ks/J �d S )Nr" ) r: r� r8 �getpid� num_handlesr? � OpenProcess�win32con�PROCESS_QUERY_INFORMATION�FALSE� CloseHandle)r< r� �before�handle�afterr r r �test_num_handles_increment� s �
z&TestProcess.test_num_handles_incrementc C s� t �| �� j�}|�tj� |�tj� |�� |� � t
� t j �� |�tj� W d � n1 s3w Y t
� t j �� |�tj� W d � d S 1 sPw Y d S r$ )
r: r� r r� r� r� Z CTRL_C_EVENTZCTRL_BREAK_EVENTr� �waitr r� �
NoSuchProcessr� r r r �test_ctrl_signals� s �"�z TestProcess.test_ctrl_signalsc C s8 t �tj�}|�d�rt�d�S t�� � � |ksJ �d S )N�$zrunning as service account)
r? Z
GetUserNameExr� ZNameSamCompatible�endswithr r r: r� r� )r< r� r r r �
test_username� s
zTestProcess.test_usernamec C sP t �ddt�� ��� }d�t�� �� �}|� dd�}|� dd�}||ks&J �d S )Nz[ ]+rZ r r! )
�re�subr? ZGetCommandLiner+ r, r: r� r r] r@ r r r � test_cmdline� s
zTestProcess.test_cmdlinec C sJ t �tjtjt�� �}| �t j|� t �
|�}t � � �
� }||ks#J �d S r$ )r? r� r� r� r� r8 r� �
addCleanupr� � win32processZGetPriorityClassr: r� �nice�r< r� rA rB r r r � test_nice� s �
zTestProcess.test_nicec C s� t �tjtj| j�}| �t j|� t� |�}t
� | j�� � }|d |j
ks'J �|d |jks0J �|d |jks9J �|d |jksBJ �|d |jksKJ �|d |jksTJ �|d |jks]J �|d |jksfJ �|j|jksnJ �|j|jksvJ �d S ) NZPeakWorkingSetSize�WorkingSetSizeZQuotaPeakPagedPoolUsageZQuotaPagedPoolUsageZQuotaPeakNonPagedPoolUsageZQuotaNonPagedPoolUsageZ
PagefileUsageZPeakPagefileUsage)r? r� r� r� r� r� r� r� r� ZGetProcessMemoryInfor: r� r� Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZpagefileZ
peak_pagefiler� �vmsr� r r r �test_memory_info� s( �
��z TestProcess.test_memory_infoc C sX t �tjtj| j�}| �t j|� t� | j�}|�
� |� � }t �
|�}||ks*J �d S r$ )r? r� r� r� r� r� r� r� r: r� r r� r� ZGetExitCodeProcess)r< r� r� rB rA r r r � test_wait s �
zTestProcess.test_waitc C s\ dd� }t �tjtj| j�}| �t j|� |t� |�d �}t
� | j�� � }||ks,J �d S )Nc s � fdd�t d�D �S )Nc s g | ]
}d |> � @ r|�qS )r" r )r& �i�r' r r r� s zGTestProcess.test_cpu_affinity.<locals>.from_bitmask.<locals>.<listcomp>�@ )�ranger� r r� r � from_bitmask s z3TestProcess.test_cpu_affinity.<locals>.from_bitmaskr )
r? r� r� r� r� r� r� r� r� ZGetProcessAffinityMaskr: r� Z cpu_affinity)r< r� r� rA rB r r r �test_cpu_affinity s � �z TestProcess.test_cpu_affinityc C s� t �tjtjt�� �}| �t j|� t �
|�}t � � �
� }|j|d ks&J �|j|d ks/J �|j|d ks8J �|j|d ksAJ �|j|d ksJJ �|j|d ksSJ �d S )NZReadOperationCountZWriteOperationCountZReadTransferCountZWriteTransferCountZOtherOperationCountZOtherTransferCount)r? r� r� r� r� r8 r� r� r� r� ZGetProcessIoCountersr: r� � io_countersZ
read_countZ write_count�
read_bytes� write_bytes� other_countZ other_bytesr� r r r �test_io_counters! s �
z TestProcess.test_io_countersc C s� dd l }dd l}d}|jj�|d| j�}| �|jjj|� |j� � }|jj�
||� |�� |j }t
�| j��� }||ks>J �d S )Nr � )r� Zctypes.wintypesr� r� r� r� r� r� ZwintypesZDWORDZGetProcessHandleCountZbyref�valuer: r� r� )r< r� r� r� ZhndcntrA rB r r r �test_num_handles/ s �
�z TestProcess.test_num_handlesc
C s� t � }d|_tjd|d��; t�d��&}t�� }t�tj�� |� � W d � n1 s-w Y W d � n1 s<w Y W d � n1 sKw Y |j
dksWJ �d S )Ni+ z psutil._psplatform.cext.proc_cwdr� z
time.sleepr> ) �OSError�winerrorr r� r: r� r r� r� �cwdZ
call_count)r< �excr� r� r r r �test_error_partial_copyA s
�� �� �z#TestProcess.test_error_partial_copyc C sV t �� d d }t j�|�}t�t j��
|�� W d � d S 1 s$w Y d S )Nr� i�� )r: rt � _psplatformr� r r� r� �exe)r< r� rF r r r �test_exeL s
"�zTestProcess.test_exeN)r r r � classmethodr� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r� � s(
r� c @ s| e Zd ZdZedd� �Zedd� �Zdd� Zej j
e dd �d
d � �Z d d
� Z
dd� Ze� dd� �Ze� dd� �Zdd� ZdS )�TestProcessWMIz%Compare Process API results with WMI.c C r� r$ r� r� r r r r� X r� zTestProcessWMI.setUpClassc C r� r$ r� r� r r r r� \ r� z TestProcessWMI.tearDownClassc C s8 t �� j| jd�d }t�| j�}|�� |jksJ �d S �Nrp r )rH rI rr r� r: r� r� ZCaption�r< rL r� r r r � test_name` s zTestProcessWMI.test_namez!unreliable path on GITHUB_ACTIONSr c C s@ t �� j| jd�d }t�| j�}|�� �� |j�� ks J �d S r ) rH rI rr r� r: r� r r^ ZExecutablePathr r r r r f s zTestProcessWMI.test_exec C sF t �� j| jd�d }t�| j�}d�|�� �|j� dd�ks!J �d S )Nrp r rZ r r! )
rH rI rr r� r: r� r, r Z CommandLiner] r r r r r� p s $zTestProcessWMI.test_cmdlinec C sR t �� j| jd�d }t�| j�}|�� \}}}|� d|� �}|�� |ks'J �d S )Nrp r rx )rH rI rr r� r: r� ZGetOwnerr� )r< rL r� �domain�_r� r r r r� u s
z TestProcessWMI.test_usernamec C sB t �� j| jd�d }t�| j�}|�� j}|t|j �ks J �d S r )
rH rI rr r� r: r� r� r� r7 r� )r< rL r� r� r r r �test_memory_rss| s
z TestProcessWMI.test_memory_rssc C s` t �� j| jd�d }t�| j�}|�� j}t|j �}|||d hvr.t
� d|� d|� ��S d S )Nrp r r zwmi=z , psutil=) rH rI rr r� r: r� r� r� r7 Z
PageFileUsager r_ )r< rL r� r� Z wmi_usager r r �test_memory_vms� s
�z TestProcessWMI.test_memory_vmsc C s\ t �� j| jd�d }t�| j�}t|j�d�d �}t �
dt � |� � ��}||ks,J �d S )Nrp r r� r� )
rH rI rr r� r: r� r� Z CreationDater/ �time�strftime� localtimer� )r< rL r� Z wmic_createZ
psutil_creater r r �test_create_time� s �z TestProcessWMI.test_create_timeN)r r r �__doc__r r� r� r r rW rX r r r� r� r r r r r r r r r U s$
�
r c @ sX e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dS )� TestDualProcessImplementationaw Certain APIs on Windows have 2 internal implementations, one
based on documented Windows APIs, another one based
NtQuerySystemInformation() which gets called as fallback in
case the first fails because of limited permission error.
Here we test that the two methods return the exact same value,
see:
https://github.com/giampaolo/psutil/issues/304.
c C r� r$ r� r� r r r r� � r� z(TestDualProcessImplementation.setUpClassc C r� r$ r� r� r r r r� � r� z+TestDualProcessImplementation.tearDownClassc C s� t �| j��� }tjdtd��G}t �| j��� }t|�t|�ks"J �tt|��D ] }|| dks2J �|| dks:J �t || || �dk sHJ �q(|j
sNJ �W d � d S 1 sYw Y d S )Nz(psutil._psplatform.cext.proc_memory_infor� r i ) r: r� r� r� r r� �PermissionErrorrO r� rf r� )r< Zmem_1�funZmem_2r� r r r r� � s � "�z.TestDualProcessImplementation.test_memory_infoc C �f t �| j��� }tjdtd��}t �| j��� |ks J �|js!J �W d � d S 1 s,w Y d S )N�"psutil._psplatform.cext.proc_timesr� )r: r� r� r� r r� r r� )r< �ctimer r r r r � � � "�z.TestDualProcessImplementation.test_create_timec C s� t �| j��� }tjdtd��.}t �| j��� }|js J �t|j |j �dk s)J �t|j
|j
�dk s5J �W d � d S 1 s@w Y d S )Nr" r� g{�G�z�?) r: r� r� rP r r� r r� rf �user�system)r< Z cpu_times_1r Z cpu_times_2r r r �test_cpu_times� s �
"�z,TestDualProcessImplementation.test_cpu_timesc C s� t �| j��� }tjdtd��-}t �| j��� }tt|��D ]}t || || �dk s.J �q |j
s4J �W d � d S 1 s?w Y d S )Nz(psutil._psplatform.cext.proc_io_countersr� r> ) r: r� r� r r r� r r� rO rf r� )r< Z
io_counters_1r Z
io_counters_2r� r r r r � s � "�z.TestDualProcessImplementation.test_io_countersc C r! )Nz(psutil._psplatform.cext.proc_num_handlesr� )r: r� r� r� r r� r r� )r< r� r r r r r � r$ z.TestDualProcessImplementation.test_num_handlesc C s� t �� D ]9}ztj|dd�}tj|dd�}W n ty6 } zt|�}t|t jt jf�s,� W Y d }~qd }~ww ||ks=J �qd S )NT)Zuse_pebF) r: rt r� Z proc_cmdliner r �
isinstancer� r� )r< r� �a�b�errr r r r� � s � ����z*TestDualProcessImplementation.test_cmdlineN)
r r r r r r� r� r� r r' r r r� r r r r r � s
r c sp e Zd ZdZedd� �ZddgZ� fdd�Z� fdd �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Z� ZS )�RemoteProcessTestCasez�Certain functions require calling ReadProcessMemory.
This trivially works when called on the current process.
Check that this works on other processes, especially when they
have a different bitness.
c C s\ d} t � d�D ]$}tj|d| gtjtjd�}|�� \}}|�� |tt �kr+| S qd S )Nz6import sys; sys.stdout.write(str(sys.maxsize > 2**32))zC:\Python*\python.exe�-c)�args�stdout�stderr) �glob�
subprocess�Popen�PIPE�STDOUT� communicater� r� r )�code�filenamerF �outputr r r r �find_other_interpreter� s � ��z,RemoteProcessTestCase.find_other_interpreterr- z import sys; sys.stdin.read()c s� t � �� | �� }|d u rt�d�S tr tj| _|| _ n|| _tj| _ t
j � � }t
t
�� �|d< | j| j g| j |tjd�| _| j| jg| j |tjd�| _d S )Nz0could not find interpreter with opposite bitness�THINK_OF_A_NUMBER)�env�stdin)�super�setUpr: r r r �sys�
executableZpython64Zpython32r8 r9 �copyr� r� r � test_argsr2 r4 �proc32�proc64)r< Z other_pythonr< �� __class__r r r? s&
�
� �zRemoteProcessTestCase.setUpc s" t � �� | j�� | j�� d S r$ )r> �tearDownrD r6 rE rQ rF r r rH ) s
z RemoteProcessTestCase.tearDownc C �@ t �| jj�}t|�� �dksJ �|�� dd � | jks J �d S �N� r" )r: r� rD r� rO r rC r� r r r �test_cmdline_32. � z%RemoteProcessTestCase.test_cmdline_32c C rI rJ )r: r� rE r� rO r rC r� r r r �test_cmdline_643 rM z%RemoteProcessTestCase.test_cmdline_64c C �& t �| jj�}|�� t�� ksJ �d S r$ )r: r� rD r� r
r8 �getcwdr� r r r � test_cwd_328 � z!RemoteProcessTestCase.test_cwd_32c C rO r$ )r: r� rE r� r
r8 rP r� r r r � test_cwd_64< rR z!RemoteProcessTestCase.test_cwd_64c C s> t �| jj�}|�� }d|v sJ �|d tt�� �ks J �d S )Nr; )r: r� rD r� r9 r� r8 r� )r< r� �er r r �test_environ_32@ s z%RemoteProcessTestCase.test_environ_32c C s4 t �| jj�}z|�� W d S t jy Y d S w r$ )r: r� rE r� r9 r� r� r r r �test_environ_64F s �z%RemoteProcessTestCase.test_environ_64)r r r r � staticmethodr: rC r? rH rL rN rQ rS rU rV �
__classcell__r r rF r r, � s
r, c @ s e Zd Zdd� Zdd� ZdS )� TestServicesc C sX h d�}h d�}h d�}t �� D ]�}|�� dkrq|�� }t|d t�s&J �|d �� s.J �t|d t�s7J �t|d t�s@J �|d |v sHJ �|d d urUt �|d � t|d
t�s^J �t|d t�sgJ �t|d t�spJ �|d |v sxJ �|d |v s�J �t|d t�s�J �|�� }|d ur�t �|�}|� � s�J �t �
|�� �}||ks�J �qd S )
N> �stop�start�paused�pause�stopped�running�continue> Z automaticZmanual�disabled> r\ r^ r_ Z
pause_pendingZcontinue_pendingZ stop_pendingZ
start_pendingZ WaaSMedicSvcr� � display_namer� �statusr� ZbinpathZ
start_type� description) r: �win_service_iterr� �as_dictr( r� r+ r� r� Z
is_running�win_service_get)r< Zvalid_statusesZvalid_start_typesZservr3 r� r� �sr r r �test_win_service_iterU s6
�z"TestServices.test_win_service_iterc C s� t jjj}t jjj}tt �� ��� }t� t j
��}t � |d � W d � n1 s*w Y |j j|d ks9J �t � |�}t
ddd�}||_tjd|d��" t� t j
�� |�� W d � n1 sdw Y W d � n1 ssw Y tjd|d��" t� t j
�� |�� W d � n1 s�w Y W d � n1 s�w Y t
ddd�}||_tjd|d��" t� t j�� |�� W d � n1 s�w Y W d � n1 s�w Y tjd|d��# t� t j�� |�� W d � n 1 �sw Y W d � n 1 �sw Y |�� t|�v �s!J �|�� t|�v �s,J �|�� t|�v �s7J �|�� t|�v �sBJ �d S )Nz???r �msgz/psutil._psplatform.cext.winservice_query_statusr� z/psutil._psplatform.cext.winservice_query_config)r: r
r� � ERROR_SERVICE_DOES_NOT_EXIST�ERROR_ACCESS_DENIED�nextre r� r r� r� rg r r r r r� rc r� r� r� rb �repr)r< rk rl r� �cmZservicer r r r �test_win_service_get� sb �
�
�
�� ��
�� � �
�� ��
�� �z!TestServices.test_win_service_getN)r r r ri rp r r r r rY S s 1rY )6r r� r� r1 r8 r� r� r r� r2 r@ r �warningsZunittestr r: r Z psutil.testsr r r r r r r
r r r
r r �catch_warnings� simplefilterr? r� r� rH Zpsutil._pswindowsr r
r� rW rX r r r7 r4 r5 rY r� r� r r r, rY r r r r �<module> sr
� ( =F LHW]