File: //usr/local/lib64/python3.9/site-packages/psutil/tests/__pycache__/test_windows.cpython-39.pyc
a
��?h � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl m
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! e
�r�e�s�e �"� �: e �#d� ddl$Z$ddl%Z%ddl&Z&ddl'Z'W d � n1 �s�0 Y e
�r�ddl(m)Z) e j*j+Z+e j,j-e
dd�e j,j-edd�e j,j-e�o�e dd�G d d � d e����Z.d d � Z/e0fd d!�Z1G d"d#� d#e.�Z2G d$d%� d%e.�Z3G d&d'� d'e.�Z4G d(d)� d)e.�Z5G d*d+� d+e.�Z6e j,j-e
dd�G d,d-� d-e��Z7e j,j-e
dd�G d.d/� d/e��Z8e j,j-e
dd�G d0d1� d1e��Z9dS )2zWindows specific tests.� N)�WINDOWS)�FileNotFoundError)�super)�which)�APPVEYOR)�GITHUB_ACTIONS)� HAS_BATTERY)�IS_64BIT)�PY3)�PYPY)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�mock)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate�ignore)�convert_oserrorz WINDOWS only��reasonz pywin32 not available on PYPYz pywin32 broken on GITHUB + PY2c @ s e Zd ZdS )�WindowsTestCaseN)�__name__�
__module__� __qualname__� r r �E/usr/local/lib64/python3.9/site-packages/psutil/tests/test_windows.pyr ; s r c C s& t d�st�d��dd| }t|�S )z�Currently not used, but available just in case. Usage:
>>> powershell(
"Get-CIMInstance Win32_PageFileUsage | Select AllocatedBaseSize")
zpowershell.exez powershell.exe not availablez?powershell.exe -ExecutionPolicy Bypass -NoLogo -NonInteractive z,-NoProfile -WindowStyle Hidden -Command "%s")r r �skipr )�cmd�cmdliner r r �
powershellE s
��r" c sj t d| |f ��� }d�|�� dd� ��� }� durbd|v rXt� fdd�|�� D ��S � |�S n|S dS )z�Currently not used, but available just in case. Usage:
>>> wmic("Win32_OperatingSystem", "FreePhysicalMemory")
2134124534
zwmic path %s get %s� � N�,c s g | ] }� |��qS r r ��.0�x�� converterr r �
<listcomp>^ � zwmic.<locals>.<listcomp>)r �strip�join�
splitlines�tuple�split)�path�whatr* �out�datar r) r �wmicT s
r6 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�� ks J �d S )Nr8 )�int�os�environ�psutil� cpu_count)�selfZnum_cpusr r r �&test_cpu_count_vs_NUMBER_OF_PROCESSORSk s z2TestCpuAPIs.test_cpu_count_vs_NUMBER_OF_PROCESSORSc C s$ t �� d }t�� }||ks J �d S )N� )�win32apiZ
GetSystemInfor< r= �r> � sys_value� psutil_valuer r r � test_cpu_count_vs_GetSystemInfou s z+TestCpuAPIs.test_cpu_count_vs_GetSystemInfoc C s2 t �� }tdd� |�� D ��}t�� |ks.J �d S )Nc s s | ] }|j V qd S �N)ZNumberOfLogicalProcessors�r' �procr r r � <genexpr>~ 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_wmi| s
�z)TestCpuAPIs.test_cpu_count_logical_vs_wmic C s6 t �� }tdd� |�� D ��}tjdd�|ks2J �d S )Nc s s | ] }|j V qd S rF )Z
NumberOfCoresrG r r r rI � r, z:TestCpuAPIs.test_cpu_count_cores_vs_wmi.<locals>.<genexpr>F)�logicalrJ )r> rO Zcoresr r r �test_cpu_count_cores_vs_wmi� s z'TestCpuAPIs.test_cpu_count_cores_vs_wmic C s t �� tt jdd��ks J �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�� jks(J �|jt�� jks<J �d S �Nr ) rK rL rN ZCurrentClockSpeedr< Zcpu_freq�currentZ
MaxClockSpeed�max)r> rO rH r r r �
test_cpu_freq� s zTestCpuAPIs.test_cpu_freqN)r r r r �mark�skipifr: r; r? rE rP rR rV rZ r r r r r7 j s �
r7 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej j
e d d �e � d
d� ��Z
e � dd� �Ze � dd� �Zdd� Zdd� Zdd� Zdd� ZdS ) �TestSystemAPIsc C sR t d�}tjdd��� }|D ]0}d|�dd��� v r6q ||vr | �d| ��q d S )Nz
ipconfig /allT)Zperniczpseudo-interface� �-z-%r nic wasn't found in 'ipconfig /all' output)r r< Znet_io_counters�keys�replace�lower�fail)r> r4 ZnicsZnicr r r �test_nic_names� s �z TestSystemAPIs.test_nic_namesc C s, t �� �� d }t|j�t�� jks(J �d S rW )rK rL ZWin32_ComputerSystemr9 ZTotalPhysicalMemoryr< �virtual_memory�total�r> rO r r r �test_total_phymem� s z TestSystemAPIs.test_total_phymemc C s4 t �� �� d }tt|j�t�� j �t k s0J �d S rW )
rK rL � Win32_PerfRawData_PerfOS_Memory�absr9 ZAvailableBytesr< re �freer
rg r r r �test_free_phymem� s
��z TestSystemAPIs.test_free_phymemc C sh t �� �� d }t|j�t�� j t�� jks2J �t�� jdkrdt�� j dksRJ �t�� j
dksdJ �d S rW ) rK rL ri r9 Z CommitLimitr< re rf � swap_memoryrk �usedrg r r r �test_total_swapmem� s ��z!TestSystemAPIs.test_total_swapmemc C s| t �� jdkrxt�� jdd�d }t|j�d t|j� }t �� j dksLJ �t
t �� j | �dk sfJ �t �� j dksxJ �d S )Nr Z_Total)�Name�d r@ ) r< rm rf rK rL Z#Win32_PerfRawData_PerfOS_PagingFiler9 Z PercentUsageZPercentUsage_Base�percentrj )r> rO Z percentSwapr r r �test_percent_swapmem� s z#TestSystemAPIs.test_percent_swapmemz test not relieable on appveyorr c C s: t �� �� }tdd� |D ��}tt�� �}||ks6J �d S )Nc S s g | ]
}|j �qS r �Z ProcessIdr& r r r r+ � r, z,TestSystemAPIs.test_pids.<locals>.<listcomp>)rK rL �
Win32_Process�setr< �pids)r> rO Zwmi_pidsZ psutil_pidsr r r � test_pids� s zTestSystemAPIs.test_pidsc
C s� t jdd�}t�� �� }|D ]�}|D ]�}|j�dd�|jkr$|jsF q d|j v rT q |j�
d�rd q zt � |j�}W n t y� Y q Y n0 |j
t|j�ks�J �t|j�}|j|ks�J �t|j| �dkr�| �d|j|f �� q q$| �d t|� ��q d S )
NT��all�\r# �cdrom�A:i � zpsutil=%s, wmi=%szcan't find partition %s)r< �disk_partitionsrK rL ZWin32_LogicalDiskZdevicera ZDeviceID�
mountpoint�opts�
startswith�
disk_usager rf r9 �SizeZ FreeSpacerk rj rc �repr)r> Zps_partsZ wmi_partsZps_partZwmi_part�usageZwmi_freer r r �
test_disks� s0
�zTestSystemAPIs.test_disksc C s� t �� D ]r}d|jv rqt�|j�}t �|j�}t|d |j �t k sJJ �t|d |j
�t k sdJ �|j |j
|j ksJ �qd S )Nr| r r$ ) r< r~ r� rA ZGetDiskFreeSpaceExr r� rj rk r rf rn )r> ZdiskrC rD 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 )r} r{ )r� r& r r r r+ s �z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>z\ c S s g | ]}|j �d �s|j �qS )r} )r r� r& r r r r+ s �Try )rA ZGetLogicalDriveStringsr1 r< r~ rB r r r �test_disk_partitions s �
�z#TestSystemAPIs.test_disk_partitionsc C s\ t t�� �}t�� �� }t � }|D ] }|�|j� |�|j� q"||@ sXJ d||f ��d S )Nzno common entries in %s, %s) rv �cextZ net_if_statsrK rL ZWin32_NetworkAdapter�addrp ZNetConnectionID)r> Zps_namesZ wmi_adaptersZ wmi_namesZ wmi_adapterr r r �test_net_if_stats
s �z TestSystemAPIs.test_net_if_statsc C s^ t �� �� }|d j�d�d }tj�|d�}tj�t� � �}t
|| � � �}|dksZJ �d S )Nr �.� %Y%m%d%H%M%Sr@ ) rK rL ZWin32_OperatingSystemZLastBootUpTimer1 �datetime�strptime�
fromtimestampr< � boot_timerj �
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 s� t jddd�� t�� dks J �W d � n1 s40 Y t jddd�� t�� dks^J �W d � n1 sr0 Y t jddd�� t�� dks�J �W d � n1 s�0 Y t jddd�� t�� dks�J �W d � n1 s�0 Y d S )Nz psutil._pswindows.cext.boot_timer@ �Z return_value� � iM )r �patchr< r� rU r r r �test_boot_time_fluctuation# s ...z)TestSystemAPIs.test_boot_time_fluctuationN)r r r rd rh rl ro rs r r[ r\ r r rx r� r� r� r� r� r� r r r r r] � s
r] 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 s2 t �� d r t�� d us.J �nt�� d u s.J �d S )NZSystemBatteriesPresent)rA ZGetPwrCapabilitiesr< �sensors_batteryrU r r r �test_has_battery5 s z#TestSensorsBattery.test_has_batteryz
no batteryr c C s: t �� }|�d�d }t�� }t|j|j �dk s6J �d S )N�select * from Win32_Batteryr r$ )rK rL �queryr< r� rj rr ZEstimatedChargeRemaining�r> rO Z battery_wmiZbattery_psutilr r r � test_percent; s ��z TestSensorsBattery.test_percentc C s6 t �� }|�d�d }t�� }|j|jdkks2J �d S )Nr� r � )rK rL r� r< r� Z
power_pluggedZ
BatteryStatusr� r r r �test_power_pluggedE s z%TestSensorsBattery.test_power_pluggedc C sL t jddd��*}t�� d u s J �|js*J �W d � n1 s>0 Y d S )N�&psutil._pswindows.cext.sensors_battery)r � r r r� )r r� r< r� �called�r> �mr r r �test_emulate_no_batteryN s �z*TestSensorsBattery.test_emulate_no_batteryc C sP t jddd��.}t�� jtjks$J �|js.J �W d � n1 sB0 Y d S )Nr� )r$ r r r r� �r r� r< r� �secsleftZPOWER_TIME_UNLIMITEDr� r� r r r � test_emulate_power_connectedV s ���z/TestSensorsBattery.test_emulate_power_connectedc C sP t jddd��.}t�� jtjks$J �|js.J �W d � n1 sB0 Y d S )Nr� )r � r r r� r� r� r r r �test_emulate_power_charging` s ���z.TestSensorsBattery.test_emulate_power_chargingc C sP t jddd��.}t�� jtjks$J �|js.J �W d � n1 sB0 Y d S )Nr� )r r r ���r� )r r� r< r� r� ZPOWER_TIME_UNKNOWNr� r� r r r � test_emulate_secs_left_unknownj s ��z1TestSensorsBattery.test_emulate_secs_left_unknownN)r r r r� r r[ r\ r r� r� r� r� r� r� r r r r r� 4 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 s t � j| _d S rF �r �pid��clsr r r �
setUpClass{ s zTestProcess.setUpClassc C s t | j� d S rF �r r� r� r r r �
tearDownClass s zTestProcess.tearDownClassc C sB t �d�}t�t j�� |�� W d � n1 s40 Y d S rW )r< �Processr �raises� AccessDenied�kill�r> �pr r r �
test_issue_24� s
zTestProcess.test_issue_24c C s� t �d�}|�� dksJ �t|� |�� |�� dks:J �z|�� d d� \}}W n& t jyx t� � d dvrt� Y n0 |dks�J �d S )Nr� �Systemg r� r$ )Zvistazwin-7Zwin7r )
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 sF t �| j�}t�t�� |�tj� W d � n1 s80 Y d S rF ) 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 ksDJ �t�
|� |�� |ks^J �d S )Nr$ ) r< r� r: �getpid� num_handlesrA � 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 sf0 Y t
� t j �� |�tj� W d � n1 s�0 Y d S rF )
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�r t�d��t�� � � |ks4J �d S )N�$zrunning as service account)
rA Z
GetUserNameExr� ZNameSamCompatible�endswithr r r< r� r� )r> r� r r r �
test_username� s
zTestProcess.test_usernamec C sf t �ddt�� ��� }d�t�� �� �}|d d krD|d krVn n|� ddd�}||ksbJ �d S )Nz[ ]+r^ r �"r# r� )
�re�subrA ZGetCommandLiner- r. r< r� r! ra rB r r r � test_cmdline� s
zTestProcess.test_cmdlinec C sJ t �tjtjt�� �}| �t j|� t �
|�}t � � �
� }||ksFJ �d S rF )rA r� r� r� r� r: r� �
addCleanupr� � win32processZGetPriorityClassr< r� �nice�r> r� rC rD r r r � test_nice� s �
zTestProcess.test_nicec C s� t �tjtj| j�}| �t j|� t� |�}t
� | j�� � }|d |j
ksNJ �|d |jks`J �|d |jksrJ �|d |jks�J �|d |jks�J �|d |jks�J �|d |jks�J �|d |jks�J �|j|jks�J �|j|jks�J �d S ) NZPeakWorkingSetSize�WorkingSetSizeZQuotaPeakPagedPoolUsageZQuotaPagedPoolUsageZQuotaPeakNonPagedPoolUsageZQuotaNonPagedPoolUsageZ
PagefileUsageZPeakPagefileUsage)rA 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 �
|�}||ksTJ �d S rF )rA r� r� r� r� r� r� r� r< r� r r� r� ZGetExitCodeProcess)r> r� r� rD rC r r r � test_wait� s �
zTestProcess.test_waitc C s\ dd� }t �tjtj| j�}| �t j|� |t� |�d �}t
� | j�� � }||ksXJ �d S )Nc s � fdd�t d�D �S )Nc s g | ]}d |> � @ r|�qS )r$ r )r' �i�r( r r r+ r, zGTestProcess.test_cpu_affinity.<locals>.from_bitmask.<locals>.<listcomp>�@ )�ranger� r r� r � from_bitmask s z3TestProcess.test_cpu_affinity.<locals>.from_bitmaskr )
rA r� r� r� r� r� r� r� r� ZGetProcessAffinityMaskr< r� Z cpu_affinity)r> r� r� rC rD r r r �test_cpu_affinity s � �z TestProcess.test_cpu_affinityc C s� t �tjtjt�� �}| �t j|� t �
|�}t � � �
� }|j|d ksLJ �|j|d ks^J �|j|d kspJ �|j|d ks�J �|j|d ks�J �|j|d ks�J �d S )NZReadOperationCountZWriteOperationCountZReadTransferCountZWriteTransferCountZOtherOperationCountZOtherTransferCount)rA r� r� r� r� r: 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 � )�ctypesZctypes.wintypes�windllZkernel32r� r� r� r� ZwintypesZDWORDZGetProcessHandleCount�byref�valuer< r� r� )r> r� r� r� ZhndcntrC rD r r r �test_num_handles" s �
�z TestProcess.test_num_handlesc
C s� t � }d|_tjd|d��v t�d��L}t�� }t�tj�� |� � W d � n1 sZ0 Y W d � n1 sx0 Y W d � n1 s�0 Y |j
dks�J �d S )Ni+ z psutil._psplatform.cext.proc_cwd�Z side_effectz
time.sleepr@ ) � WindowsError�winerrorr r� r< r� r r� r� �cwdZ
call_count)r> �excr� r� r r r �test_error_partial_copy4 s bz#TestProcess.test_error_partial_copyc C sT t �� d d }t j�|�}t�t j�� |�� W d � n1 sF0 Y d S )Nr� i�� )r< rw � _psplatformr� r r� r� �exe)r> r� rH r r r �test_exe? 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� z 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 s t � j| _d S rF r� r� r r r r� K s zTestProcessWMI.setUpClassc C s t | j� d S rF r� r� r r r r� O s z TestProcessWMI.tearDownClassc C s8 t �� j| jd�d }t�| j�}|�� |jks4J �d S �Nrt r )rK rL ru r� r< r� r� ZCaption�r> rO r� r r r � test_nameS s zTestProcessWMI.test_namez!unreliable path on GITHUB_ACTIONSr c C s@ t �� j| jd�d }t�| j�}|�� �� |j�� ks<J �d S r ) rK rL ru r� r< r� r rb ZExecutablePathr r r r r Y s zTestProcessWMI.test_exec C sF t �� j| jd�d }t�| j�}d�|�� �|j� dd�ksBJ �d S )Nrt r r^ r� r# )
rK rL ru r� r< r� r. r! Z CommandLinera r r r r r� c s zTestProcessWMI.test_cmdlinec C sP t �� j| jd�d }t�| j�}|�� \}}}d||f }|�� |ksLJ �d S )Nrt r z%s\%s)rK rL ru r� r< r� ZGetOwnerr� )r> rO r� �domain�_r� r r r r� h s
z TestProcessWMI.test_usernamec C sB t �� j| jd�d }t�| j�}|�� j}|t|j �ks>J �d S r )
rK rL ru r� r< r� r� r� r9 r� )r> rO r� r� r r r �test_memory_rsso s
z TestProcessWMI.test_memory_rssc C s\ t �� j| jd�d }t�| j�}|�� j}t|j �}|||d fvrX| �
d||f ��d S )Nrt r r� zwmi=%s, psutil=%s) rK rL ru r� r< r� r� r� r9 Z
PageFileUsagerc )r> rO r� r� Z wmi_usager r r �test_memory_vmsv s
z TestProcessWMI.test_memory_vmsc C s\ t �� j| jd�d }t�| j�}t|j�d�d �}t �
dt � |� � ��}||ksXJ �d S )Nrt r r� r� )
rK rL ru r� r< r� r� Z CreationDater1 �time�strftime� localtimer� )r> rO 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 r[ r\ r r r� r� r r
r r r r r r r H 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 s t � j| _d S rF r� r� r r r r� � s z(TestDualProcessImplementation.setUpClassc C s t | j� d S rF r� r� r r r r� � s z+TestDualProcessImplementation.tearDownClassc C s� t �| j��� }tjdttjd�d���}t �| j��� }t |�t |�ksLJ �t
t |��D ]@}|| dkslJ �|| dks|J �t || || �dk sXJ �qX|j s�J �W d � n1 s�0 Y d S )Nz(psutil._psplatform.cext.proc_memory_info�msgr� r i )
r< r� r� r� r r� �OSError�errno�EPERMrS r� rj r� )r> Zmem_1�funZmem_2r� r r r r� � s
� z.TestDualProcessImplementation.test_memory_infoc C sl t �| j��� }tjdttjd�d��2}t �| j��� |ks@J �|j sJJ �W d � n1 s^0 Y d S )N�"psutil._psplatform.cext.proc_timesr r� )
r< r� r� r� r r� r r r r� )r> �ctimer r r r r � s
�z.TestDualProcessImplementation.test_create_timec C s� t �| j��� }tjdttjd�d��Z}t �| j��� }|j sBJ �t
|j |j �dk sZJ �t
|j |j �dk srJ �W d � n1 s�0 Y d S )Nr r r� g{�G�z�?)
r< r� r� rT r r� r r r r� rj �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��� }tjdttjd�d��X}t �| j��� }t t
|��D ] }t || || �dk sDJ �qD|j spJ �W d � n1 s�0 Y d S )Nz(psutil._psplatform.cext.proc_io_countersr r� r@ )
r< r� r� r� r r� r r r r� rS rj r� )r> Z
io_counters_1r Z
io_counters_2r� r r r r� � s
� z.TestDualProcessImplementation.test_io_countersc C sl t �| j��� }tjdttjd�d��2}t �| j��� |ks@J �|j sJJ �W d � n1 s^0 Y d S )Nz(psutil._psplatform.cext.proc_num_handlesr r� )
r< r� r� r� r r� r r r r� )r> r� r r r r r� � s
�z.TestDualProcessImplementation.test_num_handlesc C s� t �� D ]r}z tj|dd�}tj|dd�}W n@ tyl } z(t|�}t|t jt jf�sX� W Y d }~qd }~0 0 ||ksJ �qd S )NT)Zuse_pebF) r< rw 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 ]H}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�filenamerH �outputr r r r �find_other_interpreter� s � z,RemoteProcessTestCase.find_other_interpreterr$ z import sys; sys.stdin.read()c s� t � �� | �� }|d u r$t�d��tr8tj| _|| _ 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)r �setUpr1 r r r �sys�
executableZpython64Zpython32r: r; �copyr� r� r � test_argsr) r+ �proc32�proc64)r> Z other_pythonr3 �� __class__r r r5 s&
�
��zRemoteProcessTestCase.setUpc s" t � �� | j�� | j�� d S rF )r �tearDownr: r- r; rU r< r r r> s
z RemoteProcessTestCase.tearDownc C s@ t �| jj�}t|�� �dks"J �|�� dd � | jks<J �d S �N� r$ )r< r� r: r� rS r! r9 r� r r r �test_cmdline_32! s z%RemoteProcessTestCase.test_cmdline_32c C s@ t �| jj�}t|�� �dks"J �|�� dd � | jks<J �d S r? )r< r� r; r� rS r! r9 r� r r r �test_cmdline_64& s z%RemoteProcessTestCase.test_cmdline_64c C s& t �| jj�}|�� t�� ks"J �d S rF )r< r� r: r� r r: �getcwdr� r r r � test_cwd_32+ s z!RemoteProcessTestCase.test_cwd_32c C s& t �| jj�}|�� t�� ks"J �d S rF )r< r� r; r� r r: rC r� r r r � test_cwd_64/ s z!RemoteProcessTestCase.test_cwd_64c C s> t �| jj�}|�� }d|v s"J �|d tt�� �ks:J �d S )Nr2 )r< r� r: r� r; r� r: r� )r> r� �er r r �test_environ_323 s z%RemoteProcessTestCase.test_environ_32c C s4 t �| jj�}z |�� W n t jy. Y n0 d S rF )r< r� r; r� r; r� r� r r r �test_environ_649 s
z%RemoteProcessTestCase.test_environ_64)r r r r � staticmethodr1 r9 r5 r> rA rB rD rE rG rH �
__classcell__r r r< r r# � s
r# c @ s e Zd Zdd� Zdd� ZdS )� TestServicesc C s` t g d��}t g d��}t g d��}t�� D �],}|�� }t|d t�sLJ �|d �� s\J �t|d t�snJ �t|d t�s�J �|d |v s�J �|d d ur�t�|d � t|d t�s�J �t|d t�s�J �t|d
t�s�J �|d
|v s�J �|d |v �sJ �t|d t��sJ �|�� }|d u�r@t�|�}|� � �s@J �t�
|� � �}||ks,J �q,d S ) N)�running�paused�start�pause�continue�stop�stopped)Z automaticZmanual�disabled)rL rM Z
start_pendingZ
pause_pendingZcontinue_pendingZ stop_pendingrR r� � display_namer� �statusr� ZbinpathZ
start_type� description) rv r< �win_service_iterZas_dictr r� r- r� r� �
is_running�win_service_getr� )r> Zvalid_statusesZvalid_start_typesZservr5 r� r� �sr r r �test_win_service_iterH s0
z"TestServices.test_win_service_iterc C s� t jjj}t jjj}tt �� ��� }t� t j
�� }t � |d � W d � n1 sT0 Y |j j|d ksrJ �t � |�}t
r�ddd|f}n|df}t|� }tjd|d��D t� t j
�� |�� W d � n1 s�0 Y W d � n1 s�0 Y tjd|d��F t� t j
�� |�� W d � n1 �s<0 Y W d � n1 �s\0 Y t
�rzddd|f}n|df}t|� }tjd|d��F t� t j�� |�� W d � n1 �s�0 Y W d � n1 �s�0 Y tjd|d��F t� t j�� |�� W d � n1 �s,0 Y W d � n1 �sL0 Y |�� t|�v �slJ �|�� t|�v �s�J �|�� t|�v �s�J �|�� t|�v �s�J �d S )Nz???r r z/psutil._psplatform.cext.winservice_query_statusr� z/psutil._psplatform.cext.winservice_query_config)r< r r� � ERROR_SERVICE_DOES_NOT_EXIST�ERROR_ACCESS_DENIED�nextrW r� r r� r� rY r� r
r� r r� rU r� r� r� rT r� )r> r\ r] r� �cmZservicer% r r r r �test_win_service_gett sP �
,
�D�H�H�Hz!TestServices.test_win_service_getN)r r r r[ r` r r r r rK F s ,rK ):r r� r r( r: r� r� r� r) r6 r �warningsr< r Zpsutil._compatr r r Z psutil.testsr r r r r
r r r
r r r r r r r �catch_warnings� simplefilterrA r� r� rK Zpsutil._pswindowsr r r� r[ r\ r r" r9 r6 r7 r] r� r� r r r# rK r r r r �<module> s|
( � ( #F OHW]