3 ��JZ/��@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddlZddlZddlm Z ddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZddlmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddlm&Z&ddlm'Z'ddlm(Z(ddlm)Z)ddlm*Z*ddlm+Z+dd lm,Z,dd!lm-Z-Gd"d#�d#e-j.�Z/e0d$k�r�e(e1�dS)%zTests for system APIS.�N)�AIX)�BSD)�FREEBSD)�LINUX)�NETBSD)�OPENBSD)�OSX)�POSIX)�SUNOS)�WINDOWS)�long)�APPVEYOR)�ASCII_FS)�check_net_address)�DEVNULL)�enum)�get_test_subprocess)�HAS_BATTERY)�HAS_CPU_FREQ)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)�mock)� reap_children)�retry_before_failing)�run_test_module_by_name)�safe_rmpath)�TESTFN)�TESTFN_UNICODE)�TRAVIS)�unittestc@sBeZdZdZdd�Zdd�Zdd�Zdd �Zd d�Zdd �Z dd�Z eje d�dd��Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Z d7d8�Z!d9d:�Z"d;d<�Z#d=d>�Z$d?d@�Z%dAdB�Z&dCdD�Z'dEdF�Z(dGdH�Z)eje*dI�dJdK��Z+eje,�o\e-j.j/dL�dM�eje0�ote1j2�dNkdO�dPdQ��Z3dRdS�Z4eje0�p�e*�o�e1j5�dT�dUdV��Z6dWdX�Z7eje8dY�dZd[��Z9d\d]�Z:eje;d^�d_d`��Zd^�eje?dc�ddde��Z@ejeAd^�dfdg��ZBdNS)h�TestSystemAPIszTests for system-related APIs.cCstt�dS)N)rr)�self�r#�#/usr/lib64/python3.6/test_system.py�setUp>szTestSystemAPIs.setUpcCs t�dS)N)r)r"r#r#r$�tearDownAszTestSystemAPIs.tearDowncCs�|jtj�dd�tj�D��t�}|j|jdd�tj�D��tj|j�}|j�|j �|j |jdd�tj�D��tjdtj tj��d��|jttj��g�WdQRXtjdtjtj��d��*|jtj��ttj��WdQRXWdQRXdS)NcSsg|] }|j�qSr#)�pid)�.0�xr#r#r$� Esz4TestSystemAPIs.test_process_iter..cSsg|] }|j�qSr#)r')r(r)r#r#r$r*GscSsg|] }|j�qSr#)r')r(r)r#r#r$r*Kszpsutil.Process)�side_effect)�assertIn�os�getpid�psutil�process_iterrr'�Process�kill�wait�assertNotInr�patchZ NoSuchProcess�assertEqual�list�AccessDenied�assertRaises)r"�sproc�pr#r#r$�test_process_iterDsz TestSystemAPIs.test_process_itercCs"x.tjdgd�D]}|jt|jj��dg�qW|jt��ttjdgd��WdQRXtj dtj dd�d��B}x:tjddgd�D]&}|j|jd�|j|jdd�q�WWdQRXtj dtj dd�d��L}t �}x>tjddg|d �D](}|j|jd|�|j|jdd�q�WWdQRXdS) Nr')�attrsZfooz$psutil._psplatform.Process.cpu_timesr�)r+� cpu_times)r=Zad_value)r/r0r6r7�info�keysr9� ValueErrorrr5r8�assertIsNone�assertGreaterEqual�object�assertIs)r"r;�m�flagr#r#r$�test_prcess_iter_w_paramsUs$ z(TestSystemAPIs.test_prcess_iter_w_paramscs��fdd�}g�t�}t�}t�}dd�|||fD�}�jttj|dd��jttj|dd�tj�}tj|d|d �\}}�jtj�|d ��j|g��jt |�d��j�g�x|D]} �j t| d��q�Wtd ��fdd��} |j �| ||�\}}�j|jdd�|D��t�r4�j|j�jtj�n�j|j�jd��j�|jg�x|D]} �j t| d��q\Wtd ��fdd��} |j �|j �| ||�\}}�jt��t|j|j|jg��x|D]} �jt| d��q�WdS)Ncs�j|j�dS)N)�appendr')r;)�pidsr#r$�callbackjsz0TestSystemAPIs.test_wait_procs..callbackcSsg|]}tj|j��qSr#)r/r1r')r(r)r#r#r$r*qsz2TestSystemAPIs.test_wait_procs..�)�timeout)rLg{�G�z�?)rNrLg�?�� returncode�cs<tj|d|d�\}}�jt|�d��jt|�d�||fS)Ng��Q��?)rNrLrM�)r/� wait_procsr6�len)�procsrL�gone�alive)r"r#r$�test~s z,TestSystemAPIs.test_wait_procs..testcSsg|] }|j�qSr#)r')r(r)r#r#r$r*�scs<tj|d|d�\}}�jt|�d��jt|�d�||fS)Ng��Q��?)rNrLrOr)r/rSr6rT)rUrLrVrW)r"r#r$rX�s ��)rr9rBr/rS� TypeError�time� assertLessr6rT�assertFalse�hasattrr� terminater,r'r �poprP�signal�SIGTERM�set� assertTrue)r"rL�sproc1�sproc2�sproc3rU�trVrWr;rXr#)rKr"r$�test_wait_procsisB zTestSystemAPIs.test_wait_procscCsNt�}t�}t�}dd�|||fD�}x|D]}|j�q,Wtj|�\}}dS)NcSsg|]}tj|j��qSr#)r/r1r')r(r)r#r#r$r*�sz=TestSystemAPIs.test_wait_procs_no_timeout..)rr_r/rS)r"rerfrgrUr;rVrWr#r#r$�test_wait_procs_no_timeout�s z)TestSystemAPIs.test_wait_procs_no_timeoutcCs4tj�}|j|t�|j|d�|j|tj��dS)Nr)r/Z boot_time�assertIsInstance�float� assertGreaterr\r[)r"Zbtr#r#r$�test_boot_time�szTestSystemAPIs.test_boot_timez POSIX onlycCs"ddl}|jtjd�|j��dS)Nr�SC_PAGE_SIZE)�resourcer6r-�sysconfZgetpagesize)r"rpr#r#r$� test_PAGESIZE�szTestSystemAPIs.test_PAGESIZEcCs�tj�}xt|jD]j}t||�}|dkr6|j|ttf�|dkr|dksX|jd||f�||jkr|jd||j||f�qWdS)N�percent�totalrz%r < 0 (%s)z%r > total (total=%s, %s=%s)) r/Zvirtual_memory�_fields�getattrrk�intr�failrt)r"�mem�name�valuer#r#r$�test_virtual_memory�s z"TestSystemAPIs.test_virtual_memorycCs&tj�}|j|jd�|jdkr"ndS) Nrt�used�freers�sin�soutr)rtr}r~rsrr�)r/Zswap_memoryr6rurt)r"ryr#r#r$�test_swap_memory�s zTestSystemAPIs.test_swap_memorycCstt�}|jtj|j��tj|j�}|j�|j�|jtj|j��|jtjd��|j tjd�dtj �k�dS)NrMrrY)rrdr/� pid_existsr'r1r2r3r]r6rK)r"r:r;r#r#r$�test_pid_exists�szTestSystemAPIs.test_pid_existscCs�t�tj�}xF|D]>}yWqtk rPtjd�|tj�krL|j|�YqXqWtt|�dt|�d�}x |D]}|j tj |�|d�qvWdS)Ng��?i�ip)�msg)rr/rK�AssertionErrorr[�sleeprx�range�maxr]r�)r"rKr'r#r#r$�test_pid_exists_2�s z TestSystemAPIs.test_pid_exists_2cCsJdd�tj�D�}tj�}|j|j�|j��|jt|�tt|��dS)NcSsg|] }|j�qSr#)r')r(r)r#r#r$r*�sz,TestSystemAPIs.test_pids..)r/r0rKr6�sortrTrc)r"ZplistZpidlistr#r#r$� test_pids�szTestSystemAPIs.test_pidscCs&tj}tt_ztj�Wd|t_XdS)N)�sys�stdoutrr/rX)r"r�r#r#r$� test_test�s zTestSystemAPIs.test_testc Cs�tj�}|j|ttjdd��|j|d�tjjd�rft d��}|j �}WdQRXd|krftjd��tjdd�}|j|d�|j||�dS) NT)�percpurMz /proc/cpuinfozphysical idz#cpuinfo doesn't include physical idF)�logical) r/� cpu_countr6rTr?rDr-�path�exists�open�readr ZSkipTest)r"r��fdZcpuinfo_dataZphysicalr#r#r$�test_cpu_counts zTestSystemAPIs.test_cpu_countcCsfx`d D]X}tjd|d��}|jtj��WdQRXtjd|d��}|jtjdd��WdQRXqWdS) NrMrz$psutil._psplatform.cpu_count_logical)�return_valuez%psutil._psplatform.cpu_count_physicalF)r�rY)rYrN)rr5rCr/r�)r"�valrGr#r#r$�test_cpu_count_nones z"TestSystemAPIs.test_cpu_count_nonecCs^d}tj�}t|�x,|D]$}|j|t�|j|d�||7}qW|j|t|��t|�dS)Nrg)r/r?�sumrkrlrDr6�str)r"rt�times�cp_timer#r#r$�test_cpu_timess zTestSystemAPIs.test_cpu_timescCsDttj��}tjd�ttj��}||}|dks@|jd|�dS)Ng��?g��?z difference %s)r�r/r?r[r�rx)r"�t1�t2� differencer#r#r$�test_cpu_times_time_increases:s z,TestSystemAPIs.test_cpu_times_time_increasescCs�xftjdd�D]V}d}t|�x,|D]$}|j|t�|j|d�||7}q$W|j|t|��t|�qW|jttjdd�d�ttjdd��dS)NT)r�rgF) r/r?r�rkrlrDr6r�rT)r"r�rtr�r#r#r$�test_per_cpu_timesCs z!TestSystemAPIs.test_per_cpu_timescCs�tjdd�}tj�d}xtj�|krPqWtjdd�}x.rM)�delta)r/r?�_maker�ru�assertAlmostEqualrv)r"�baseZper_cpuZ summed_valuesZfieldr#r#r$�test_cpu_times_comparisontsz(TestSystemAPIs.test_cpu_times_comparisoncCs�y<|j|t�|j|d�|j|d�|j|dtj��Wn@tk r|}z$td|tj |�tj |�f��WYdd}~XnXdS)NggY@z %s last=%s new=%sg�) rkrlrDZassertIsNot�assertLessEqualr/r�r��pprintZpformat)r"rsZlast_retZnew_ret�errr#r#r$�_test_cpu_percent~sz TestSystemAPIs._test_cpu_percentc Csbtjdd�}x.td�D]"}tjdd�}|j|||�|}qW|jt��tjdd�WdQRXdS)Ng��MbP?)�interval�drMrY)r/�cpu_percentr�r�r9rB)r"�lastr)�newr#r#r$�test_cpu_percent�szTestSystemAPIs.test_cpu_percentcCs�tjddd�}|jt|�tj��x>td�D]2}tjddd�}x|D]}|j|||�qDW|}q,W|jt��tjddd�WdQRXdS)Ng��MbP?T)r�r�r�rMrY) r/r�r6rTr�r�r�r9rB)r"r�r)r�rsr#r#r$�test_per_cpu_percent�s z#TestSystemAPIs.test_per_cpu_percentcCs`tjdd�}xNtd�D]B}tjdd�}x|D]}|j|||�q,W|jt|�||�|}qWdS)Ng��MbP?)r�r�)r/�cpu_times_percentr�r�r�)r"r�r)r�rsr#r#r$�test_cpu_times_percent�s z%TestSystemAPIs.test_cpu_times_percentcCs�tjddd�}|jt|�tj��x^td�D]R}tjddd�}x:|D]2}x|D]}|j|||�qNW|jt|�||�qDW|}q,WdS)Ng��MbP?T)r�r�r�)r/r�r6rTr�r�r�r�)r"r�r)r��cpursr#r#r$�test_per_cpu_times_percent�s z)TestSystemAPIs.test_per_cpu_times_percentc Csrtjdd�dd�tjdd�D�}tjd|d��8x0tjdd�D] }x|D]}|j|dd�qJWq@WWdQRXdS)NT)r�cSs*g|]"}|jdd�tt|j��D��qS)cSsg|]}d�qS)rr#)r(r)r#r#r$r*�szQTestSystemAPIs.test_per_cpu_times_percent_negative...)r�r�rTru)r(r)r#r#r$r*�szFTestSystemAPIs.test_per_cpu_times_percent_negative..zpsutil.cpu_times)r�)r/r�r?rr5r�)r"Z zero_timesr�rsr#r#r$�#test_per_cpu_times_percent_negative�s z2TestSystemAPIs.test_per_cpu_times_percent_negativec Cs�tjtj��}|j|jd �ttd�rptjtj��}d}|j|j|j�|j |j |j |d�|j |j|j|d�tj �}|jt��}tj|�WdQRX|j|jjtj�dS)Nrtr}r~rs� disk_usage�i)r�)rtr}r~rsiiP)r/r�r-�getcwdr6rur^�shutilrtr�r~r}�tempfileZmktempr9�OSErrorZ exception�errno�ENOENT)r"ZusageZshutil_usageZ toleranceZfname�excr#r#r$�test_disk_usage�s zTestSystemAPIs.test_disk_usagec Cs(tr$|jt��tjt�WdQRXdS)N)rr9�UnicodeEncodeErrorr/r�r)r"r#r#r$�test_disk_usage_unicode�sz&TestSystemAPIs.test_disk_usage_unicodecCstjd�dS)N�.)r/r�)r"r#r#r$�test_disk_usage_bytes�sz$TestSystemAPIs.test_disk_usage_bytescCs�tjdd�}|j||d�xj|D]b}|j|jt�|j|jt�|j|jt�|j|jt�t rld|jkrlq t srn|jts�tr q q Wtjdd�}|j||d�x�tjdd�D]�}t �s(yt j|j�WnPtk �r}z2tr�tr�|jtjkr�w�|jtjtjfk�r�WYdd}~XnXt�s(t�r(n|j|jt�|j|jt�q�Wdd�}|t�}dd �tjdd�D�}|j||�tj|�dS) NF)�all)r�ZcdromTcSs2tjj|�}xtjj|�s(tjj|�}qW|j�S)N)r-r��abspath�ismount�dirname�lower)r�r#r#r$�find_mount_pointsz=TestSystemAPIs.test_disk_partitions..find_mount_pointcSsg|]}|jj��qSr#)� mountpointr�)r(r)r#r#r$r*sz7TestSystemAPIs.test_disk_partitions..)r/Zdisk_partitionsrdrkZdevicer�r�ZfstypeZoptsrr r rr-�statr�rr�ZEIOZEPERMZEACCES�__file__r,r�)r"�lsZdiskr�r�ZmountZmountsr#r#r$�test_disk_partitions�sH z#TestSystemAPIs.test_disk_partitionscsl�fdd�}tjdd�}||�tjdd�}�j|g�x.|D]&}�j|��j|t�|||�q>WdS)Ncs��j|d|j��j|d|j��j|d|j��j|d|j��j|d|j��j|d|j��j|d|j��j|d|j�dS) NrrMrRrO�r��) r6Z bytes_sentZ bytes_recvZpackets_sentZpackets_recvZerrinZerroutZdropinZdropout)�nt)r"r#r$�check_ntuplesz9TestSystemAPIs.test_net_io_counters..check_ntupleF)�pernicT)r/�net_io_counters�assertNotEqualrdrkr�)r"r��ret�keyr#)r"r$�test_net_io_counterss z#TestSystemAPIs.test_net_io_counterscCsDtjdid��,}|jtjdd��|jtjdd�i�WdQRXdS)Nz"psutil._psplatform.net_io_counters)r�F)r�T)rr5rCr/r�r6)r"rGr#r#r$�test_net_io_counters_no_nics8s z+TestSystemAPIs.test_net_io_counters_no_nicsc Cshtj�}tj�}ttjtjtjg�}�x�|j�D�]�\}}|j |t �|jtt|��t|��x�|D�]�}|j |j t�|j |jt �|j |jt td�f�|j |jt td�f�|j|j |�tjdkr�|j |j tj�||j�r�|j tjk�r*tj|j �}tj|��|j|jdf�WdQRXnj|j tjk�r�tj|jdtjtjdtj�d}|\} } }}} tj| | |�}tj|��|j| �WdQRXxB|j|j|j|jfD]*}|dk �r�|j tjk�r�t ||j ��q�W|j�r�|j!|j�qf|jrf|j!|j�qfWq0Wt"�st#�st$�r8t%td��rd|jtjtj�n,t&�rP|jtjtj'�nt(�rd|jtjd�dS)NrOr�r�AF_LINKrM)rOr�rY))r/�net_if_addrs�net_if_statsrc�socketZAF_INETZAF_INET6r��itemsrkr�r6rTZfamilyrw�addressZnetmask�typeZ broadcastr,r��version_infor�IntEnum�isup� contextlib�closingZbindZgetaddrinfoZSOCK_STREAMZ AI_PASSIVEZptprrCrrr r^rZ AF_PACKETr)r"�nicsZ nic_statsZfamiliesZnicZaddrs�addr�sr@ZafZsocktype�protoZ canonnameZsaZipr#r#r$�test_net_if_addrsAsV z TestSystemAPIs.test_net_if_addrsc Csptrdtjddddfg}ndg}tjd|d��8}tj�dd}trT|j|jd�n|j|jd �WdQRXdS)N�em1z06:3d:29rM�06-3d-29zpsutil._psplatform.net_if_addrs)r�rz06:3d:29:00:00:00z06-3d-29-00-00-00rY)r�rYr�NNN)r r/r�rr5r�r6r�)r"r�rGr�r#r#r$� test_net_if_addrs_mac_null_bytes}s z/TestSystemAPIs.test_net_if_addrs_mac_null_byteszunreliable on TRAVISc Cs�tj�}tjtjtjf}xh|j�D]\\}}|j|t�|\}}}}|j|t�|j ||�|j ||�|j |d�|j |d�q"WdS)Nr)r/r�ZNIC_DUPLEX_FULLZNIC_DUPLEX_HALFZNIC_DUPLEX_UNKNOWNr�rkr��boolr,rD) r"r�Zall_duplexesrzZstatsr�ZduplexZspeedZmtur#r#r$�test_net_if_stats�sz TestSystemAPIs.test_net_if_statsz/proc/diskstatsz3/proc/diskstats not available on this linux versionNzunreliable on APPVEYORcs��fdd�}tjdd�}||�tjdd�}�jt|�tt|��xV|D]N}|||�trJ|dj�rJx|dj�r�|dd �}qlW�j||j��qJWdS) Ncs��j|d|j��j|d|j��j|d|j��j|d|j�tpNtsƈj|d|j��j|d|j�t r��j|d|j ��j|d|j��j|d |j�nt rƈj|d|j�x|jD]}q�WdS) NrrMrRrOr�r�r�r��)r6Z read_countZwrite_countZ read_bytesZwrite_bytesrrZ read_timeZ write_timerZread_merged_countZwrite_merged_countZ busy_timerru)r�rz)r"r#r$r��sz:TestSystemAPIs.test_disk_io_counters..check_ntupleF)�perdiskTrMrYrYrY) r/�disk_io_countersr6rTrcr�isdigitr4rA)r"r�r�r�r#)r"r$�test_disk_io_counters�s z$TestSystemAPIs.test_disk_io_counterscCsDtjdid��,}|jtjdd��|jtjdd�i�WdQRXdS)Nz#psutil._psplatform.disk_io_counters)r�F)r�T)rr5rCr/r�r6)r"rGr#r#r$�test_disk_io_counters_no_disks�s z-TestSystemAPIs.test_disk_io_counters_no_disksz unreliable on APPVEYOR or TRAVIScCs�tj�}|j|g�x�|D]�}|j|jt�|j|jttd�f�|jdk rb|j|jttd�f�|j|jt j j |j�ts�t r�|j|j�qtj|j�qWdS)N)r/�usersr�rkrzr�Zterminalr��host�datetimeZ fromtimestampZstartedrrrCr'r1)r"r��userr#r#r$� test_users�s zTestSystemAPIs.test_userscCsZtj�}|j|jd�x>|jD]4}t||�}|j|d�tr|dkr|j|d�qWdS)N�ctx_switches� interrupts�soft_interrupts�syscallsr)rrrr)rr)r/Z cpu_statsr6rurvrDrrm)r"Zinfosrzr{r#r#r$�test_cpu_stats�s zTestSystemAPIs.test_cpu_statsznot suportedcsT�fdd�}tjdd�}tr&|r&dS|tjdd�g�trP�jt|�tj��dS)Ncshxb|D]Z}�j|jd��j|j|j�x6|jD],}t||�}�j|ttt f��j |d�q0WqWdS)N�current�minr�r)rrr�)r6rur�rr�rvrkrwrrlrD)r�r�rzr{)r"r#r$�check_ls�s z.TestSystemAPIs.test_cpu_freq..check_lsT)r�F)r/Zcpu_freqrrr6rTr�)r"r r�r#)r"r$� test_cpu_freq�s zTestSystemAPIs.test_cpu_freqc Cs8ddddddddd g }x"|D]}|jtt|�t|d �qWtjdk�r|jd�dtjj �krj|jd�n�d tjj �kr�|j tjtjtj gjd�d�|jd�|jd�|jd�|jd�nBdtjj �ks�dtjj �kr�|jd �ndtjj �k�r|jd�n |jd�x$|D]}|jtt|�d|d ��qWdS)Nr rrrrrrrr )r��posix�linuxZbsdTrM�sunos�solaris�darwinF)rkrvr/r�r-rz�remover��platformr�r6rrr�countrF)r"�namesrzr#r#r$�test_os_constantss. z TestSystemAPIs.test_os_constantsz not supportedcCs�tj�}x�|j�D]x\}}|j|t�xb|D]Z}|j|jt�|jdk rV|j|jd�|jdk rn|j|jd�|j dk r,|j|j d�q,WqWdS)Nr) r/�sensors_temperaturesr�rkr��labelrrD�high�critical)r"�tempsrz�entries�entryr#r#r$�test_sensors_temperatures)s z(TestSystemAPIs.test_sensors_temperaturesc Csfddgi}tjd|d��D}tjdd �dd }|j|jd�|j|jd�|j|jd �WdQRXdS)NZcoretempr�I@�N@��Q@z'psutil._psplatform.sensors_temperatures)r�T)Z fahrenheitrg�^@g�a@g�c@)rrrr)rr5r/rr6rrr)r"�drGrr#r#r$�#test_sensors_temperatures_fahreneit7s z2TestSystemAPIs.test_sensors_temperatures_fahreneitz no batterycCsptj�}|j|jd�|j|jd�|jtjtjfkrF|j|jd�n|jtjkr^|j|j �|j |j t�dS)Nrr�)r/Zsensors_batteryrDrsr�ZsecsleftZPOWER_TIME_UNKNOWNZPOWER_TIME_UNLIMITEDrdZ power_pluggedrkr�)r"r�r#r#r$�test_sensors_batteryCs z#TestSystemAPIs.test_sensors_batterycCsjtj�}x\|j�D]P\}}|j|t�x:|D]2}|j|jt�|j|jttf�|j |jd�q,WqWdS)Nr) r/Zsensors_fansr�rkr�rrrwrrD)r"Zfansrzrrr#r#r$�test_sensors_fansQs z TestSystemAPIs.test_sensors_fans)C�__name__� __module__�__qualname__�__doc__r%r&r<rIrirjrnr ZskipIfr rrr|r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rr�rr-r�r�r r/r�r�r�r�rrrr rrrr!rrr"rr#r#r#r#r$r!;sh7 ! > <# % r!�__main__)2r'r�r�r�r-r�r�rar�r�r�r[r/rrrrrrrr r rZpsutil._compatrZpsutil.testsr rrrrrrrrrrrrrrrrrrr ZTestCaser!r$r�r#r#r#r$�sh'