관리-도구
편집 파일: test_windows.cpython-38.pyc
U �Mg � @ 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 �"� �0 e �#d� ddl$Z$ddl%Z%ddl&Z&ddl'Z'W 5 Q R X 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 �I/opt/hc_python/lib64/python3.8/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� ��� }� dk rbd|krXt� 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>^ s zwmic.<locals>.<listcomp>)r �strip�join� splitlines�tuple�split)�path�whatr* �out�datar r) r �wmicT s r5 c @ sR e Zd Zejjdejkdd�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 t�d S )Nr7 )�int�os�environ�psutil� cpu_count�AssertionError)�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 t�d S )N� )�win32apiZ GetSystemInfor; r< 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.t�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= )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�|ks2t�d S )Nc s s | ] }|j V qd S rF )Z NumberOfCoresrG r r r rI � s z:TestCpuAPIs.test_cpu_count_cores_vs_wmi.<locals>.<genexpr>F)ZlogicalrJ )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 t�d S )NT)Zpercpu)r; r< �len� cpu_timesr= �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(t�|jt�� j ks<t�d S �Nr ) rK rL rN ZCurrentClockSpeedr; Zcpu_freq�currentr= Z MaxClockSpeed�max)r> rO rH r r r � test_cpu_freq� s zTestCpuAPIs.test_cpu_freqN)r r r r �mark�skipifr9 r: r? rE rP rQ rU rY r r r r r6 j s � r6 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��� kr6q ||kr | �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> r3 ZnicsZnicr r r �test_nic_names� s �z TestSystemAPIs.test_nic_namesc C s, t �� �� d }t|j�t�� jks(t�d S rV ) rK rL ZWin32_ComputerSystemr8 ZTotalPhysicalMemoryr; �virtual_memory�totalr= �r> rO r r r �test_total_phymem� s z TestSystemAPIs.test_total_phymemc C s4 t �� �� d }tt|j�t�� j �t k s0t �d S rV ) rK rL � Win32_PerfRawData_PerfOS_Memory�absr8 ZAvailableBytesr; rd �freer r= rf r r r �test_free_phymem� s ��z TestSystemAPIs.test_free_phymemc C sh t �� �� d }t|j�t�� j t�� jks2t �t�� jdkrdt�� j dksRt �t�� j dksdt �d S rV ) rK rL rh r8 Z CommitLimitr; rd re � swap_memoryr= rj �usedrf 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 dksLt �t t �� j | �dk sft �t �� j dksxt �d S )Nr Z_Total)�Name�d r@ ) r; rl re rK rL Z#Win32_PerfRawData_PerfOS_PagingFiler8 Z PercentUsageZPercentUsage_Base�percentr= ri )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�� �}||ks6t�d S )Nc S s g | ] }|j �qS r �Z ProcessIdr& r r r r+ � s z,TestSystemAPIs.test_pids.<locals>.<listcomp>)rK rL � Win32_Process�setr; �pidsr= )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 krT q |j� d�rd q zt � |j�}W n t k r� Y q Y nX |j t|j�ks�t�t|j�}|j|ks�t�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_LogicalDiskZdevicer` ZDeviceID� mountpoint�opts� startswith� disk_usager re r8 �Sizer= Z FreeSpacerj ri rb �repr)r> Zps_partsZ wmi_partsZps_partZwmi_part�usageZwmi_freer r r � test_disks� s0 �zTestSystemAPIs.test_disksc C s� t �� D ]r}d|jkrqt�|j�}t �|j�}t|d |j �t k sJt �t|d |j �t k sdt �|j |j |j kst �qd S )Nr{ r r$ ) r; r} r rA ZGetDiskFreeSpaceExr~ r� ri rj r r= re rm )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:t�d S )Nc S s" g | ]}|r|� d �s|d �qS )r| rz )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 �Trx )rA ZGetLogicalDriveStringsr0 r; 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"||@ sXt d||f ��d S )Nzno common entries in %s, %s) ru �cextZ net_if_statsrK rL ZWin32_NetworkAdapter�addro ZNetConnectionIDr= )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 || � � �}|dksZt �d S )Nr �.� %Y%m%d%H%M%Sr@ ) rK rL ZWin32_OperatingSystemZLastBootUpTimer0 �datetime�strptime� fromtimestampr; � boot_timeri � total_secondsr= )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 t�W 5 Q R X t jddd�� t�� dksJt�W 5 Q R X t jddd�� t�� dkstt�W 5 Q R X t jddd�� t�� dks�t�W 5 Q R X d S )Nz psutil._pswindows.cext.boot_timer@ �Z return_value� � iM )r �patchr; r� r= rT r r r �test_boot_time_fluctuation# s z)TestSystemAPIs.test_boot_time_fluctuationN)r r r rc rg rk rn rr r rZ r[ r r rw 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 k s.t�nt�� d ks.t�d S )NZSystemBatteriesPresent)rA ZGetPwrCapabilitiesr; �sensors_batteryr= rT 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 s6t�d S )N�select * from Win32_Batteryr r$ ) rK rL �queryr; r� ri rq ZEstimatedChargeRemainingr= �r> rO Z battery_wmiZbattery_psutilr r r � test_percent; s ��z TestSensorsBattery.test_percentc C s6 t �� }|�d�d }t�� }|j|jdkks2t�d S )Nr� r � )rK rL r� r; r� Z power_pluggedZ BatteryStatusr= r� r r r �test_power_pluggedE s z%TestSensorsBattery.test_power_pluggedc C s8 t jddd�� }t�� d ks t�|js*t�W 5 Q R X d S )N�&psutil._pswindows.cext.sensors_battery)r � 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 s< t jddd��$}t�� jtjks$t�|js.t�W 5 Q R X d S )Nr� )r$ r r r r� �r r� r; r� �secsleftZPOWER_TIME_UNLIMITEDr= r� r� r r r � test_emulate_power_connectedV s ���z/TestSensorsBattery.test_emulate_power_connectedc C s< t jddd��$}t�� jtjks$t�|js.t�W 5 Q R X d S )Nr� )r � r r r� r� r� r r r �test_emulate_power_charging` s ���z.TestSensorsBattery.test_emulate_power_chargingc C s< t jddd��$}t�� jtjks$t�|js.t�W 5 Q R X d S )Nr� )r r r ���r� )r r� r; r� r� ZPOWER_TIME_UNKNOWNr= r� r� r r r � test_emulate_secs_left_unknownj s ��z1TestSensorsBattery.test_emulate_secs_left_unknownN)r r r r� r rZ 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 s. t �d�}t�t j�� |�� W 5 Q R X d S rV )r; �Processr �raises� AccessDenied�kill�r> �pr r r � test_issue_24� s zTestProcess.test_issue_24c C s� t �d�}|�� dkst�t|� |�� |�� dks:t�z|�� d d� \}}W n( t jk rz t � � d dkrv� Y nX |dks�t�d S )Nr� �Systemg r� r$ )Zvistazwin-7Zwin7r ) r; r� �namer= �str�username� create_time� memory_infor� �platform�uname)r> r� �rssZ_vmsr r r �test_special_pid� s z TestProcess.test_special_pidc C s2 t �| j�}t�t�� |�tj� W 5 Q R X 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 ksDt �t� |� |�� |ks^t �d S )Nr$ ) r; r� r9 �getpid� num_handlesrA � OpenProcess�win32con�PROCESS_QUERY_INFORMATION�FALSEr= � 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 5 Q R X t � t j �� |�tj� W 5 Q R X 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�� � � |ks4t �d S )N�$zrunning as service account) rA Z GetUserNameExr� ZNameSamCompatible�endswithr r 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�}||ksbt �d S )Nz[ ]+r] r �"r# r� ) �re�subrA ZGetCommandLiner, r- r; r� r! r` r= rB r r r � test_cmdline� s zTestProcess.test_cmdlinec C sJ t �tjtjt�� �}| �t j|� t � |�}t � � � � }||ksFt�d S rF )rA r� r� r� r� r9 r� � addCleanupr� � win32processZGetPriorityClassr; r� �nicer= �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 ksNt�|d |jks`t�|d |jksrt�|d |jks�t�|d |jks�t�|d |jks�t�|d |jks�t�|d |jks�t�|j|jks�t�|j|jks�t�d S ) NZPeakWorkingSetSize�WorkingSetSizeZQuotaPeakPagedPoolUsageZQuotaPagedPoolUsageZQuotaPeakNonPagedPoolUsageZQuotaNonPagedPoolUsageZ PagefileUsageZPeakPagefileUsage)rA r� r� r� r� r� r� r� r� ZGetProcessMemoryInfor; r� r� Z peak_wsetr= ZwsetZpeak_paged_poolZ paged_poolZpeak_nonpaged_poolZ nonpaged_poolZpagefileZ peak_pagefiler� �vmsr� r r r �test_memory_info� s2 � ���� �z TestProcess.test_memory_infoc C sX t �tjtj| j�}| �t j|� t� | j�}|� � |� � }t � |�}||ksTt�d S rF )rA r� r� r� r� r� r� r� r; r� r r� r� ZGetExitCodeProcessr= )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�� � }||ksXt �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 )rA r� r� r� r� r� r� r� r� ZGetProcessAffinityMaskr; r� Z cpu_affinityr= )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 ksLt�|j|d ks^t�|j|d kspt�|j|d ks�t�|j|d ks�t�|j|d ks�t�d S )NZReadOperationCountZWriteOperationCountZReadTransferCountZWriteTransferCountZOtherOperationCountZOtherTransferCount)rA r� r� r� r� r9 r� r� r� r� ZGetProcessIoCountersr; r� � io_countersZ read_countr= Z 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|t�d S )Nr � )�ctypesZctypes.wintypes�windll�kernel32r� r� r� r� ZwintypesZDWORDZGetProcessHandleCountZbyref�valuer; r� r� r= )r> r� r� r� ZhndcntrC rD r r r �test_num_handles" s" � �z TestProcess.test_num_handlesc C sv t � }d|_tjd|d��D t�d��.}t�� }t�tj�� |� � W 5 Q R X W 5 Q R X W 5 Q R X |j dksrt �d S )Ni+ z psutil._psplatform.cext.proc_cwd�Z side_effectz time.sleepr@ ) � WindowsError�winerrorr r� r; r� r r� r� �cwdZ call_countr= )r> �excr� r� r r r �test_error_partial_copy4 s &z#TestProcess.test_error_partial_copyc C s@ t �� d d }t j�|�}t�t j�� |�� W 5 Q R X d S )Nr� i�� )r; rv � _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�}|�� |jks4t�d S �Nrs r ) rK rL rt r� r; r� r� ZCaptionr= �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<t �d S r ) rK rL rt r� r; r� r ra ZExecutablePathr= r r r r r Y s zTestProcessWMI.test_exec C sF t �� j| jd�d }t�| j�}d�|�� �|j� dd�ksBt �d S )Nrs r r] r� r# ) rK rL rt r� r; r� r- r! Z CommandLiner` r= r r r r r� c s zTestProcessWMI.test_cmdlinec C sP t �� j| jd�d }t�| j�}|�� \}}}d||f }|�� |ksLt�d S )Nrs r z%s\%s) rK rL rt r� r; r� ZGetOwnerr� r= )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>t �d S r ) rK rL rt r� r; r� r� r� r8 r� 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 fkrX| � d||f ��d S )Nrs r r� zwmi=%s, psutil=%s) rK rL rt r� r; r� r� r� r8 Z PageFileUsagerb )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 � |� � ��}||ksXt �d S )Nrs r r� r� )rK rL rt r� r; r� r� Z CreationDater0 �time�strftime� localtimer� r= )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 rZ 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 |�ksLt �t t |��D ]@}|| dkslt �|| dks|t �t || || �dk sXt �qX|j s�t �W 5 Q R X d S )Nz(psutil._psplatform.cext.proc_memory_info�msgr� r i )r; r� r� r� r r� �OSError�errno�EPERMrR r= r� ri r� )r> Zmem_1�funZmem_2r� r r r r� � s � z.TestDualProcessImplementation.test_memory_infoc C sX t �| j��� }tjdttjd�d��(}t �| j��� |ks@t �|j sJt �W 5 Q R X d S )N�"psutil._psplatform.cext.proc_timesr r� ) 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��P}t �| j��� }|j sBt �t |j |j �dk sZt �t |j |j �dk srt �W 5 Q R X d S )Nr r r� g{�G�z�?)r; r� r� rS r r� r r r r� r= ri �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��N}t �| j��� }t t |��D ] }t || || �dk sDt �qD|j spt �W 5 Q R X d S )Nz(psutil._psplatform.cext.proc_io_countersr r� r@ )r; r� r� r� r r� r r r r� rR ri r= r� )r> Z io_counters_1r Z io_counters_2r� r r r r� � s � z.TestDualProcessImplementation.test_io_countersc C sX t �| j��� }tjdttjd�d��(}t �| j��� |ks@t �|j sJt �W 5 Q R X 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 r� � s �z.TestDualProcessImplementation.test_num_handlesc C s~ t �� D ]p}z tj|dd�}tj|dd�}W n> tk rj } z t|�}t|t jt jf�sZ� W 5 d }~X Y qX ||kst �qd S )NT)Zuse_pebF) r; rv r� Z proc_cmdliner r � isinstancer� r� 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 kr$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 �setUpr0 r r r �sys� executableZpython64Zpython32r9 r: �copyr� r� r � test_argsr( r* �proc32�proc64)r> Z other_pythonr2 �� __class__r r r4 s. � � �zRemoteProcessTestCase.setUpc s" t � �� | j�� | j�� d S rF )r �tearDownr9 r, r: rT r; r r r= s z RemoteProcessTestCase.tearDownc C s@ t �| jj�}t|�� �dks"t�|�� dd � | jks<t�d S �N� r$ )r; r� r9 r� rR r! r= r8 r� r r r �test_cmdline_32! s z%RemoteProcessTestCase.test_cmdline_32c C s@ t �| jj�}t|�� �dks"t�|�� dd � | jks<t�d S r> )r; r� r: r� rR r! r= r8 r� r r r �test_cmdline_64& s z%RemoteProcessTestCase.test_cmdline_64c C s& t �| jj�}|�� t�� ks"t�d S rF )r; r� r9 r� r� r9 �getcwdr= r� r r r � test_cwd_32+ s z!RemoteProcessTestCase.test_cwd_32c C s& t �| jj�}|�� t�� ks"t�d S rF )r; r� r: r� r� r9 rB r= r� r r r � test_cwd_64/ s z!RemoteProcessTestCase.test_cwd_64c C s> t �| jj�}|�� }d|ks"t�|d tt�� �ks:t�d S )Nr1 ) r; r� r9 r� r: r= r� r9 r� )r> r� �er r r �test_environ_323 s z%RemoteProcessTestCase.test_environ_32c C s6 t �| jj�}z |�� W n t jk r0 Y nX 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 � staticmethodr0 r8 r4 r= r@ rA rC rD rF rG � __classcell__r r r; r r" � s r" c @ s e Zd Zdd� Zdd� ZdS )� TestServicesc C sx t dddddddg�}t dd d g�}t ddd d d ddg�}t�� D �].}|�� }t|d t�sbt�|d �� srt�t|d t�s�t�t|d t�s�t�|d |ks�t�|d d k r�t�|d � t|d t�s�t�t|d t�s�t�t|d t�s�t�|d |k�st�|d |k�st�t|d t��s.t�|� � }|d k �rXt�|�}|� � �sXt�t� |� � �}||ksBt�qBd S )N�runningZpaused�start�pause�continue�stop�stoppedZ automaticZmanual�disabledZ start_pendingZ pause_pendingZcontinue_pendingZ stop_pendingr� � display_namer� �statusr� ZbinpathZ start_type� description) ru r; �win_service_iter�as_dictr r� r= r, r� r� Z is_running�win_service_getr� )r> Zvalid_statusesZvalid_start_typesZservr4 r� r� �sr r r �test_win_service_iterH sP � � z"TestServices.test_win_service_iterc C s� t jjj}t jjj}tt �� ��� }t� t j ��}t � |d � W 5 Q R X |j j|d ks^t �t � |�}trzddd|f}n|df}t|� }tjd|d��& t� t j �� |�� W 5 Q R X W 5 Q R X tjd|d��& t� t j �� |�� W 5 Q R X W 5 Q R X t�rddd|f}n|df}t|� }tjd|d��& t� t j�� |�� W 5 Q R X W 5 Q R X tjd|d��& t� t j�� |�� W 5 Q R X W 5 Q R X |�� t|�k�s�t �|�� t|�k�s�t �|�� t|�k�s�t �|�� t|�k�s�t �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�nextrU r� r r� r� rW r� r= r r� r r� rS r� r� r� rR r� )r> rZ r[ r� �cmZservicer$ r r r r �test_win_service_gett sX � � � � � z!TestServices.test_win_service_getN)r r r rY r^ r r r r rJ F s ,rJ ):r r� r r' r9 r� r� r� r( r5 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� rZ r[ r r" r8 r5 r6 r\ r� r� r r r" rJ r r r r �<module> s~ � ( #F OHW]