관리-도구
편집 파일: test_windows.cpython-37.pyc
B �=�_M � @ 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 e �rze�sze � � �0 e �!d� ddl"Z"ddl#Z#ddl$Z$ddl%Z%W dQ R X e j&j'Z'dd� Z(e �)e d�e �)ed�e �)e�o�e d�G dd� de����Z*G d d � d e*�Z+G d d � d e*�Z,G d d!� d!e*�Z-G d"d#� d#e*�Z.G d$d%� d%e*�Z/e �)e d�G d&d'� d'e��Z0e �)e d�G d(d)� d)e��Z1e �)e d�G d*d+� d+e��Z2e3d,k�r�dd-l4m5Z5 e5e6� dS ).zWindows specific tests.� N)�WINDOWS)�FileNotFoundError)�super)�APPVEYOR)�GITHUB_ACTIONS)� HAS_BATTERY)�IS_64BIT)�mock)�PsutilTestCase)�PY3)�PYPY)�retry_on_failure)�sh)�spawn_testproc)� terminate)�TOLERANCE_DISK_USAGE)�unittest�ignorec s � fdd�}|S )Nc st y� | f|�|�S t k rn } z@ddlm} |j|krDt�d d ��|jtjkr\t�d d ��� W d d }~X Y nX d S )Nr )�ACCESS_DENIED_SET)�OSError�psutil._pswindowsr �errno�psutil� AccessDeniedZESRCH� NoSuchProcess)�self�args�kwargs�errr )�fun� �L/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_windows.py�wrapper7 s z wrap_exceptions.<locals>.wrapperr )r r" r )r r! �wrap_exceptions6 s r# z WINDOWS onlyz pywin32 not available on PYPYz pywin32 broken on GITHUB + PY2c @ s e Zd ZdS )�WindowsTestCaseN)�__name__� __module__� __qualname__r r r r! r$ D s r$ c @ sN e Zd Ze�dejkd�dd� �Zdd� Zdd� Z d d � Z d d � Z d d� Z dS )� TestCpuAPIs�NUMBER_OF_PROCESSORSz-NUMBER_OF_PROCESSORS env var is not availablec C s" t tjd �}| �|t�� � d S )Nr) )�int�os�environ� assertEqualr � cpu_count)r Znum_cpusr r r! �&test_cpu_count_vs_NUMBER_OF_PROCESSORSS s z2TestCpuAPIs.test_cpu_count_vs_NUMBER_OF_PROCESSORSc C s$ t �� d }t�� }| �||� d S )N� )�win32apiZ GetSystemInfor r. r- )r � sys_value� psutil_valuer r r! � test_cpu_count_vs_GetSystemInfo[ s z+TestCpuAPIs.test_cpu_count_vs_GetSystemInfoc C s* t �� }|�� d }| �t�� |j� d S )Nr )�wmi�WMI�Win32_Processorr- r r. ZNumberOfLogicalProcessors)r �w�procr r r! � test_cpu_count_logical_vs_wmib s z)TestCpuAPIs.test_cpu_count_logical_vs_wmic C s. t �� }|�� d }| �tjdd�|j� d S )Nr F)Zlogical)r5 r6 r7 r- r r. Z NumberOfCores)r r8 r9 r r r! �test_cpu_count_phys_vs_wmig s z&TestCpuAPIs.test_cpu_count_phys_vs_wmic C s | � t�� ttjdd��� d S )NT)Zpercpu)r- r r. �len� cpu_times)r r r r! �test_cpu_count_vs_cpu_timesl s z'TestCpuAPIs.test_cpu_count_vs_cpu_timesc C s@ t �� }|�� d }| �|jt�� j� | �|jt�� j � d S )Nr ) r5 r6 r7 r- ZCurrentClockSpeedr Zcpu_freq�currentZ MaxClockSpeed�max)r r8 r9 r r r! � test_cpu_freqp s zTestCpuAPIs.test_cpu_freqN) r% r& r' r �skipIfr+ r, r/ r4 r: r; r> rA r r r r! r( Q s r( c @ sr e Zd Zdd� Zdd� Ze�ed�e� dd� ��Z e� dd � �Z e� d d � �Z d d � Z dd� Z dd� Zdd� ZdS )�TestSystemAPIsc C sV t d�}tjdd��� }x8|D ]0}d|�dd��� kr8q ||kr | �d| � 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_namesy s z TestSystemAPIs.test_nic_namesc C s, t �� �� d }| �t|j�t�� j� d S )Nr ) r5 r6 ZWin32_ComputerSystemr- r* ZTotalPhysicalMemoryr Zvirtual_memory�total)r r8 r r r! �test_total_phymem� s z TestSystemAPIs.test_total_phymemz test not relieable on appveyorc C s: t �� �� }tdd� |D ��}tt�� �}| �||� 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 �pidsr- )r r8 Zwmi_pidsZ psutil_pidsr r r! � test_pids� s zTestSystemAPIs.test_pidsc C s� t jdd�}t�� �� }x�|D ]�}x�|D ]�}|j�dd�|jkr(|jsHP d|j krTP |j� d�rbP yt � |j�}W n t k r� P Y nX | � |jt|j�� t|j�}| � |j|� t|j| �dkr�| �d|j|f � P q(W | �d t|� � q W d S ) NT)�all�\� �cdromzA:i � zpsutil=%s, wmi=%szcan't find partition %s)r �disk_partitionsr5 r6 ZWin32_LogicalDisk�devicerG ZDeviceID� mountpoint�opts� startswith� disk_usager r- rL r* �SizeZ FreeSpace�free�absrI �repr)r Zps_partsZ wmi_partsZps_partZwmi_part�usageZwmi_freer r r! � test_disks� s. zTestSystemAPIs.test_disksc C s| xvt �� D ]j}d|jkrq t�|j�}t �|j�}| j|d |jt d� | j|d |j t d� | � |j |j |j � q W d S )NrY r )�delta� ) r rZ r] r1 ZGetDiskFreeSpaceExr\ r_ �assertAlmostEqualra r rL r- �used)r Zdiskr2 r3 r r r! �test_disk_usage� s z TestSystemAPIs.test_disk_usagec C s> dd� t �� �d�D �}dd� tjdd�D �}| �||� d S )Nc S s" g | ]}|r|� d �s|d �qS )zA:rW )r^ )rO rP r r r! rQ � s z7TestSystemAPIs.test_disk_partitions.<locals>.<listcomp>z\ c S s g | ]}|j �d �s|j �qS )zA:)r\ r^ )rO rP r r r! rQ � s T)rV )r1 ZGetLogicalDriveStrings�splitr rZ r- )r r2 r3 r r r! �test_disk_partitions� s z#TestSystemAPIs.test_disk_partitionsc C s` t t�� �}t�� �� }t � }x$|D ] }|�|j� |�|j� q$W | � ||@ d||f � d S )Nzno common entries in %s, %s) rS �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 �� �� }|d j�d�d }tj�|d�}tj�t� � �}t || � � �}| � |d� d S )Nr �.z %Y%m%d%H%M%S� ) r5 r6 ZWin32_OperatingSystemZLastBootUpTimerk �datetime�strptime� fromtimestampr � boot_timerb � 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�� | �t�� d� W d Q R X t jddd�� | �t�� d� W d Q R X t jddd�� | �t�� d� W d Q R X t jddd�� | �t�� d� W d Q R X d S )Nz psutil._pswindows.cext.boot_timer0 )� return_value� � iM )r �patchr- r rw )r r r r! �test_boot_time_fluctuation� s z)TestSystemAPIs.test_boot_time_fluctuationN)r% r& r' rK rM r rB r r rU re rj rl rq ry r~ r r r r! rC w s rC c @ s` e Zd Zdd� Ze�e d�dd� �Ze�e d�dd� �Zdd � Z d d � Z d d � Z dd� Z dS )�TestSensorsBatteryc C s. t �� d r | �t�� � n| �t�� � 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 �� }|�d�d }t�� }| j|j|jdd� d S )Nzselect * from Win32_Batteryr rg )rf )r5 r6 �queryr r� rh �percentZEstimatedChargeRemaining)r r8 � battery_wmi�battery_psutilr r r! � test_percent� s z TestSensorsBattery.test_percentc C s6 t �� }|�d�d }t�� }| �|j|jdk� d S )Nzselect * from Win32_Batteryr � )r5 r6 r� r r� r- � power_pluggedZ BatteryStatus)r r8 r� r� r r r! �test_power_plugged s z%TestSensorsBattery.test_power_pluggedc C s6 t jddd�� }| �t�� � |js(t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_battery)r � r r )rz )r r} r� r r� �called�AssertionError)r �mr r r! �test_emulate_no_battery s z*TestSensorsBattery.test_emulate_no_batteryc C s< t jddd��$}| �t�� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_battery)rg r r r )rz ) r r} r- r r� �secsleft�POWER_TIME_UNLIMITEDr� r� )r r� r r r! � test_emulate_power_connected s z/TestSensorsBattery.test_emulate_power_connectedc C s< t jddd��$}| �t�� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_battery)r � r r )rz ) r r} r- r r� 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�� jtj� |js.t�W d Q R X d S )Nz&psutil._pswindows.cext.sensors_battery)r r r ���)rz ) r r} r- r r� r� �POWER_TIME_UNKNOWNr� r� )r r� r r r! � test_emulate_secs_left_unknown% s z1TestSensorsBattery.test_emulate_secs_left_unknownN) r% r& r' r� r rB r r� r� r� r� r� r� r r r r! r � s r c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d � Zd d � Z e � e j d k d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )&� TestProcessc C s t � j| _d S )N)r �pid)�clsr r r! � setUpClass4 s zTestProcess.setUpClassc C s t | j� d S )N)r r� )r� r r r! � tearDownClass8 s zTestProcess.tearDownClassc C s t �d�}| �t j|j� d S )Nr )r �Process� assertRaisesr �kill)r �pr r r! � test_issue_24< s zTestProcess.test_issue_24c C s� t �d�}| �|�� d� t|� |�� | �|�� dk� y|�� d d� \}}W n( t j k r| t � � d dkrx� Y nX | �|dk� d S )Nr{ �Systemg r� rg )Zvistazwin-7Zwin7r ) r r� r- �name�str�usernamerp � 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�}| �t|jtj� d S )N)r r� r� r� � ValueError� send_signal�signal�SIGINT)r r� r r r! �test_send_signalQ s z TestProcess.test_send_signalc C sb t �t�� �}|�� }t�tjtj t�� �}|�� }| � ||d � t� |� | � |�� |� d S )Nrg ) r r� r+ �getpid� num_handlesr1 � OpenProcess�win32con�PROCESS_QUERY_INFORMATION�FALSEr- � CloseHandle)r r� �before�handle�afterr r r! �test_num_handles_incrementU s z&TestProcess.test_num_handles_increment)r� � z CTRL_* signals not supportedc C sd t �| �� j�}|�tj� |�tj� |�� |� � | � t j |jtj� | � t j |jtj� d S )N) r r� r r� r� r� Z CTRL_C_EVENTZCTRL_BREAK_EVENTr� �waitr� r )r r� r r r! �test_ctrl_signals_ s z TestProcess.test_ctrl_signalsc C s | � t�� �� t�tj�� d S )N)r- r r� r� r1 Z GetUserNameExr� ZNameSamCompatible)r r r r! � test_usernamel s zTestProcess.test_usernamec C s8 t �ddt�� ��� }d�t�� �� �}| � ||� d S )Nz +rD ) �re�subr1 ZGetCommandLine�strip�joinr r� �cmdliner- )r r2 r3 r r r! � test_cmdlinep s zTestProcess.test_cmdlinec C sJ t �tjtjt�� �}| �t j|� t � |�}t � � � � }| �||� d S )N)r1 r� r� r� r� r+ r� � addCleanupr� � win32processZGetPriorityClassr r� �nicer- )r r� r2 r3 r r r! � test_nice� s zTestProcess.test_nicec C s� t �tjtj| j�}| �t j|� t� |�}t � | j�� � }| � |d |j� | � |d |j� | � |d |j� | � |d |j� | � |d |j� | � |d |j� | � |d |j� | � |d |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 �tjtj| j�}| �t j|� t� | j�}|� � |� � }t � |�}| �||� 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 �tjtj| j�}| �t j|� |t� |�d �}t � | j�� � }| � ||� d S )Nc s � fdd�t d�D �S )Nc s g | ]}d |> � @ r|�qS )rg r )rO �i)rP r r! rQ � s zGTestProcess.test_cpu_affinity.<locals>.from_bitmask.<locals>.<listcomp>�@ )�range)rP r )rP 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 �tjtjt�� �}| �t j|� t � |�}t � � � � }| �|j|d � | �|j|d � | �|j|d � | �|j|d � | �|j|d � | �|j|d � d S )NZReadOperationCountZWriteOperationCountZReadTransferCountZWriteTransferCountZOtherOperationCountZOtherTransferCount)r1 r� r� r� r� r+ r� r� r� r� ZGetProcessIoCountersr r� � io_countersr- � read_count� write_count� read_bytes� 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}|jj�|d| j�}| �|jjj|� |j� � }|jj� ||� |�� |j }t �| j��� }| �||� d S )Nr i )�ctypesZctypes.wintypes�windll�kernel32r� r� r� r� ZwintypesZDWORDZGetProcessHandleCount�byref�valuer r� r� r- )r r� r� r� Zhndcntr2 r3 r r r! �test_num_handles� s z TestProcess.test_num_handlesc C sf t � }d|_tjd|d��4 t�d�� }t�� }| �tj|j� W d Q R X W d Q R X | � |j d� d S )Ni+ z psutil._psplatform.cext.proc_cwd)� side_effectz time.sleepr0 ) � WindowsError�winerrorr r} r r� r� r �cwd�assertGreaterEqualZ call_count)r �excr� r� r r r! �test_error_partial_copy� s $z#TestProcess.test_error_partial_copyc C s0 t �� d d }t j�|�}| �t j|j� d S )Nr� i�� )r rT � _psplatformr� r� r �exe)r r� r9 r r r! �test_exe� s zTestProcess.test_exeN)r% r& r' � classmethodr� r� r� r� r� r� r rB �sys� version_infor� r� r� r� r� r� r� r� r� r� r� r r r r! r� 2 s$ r� c @ sx e Zd ZdZedd� �Zedd� �Zdd� Ze� e d�d d � �Z d d � Z d d� Z e� dd� �Ze� dd� �Zdd� ZdS )�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 | j� d S )N)r r� )r� r r r! r� � s z TestProcessWMI.tearDownClassc C s8 t �� j| jd�d }t�| j�}| �|�� |j� d S )N)rN r ) r5 r6 rR r� r r� r- r� ZCaption)r r8 r� r r r! � test_name� s zTestProcessWMI.test_namez!unreliable path on GITHUB_ACTIONSc C s@ t �� j| jd�d }t�| j�}| �|�� �� |j �� � d S )N)rN r ) r5 r6 rR r� r r� r- r� rH ZExecutablePath)r r8 r� r r r! r� s zTestProcessWMI.test_exec C sF t �� j| jd�d }t�| j�}| �d�|�� �|j � dd�� d S )N)rN r rD �"rX ) r5 r6 rR r� r r� r- r� r� Z CommandLinerG )r r8 r� r r r! r� s zTestProcessWMI.test_cmdlinec C sP t �� j| jd�d }t�| j�}|�� \}}}d||f }| �|�� |� d S )N)rN r z%s\%s) r5 r6 rR r� r r� ZGetOwnerr- r� )r r8 r� �domain�_r� r r r! r� s z TestProcessWMI.test_usernamec C sB t �� j| jd�d }t�| j�}|�� j}| �|t |j �� d S )N)rN r ) r5 r6 rR 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 s` t �� j| jd�d }t�| j�}|�� j}t|j �}||kr\||d kr\| � d||f � d S )N)rN r i zwmi=%s, psutil=%s) r5 r6 rR r� r r� r� r� r* Z PageFileUsagerI )r r8 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 � |� � ��}| � ||� d S )N)rN r rr z %Y%m%d%H%M%S)r5 r6 rR r� r r� r� Z CreationDaterk �time�strftime� localtimer� r- )r r8 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 rB 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� C s z(TestDualProcessImplementation.setUpClassc C s t | j� d S )N)r r� )r� r r r! r� G s z+TestDualProcessImplementation.tearDownClassc C s� t �| j��� }tjdttjd�d���}t �| j��� }| � t |�t |�� xLt t |��D ]<}| � || d� | � || d� | j || || dd� qZW |js�t�W d Q R X d S )Nz(psutil._psplatform.cext.proc_memory_info�msg)r� r i )rf )r r� r� r� r r} r r �EPERMr- r< r� r� rh r� r� )r Zmem_1r Zmem_2r� r r r! r� K s z.TestDualProcessImplementation.test_memory_infoc C sX t �| j��� }tjdttjd�d��(}| � t �| j��� |� |j sJt �W d Q R X d S )Nz"psutil._psplatform.cext.proc_timesr� )r� ) r r� r� r� r r} r r r� r- r� r� )r �ctimer r r r! r� W s z.TestDualProcessImplementation.test_create_timec C sx t �| j��� }tjdttjd�d��H}t �| j��� }|j sBt �| j |j |j dd� | j |j |j dd� W d Q R X d S )Nz"psutil._psplatform.cext.proc_timesr� )r� g{�G�z�?)rf )r r� r� r= r r} r r r� r� r� rh �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��� }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 )rf )r r� r� r� r r} r r r� r� r< rh r� r� )r Z io_counters_1r Z io_counters_2r� r r r! r� i s z.TestDualProcessImplementation.test_io_countersc C sX t �| j��� }tjdttjd�d��(}| � t �| j��� |� |j sJt �W d 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! r� s s z.TestDualProcessImplementation.test_num_handlesc C s� ddl m} x|t�� D ]p}y tj|dd�}tj|dd�}W n> tk rx } z ||�}t|tjtj f�sh� W d d }~X Y qX | � ||� qW d S )Nr )�convert_oserrorT)Zuse_pebF) r r r rT rm Z proc_cmdliner � isinstancer r r- )r r r� �a�br r 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� 7 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} xRt � d�D ]D}tj|d| gtjtjd�}|�� \}}|�� |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�PIPE�STDOUT� communicater� r� r )�code�filenamer9 �outputr� r r r! �find_other_interpreter� s z,RemoteProcessTestCase.find_other_interpreterz-cz 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 �setUpr r ZSkipTestr r� � executableZpython64Zpython32r+ r, �copyr� r� r � test_argsr r �proc32�proc64)r Z other_pythonr )� __class__r r! r � s( zRemoteProcessTestCase.setUpc s" t � �� | j�� | j�� d S )N)r �tearDownr r r )r )r r r! r � s z RemoteProcessTestCase.tearDownc C s@ t �| jj�}| �t|�� �d� | �|�� dd � | j� d S )Nrs rg )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 �| jj�}| �t|�� �d� | �|�� dd � | j� d S )Nrs rg )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 �| jj�}| �|�� t�� � d S )N)r r� r r� r- r� r+ �getcwd)r r� r r r! � test_cwd_32� s z!RemoteProcessTestCase.test_cwd_32c C s& t �| jj�}| �|�� t�� � 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 �| jj�}|�� }| �d|� | �|d tt� � �� d S )Nr ) r r� r r� r, �assertInZ assertEqualsr� r+ r� )r r� �er r r! �test_environ_32� s z%RemoteProcessTestCase.test_environ_32c C s6 t �| jj�}y |�� W n t jk r0 Y nX d S )N)r r� r r� r, r )r r� r r r! �test_environ_64� s z%RemoteProcessTestCase.test_environ_64)r% r& r' r� � staticmethodr r r r r r r r r# r$ � __classcell__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,t�� D �] }|�� }| �|d t� | �|d �� d� | �|d t� | �|d t� | �|d |� |d d k r�t� |d � | �|d t� | �|d t� | �|d t� | �|d |� | �|d |� | �|d t� |� � }|d k �rLt� |�}| � |� � � t� |�� �}| �||� qFW d S )N�runningZpaused�start�pause�continue�stop�stoppedZ automaticZmanual�disabledZ start_pendingZ pause_pendingZcontinue_pendingZ stop_pendingr� rX � display_namer� �statusr� ZbinpathZ start_type� description)rS r �win_service_iterZas_dictZassertIsInstancer� ZassertNotEqualr� r! r� r� rp 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 s� t jjj}t jjj}tt �� ��� }| �t j ��}t � |d � W d Q R X | � |j j|d � t � |�}t rzddd|f}n|df}t|� }tjd|d�� | �t j |j� W d Q R X tjd|d�� | �t j |j� W d Q R X t r�ddd|f}n|df}t|� }tjd|d�� | �t j|j� W d Q R X tjd|d�� | �t j|j� W d Q R X | �|�� t|�� | �|�� t|�� | �|�� t|�� | �|�� t|�� d S )Nz???r r� z/psutil._psplatform.cext.winservice_query_status)r� z/psutil._psplatform.cext.winservice_query_config)r r� rm � ERROR_SERVICE_DOES_NOT_EXIST�ERROR_ACCESS_DENIED�nextr2 r� r� r r3 r- � exceptionr r� r r} r0 r� r r! r� r/ rc )r r7 r8 r� �cmZservicer r� r r r! �test_win_service_get s> z!TestServices.test_win_service_getN)r% r& r' r6 r< r r r r! r' � s 0r' �__main__)� run_from_name)7r� rt r r r+ r� r� r� r r� r� �warningsr r Zpsutil._compatr r Z psutil.testsr r r r r r r r r r r r r r �catch_warnings� simplefilterr1 r� r� r5 r� rm r# rB r$ r( rC r r� r� r� r r' r% Zpsutil.tests.runnerr> �__file__r r r r! �<module> sr &< @FR\^