Your IP : 18.117.184.125
3
��JZK � @ 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 e j� �Z e jd� y$ddl Z ddl Z ddl Z ddl Z W n$ e k
�rD ej!dk�r@� Y nX W dQ R X e j"j#Z#e j$d$kZ%dd� Z&ej'e
d�G dd� dej(��Z)ej'e
d�G dd� dej(��Z*ej'e
d�G dd� dej(��Z+ej'e
d�G dd � d ej(��Z,ej'e
d�G d d � d ej(��Z-ej'e
d�G d d � d ej(��Z.ej'e
d�G d!d"� d"ej(��Z/e0d#k�r\ee1� dS )%zWindows specific tests.� N)�WINDOWS)�callable)�APPVEYOR)�get_test_subprocess)� HAS_BATTERY)�mock)�
reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�unittest�ignore�nt� � c s � fdd�}|S )Nc st y� | f|�|�S t k
rn } zBddlm} |j|krDtjd d ��|jtjkr\tjd d ��� W Y d d }~X nX d S )Nr )�ACCESS_DENIED_SET)�OSErrorZpsutil._pswindowsr �errno�psutil� AccessDeniedZESRCH�
NoSuchProcess)�self�args�kwargs�errr )�fun� �$/usr/lib64/python3.6/test_windows.py�wrapper5 s
z wrap_exceptions.<locals>.wrapperr )r r r )r r �wrap_exceptions4 s
r z WINDOWS onlyc @ s� e Zd Zdd� Zejdejkd�dd� �Zdd� Z d d
� Z
d d � Z eje d
�e
� dd� ��Ze
� dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zd S ) �TestSystemAPIsc C sV t d�}tjdd�j� }x8|D ]0}d|jdd�j� kr8q ||kr | jd| � q W 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 �outZnicsZnicr r r �test_nic_namesJ s
z TestSystemAPIs.test_nic_names�NUMBER_OF_PROCESSORSz-NUMBER_OF_PROCESSORS env var is not availablec C s" t tjd �}| j|tj� � d S )Nr) )�int�os�environ� assertEqualr � cpu_count)r Znum_cpusr r r �test_cpu_countT s z TestSystemAPIs.test_cpu_countc C s$ t j� d }tj� }| j||� d S )N� )�win32apiZ
GetSystemInfor r. r- )r � sys_value� psutil_valuer r r �test_cpu_count_2Z s z TestSystemAPIs.test_cpu_count_2c C s@ t j� }|j� d }| j|jtj� j� | j|jtj� j � d S )Nr )
�wmi�WMIZWin32_Processorr- ZCurrentClockSpeedr Zcpu_freqZcurrentZ
MaxClockSpeed�max)r �w�procr r r �
test_cpu_freq_ s z TestSystemAPIs.test_cpu_freqc C s, t j� j� d }| jt|j�tj� j� d S )Nr ) r5 r6 ZWin32_ComputerSystemr- r* ZTotalPhysicalMemoryr Zvirtual_memory�total)r r8 r r r �test_total_phymeme s z TestSystemAPIs.test_total_phymemz test not relieable on appveyorc C s: t j� j� }tdd� |D ��}ttj� �}| j||� d S )Nc S s g | ]
}|j �qS r )� ProcessId)�.0�xr r r �
<listcomp>| s z,TestSystemAPIs.test_pids.<locals>.<listcomp>)r5 r6 �
Win32_Process�setr Zpidsr- )r r8 Zwmi_pidsZ psutil_pidsr r r � test_pidsv s zTestSystemAPIs.test_pidsc C s t jdd�}tj� j� }x�|D ]�}x�|D ]�}|jjdd�|jkr(|jsHP yt j |j�}W n4 t
k
r� } z|j t j krzP n� W Y d d }~X nX | j
|jt|j�� t|j�}| j
|j|� t|j| �d
kr�| jd|j|f � P q(W | jdt|� � q W d S ) NT)�all�\� �
i zpsutil=%s, wmi=%szcan't find partition %si ( i � )r �disk_partitionsr5 r6 ZWin32_LogicalDiskZdevicer$ ZDeviceID�
mountpoint�
disk_usager r �ENOENTr- r; r* ZSizeZ FreeSpace�free�absr& �repr)r Zps_partsZ wmi_partsZps_partZwmi_partZusager Zwmi_freer r r �
test_disks� s*
zTestSystemAPIs.test_disksc C sp xjt j� D ]^}tj|j�}t j|j�}| j|d |jdd� | j|d |jdd� | j |j
|j|j � q
W d S )Nr i )�delta� i i ) r rH r1 ZGetDiskFreeSpaceExrI rJ �assertAlmostEqualrL r; r- Zused)r Zdiskr2 r3 r r r �test_disk_usage� s z TestSystemAPIs.test_disk_usagec C s> dd� t j� jd�D �}dd� tjdd�D �}| j||� d S )Nc S s$ g | ] }|r|j d � r|d �qS )zA:rE )�
startswith)r> r? r r r r@ � s z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>z\ c S s g | ]
}|j �qS r )rI )r> r? r r r r@ � s T)rD )r1 ZGetLogicalDriveStrings�splitr rH r- )r r2 r3 r r r �test_disk_partitions� s z#TestSystemAPIs.test_disk_partitionsc C s` t tj� �}tj� j� }t � }x$|D ] }|j|j� |j|j� q$W | j ||@ d||f � d S )Nzno common entries in %s, %s)
rB �cextZ net_if_statsr5 r6 ZWin32_NetworkAdapter�add�NameZNetConnectionID�
assertTrue)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 j� j� }|d jjd�d }tjj|d�}tjjtj � �}t
|| j � �}| j |d� d S )Nr �.z %Y%m%d%H%M%S� )
r5 r6 ZWin32_OperatingSystemZLastBootUpTimerU �datetimeZstrptimeZ
fromtimestampr � boot_timerM Z
total_secondsZassertLessEqual)r Zwmi_osZ
wmi_btime_strZ wmi_btime_dtZ psutil_dtZdiffr r r �test_boot_time� s z TestSystemAPIs.test_boot_timec C s� t jddd�� | jtj� d� W d Q R X t jddd�� | jtj� d� W d Q R X t jddd�� | jtj� d� W d Q R X t jddd�� | jtj� d� W d Q R X d S )Nz psutil._pswindows.cext.boot_timer0 )� return_value� � iM )r �patchr- r r_ )r r r r �test_boot_time_fluctuation� s z)TestSystemAPIs.test_boot_time_fluctuationN)�__name__�
__module__� __qualname__r( r �skipIfr+ r, r/ r4 r: r< r r rC rO rS rV r[ r` re r r r r r G s
r c @ s` e Zd Zdd� Zeje d�dd� �Zeje d�dd� �Zdd � Z d
d � Z
d d
� Z dd� Z dS )�TestSensorsBatteryc C s. t j� d r | jtj� � n| jtj� � d S )NZSystemBatteriesPresent)r1 ZGetPwrCapabilitiesZassertIsNotNoner �sensors_battery� assertIsNone)r r r r �test_has_battery� s z#TestSensorsBattery.test_has_batteryz
no batteryc C s6 t j� }|jd�d }tj� }| j|j|jdd� d S )Nzselect * from Win32_Batteryr rQ )rP )r5 r6 �queryr rk rR ZpercentZEstimatedChargeRemaining)r r8 � battery_wmi�battery_psutilr r r � test_percent� s z TestSensorsBattery.test_percentc C s6 t j� }|jd�d }tj� }| j|j|jdk� d S )Nzselect * from Win32_Batteryr r )r5 r6 rn r rk r- Z
power_pluggedZ
BatteryStatus)r r8 ro rp r r r �test_power_plugged� s
z%TestSensorsBattery.test_power_pluggedc C s6 t jddd�� }| jtj� � |js(t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_batteryr � )ra )r rs r r )r rd rl r rk �called�AssertionError)r �mr r r �test_emulate_no_battery� s
z*TestSensorsBattery.test_emulate_no_batteryc
C s< t jddd��$}| jtj� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_batteryrQ r )ra )rQ r r r ) r rd r- r rk �secsleft�POWER_TIME_UNLIMITEDrt ru )r rv r r r � test_emulate_power_connected� s
z/TestSensorsBattery.test_emulate_power_connectedc
C s< t jddd��$}| jtj� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_batteryr � )ra )r r{ r r ) r rd r- r rk rx ry rt ru )r rv r r r �test_emulate_power_charging� s
z.TestSensorsBattery.test_emulate_power_chargingc
C s< t jddd��$}| jtj� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_batteryr rQ )ra ���)r r r r} ) r rd r- r rk rx ZPOWER_TIME_UNKNOWNrt ru )r rv r r r � test_emulate_secs_left_unknown s
z1TestSensorsBattery.test_emulate_secs_left_unknownN)
rf rg rh rm r ri r rq rr rw rz r| r~ r r r r rj � s
rj c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z e
jejd+k d�dd� �Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*S ),� TestProcessc C s t � j| _d S )N)r �pid)�clsr r r �
setUpClass s zTestProcess.setUpClassc C s
t � d S )N)r )r� r r r �
tearDownClass s zTestProcess.tearDownClassc C s t jd�}| jt j|j� d S )Nr )r �Process� assertRaisesr �kill)r �pr r r �
test_issue_24 s
zTestProcess.test_issue_24c C s� t jd�}| j|j� d� t|� |j� | j|j� dk� y|j� d d� \}}W n( t j k
r| t
j � d d
krx� Y nX | j|d k� d S ) Nrb ZSystemg r rQ �vista�win-7�win7r )r� r� r� ) r r� r- �name�str�usernamerZ � create_time� memory_infor �platform�uname)r r� �rss�vmsr r r �test_special_pid s
z TestProcess.test_special_pidc C s" t j| j�}| jt|jtj� d S )N)r r� r� r� �
ValueError� send_signal�signal�SIGINT)r r� r r r �test_send_signal1 s z TestProcess.test_send_signalc C sN xHt j� D ]<}y | jtjj|j� �|j� � W q
t jk
rD Y q
X q
W d S )N) r � process_iterr- r+ �path�basename�exer� �Error)r r� r r r �test_exe5 s
zTestProcess.test_exec C sb t jtj� �}|j� }tjtjtj tj� �}|j� }| j
||d � tj |� | j
|j� |� d S )NrQ ) r r� r+ �getpid� num_handlesr1 � OpenProcess�win32con�PROCESS_QUERY_INFORMATION�FALSEr- � CloseHandle)r r� Zbefore�handle�afterr r r �test_num_handles_increment< s
z&TestProcess.test_num_handles_incrementc
s� � fdd�}t j| j�}g }x�tt j�D ]�� � jd�s(� dkrBq(q(y(||� � |j� }||� � |j� }W n t jt jfk
r� Y q(X ||kr(d� ||f }|j|� q(W |r�| j ddj
|� � d S )Nc s, t | � d �}|d k r$t|�r$|� n| d S )N)�getattrr )r� �attr)r� r r �callJ s z+TestProcess.test_handles_leak.<locals>.call�_� terminater� �suspend�resume�nicer� �wait�children�as_dict�memory_info_exz@failure while processing Process.%s method (before=%s, after=%s)�
)
r� r� r� r� r� r� r� r� r� r� ) r r� r� �dirrT r� r r �appendr& �join)r r� r� ZfailuresZnum1Znum2r& r )r� r �test_handles_leakF s,
z TestProcess.test_handles_leakc C s: x4t j� D ](}y |j� W q
t jk
r0 Y q
X q
W d S )N)r r� r� r )r r� r r r �test_name_always_availablej s
z&TestProcess.test_name_always_availabler � z CTRL_* signals not supportedc C sb t jt� j�}|jtj� |jtj� |j� |j � | j
t j |jtj� | j
t j |jtj� d S )N) r r� r r� r� r� Z CTRL_C_EVENTZCTRL_BREAK_EVENTr� r� r� r )r r� r r r �test_ctrl_signalss s z TestProcess.test_ctrl_signalsc
C s\ xVt j� D ]J}y tjj|j� �}|j� }W n t jt jfk
rF Y q
X | j ||� q
W d S )N)
r r� r+ r� r� r� r� r r r- )r r� �a�br r r �test_compare_name_exe� s z!TestProcess.test_compare_name_exec C s | j tj� j� tjtj�� d S )N)r- r r� r� r1 Z
GetUserNameExr� ZNameSamCompatible)r r r r �
test_username� s zTestProcess.test_usernamec C s8 t jddtj� �j� }djtj� j� �}| j ||� d S )Nz +r! )
�re�subr1 ZGetCommandLine�stripr� r r� �cmdliner- )r r2 r3 r r r � test_cmdline� s zTestProcess.test_cmdlinec C sJ t jtjtjtj� �}| jt j|� t j
|�}t j � j
� }| j||� d S )N)r1 r� r� r� r� r+ r� �
addCleanupr� � win32processZGetPriorityClassr r� r� r- )r r� r2 r3 r r r � test_nice� s
zTestProcess.test_nicec C s� t jtjtj| j�}| jt j|� tj |�}t
j | j�j � }| j
|d |j� | j
|d |j� | j
|d |j� | j
|d |j� | j
|d |j� | j
|d |j� | j
|d |j� | j
|d |j� | j
|j|j� | j
|j|j� d S ) NZPeakWorkingSetSize�WorkingSetSizeZQuotaPeakPagedPoolUsageZQuotaPagedPoolUsageZQuotaPeakNonPagedPoolUsageZQuotaNonPagedPoolUsageZ
PagefileUsageZPeakPagefileUsage)r1 r� r� r� r� r� r� r� r� ZGetProcessMemoryInfor r� r� r- Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZpagefileZ
peak_pagefiler� r� )r r� r2 r3 r r r �test_memory_info� s0
z TestProcess.test_memory_infoc C sX t jtjtj| j�}| jt j|� tj | j�}|j
� |j � }t j
|�}| j||� d S )N)r1 r� r� r� r� r� r� r� r r� r� r� r� ZGetExitCodeProcessr- )r r� r� r3 r2 r r r � test_wait� s
zTestProcess.test_waitc C s\ dd� }t jtjtj| j�}| jt j|� |tj |�d �}t
j | j�j � }| j
||� d S )Nc s � fdd�t d�D �S )Nc s g | ]}d |> � @ r|�qS )rQ r )r> �i)r? r r r@ � s zGTestProcess.test_cpu_affinity.<locals>.from_bitmask.<locals>.<listcomp>�@ )�range)r? r )r? r � from_bitmask� s z3TestProcess.test_cpu_affinity.<locals>.from_bitmaskr )r1 r� r� r� r� r� r� r� r� ZGetProcessAffinityMaskr r� Z cpu_affinityr- )r r� r� r2 r3 r r r �test_cpu_affinity� s z TestProcess.test_cpu_affinityc C s� t jtjtjtj� �}| jt j|� t j
|�}t j � j
� }| j|j|d � | j|j|d � | j|j|d � | j|j|d � | j|j|d � | j|j|d � d S )NZReadOperationCountZWriteOperationCountZReadTransferCountZWriteTransferCountZOtherOperationCountZOtherTransferCount)r1 r� r� r� r� r+ r� r� r� r� ZGetProcessIoCountersr r� � io_countersr- Z
read_countZ write_countZ
read_bytesZ write_bytes� other_countZ other_bytes)r r� r2 r3 r r r �test_io_counters� s"
z TestProcess.test_io_countersc C s� dd l }dd l}d}|jjj|dtj� �}| j|jjj|� |j j
� }|jjj ||j |�� |j
}tj� j� }|jjj|� | j||d � d S )Nr i rQ )�ctypesZctypes.wintypesZwindllZkernel32r� r+ r� r� r� ZwintypesZDWORDZGetProcessHandleCountZbyref�valuer r� r� r- )r r� r� r� Zhndcntr2 r3 r r r �test_num_handles� s
z TestProcess.test_num_handlesN)r r� )rf rg rh � classmethodr� r� r� r� r� r� r� r� r� r ri �sys� version_infor� r� r� r� r� r� r� r� r� r� r r r r r s(
$
r c @ s` e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )�TestProcessWMIz%Compare Process API results with WMI.c C s t � j| _d S )N)r r� )r� r r r r� s zTestProcessWMI.setUpClassc C s
t � d S )N)r )r� r r r r� s z TestProcessWMI.tearDownClassc C s8 t j� j| jd�d }tj| j�}| j|j� |j� d S )N)r= r ) r5 r6 rA r� r r� r- r� ZCaption)r r8 r� r r r � test_name
s zTestProcessWMI.test_namec C s@ t j� j| jd�d }tj| j�}| j|j� j� |j j� � d S )N)r= r )
r5 r6 rA r� r r� r- r� r% ZExecutablePath)r r8 r� r r r r� s zTestProcessWMI.test_exec C sF t j� j| jd�d }tj| j�}| jdj|j� �|j j
dd�� d S )N)r= r r! �"rF ) r5 r6 rA r� r r� r- r� r� Z CommandLiner$ )r r8 r� r r r r� s zTestProcessWMI.test_cmdlinec C sP t j� j| jd�d }tj| j�}|j� \}}}d||f }| j|j� |� d S )N)r= r z%s\%s) r5 r6 rA r� r r� ZGetOwnerr- r� )r r8 r� Zdomainr� r� r r r r� s
z TestProcessWMI.test_usernamec C sL t jd� tj� j| jd�d }tj| j�}|j� j }| j
|t |j �� d S )Ng�������?)r= r )
�time�sleepr5 r6 rA r� r r� r� r� r- r* r� )r r8 r� r� r r r �test_memory_rss# s
z TestProcessWMI.test_memory_rssc C sj t jd� tj� j| jd�d }tj| j�}|j� j }t
|j �}||krf||d krf| j d||f � d S )Ng�������?)r= r i zwmi=%s, psutil=%s)
r� r� r5 r6 rA r� r r� r� r� r* Z
PageFileUsager& )r r8 r� r� Z wmi_usager r r �test_memory_vms* s
z TestProcessWMI.test_memory_vmsc C s\ t j� j| jd�d }tj| j�}t|jjd�d �}t j
dt j |j � ��}| j
||� d S )N)r= r r\ z %Y%m%d%H%M%S)r5 r6 rA r� r r� r� Z CreationDaterU r� ZstrftimeZ localtimer� r- )r r8 r� Z wmic_createZ
psutil_creater r r �test_create_time7 s z TestProcessWMI.test_create_timeN)rf rg rh �__doc__r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ sX e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dS )� TestDualProcessImplementationa{
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 )N)r r� )r� r r r r� L s z(TestDualProcessImplementation.setUpClassc C s
t � d S )N)r )r� r r r r� P s z+TestDualProcessImplementation.tearDownClassc
C sZ t j| j�j� }tjdt jtj� �d��(}| j t j| j�j� |� |j
sLt �W d Q R X d S )Nz psutil._psplatform.cext.proc_exe)� side_effect) r r� r� r� r rd r r+ r� r- rt ru )r r� r r r r r� W s
z'TestDualProcessImplementation.test_namec
C s� t j| j�j� }tjdttjd�d���}t j| j�j� }| j t
|�t
|�� xLt t
|��D ]<}| j || d� | j || d� | j
|| || dd� qZW |js�t�W d Q R X d S )Nz(psutil._psplatform.cext.proc_memory_info�msg)r� r i )rP )r r� r� r� r rd r r �EPERMr- �lenr� ZassertGreaterEqualrR rt ru )r Zmem_1r Zmem_2r� r r r r� ^ s z.TestDualProcessImplementation.test_memory_infoc
C sX t j| j�j� }tjdttjd�d��(}| j t j| j�j� |� |j
sJt �W d Q R X d S )Nz(psutil._psplatform.cext.proc_create_timer� )r� ) r r� r� r� r rd r r r� r- rt ru )r Zctimer r r r r� j s
z.TestDualProcessImplementation.test_create_timec C sx t j| j�j� }tjdttjd�d��H}t j| j�j� }|j sBt
�| j |j |j dd� | j |j
|j
dd� W d Q R X d S )Nz&psutil._psplatform.cext.proc_cpu_timesr� )r� g{�G�z�?)rP )r r� r� Z cpu_timesr rd r r r� rt ru rR �user�system)r Z cpu_times_1r Z cpu_times_2r r r �test_cpu_timesq s
z,TestDualProcessImplementation.test_cpu_timesc
C s~ t j| j�j� }tjdttjd�d��N}t j| j�j� }x,t t
|��D ] }| j || || dd� qFW |j spt
�W d Q R X d S )Nz(psutil._psplatform.cext.proc_io_countersr� )r� r0 )rP )r r� r� r� r rd r r r� r� r� rR rt ru )r Z
io_counters_1r Z
io_counters_2r� r r r r� | s z.TestDualProcessImplementation.test_io_countersc
C sX t j| j�j� }tjdttjd�d��(}| j t j| j�j� |� |j
sJt �W d Q R X d S )Nz(psutil._psplatform.cext.proc_num_handlesr� )r� ) r r� r� r� r rd r r r� r- rt ru )r r� r r r r r� � s z.TestDualProcessImplementation.test_num_handlesN)
rf rg rh r� r� r� r� r� r� r� r� r� r� r r r r r� @ s
r� c @ s| e Zd ZdZedd� �Zedd� �ZddgZdd � Z d
d � Z
ed d
� �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS )�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 sT d} xJt j d�D ]<}tj|d| gtjtjd�}|j� \}}|tt �kr|S qW d S )Nz6import sys; sys.stdout.write(str(sys.maxsize > 2**32))zC:\Python*\python.exez-c)r �stdout�stderr)�glob�
subprocess�Popen�PIPEZSTDOUT� communicater� � IS_64_BIT)�code�filenamer9 �outputr� r r r �find_other_interpreter� s
z,RemoteProcessTestCase.find_other_interpreterc C s@ | j � }|d krtjd��tr.tj| _|| _n|| _tj| _d S )Nz0could not find interpreter with opposite bitness)r� r ZSkipTestr� r� �
executable�python64�python32)r� Z other_pythonr r r r� � s z RemoteProcessTestCase.setUpClassz-cz import sys; sys.stdin.read()c C sV t jj� }tt j� �|d< t| jg| j |tj d�| _
t| j g| j |tj d�| _ d S )N�THINK_OF_A_NUMBER)�env�stdin)
r+ r, �copyr� r� r r� � test_argsr� r� �proc32r� �proc64)r r� r r r �setUp� s
zRemoteProcessTestCase.setUpc C s | j j� | jj� t� d S )N)r r� r r )r r r r �tearDown� s
z RemoteProcessTestCase.tearDownc C s
t � d S )N)r )r� r r r r� � s z#RemoteProcessTestCase.tearDownClassc C s@ t j| jj�}| jt|j� �d� | j|j� dd � | j� d S )Nr] rQ )r r� r r� r- r� r� r )r r� r r r �test_cmdline_32� s z%RemoteProcessTestCase.test_cmdline_32c C s@ t j| jj�}| jt|j� �d� | j|j� dd � | j� d S )Nr] rQ )r r� r r� r- r� r� r )r r� r r r �test_cmdline_64� s z%RemoteProcessTestCase.test_cmdline_64c C s& t j| jj�}| j|j� tj� � d S )N)r r� r r� r- �cwdr+ �getcwd)r r� r r r � test_cwd_32� s z!RemoteProcessTestCase.test_cwd_32c C s& t j| jj�}| j|j� tj� � d S )N)r r� r r� r- r r+ r )r r� r r r � test_cwd_64� s z!RemoteProcessTestCase.test_cwd_64c C s> t j| jj�}|j� }| jd|� | j|d ttj � �� d S )Nr� )
r r� r r� r, �assertIn� assertEqualsr� r+ r� )r r� �er r r �test_environ_32� s z%RemoteProcessTestCase.test_environ_32c C s> t j| jj�}|j� }| jd|� | j|d ttj � �� d S )Nr� )
r r� r r� r, r r
r� r+ r� )r r� r r r r �test_environ_64� s z%RemoteProcessTestCase.test_environ_64N)rf rg rh r� � staticmethodr� r� r� r r r r� r r r
r r r r r r r r� � s
r� c @ s e Zd Zdd� Zdd� ZdS )� TestServicesc C sn t dddddddg�}t dd d
g�}t ddd d d
ddg�}�x,tj� D �] }|j� }| j|d t� | j|d j� d� | j|d t� | j|d t� | j|d |� |d d k r�tj |d � | j|d t� | j|d t� | j|d t� | j|d |� | j|d |� | j|d t� |j
� }|d k �rLtj |�}| j |j � � tj
|j� �}| j||� qFW d S )NZrunningZpaused�start�pause�continue�stopZstoppedZ automaticZmanualZdisabledZ
start_pendingZ
pause_pendingZcontinue_pendingZ stop_pendingr� rF � display_namer� �statusr� ZbinpathZ
start_type� description)rB r �win_service_iterr� ZassertIsInstancer� ZassertNotEqualr� r r� r� rZ Z
is_running�win_service_getr� r- )r Zvalid_statusesZvalid_start_typesZserv�datar� r� �sr r r �test_win_service_iter� sR
z"TestServices.test_win_service_iterc ' C sp t tj� �j� }| jtj��}tj|d � W d Q R X | j|jj|d � tj|�}t tj
j j d�}t
jd|d�� | jtj|j� W d Q R X t
jd|d�� | jtj|j� W d Q R X t tj
j jd�}t
jd|d�� | jtj|j� W d Q R X t
jd|d�� | jtj|j� W d Q R X | j|j� t|�� | j|j� t|�� | j|j� t|�� | j|j� t|�� d S )Nz???rF z/psutil._psplatform.cext.winservice_query_status)r� z/psutil._psplatform.cext.winservice_query_config)�nextr r r� r� r r r- Z exceptionZ WindowsError� _psplatformrW Z ERROR_SERVICE_DOES_NOT_EXISTr rd r r� ZERROR_ACCESS_DENIEDr r r� r rN )r r� �cmZservice�excr r r �test_win_service_get! s2
z!TestServices.test_win_service_getN)rf rg rh r r# r r r r r � s 0r �__main__l )2r� r^ r r� r+ r� r� r� r� r� r� �warningsr r Zpsutil._compatr Z psutil.testsr r r r r r r
r r �catch_warnings� simplefilterr1 r� r� r5 � ImportErrorr� r rW �maxsizer� r ri ZTestCaser rj r r� r� r� r rf �__file__r r r r �<module> sl
<
m
A
N
^
U