U
Ã|ýf㉠ã @ sÆ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddl
mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! erŽesŽe "¡ 0 e #d¡ ddl$Z$ddl%Z%ddl&Z&ddl'Z'W 5 Q R X er ddl(m)Z) e
j*j+Z+e ,e d¡e ,ed¡e ,eoÌe d¡G dd„ de ƒƒƒƒZ-d d „ Z.e/fd d „Z0G d d!„ d!e-ƒZ1G d"d#„ d#e-ƒZ2G d$d%„ d%e-ƒZ3G d&d'„ d'e-ƒZ4G d(d)„ d)e-ƒZ5e ,e d¡G d*d+„ d+e ƒƒZ6e ,e d¡G d,d-„ d-e ƒƒZ7e ,e d¡G d.d/„ d/e ƒƒZ8e9d0krÂdd1l:m;Z; e;e<ƒ dS )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)Úretry_on_failure)Úsh)Úspawn_testproc)Ú terminateÚignore)Úconvert_oserrorz WINDOWS onlyz pywin32 not available on PYPYz pywin32 broken on GITHUB + PY2c @ s e Zd ZdS )ÚWindowsTestCaseN)Ú__name__Ú
__module__Ú __qualname__© r r úL/opt/alt/python38/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 ÚunittestÚSkipTestr )ÚcmdÚcmdliner r r Ú
powershellC 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 Ú
\ s zwmic..)r ÚstripÚjoinÚ
splitlinesÚtupleÚsplit)ÚpathÚwhatr( ÚoutÚdatar r' r ÚwmicR s
r3 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 )Nr5 )ÚintÚosÚenvironÚ assertEqualÚpsutilÚ cpu_count)ÚselfZnum_cpusr r r Ú&test_cpu_count_vs_NUMBER_OF_PROCESSORSi s z2TestCpuAPIs.test_cpu_count_vs_NUMBER_OF_PROCESSORSc C s$ t ¡ d }t ¡ }| ||¡ d S )Né )Úwin32apiZ
GetSystemInfor: r; r9 ©r< Ú sys_valueÚ psutil_valuer r r Ú test_cpu_count_vs_GetSystemInfos s z+TestCpuAPIs.test_cpu_count_vs_GetSystemInfoc C s2 t ¡ }tdd„ | ¡ D ƒƒ}| t ¡ |¡ d S )Nc s s | ] }|j V qd S ©N)ZNumberOfLogicalProcessors©r% Úprocr r r Ú | s z.©ÚwmiÚWMIÚsumÚWin32_Processorr9 r: r; )r< ÚwZprocsr r r Ú test_cpu_count_logical_vs_wmiz s
ÿz)TestCpuAPIs.test_cpu_count_logical_vs_wmic C s6 t ¡ }tdd„ | ¡ D ƒƒ}| tjdd|¡ d S )Nc s s | ] }|j V qd S rD )Z
NumberOfCoresrE r r r rG ƒ s z:TestCpuAPIs.test_cpu_count_cores_vs_wmi..F)ZlogicalrH )r< rM Zcoresr r r Útest_cpu_count_cores_vs_wmi s z'TestCpuAPIs.test_cpu_count_cores_vs_wmic C s | t ¡ ttjddƒ¡ d S )NT)Zpercpu)r9 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 ¡ j¡ | |jt ¡ j ¡ d S ©Nr )
rI rJ rL r9 ZCurrentClockSpeedr: Zcpu_freqÚcurrentZ
MaxClockSpeedÚmax)r< rM rF r r r Ú
test_cpu_freq‹ s zTestCpuAPIs.test_cpu_freqN)
r r r r ÚskipIfr7 r8 r= rC rN rO rS rW r r r r r4 h s þ
r4 c @ sŠ e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Ze e
d ¡e ƒ d d
„ ƒƒZ e ƒ dd„ ƒZ
e ƒ 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< r1 ZnicsZnicr r r Útest_nic_names“ s ÿz TestSystemAPIs.test_nic_namesc C s, t ¡ ¡ d }| t|jƒt ¡ j¡ d S rT ) rI rJ ZWin32_ComputerSystemr9 r6 ZTotalPhysicalMemoryr: Úvirtual_memoryÚtotal©r< rM r r r Útest_total_phymemž s
ÿz TestSystemAPIs.test_total_phymemc C s0 t ¡ ¡ d }| jt|jƒt ¡ jt d d S )Nr ©Údelta)
rI rJ Ú Win32_PerfRawData_PerfOS_MemoryÚassertAlmostEqualr6 ZAvailableBytesr: ra Úfreer
rc r r r Útest_free_phymem¤ s ýz TestSystemAPIs.test_free_phymemc C sh t ¡ ¡ d }| t|jƒt ¡ j t ¡ j¡ t ¡ jdkrd| dt ¡ j
¡ | dt ¡ j ¡ d S rT ) rI rJ rg r9 r6 Z CommitLimitr: ra rb Ú swap_memoryri Úusedrc r r r Útest_total_swapmem¬ s þz!TestSystemAPIs.test_total_swapmemc C sx t ¡ jdkrtt ¡ jddd }t|jƒd t|jƒ }| t ¡ j
d¡ | j t ¡ j
|dd | t ¡ j
d¡ d S )Nr Z_Total)ÚNameéd r> re )
r: rk rb rI rJ Z#Win32_PerfRawData_PerfOS_PagingFiler6 Z PercentUsageZPercentUsage_BaseÚassertGreaterEqualÚpercentrh ÚassertLessEqual)r< rM Z percentSwapr r r Útest_percent_swapmem¶ s ÿz#TestSystemAPIs.test_percent_swapmemz test not relieable on appveyorc C s: t ¡ ¡ }tdd„ |D ƒƒ}tt ¡ ƒ}| ||¡ d S )Nc S s g | ]
}|j ‘qS r ©Z ProcessIdr$ r r r r) Õ s z,TestSystemAPIs.test_pids..)rI rJ Ú
Win32_ProcessÚsetr: Úpidsr9 )r< rM 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 |
|jt|jƒ¡ t|jƒ}|
|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_partitionsrI rJ ZWin32_LogicalDiskZdevicer] ZDeviceIDÚ
mountpointÚoptsÚ
startswithÚ
disk_usager r9 rb r6 ÚSizeZ FreeSpaceri Úabsr_ Úrepr)r< Zps_partsZ wmi_partsZps_partZwmi_partÚusageZwmi_freer r r Ú
test_disksÙ s0
ÿzTestSystemAPIs.test_disksc C sx t ¡ D ]j}d|jkrqt |j¡}t |j¡}| j|d |jt d | j|d |j
t d | |j |j
|j ¡ qd S )Nr| r re r" )
r: r~ r€ r? ZGetDiskFreeSpaceExr r‚ rh ri r rb r9 rl )r< ZdiskrA rB r r r Útest_disk_usageø s&
ÿ ÿ
ÿz TestSystemAPIs.test_disk_usagec C s> dd„ t ¡ d¡D ƒ}dd„ tjddD ƒ}| ||¡ d S )Nc S s" g | ]}|r| d ¡s|d ‘qS )r} r{ )r r$ r r r r)
s
þz7TestSystemAPIs.test_disk_partitions..z\ c S s g | ]}|j d ¡s|j ‘qS )r} )r r r$ r r r r) s þTry )r? ZGetLogicalDriveStringsr. r: r~ r9 r@ r r r Útest_disk_partitions s þ
þz#TestSystemAPIs.test_disk_partitionsc C s\ t t ¡ ƒ}t ¡ ¡ }t ƒ }|D ] }| |j¡ | |j¡ q"| ||@ d||f ¡ d S )Nzno common entries in %s, %s)
rv ÚcextZ net_if_statsrI rJ ZWin32_NetworkAdapterÚaddrn ZNetConnectionIDÚ
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 Ú.ú %Y%m%d%H%M%Sr> )
rI rJ ZWin32_OperatingSystemZLastBootUpTimer. ÚdatetimeÚstrptimeÚ
fromtimestampr: Ú boot_timer„ Ú
total_secondsrr )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 ¡ d¡ W 5 Q R X t jddd | t ¡ d¡ W 5 Q R X t jddd | t ¡ d¡ W 5 Q R X t jddd | t ¡ d¡ W 5 Q R X d S )Nz psutil._pswindows.cext.boot_timer> ©Z return_valueé é iM )r Úpatchr9 r: r“ rR r r r Útest_boot_time_fluctuation, s z)TestSystemAPIs.test_boot_time_fluctuationN)r r r r` rd rj rm rs r rX r r rx r‡ rˆ r‰ r r– r› r r r r rY ’ s
rY 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)r? ZGetPwrCapabilitiesZassertIsNotNoner: Úsensors_batteryÚ assertIsNonerR 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 )Núselect * from Win32_Batteryr r" re )rI rJ Úqueryr: r rh rq ZEstimatedChargeRemaining©r< rM Z battery_wmiZbattery_psutilr r r Ú test_percentD s ýz TestSensorsBattery.test_percentc C s6 t ¡ }| d¡d }t ¡ }| |j|jdk¡ d S )Nr r é )rI rJ r¡ r: r r9 Z
power_pluggedZ
BatteryStatusr¢ r r r Útest_power_pluggedO s ÿz%TestSensorsBattery.test_power_pluggedc C s6 t jddd }| 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ÚAssertionError©r< Úmr r r Útest_emulate_no_batteryZ s þz*TestSensorsBattery.test_emulate_no_batteryc C s< t jddd$}| t ¡ jtj¡ |js.t‚W 5 Q R X d S )Nr¦ )r" r r r r— © r rš r9 r: r ÚsecsleftZPOWER_TIME_UNLIMITEDr¨ r© rª r r r Ú test_emulate_power_connectedb s ÿ ÿz/TestSensorsBattery.test_emulate_power_connectedc C s< t jddd$}| t ¡ jtj¡ |js.t‚W 5 Q R X d S )Nr¦ )r é r r r— r rª r r r Útest_emulate_power_chargingk s ÿ ÿz.TestSensorsBattery.test_emulate_power_chargingc C s< t jddd$}| t ¡ jtj¡ |js.t‚W 5 Q R X d S )Nr¦ )r r r éÿÿÿÿr— ) r rš r9 r: r r® ZPOWER_TIME_UNKNOWNr¨ r© rª r r r Ú test_emulate_secs_left_unknownt s þ ÿz1TestSensorsBattery.test_emulate_secs_left_unknownN)
r r r rŸ r rX 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 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 rD ©r Úpid©Úclsr r r Ú
setUpClass… s zTestProcess.setUpClassc C s t | jƒ d S rD ©r r¶ r· r r r Ú
tearDownClass‰ s zTestProcess.tearDownClassc C s t d¡}| t j|j¡ d S rT )r: ÚProcessÚ assertRaisesÚ AccessDeniedÚkill©r< Úpr r r Ú
test_issue_24 s
zTestProcess.test_issue_24c C sŒ t d¡}| | ¡ d¡ t|ƒ | ¡ | | ¡ d¡ z| ¡ d d… \}}W n( t j k
rz t
¡ d dkrv‚ Y nX | |d¡ d S )Nr˜ ÚSystemg r¤ r" )Zvistazwin-7Zwin7r )
r: r¼ r9 ÚnameÚstrÚusernamerp Ú create_timeÚ memory_infor¾ ÚplatformÚunameZ
assertGreater)r< rÁ ÚrssZ_vmsr r r Útest_special_pid‘ s
z TestProcess.test_special_pidc C s" t | j¡}| t|jtj¡ d S rD )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 ¡ t |¡ |
| ¡ |¡ d S )Nr" ) r: r¼ r7 ÚgetpidÚ num_handlesr? Ú OpenProcessÚwin32conÚPROCESS_QUERY_INFORMATIONÚFALSEr9 Ú CloseHandle)r< rÁ ÚbeforeÚhandleÚafterr r r Útest_num_handles_increment¦ s ÿ
z&TestProcess.test_num_handles_incrementc C sd t | ¡ j¡}| tj¡ | tj¡ | ¡ | ¡ |
t j |jtj¡ |
t j |jtj¡ d S rD ) r: r¼ r r¶ rÎ rÏ Z CTRL_C_EVENTZCTRL_BREAK_EVENTr¿ Úwaitr½ Ú
NoSuchProcessrÀ r r r Útest_ctrl_signals± s ÿ ÿz TestProcess.test_ctrl_signalsc C s8 t tj¡}| d¡r t d¡‚| t ¡
¡ |¡ d S )Nú$zrunning as service account) r? Z
GetUserNameExrÕ ZNameSamCompatibleÚendswithr r r9 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¡}|
||¡ d S )Nz[ ]+rZ r ú"r! r¤ ) ÚreÚsubr? ZGetCommandLiner* r+ r: r¼ r r] r9 r@ r r r Ú test_cmdlineÇ s
zTestProcess.test_cmdlinec C sJ t tjtjt ¡ ¡}| t j|¡ t
|¡}t ¡
¡ }| ||¡ d S rD )r? rÔ rÕ rÖ r× r7 rÒ Ú
addCleanuprØ Ú win32processZGetPriorityClassr: r¼ Únicer9 ©r< rÚ rA rB 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)r? rÔ rÕ rÖ r× r¶ rç rØ rè ZGetProcessMemoryInfor: r¼ rÈ r9 Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZpagefileZ
peak_pagefilerË Úvmsrê r r r Útest_memory_infoê sH ÿ
ÿ ÿ ÿþ ÿ ÿz TestProcess.test_memory_infoc C sX t tjtj| j¡}| t j|¡ t | j¡}|
¡ | ¡ }t
|¡}| ||¡ d S rD )r? rÔ rÕ rÖ r× r¶ rç rØ r: r¼ r rÝ rè ZGetExitCodeProcessr9 )r< rÚ rÁ rB rA 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 )r" r )r% Úi©r& r r r) s zGTestProcess.test_cpu_affinity..from_bitmask..é@ )Úrangerñ r rñ r Ú from_bitmask s z3TestProcess.test_cpu_affinity..from_bitmaskr )r? rÔ rÕ rÖ r× r¶ rç rØ rè ZGetProcessAffinityMaskr: r¼ Z cpu_affinityr9 )r< rô rÚ rA rB 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)r? rÔ rÕ rÖ r× r7 rÒ rç rØ rè ZGetProcessIoCountersr: r¼ Ú io_countersr9 Z
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¡ ¡ }| ||¡ d S )Nr é )ÚctypesZctypes.wintypesÚwindllÚkernel32rÔ r¶ rç rØ ZwintypesZDWORDZGetProcessHandleCountZbyrefÚvaluer: r¼ rÓ r9 )r< rü rÖ rÚ ZhndcntrA rB r r r Útest_num_handles= s" ÿ
ÿz TestProcess.test_num_handlesc
C sf t ƒ }d|_tjd|d4 t d¡ }t ¡ }| tj|j¡ W 5 Q R X W 5 Q R X | |j
d¡ d S )Ni+ z psutil._psplatform.cext.proc_cwd©Z side_effectz
time.sleepr> ) Ú WindowsErrorÚwinerrorr rš r: r¼ r½ r¾ Úcwdrp Z
call_count)r< Úexcr« rÁ r r r Útest_error_partial_copyO s $z#TestProcess.test_error_partial_copyc C s0 t ¡ d d }t j |¡}| t j|j¡ d S )Nr² iŸ† )r: rw Ú _psplatformr¼ r½ rÞ Úexe)r< r¶ rF r r r Útest_exeY 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´ „ 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 rD rµ r· r r r r¹ d s zTestProcessWMI.setUpClassc C s t | jƒ d S rD rº r· r r r r» h s z TestProcessWMI.tearDownClassc C s8 t ¡ j| jdd }t | j¡}| | ¡ |j¡ d S ©Nrt r ) rI rJ ru r¶ r: r¼ r9 rÄ ZCaption©r< rM rÁ r r r Ú test_namel s zTestProcessWMI.test_namez!unreliable path on GITHUB_ACTIONSc C s@ t ¡ j| jdd }t | j¡}| | ¡ ¡ |j ¡ ¡ d S r )
rI rJ ru r¶ r: r¼ r9 r r^ ZExecutablePathr
r r r r r s zTestProcessWMI.test_exec C sF t ¡ j| jdd }t | j¡}| d | ¡ ¡|j
dd¡¡ d S )Nrt r rZ rã r! ) rI rJ ru r¶ r: r¼ r9 r+ r Z CommandLiner] r
r r r ræ z s zTestProcessWMI.test_cmdlinec C sP t ¡ j| jdd }t | j¡}| ¡ \}}}d||f }| | ¡ |¡ d S )Nrt r z%s\%s) rI rJ ru r¶ r: r¼ ZGetOwnerr9 rÆ )r< rM rÁ ÚdomainÚ_rÆ r r r râ s
z TestProcessWMI.test_usernamec C sB t ¡ j| jdd }t | j¡}| ¡ j}| |t |j
ƒ¡ d S r ) rI rJ ru r¶ r: r¼ rÈ rË r9 r6 rì )r< rM rÁ rË r r r Útest_memory_rss† s
z TestProcessWMI.test_memory_rssc C s\ t ¡ j| jdd }t | j¡}| ¡ j}t|j ƒ}|||d fkrX|
d||f ¡‚d S )Nrt r rû zwmi=%s, psutil=%s) rI rJ ru r¶ r: r¼ rÈ rí r6 Z
PageFileUsager_ )r< rM rÁ rí Z wmi_usager r r Útest_memory_vms s
z TestProcessWMI.test_memory_vmsc C s\ t ¡ j| jdd }t | j¡}t|j d¡d ƒ}t
dt | ¡ ¡¡}|
||¡ d S )Nrt r rŽ r )rI rJ ru r¶ r: r¼ rÅ Z CreationDater. ÚtimeÚstrftimeÚ localtimerÇ r9 )r< rM 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 rX r r ræ râ r r r r r r r r r a 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 rD rµ r· r r r r¹ ² s z(TestDualProcessImplementation.setUpClassc C s t | jƒ d S rD rº r· r r r r» ¶ s z+TestDualProcessImplementation.tearDownClassc C s® t | j¡ ¡ }tjdttjdƒd~}t | j¡ ¡ }| t
|ƒt
|ƒ¡ t t
|ƒƒD ]<}| || d¡ | || d¡ | j
|| || dd qX|js t‚W 5 Q R X d S )Nz(psutil._psplatform.cext.proc_memory_infoÚmsgr r i re )r: r¼ r¶ rÈ r rš ÚOSErrorÚerrnoÚEPERMr9 rP ró rp rh r¨ r© )r< Zmem_1ÚfunZmem_2rð r r r rî º s
þz.TestDualProcessImplementation.test_memory_infoc C sX t | j¡ ¡ }tjdttjdƒd(}| t | j¡ ¡ |¡ |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 r9 r¨ r© )r< Úctimer r r r r È s
þz.TestDualProcessImplementation.test_create_timec C sx t | j¡ ¡ }tjdttjdƒdH}t | j¡ ¡ }|j sBt
‚| j |j |j dd | j |j
|j
dd W 5 Q R X d S )Nr r r g{®Gáz„?re )r: r¼ r¶ rQ 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 sz t | j¡ ¡ }tjdttjdƒdJ}t | j¡ ¡ }t t
|ƒƒD ] }| j || || dd qD|j slt
‚W 5 Q R X d S )Nz(psutil._psplatform.cext.proc_io_countersr r r> re )r: r¼ r¶ rö r rš r r r ró rP rh 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¡ ¡ |¡ |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 r9 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 | ||¡ qd S )NT)Zuse_pebF)
r: rw rŠ Z proc_cmdliner r Ú
isinstancer¾ rÞ r9 )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ÚfilenamerF Ú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 ÚsetUpr5 r r r ÚsysÚ
executableZpython64Zpython32r7 r8 ÚcopyrÅ rÒ r Ú test_argsr- r/ Úproc32Úproc64)r< Z other_pythonr7 ©Ú __class__r r r9 # s.
ÿ
ÿ ÿzRemoteProcessTestCase.setUpc s" t ƒ ¡ | j ¡ | j ¡ d S rD )r ÚtearDownr> r1 r? rR r@ r r rB ; s
z RemoteProcessTestCase.tearDownc C s@ t | jj¡}| t| ¡ ƒd¡ | | ¡ dd … | j¡ d S ©Né r" )r: r¼ r> r¶ r9 rP 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 rC )r: r¼ r? r¶ r9 rP r r= rÀ r r r Útest_cmdline_64E s z%RemoteProcessTestCase.test_cmdline_64c C s& t | jj¡}| | ¡ t ¡ ¡ d S rD )r: r¼ r> r¶ r9 r r7 ÚgetcwdrÀ r r r Ú test_cwd_32J s z!RemoteProcessTestCase.test_cwd_32c C s& t | jj¡}| | ¡ t ¡ ¡ d S rD )r: r¼ r? r¶ r9 r r7 rG rÀ r r r Ú test_cwd_64N s z!RemoteProcessTestCase.test_cwd_64c C s> t | jj¡}| ¡ }| d|¡ | |d tt ¡ ƒ¡ d S )Nr6 )
r: r¼ r> r¶ r8 ÚassertInr9 rÅ r7 rÒ )r< rÁ Úer r r Útest_environ_32R s z%RemoteProcessTestCase.test_environ_32c C s6 t | jj¡}z | ¡ W n t jk
r0 Y nX d S rD )r: r¼ r? r¶ r8 r¾ rÀ r r r Útest_environ_64X s
z%RemoteProcessTestCase.test_environ_64)r r r r Ú staticmethodr5 r= r9 rB rE rF rH rI rL rM Ú
__classcell__r r r@ r r' s
r' c @ s e Zd Zdd„ Zdd„ ZdS )Ú TestServicesc C sh t dddddddgƒ}t dd d
gƒ}t ddd d d
ddgƒ}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 rHt |¡}| | ¡ ¡ t
| ¡ ¡}| ||¡ qBd S )NÚrunningÚpausedÚstartÚpauseÚcontinueÚstopÚstoppedZ automaticZmanualÚdisabledZ
start_pendingZ
pause_pendingZcontinue_pendingZ stop_pendingrÄ r! Ú display_namerÆ Ústatusr¶ ZbinpathZ
start_typeÚ description)rv r: Úwin_service_iterÚas_dictZassertIsInstancerÅ ZassertNotEqualr* rJ r¼ r¶ rŒ Ú
is_runningÚwin_service_getrÄ r9 )r< Zvalid_statusesZvalid_start_typesZservr2 r¶ rÁ Úsr r r Útest_win_service_iterg sP ù ù
z"TestServices.test_win_service_iterc C s¨ t jjj}t jjj}tt ¡ ƒ ¡ }| t j ¡}t
|d ¡ W 5 Q R X | |j j|d ¡ t
|¡}t
rzddd|f}n|df}t|Ž }tjd|d | t j |j¡ W 5 Q R X tjd|d | t j |j¡ W 5 Q R X t
rðddd|f}n|df}t|Ž }tjd|d | t j|j¡ W 5 Q R X tjd|d | t j|j¡ W 5 Q R X | | ¡ t|ƒ¡ | | ¡ t|ƒ¡ | | ¡ t|ƒ¡ | | ¡ 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Únextr\ rÄ r½ rÞ r_ r9 Ú exceptionr
r r rš rZ rÆ r¾ rJ rÅ rY r… )r< rb rc rÄ ÚcmZservicer) r r r r Útest_win_service_get“ sP ÿ
ÿ ÿ ÿ ÿz!TestServices.test_win_service_getN)r r r ra rg r r r r rP e s ,rP Ú__main__)Ú
run_from_name)=r r r r, r7 rÉ rä rÏ r- r: r r Úwarningsr: r Zpsutil._compatr r r Z psutil.testsr r r r r
r r r
r r r r r r Úcatch_warningsÚ simplefilterr? rÕ rè rI Zpsutil._pswindowsr r rŠ rX r r r6 r3 r4 rY rœ r´ r r r' rP r Zpsutil.tests.runnerri Ú__file__r r r r Ú s~
* ,G ^F _ ] ^