← Back
Editing: test_windows.cpython-36.pyc
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
Save File
Cancel