o
    ˆ¶¹aÂi  ã                   @   sj  d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl	m
Z
 ddl	mZ ddl	mZ ddl	mZ ddl	mZ ddl	mZ dd	l	mZ dd
l	mZ ddl	mZ ddl	mZ ddl	mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z  ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddlm'Z' ddlm(Z( dd lm)Z) dd!lm*Z* dd"lm+Z+ dd#lm,Z, G d$d%„ d%e$ƒZ-G d&d'„ d'e$ƒZ.G d(d)„ d)e$ƒZ/G d*d+„ d+e$ƒZ0G d,d-„ d-e$ƒZ1d.d/„ Z2e+G d0d1„ d1e$ƒƒZ3e4d2kr3dd3l5m6Z6 e6e7ƒ dS dS )4z°Contracts tests. These tests mainly check API sanity in terms of
returned types and APIs availability.
Some of these are duplicates of tests test_system.py and test_process.py
é    N)ÚAIX)ÚBSD)ÚFREEBSD)ÚLINUX)ÚMACOS)ÚNETBSD)ÚOPENBSD)ÚOSX)ÚPOSIX)ÚSUNOS)ÚWINDOWS)ÚFileNotFoundError)Úlong)Úrange)ÚAPPVEYOR)Ú
CI_TESTING)ÚGITHUB_ACTIONS)ÚHAS_CPU_FREQ)ÚHAS_NET_IO_COUNTERS)ÚHAS_SENSORS_FANS)ÚHAS_SENSORS_TEMPERATURES)ÚPYPY)ÚSKIP_SYSCONS)ÚVALID_PROC_STATUSES)ÚPsutilTestCase)Úcheck_connection_ntuple)Úcreate_sockets)Úenum)Úis_namedtuple)Úkernel_version)Úprocess_namespace)Ú	serialrun)Úunittestc                   @   sD   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Ze e	oe
d	¡d
d„ ƒZdS )ÚTestAvailConstantsAPIsc                 C   s   |   ttdƒtptpt¡ d S )NZPROCFS_PATH)ÚassertEqualÚhasattrÚpsutilr   r   r   ©Úself© r)   ú=/usr/lib/python3/dist-packages/psutil/tests/test_contracts.pyÚtest_PROCFS_PATHB   s   
ÿz'TestAvailConstantsAPIs.test_PROCFS_PATHc                 C   sj   | j }|ttdƒtƒ |ttdƒtƒ |ttdƒtƒ |ttdƒtƒ |ttdƒtƒ |ttdƒtƒ d S )NZABOVE_NORMAL_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZIDLE_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZREALTIME_PRIORITY_CLASS©r$   r%   r&   r   ©r(   Zaer)   r)   r*   Útest_win_priorityF   s   z(TestAvailConstantsAPIs.test_win_priorityc                 C   óJ   | j }|ttdƒtƒ |ttdƒtƒ |ttdƒtƒ |ttdƒtƒ d S )NZIOPRIO_CLASS_NONEZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLE©r$   r%   r&   r   r-   r)   r)   r*   Útest_linux_ioprio_linuxO   ó
   z.TestAvailConstantsAPIs.test_linux_ioprio_linuxc                 C   r/   )NÚIOPRIO_HIGHÚIOPRIO_NORMALÚ
IOPRIO_LOWÚIOPRIO_VERYLOWr,   r-   r)   r)   r*   Útest_linux_ioprio_windowsV   r2   z0TestAvailConstantsAPIs.test_linux_ioprio_windowsú%unsupported on GITHUB_ACTIONS + LINUXc                 C   s¬  | j }|ttdƒtptƒ |ttdƒtptƒ |ttdƒtptƒ |ttdƒtp)tƒ |ttdƒtp3tƒ |ttdƒtp=tƒ |ttdƒtpGtƒ |ttdƒtpQtƒ |ttd	ƒtp[tƒ |ttd
ƒtpetƒ |ttdƒtpotƒ |ttdƒtƒ tr¼tƒ dkrˆ|ttdƒtƒ tƒ dkr•|ttdƒtƒ tƒ dkr¢|ttdƒtƒ tƒ dkr¯|ttdƒtƒ tƒ dkr¼|ttdƒtƒ |ttdƒtƒ |ttdƒtƒ |ttdƒtƒ d S )NZRLIM_INFINITYZ	RLIMIT_ASZRLIMIT_COREZ
RLIMIT_CPUZRLIMIT_DATAZRLIMIT_FSIZEZRLIMIT_MEMLOCKZRLIMIT_NOFILEZRLIMIT_NPROCZ
RLIMIT_RSSZRLIMIT_STACKZRLIMIT_LOCKS)é   é   é   ZRLIMIT_MSGQUEUE)r9   r:   é   ZRLIMIT_NICEZRLIMIT_RTPRIO)r9   r:   é   ZRLIMIT_RTTIMEZRLIMIT_SIGPENDINGZRLIMIT_SWAPZRLIMIT_SBSIZEZRLIMIT_NPTS)r$   r%   r&   r   r   r
   r   r-   r)   r)   r*   Útest_rlimit]   s6   




z"TestAvailConstantsAPIs.test_rlimitN)Ú__name__Ú
__module__Ú__qualname__r+   r.   r1   r7   r"   ÚskipIfr   r   r>   r)   r)   r)   r*   r#   @   s    	
ÿr#   c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestAvailSystemAPIsc                 C   ó   |   ttdƒt¡ d S )NZwin_service_iterr,   r'   r)   r)   r*   Útest_win_service_iter   ó   z)TestAvailSystemAPIs.test_win_service_iterc                 C   rD   )NZwin_service_getr,   r'   r)   r)   r*   Útest_win_service_get„   rF   z(TestAvailSystemAPIs.test_win_service_getc                 C   ó"   |   ttdƒtptptpt¡ d S )NÚcpu_freq)r$   r%   r&   r   r   r   r   r'   r)   r)   r*   Útest_cpu_freq‡   ó   ÿz!TestAvailSystemAPIs.test_cpu_freqc                 C   s   |   ttdƒtp	t¡ d S )NÚsensors_temperatures)r$   r%   r&   r   r   r'   r)   r)   r*   Útest_sensors_temperatures‹   s   ÿz-TestAvailSystemAPIs.test_sensors_temperaturesc                 C   rD   )NÚsensors_fansr0   r'   r)   r)   r*   Útest_sensors_fans   rF   z%TestAvailSystemAPIs.test_sensors_fansc                 C   rH   )NZsensors_battery)r$   r%   r&   r   r   r   r   r'   r)   r)   r*   Útest_battery’   rK   z TestAvailSystemAPIs.test_batteryN)	r?   r@   rA   rE   rG   rJ   rM   rO   rP   r)   r)   r)   r*   rC      s    rC   c                   @   s|   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Ze 	e
oed¡dd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )ÚTestAvailProcessAPIsc                 C   s4   |   ttjdƒtptptptptpt	pt
pt¡ d S )NÚenviron)r$   r%   r&   ÚProcessr   r   r   r   r   r   r   r   r'   r)   r)   r*   Útest_environ™   s   ÿÿþz!TestAvailProcessAPIs.test_environc                 C   ó   |   ttjdƒt¡ d S ©NÚuids©r$   r%   r&   rS   r
   r'   r)   r)   r*   Ú	test_uidsž   ó   zTestAvailProcessAPIs.test_uidsc                 C   rU   rV   rX   r'   r)   r)   r*   Ú	test_gids¡   rZ   zTestAvailProcessAPIs.test_gidsc                 C   rU   )NÚterminalrX   r'   r)   r)   r*   Útest_terminal¤   rZ   z"TestAvailProcessAPIs.test_terminalc                 C   ó   |   ttjdƒtp
t¡ d S )NÚionice)r$   r%   r&   rS   r   r   r'   r)   r)   r*   Útest_ionice§   s   z TestAvailProcessAPIs.test_ionicer8   c                 C   r^   )NÚrlimit)r$   r%   r&   rS   r   r   r'   r)   r)   r*   r>   ª   s   z TestAvailProcessAPIs.test_rlimitc                 C   s.   t tjdƒ}|  |tstrd¡ d S d¡ d S )NÚio_countersFT)r%   r&   rS   r$   r   r   ©r(   Zhasitr)   r)   r*   Útest_io_counters¯   s   "z%TestAvailProcessAPIs.test_io_countersc                 C   rU   )NÚnum_fdsrX   r'   r)   r)   r*   Útest_num_fds³   rZ   z!TestAvailProcessAPIs.test_num_fdsc                 C   rU   )NÚnum_handles)r$   r%   r&   rS   r   r'   r)   r)   r*   Útest_num_handles¶   rZ   z%TestAvailProcessAPIs.test_num_handlesc                 C   ó    |   ttjdƒtptpt¡ d S )NÚcpu_affinity)r$   r%   r&   rS   r   r   r   r'   r)   r)   r*   Útest_cpu_affinity¹   ó   
ÿz&TestAvailProcessAPIs.test_cpu_affinityc                 C   ri   )NÚcpu_num)r$   r%   r&   rS   r   r   r   r'   r)   r)   r*   Útest_cpu_num½   rl   z!TestAvailProcessAPIs.test_cpu_numc                 C   s6   t tjdƒ}|  |tstststrd¡ d S d¡ d S )NÚmemory_mapsFT)r%   r&   rS   r$   r   r   r   r   rc   r)   r)   r*   Útest_memory_mapsÁ   s   ÿÿz%TestAvailProcessAPIs.test_memory_mapsN)r?   r@   rA   rT   rY   r[   r]   r`   r"   rB   r   r   r>   rd   rf   rh   rk   rn   rp   r)   r)   r)   r*   rQ   —   s     
ÿ
rQ   c                   @   sæ   e Zd ZdZedd„ ƒZedfdd„Zdd„ Zd	d
„ Z	dd„ Z
dd„ Ze e d¡dd„ ƒZdd„ Zdd„ Ze ed¡dd„ ƒZdd„ Zdd„ Ze e d¡dd„ ƒZe e d¡dd „ ƒZe e d¡d!d"„ ƒZd#d$„ Zd%d&„ Zd'S )(ÚTestSystemAPITypesz¬Check the return types of system related APIs.
    Mainly we want to test we never return unicode on Python 2, see:
    https://github.com/giampaolo/psutil/issues/1039
    c                 C   ó   t  ¡ | _d S ©N)r&   rS   Úproc)Úclsr)   r)   r*   Ú
setUpClassÒ   s   zTestSystemAPITypes.setUpClassTc                 C   s6   t |ƒsJ ‚|D ]}|  ||¡ |r|  |d¡ qd S ©Nr   )r   ÚassertIsInstanceÚassertGreaterEqual)r(   ÚntÚtype_ZgezeroÚnr)   r)   r*   Úassert_ntuple_of_numsÖ   s   €ýz(TestSystemAPITypes.assert_ntuple_of_numsc                 C   s.   |   t ¡ ¡ tjddD ]}|   |¡ qd S )NT)Zpercpu)r}   r&   Ú	cpu_times)r(   rz   r)   r)   r*   Útest_cpu_timesÝ   s   ÿz!TestSystemAPITypes.test_cpu_timesc                 C   s,   |   tjd dt¡ |   tjddt¡ d S )N©Úintervalgñhãˆµøä>)rx   r&   Úcpu_percentÚfloatr'   r)   r)   r*   Útest_cpu_percentâ   s   z#TestSystemAPITypes.test_cpu_percentc                 C   s(   |   tjd d¡ |   tjdd¡ d S )Nr€   g-Cëâ6?)r}   r&   Zcpu_times_percentr'   r)   r)   r*   Útest_cpu_times_percentæ   s   z)TestSystemAPITypes.test_cpu_times_percentc                 C   ó   |   t ¡ t¡ d S rs   )rx   r&   Ú	cpu_countÚintr'   r)   r)   r*   Útest_cpu_countê   s   z!TestSystemAPITypes.test_cpu_countznot supportedc                 C   s2   t  ¡ d u r|  d¡‚| jt  ¡ tttfd d S )Nzcpu_freq() returns None©r{   )r&   rI   ZskipTestr}   rƒ   rˆ   r   r'   r)   r)   r*   rJ   í   s   
z TestSystemAPITypes.test_cpu_freqc                 C   s<   t jdd ¡ D ]\}}|  |t¡ | j|ttfd qd S )NT)ZperdiskrŠ   )r&   Zdisk_io_countersÚitemsrx   Ústrr}   rˆ   r   )r(   ÚkÚvr)   r)   r*   Útest_disk_io_countersó   s   þz(TestSystemAPITypes.test_disk_io_countersc                 C   sf   t  ¡ D ],}|  |jt¡ |  |jt¡ |  |jt¡ |  |jt¡ |  |jt	¡ |  |j
t	¡ qd S rs   )r&   Zdisk_partitionsrx   ZdevicerŒ   Z
mountpointZfstypeZoptsZmaxfilerˆ   Zmaxpath)r(   Zdiskr)   r)   r*   Útest_disk_partitionsù   s   úz'TestSystemAPITypes.test_disk_partitionszrequires rootc                 C   sd   t ƒ % t d¡}|  t|ƒtt|ƒƒ¡ |D ]}t|ƒsJ ‚qW d   ƒ d S 1 s+w   Y  d S )NÚall)r   r&   Znet_connectionsr$   ÚlenÚsetr   )r(   ÚretÚconnr)   r)   r*   Útest_net_connections  s   
ÿ"ýz'TestSystemAPITypes.test_net_connectionsc                 C   s–   t  ¡  ¡ D ]B\}}|  |t¡ |D ]5}td ur#ts#|  |jtj¡ n|  |jt	¡ |  |j
t¡ |  |jttd ƒf¡ |  |jttd ƒf¡ qqd S rs   )r&   Znet_if_addrsr‹   rx   rŒ   r   r   ZfamilyÚIntEnumrˆ   ZaddressZnetmaskÚtypeZ	broadcast)r(   ÚifnameZaddrsÚaddrr)   r)   r*   Útest_net_if_addrs  s   ùþz$TestSystemAPITypes.test_net_if_addrsc                 C   sx   t  ¡  ¡ D ]3\}}|  |t¡ |  |jt¡ td ur$|  |jtj	¡ n|  |jt
¡ |  |jt
¡ |  |jt
¡ qd S rs   )r&   Znet_if_statsr‹   rx   rŒ   ZisupÚboolr   Zduplexr—   rˆ   ZspeedZmtu)r(   r™   Úinfor)   r)   r*   Útest_net_if_stats  s   øz$TestSystemAPITypes.test_net_if_statsc                 C   s*   t jdd ¡ D ]
\}}|  |t¡ qd S )NT)Zpernic)r&   Znet_io_countersr‹   rx   rŒ   )r(   r™   Ú_r)   r)   r*   Útest_net_io_counters$  s   ÿz'TestSystemAPITypes.test_net_io_countersc              	   C   sV   t  ¡  ¡ D ]"\}}|  |t¡ |D ]}|  |jt¡ |  |jttt	d ƒf¡ qqd S rs   )
r&   rN   r‹   rx   rŒ   ÚlabelÚcurrentrƒ   rˆ   r˜   ©r(   ÚnameZunitsZunitr)   r)   r*   rO   *  s   þþz$TestSystemAPITypes.test_sensors_fansc              	   C   s†   t  ¡  ¡ D ]:\}}|  |t¡ |D ]-}|  |jt¡ |  |jttt	d ƒf¡ |  |j
ttt	d ƒf¡ |  |jttt	d ƒf¡ qqd S rs   )r&   rL   r‹   rx   rŒ   r¡   r¢   rƒ   rˆ   r˜   ZhighZcriticalr£   r)   r)   r*   rM   3  s   üþz,TestSystemAPITypes.test_sensors_temperaturesc                 C   r†   rs   )rx   r&   Z	boot_timerƒ   r'   r)   r)   r*   Útest_boot_time>  s   z!TestSystemAPITypes.test_boot_timec                 C   sb   t  ¡ D ]*}|  |jt¡ |  |jttd ƒf¡ |  |jttd ƒf¡ |  |jt	td ƒf¡ qd S rs   )
r&   Zusersrx   r¤   rŒ   r\   r˜   ZhostÚpidrˆ   )r(   Úuserr)   r)   r*   Ú
test_usersB  s   üzTestSystemAPITypes.test_usersN)r?   r@   rA   Ú__doc__Úclassmethodrv   rƒ   r}   r   r„   r…   r‰   r"   rB   r   rJ   r   r   r   r–   r›   rž   r   r    r   rO   r   rM   r¥   r¨   r)   r)   r)   r*   rq   Ì   s2    








rq   c                   @   s"   e Zd Ze e d¡dd„ ƒZdS )ÚTestProcessWaitTypez	not POSIXc                 C   sZ   t  |  ¡ j¡}| ¡  | ¡ }|  |tj ¡ t	d ur%|  
|t	j¡ d S |  
|t¡ d S rs   )r&   rS   Zspawn_testprocr¦   Ú	terminateÚwaitr$   ÚsignalÚSIGTERMr   rx   r—   rˆ   )r(   ÚpÚcoder)   r)   r*   Útest_negative_signalM  s   z(TestProcessWaitType.test_negative_signalN)r?   r@   rA   r"   rB   r
   r²   r)   r)   r)   r*   r«   K  s    r«   c                    sè   t ƒ ‰‡‡fdd„‰ ‡ ‡‡‡‡fdd„}zt ˆ¡‰ˆ ddg¡}W n tjy.   i  Y S w |d |d ‰‰dˆji}tˆƒ}|j|jdd	D ]%\}}z|ƒ ||< W qI tj	yn } zˆ |ˆˆˆƒ W Y d }~qId }~ww |ƒ  |S )
Nc                    sr   ˆ  | jˆ ¡ ˆ  | j|¡ t| tjƒr(| jd ur'ˆ | jd¡ ˆ  | j|¡ nt| tjƒr3ˆ 	|¡ t
| ƒ d S rw   )r$   r¦   r¤   Ú
isinstancer&   ZZombieProcessÚppidry   ÚNoSuchProcessZassertProcessGonerŒ   )Úexcrt   r¤   r´   )r¦   Útcaser)   r*   Úcheck_exceptiona  s   
€
z"proc_info.<locals>.check_exceptionc               
      sT   ˆdkr(zˆ  d¡ W d S  tjy' }  zˆ | ˆˆˆƒ W Y d } ~ d S d } ~ ww d S rw   )r­   r&   ÚError)r¶   )r¸   r¤   r¦   r´   rt   r)   r*   Údo_waitl  s   €ÿýzproc_info.<locals>.do_waitr´   r¤   r¦   F)Zclear_cache)
r   r&   rS   Zas_dictrµ   r¦   r    ÚiterZgettersr¹   )r¦   rº   Údr   ÚnsZfunZfun_namer¶   r)   )r¸   r¤   r¦   r´   rt   r·   r*   Ú	proc_info^  s,   
ÿ
€þr¾   c                   @   s8  e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Zd4d5„ Zd6d7„ Zd8d9„ Zd:d;„ Z d<d=„ Z!d>d?„ Z"d@dA„ Z#dBdC„ Z$dDdE„ Z%dFdG„ Z&dHdI„ Z'dJdK„ Z(dLS )MÚTestFetchAllProcessesz·Test which iterates over all running processes and performs
    some sanity checks against Process API's returned values.
    Uses a process pool to get info about all processes.
    c                 C   rr   rs   )ÚmultiprocessingZPoolÚpoolr'   r)   r)   r*   ÚsetUp  s   zTestFetchAllProcesses.setUpc                 C   s   | j  ¡  | j  ¡  d S rs   )rÁ   r¬   Újoinr'   r)   r)   r*   ÚtearDown’  s   
zTestFetchAllProcesses.tearDownc                 C   s   ddl m} | j |t ¡ ¡S )Nr   )r¾   )Zpsutil.tests.test_contractsr¾   rÁ   Zimap_unorderedr&   Zpids)r(   r¾   r)   r)   r*   Úiter_proc_info–  s   z$TestFetchAllProcesses.iter_proc_infoc                 C   sè   g }|   ¡ D ]a}| ¡ D ]Z\}}t| |ƒ}z|||ƒ W n9 tyU   d}|d||d t|ƒf 7 }|d7 }|dt ¡  7 }d dd„ | ¡ D ƒ¡}|d7 }| 	|¡ Y qw |d	d
g d di fvrf|sfJ |ƒ‚qq|rr|  
d |¡¡‚d S )NzH
======================================================================
zFAIL: test_%s pid=%s, ret=%s
r¦   zF----------------------------------------------------------------------z
%sÚ
c                 s   s    | ]}d | V  qdS )z    Nr)   )Ú.0Úir)   r)   r*   Ú	<genexpr>©  s   € z1TestFetchAllProcesses.test_all.<locals>.<genexpr>r   ç        Ú )rÅ   r‹   ÚgetattrÚAssertionErrorÚreprÚ	tracebackÚ
format_excrÃ   Ú
splitlinesÚappendZfail)r(   Zfailuresr   r¤   ÚvalueZmethÚsr)   r)   r*   Útest_allœ  s0   
ÿø
€ñÿzTestFetchAllProcesses.test_allc                 C   s&   |   |t¡ |D ]}|   |t¡ qd S rs   )rx   ÚlistrŒ   )r(   r”   r   Úpartr)   r)   r*   Úcmdline²  s   ÿzTestFetchAllProcesses.cmdlinec                 C   sÈ   |   |ttd ƒf¡ |s|  |d¡ d S tr| d¡sd S tj |¡s'J |ƒ‚t	r\tj 
|¡r^ttdƒr`ttdƒrbzt |tj¡sCJ ‚W d S  ty[   tj |¡rUtsX‚ Y d S Y d S w d S d S d S d S )NrË   z.exeÚaccessÚX_OK)rx   rŒ   r˜   r$   r   ÚendswithÚosÚpathÚisabsr
   Úisfiler%   rÙ   rÚ   rÍ   Úexistsr   ©r(   r”   r   r)   r)   r*   Úexe·  s"   ÿÿûzTestFetchAllProcesses.exec                 C   ó   |   |t¡ |  |d¡ d S rw   ©rx   rˆ   ry   rá   r)   r)   r*   r¦   Ë  ó   zTestFetchAllProcesses.pidc                 C   s    |   |ttf¡ |  |d¡ d S rw   )rx   rˆ   r   ry   rá   r)   r)   r*   r´   Ï  s   zTestFetchAllProcesses.ppidc                 C   s8   |   |t¡ tr|s|d dkrd S ts|sJ ‚d S d S )NÚstatusÚstopped)rx   rŒ   r   r   rá   r)   r)   r*   r¤   Ó  s   ÿzTestFetchAllProcesses.namec                 C   s\   |   |t¡ z|  |d¡ W n ty"   tr|d tjkrn‚ Y nw t dt 	|¡¡ d S )Nr   ræ   z%Y %m %d %H:%M:%S)
rx   rƒ   ry   rÍ   r   r&   ZSTATUS_ZOMBIEÚtimeÚstrftimeÚ	localtimerá   r)   r)   r*   Úcreate_timeÛ  s   þý
z!TestFetchAllProcesses.create_timec                 C   ó2   t |ƒsJ ‚|D ]}|  |t¡ |  |d¡ qd S rw   )r   rx   rˆ   ry   )r(   r”   r   Úuidr)   r)   r*   rW   ë  ó
   þzTestFetchAllProcesses.uidsc                 C   s:   t |ƒsJ ‚|D ]}|  |t¡ tsts|  |d¡ qd S rw   )r   rx   rˆ   r   r   ry   )r(   r”   r   Úgidr)   r)   r*   Úgidsñ  s   €ýzTestFetchAllProcesses.gidsc                 C   s   |   |t¡ |s
J ‚d S rs   )rx   rŒ   rá   r)   r)   r*   Úusernameú  s   zTestFetchAllProcesses.usernamec                 C   s0   |   |t¡ |s
J ‚|  |d¡ |  |t¡ d S )Nú?)rx   rŒ   ZassertNotEqualÚassertInr   rá   r)   r)   r*   ræ   þ  s   zTestFetchAllProcesses.statusc                 C   s>   t |ƒsJ ‚|D ]}|  |ttf¡ |dkr|  |d¡ qd S )Néÿÿÿÿr   ©r   rx   rˆ   r   ry   )r(   r”   r   Úfieldr)   r)   r*   rb     s   €ýz!TestFetchAllProcesses.io_countersc                 C   s|   t r |  |jt¡ |  |jt¡ |  |jd¡ |  |jd¡ d S tjtjtj	tj
g}|  |t¡ |  |d¡ |  ||¡ d S rw   )r   rx   Zioclassrˆ   rÓ   ry   r&   r6   r5   r4   r3   ró   )r(   r”   r   Úchoicesr)   r)   r*   r_     s   üzTestFetchAllProcesses.ionicec                 C   s4   |   |t¡ tr|s|d dkrd S |  |d¡ d S )Nræ   rç   é   )rx   rˆ   r   ry   rá   r)   r)   r*   Únum_threads  s   z!TestFetchAllProcesses.num_threadsc                 C   sj   |   |t¡ |D ]*}t|ƒsJ ‚|  |jd¡ |  |jd¡ |  |jd¡ |D ]
}|   |ttf¡ q'qd S rw   )	rx   rÖ   r   ry   ÚidZ	user_timeZsystem_timerˆ   rƒ   )r(   r”   r   Útrö   r)   r)   r*   Úthreads!  s   ÿûzTestFetchAllProcesses.threadsc                 C   rì   rw   )r   rx   rƒ   ry   )r(   r”   r   r|   r)   r)   r*   r~   +  rî   zTestFetchAllProcesses.cpu_timesc                 C   ó4   |   |t¡ d|  krdksJ |ƒ‚ J |ƒ‚d S )NrÊ   g      Y@©rx   rƒ   rá   r)   r)   r*   r‚   2  ó   (z!TestFetchAllProcesses.cpu_percentc                 C   s\   |   |t¡ tr|dkrd S |  |d¡ t ¡ dkr |  |d¡ |  |tt	t ¡ ƒƒ¡ d S )Nrô   r   rø   )
rx   rˆ   r   ry   r&   r‡   r$   ró   rÖ   r   rá   r)   r)   r*   rm   6  s   zTestFetchAllProcesses.cpu_numc                 C   s~   t |ƒsJ ‚|D ]}|  |ttf¡ |  |d¡ qtr=|  |j|j¡ |  |j|j	¡ |  |j
|j¡ |  |j|j¡ d S d S rw   )r   rx   rˆ   r   ry   r   Z	peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZnonpaged_poolZpeak_pagefileZpagefile©r(   r”   r   rÓ   r)   r)   r*   Úmemory_info?  s   üz!TestFetchAllProcesses.memory_infoc                 C   s”   t |ƒsJ ‚t ¡ j}|jD ]-}t||ƒ}|  |ttf¡ | j	|d||fd t
s/tr0|dv r0q| j|||||fd qt
rH|  	|j|j¡ d S d S )Nr   )Úmsg)ZvmsÚdata)r   r&   Zvirtual_memoryÚtotalÚ_fieldsrÌ   rx   rˆ   r   ry   r   r	   ZassertLessEqualZpssZuss)r(   r”   r   r  r¤   rÓ   r)   r)   r*   Úmemory_full_infoJ  s   


ÿz&TestFetchAllProcesses.memory_full_infoc              	   C   sþ   |   |t¡ |D ]t}|   |jt¡ |   |jt¡ tr"|  |jd¡ n3trO|   |j	t¡ |   |j
t¡ |   |jt¡ |  |j	d¡ |  |j
d¡ |  |jd¡ ntrU|jsUqtj |j¡s`J |ƒ‚zt |j¡}W n	 tyq   Y qw t |j¡s|J |ƒ‚qd S )Nrô   r   )ÚrÚwÚazr+za+)rx   rÖ   Úfdrˆ   rÝ   rŒ   r   r$   r   ZpositionÚmodeÚflagsry   ró   ZassertGreaterr   rÜ   rÞ   Ústatr   ÚS_ISREGÚst_mode)r(   r”   r   ÚfÚstr)   r)   r*   Ú
open_filesZ  s.   
ÿëz TestFetchAllProcesses.open_filesc                 C   rã   rw   rä   rá   r)   r)   r*   re   s  rå   zTestFetchAllProcesses.num_fdsc                 C   sb   t ƒ $ |  t|ƒtt|ƒƒ¡ |D ]}t|ƒsJ ‚t|ƒ qW d   ƒ d S 1 s*w   Y  d S rs   )r   r$   r’   r“   r   r   )r(   r”   r   r•   r)   r)   r*   Úconnectionsw  s   
þ"þz!TestFetchAllProcesses.connectionsc              
   C   s¢   |rM|   |t¡ tj |¡sJ |ƒ‚zt |¡}W n+ tyD } ztr+|jt	j
jv r+n|jtjkr2‚ W Y d }~d S W Y d }~d S d }~ww t |j¡sOJ ‚d S d S rs   )rx   rŒ   rÜ   rÝ   rÞ   r  ÚOSErrorr   Úerrnor&   Z_psplatformZACCESS_DENIED_SETÚENOENTÚS_ISDIRr  )r(   r”   r   r  Úerrr)   r)   r*   Úcwd~  s&   ÿÿþ€ýózTestFetchAllProcesses.cwdc                 C   rý   )Nr   éd   rþ   rá   r)   r)   r*   Úmemory_percentŽ  rÿ   z$TestFetchAllProcesses.memory_percentc                 C   s   |   |t¡ d S rs   )rx   rœ   rá   r)   r)   r*   Ú
is_running’  s   z TestFetchAllProcesses.is_runningc                 C   sR   |   |t¡ |g ksJ |ƒ‚ttt ¡ ƒƒ}|D ]}|   |t¡ |  ||¡ qd S rs   )rx   rÖ   r   r&   r‡   rˆ   ró   )r(   r”   r   Zcpusr|   r)   r)   r*   rj   •  s   þz"TestFetchAllProcesses.cpu_affinityc                 C   sL   |   |ttd ƒf¡ |d ur"tj |¡sJ |ƒ‚tj |¡s$J |ƒ‚d S d S rs   )rx   rŒ   r˜   rÜ   rÝ   rÞ   rà   rá   r)   r)   r*   r\     s   þzTestFetchAllProcesses.terminalc                 C   sÎ   |D ]b}|   |jt¡ |   |jt¡ |   |jt¡ |jD ]G}t||ƒ}|dkr9| d¡s8tj 	|j¡s8J |jƒ‚q|dkrF|sEJ t
|ƒƒ‚q|dkrUtsT|sTJ t
|ƒƒ‚q|   |ttf¡ |  |d¡ qqd S )NrÝ   ú[rš   Úpermsr   )rx   rš   rŒ   r  rÝ   r  rÌ   Ú
startswithrÜ   rÞ   rÎ   r   rˆ   r   ry   )r(   r”   r   rz   ÚfnamerÓ   r)   r)   r*   ro   £  s(   


€€ñüz!TestFetchAllProcesses.memory_mapsc                 C   rã   rw   rä   rá   r)   r)   r*   rg   ¹  rå   z!TestFetchAllProcesses.num_handlesc                 C   s‚   |   |t¡ trd|  krdksJ |ƒ‚ J |ƒ‚d S dd„ ttƒD ƒ}|  ||¡ tjdkr9|   |tj	¡ d S |   |t¡ d S )Niìÿÿÿé   c                 S   s    g | ]}|  d ¡rtt|ƒ‘qS )Z_PRIORITY_CLASS)rÛ   rÌ   r&   )rÇ   Úxr)   r)   r*   Ú
<listcomp>Â  s    ÿz.TestFetchAllProcesses.nice.<locals>.<listcomp>)é   é   )
rx   rˆ   r
   Údirr&   ró   ÚsysÚversion_infor   r—   )r(   r”   r   Z
prioritiesr)   r)   r*   Únice½  s   (
zTestFetchAllProcesses.nicec                 C   s6   t |ƒsJ ‚|D ]}|  |ttf¡ |  |d¡ qd S rw   rõ   r   r)   r)   r*   Únum_ctx_switchesÊ  s
   þz&TestFetchAllProcesses.num_ctx_switchesc                 C   s@   |   |t¡ |  t|ƒd¡ |  |d d¡ |  |d d¡ d S )Nr9   r   rô   rø   )rx   Útupler$   r’   ry   rá   r)   r)   r*   ra   Ð  s   zTestFetchAllProcesses.rlimitc                 C   s:   |   |t¡ | ¡ D ]\}}|   |t¡ |   |t¡ q
d S rs   )rx   Údictr‹   rŒ   )r(   r”   r   r   rŽ   r)   r)   r*   rR   Ö  s
   þzTestFetchAllProcesses.environN))r?   r@   rA   r©   rÂ   rÄ   rÅ   rÕ   rØ   râ   r¦   r´   r¤   rë   rW   rð   rñ   ræ   rb   r_   rù   rü   r~   r‚   rm   r  r  r  re   r  r  r  r  rj   r\   ro   rg   r)  r*  ra   rR   r)   r)   r)   r*   r¿   ˆ  sN    	
	r¿   Ú__main__)Úrun_from_name)8r©   r  rÀ   rÜ   r®   r  r'  rè   rÏ   r&   r   r   r   r   r   r   r   r	   r
   r   r   Zpsutil._compatr   r   r   Zpsutil.testsr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   rC   rQ   rq   r«   r¾   r¿   r?   Zpsutil.tests.runnerr.  Ú__file__r)   r)   r)   r*   Ú<module>   sr   
?5*  
Vþ