Cs[fI @ s9 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 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 e rQe j d Z e j d k rHe d Z n d Z n d Z d d Z d d Z e j e d Gd d d e j Z e j e d Gd d d e j Z e j e d Gd d d e j Z e j e d Gd d! d! e j Z e d" k r5e e! d S)#z$Tests specific to all BSD platforms. N)BSD)FREEBSD)NETBSD)OPENBSD)get_test_subprocess) HAS_BATTERY)MEMORY_TOLERANCE)
reap_children)retry_before_failing)run_test_module_by_name)sh)unittest)which SC_PAGE_SIZEmuseFc C s t d | } t r6 | | j d d d } n) t sB t r_ | | j d d d } y t | SWn t k
r | SYn Xd S)zmExpects a sysctl command with an argument and parse the result
returning only the value of interest.
zsysctl z: N= )r r findr r int
ValueError)cmdlineresult r /test_bsd.pysysctl, s
r c C sV t d } x3 | j d D] } | j | r Pq Wt d t | j d S)z+Thin wrapper around 'muse' cmdline utility.r
zline not foundr )r split
startswithr r )Zfieldoutliner r r r ; s zBSD onlyc @ s e Z d Z d Z e d d Z e d d Z e j e d d d Z
d d
Z e j e d d d
d Z
e j e d d d d Z d d Z d S)BSDSpecificTestCasez)Generic tests common to all BSD variants.c C s t j | _ d S)N)r pid)clsr r r
setUpClassO s z BSDSpecificTestCase.setUpClassc C s t d S)N)r )r# r r r
tearDownClassS s z!BSDSpecificTestCase.tearDownClassz -o lstart doesn't work on NETBSDc C sr t d | j } | j d d j } t j | j j } t j d t j | } | j
| | d S)Nzps -o lstart -p %sZSTARTED z%a %b %e %H:%M:%S %Y) r r" replacestrippsutilProcessZ create_timetimestrftime localtime assertEqual)selfoutputZstart_psZ start_psutilr r r test_process_create_timeW s z,BSDSpecificTestCase.test_process_create_timec C s d d } x t j d d D] } t j | j } | | j \ } } } } | j | j | | j | j | t | j | d k r | j d | j | f t | j
| d k r | j d | j
| f q Wd S) Nc S s t d | j } | j d } | j d | j d } | j d d \ } } } } | d k ru d } t | d } t | d } t | d } | | | | f S)Nz
df -k "%s"r r Znoner& i )r r( r popr )pathr linesr devtotalusedfreer r r dfc s
" z*BSDSpecificTestCase.test_disks.<locals>.dfallF
i zpsutil=%s, df=%si ( i i ( i ) r) Zdisk_partitions
disk_usageZ
mountpointr. Zdevicer7 absr9 Zfailr8 )r/ r: partusager6 r7 r8 r9 r r r
test_disks` s
z BSDSpecificTestCase.test_disksr zsysctl cmd not availablec C s, t d } | j t j d d | d S)Nzhw.ncpuZlogicalT)r r. r) cpu_count)r/ systr r r test_cpu_count_logical{ s z*BSDSpecificTestCase.test_cpu_count_logicalc C s) t d } | j | t j j d S)Nz
hw.physmem)r r. r) virtual_memoryr7 )r/ numr r r test_virtual_memory_total s z-BSDSpecificTestCase.test_virtual_memory_totalc C s x t j j D] \ } } y t d | } Wn t k
rG Yq X| j | j d | k d | d | k r | j | j t t j
d | d q Wd S)Nz ifconfig %sRUNNINGmsgmtuz mtu (\d+)r ) r) Z net_if_statsitemsr RuntimeErrorr. ZisuprJ r refindall)r/ nameZstatsr r r r test_net_if_stats s
z%BSDSpecificTestCase.test_net_if_statsN)__name__
__module__ __qualname____doc__ classmethodr$ r% r
skipIfr r1 rA r rD rG rP r r r r r! K s %%r! z FREEBSD onlyc @ s e Z d Z e d d Z e d d Z e d d Z e d d Z d d
Z
d d Z d
d Z e d d Z
e d d Z e d d Z e d d Z e d d Z e d d Z e d d Z e d d Z e j e d d d! Z e j e d e d" d# Z e j e d e d$ d% Z e j e d e d& d' Z e j e d e d( d) Z e j e d e d* d+ Z e j e d e d, d- Z d. d/ Z d0 d1 Z d2 d3 Z! d4 d5 Z" d6 d7 Z# d8 d9 Z$ d: d; Z% d< d= Z& e j e' d> d? d@ Z( e j e' d> dA dB Z) e j e' dC dD dE Z* dF S)GFreeBSDSpecificTestCasec C s t j | _ d S)N)r r" )r# r r r r$ s z"FreeBSDSpecificTestCase.setUpClassc C s t d S)N)r )r# r r r r% s z%FreeBSDSpecificTestCase.tearDownClassc C sq t d j d } t j d | } | s> t d | d d | d d D \ } } } | | | f S) Nz swapinfo -kr z\s+zCan't parse swapinfo: %sc s s | ] } t | d Vq d S)i N)r ).0pr r r <genexpr> s z9FreeBSDSpecificTestCase.parse_swapinfo.<locals>.<genexpr>r2 )r
splitlinesrM r r )r0 partsr7 r8 r9 r r r parse_swapinfo s &z&FreeBSDSpecificTestCase.parse_swapinfoc C s t d | j } t j | j j d d } | j d d d } x | r | j } | j } | d d \ } } } } }
| j } | j d | | f | j | j t |
| j
| j j d sM | j | d | j qM Wd S)
Nzprocstat -v %sZgroupedFr r z%s-%s[r< )
r r" r) r* Z memory_mapsr r3 r. addrr Zrssr4 r ) r/ r mapsr5 r fields_startstopZpermsresmapr r r test_proc_memory_maps s z-FreeBSDSpecificTestCase.test_proc_memory_mapsc C sP t d | j } | j t j | j j | j d d j d d S)Nzprocstat -b %sr r r[ )r r" r. r) r* Zexer )r/ r r r r
test_proc_exe s z%FreeBSDSpecificTestCase.test_proc_exec C sh t d | j } | j d j t j | j j d j | j d d j d d d S)Nzprocstat -c %s r r r )r r" r. joinr) r* r r )r/ r r r r test_proc_cmdline s $z)FreeBSDSpecificTestCase.test_proc_cmdlinec C s t d | j } | j d d j d d \ } } } } } } t j | j } | j } | j }
| j | j t | | j | j
t | | j | j t | | j |
j t | | j |
j
t | | j |
j t | d S)Nzprocstat -s %sr r r ) r r" r r) r* uidsgidsr. realr Z effectiveZsaved) r/ r ZeuidZruidZsuidZegidZrgidZsgidrY ro rp r r r test_proc_uids_gids s 5 z+FreeBSDSpecificTestCase.test_proc_uids_gidsc C s g } t d | j } t j | j } x | j d D] } | j j } d | k r t | j d } | j j } | j
| | | j d q; d | k r; t | j d } | j j } | j
| | | j d q; Wt
| d k rt d d S)
Nzprocstat -r %sr z voluntary contextr z involuntary contextr z)couldn't find lines match in procstat outr[ r[ )r r" r) r* r lowerr( r Znum_ctx_switchesZ voluntaryr. appendZ involuntarylenrL )r/ testedr rY r pstat_value psutil_valuer r r test_proc_ctx_switches s" z.FreeBSDSpecificTestCase.test_proc_ctx_switchesc C s: g } t d | j } t j | j } x | j d D] } | j j } d | k r t d | j d
j d d } | j j } | j
| | | j d q; d | k r; t d | j d j d d
} | j j } | j
| | | j d q; Wt
| d k r6t d d S)Nzprocstat -r %sr z user timez0.r .z system timer z)couldn't find lines match in procstat outr[ r[ r[ r[ )r r" r) r* r rs r( floatZ cpu_timesuserr. rt systemru rL )r/ rv r rY r rw rx r r r test_proc_cpu_times s" ' 'z+FreeBSDSpecificTestCase.test_proc_cpu_timesc C s3 t d t } | j t j j | d t d S)Nzvm.stats.vm.v_active_countdelta)r PAGESIZEassertAlmostEqualr) rE activer )r/ rC r r r test_vmem_active s z(FreeBSDSpecificTestCase.test_vmem_activec C s3 t d t } | j t j j | d t d S)Nz vm.stats.vm.v_inactive_countr )r r r r) rE inactiver )r/ rC r r r test_vmem_inactive s z*FreeBSDSpecificTestCase.test_vmem_inactivec C s3 t d t } | j t j j | d t d S)Nzvm.stats.vm.v_wire_countr )r r r r) rE wiredr )r/ rC r r r test_vmem_wired s z'FreeBSDSpecificTestCase.test_vmem_wiredc C s3 t d t } | j t j j | d t d S)Nzvm.stats.vm.v_cache_countr )r r r r) rE cachedr )r/ rC r r r test_vmem_cached s z(FreeBSDSpecificTestCase.test_vmem_cachedc C s3 t d t } | j t j j | d t d S)Nzvm.stats.vm.v_free_countr )r r r r) rE r9 r )r/ rC r r r test_vmem_free s z&FreeBSDSpecificTestCase.test_vmem_freec C s/ t d } | j t j j | d t d S)Nz vfs.bufspacer )r r r) rE buffersr )r/ rC r r r test_vmem_buffers s z)FreeBSDSpecificTestCase.test_vmem_bufferszmuse not installedc C s) t d } | j t j j | d S)NZTotal)r r. r) rE r7 )r/ rF r r r test_muse_vmem_total% s z,FreeBSDSpecificTestCase.test_muse_vmem_totalc C s/ t d } | j t j j | d t d S)NZActiver )r r r) rE r r )r/ rF r r r test_muse_vmem_active* s z-FreeBSDSpecificTestCase.test_muse_vmem_activec C s/ t d } | j t j j | d t d S)NZInactiver )r r r) rE r r )r/ rF r r r test_muse_vmem_inactive1 s z/FreeBSDSpecificTestCase.test_muse_vmem_inactivec C s/ t d } | j t j j | d t d S)NZWiredr )r r r) rE r r )r/ rF r r r test_muse_vmem_wired8 s z,FreeBSDSpecificTestCase.test_muse_vmem_wiredc C s/ t d } | j t j j | d t d S)NZCacher )r r r) rE r r )r/ rF r r r test_muse_vmem_cached? s z-FreeBSDSpecificTestCase.test_muse_vmem_cachedc C s/ t d } | j t j j | d t d S)NZFreer )r r r) rE r9 r )r/ rF r r r test_muse_vmem_freeF s z+FreeBSDSpecificTestCase.test_muse_vmem_freec C s/ t d } | j t j j | d t d S)NZBufferr )r r r) rE r r )r/ rF r r r test_muse_vmem_buffersM s z.FreeBSDSpecificTestCase.test_muse_vmem_buffersc C s) | j t j j t d d d d S)Nzvm.stats.sys.v_swtchr i )r r) cpu_stats ctx_switchesr )r/ r r r test_cpu_stats_ctx_switchesT s z3FreeBSDSpecificTestCase.test_cpu_stats_ctx_switchesc C s) | j t j j t d d d d S)Nzvm.stats.sys.v_intrr i )r r) r
interruptsr )r/ r r r test_cpu_stats_interruptsX s z1FreeBSDSpecificTestCase.test_cpu_stats_interruptsc C s) | j t j j t d d d d S)Nzvm.stats.sys.v_softr i )r r) r Zsoft_interruptsr )r/ r r r test_cpu_stats_soft_interrupts\ s z6FreeBSDSpecificTestCase.test_cpu_stats_soft_interruptsc C s) | j t j j t d d d d S)Nzvm.stats.sys.v_syscallr i )r r) r Zsyscallsr )r/ r r r test_cpu_stats_syscalls` s z/FreeBSDSpecificTestCase.test_cpu_stats_syscallsc C s8 | j \ } } } | j t j j | d t d S)Nr )r^ r r) swap_memoryr9 r )r/ r7 r8 r9 r r r test_swapmem_freej s z)FreeBSDSpecificTestCase.test_swapmem_freec C s8 | j \ } } } | j t j j | d t d S)Nr )r^ r r) r r8 r )r/ r7 r8 r9 r r r test_swapmem_usedo s z)FreeBSDSpecificTestCase.test_swapmem_usedc C s8 | j \ } } } | j t j j | d t d S)Nr )r^ r r) r r7 r )r/ r7 r8 r9 r r r test_swapmem_totalt s z*FreeBSDSpecificTestCase.test_swapmem_totalc C sh t d } | | j d d d } | d | j d } t | } | j | t j d S)Nzsysctl kern.boottimez sec = ,)r r r r. r) boot_time)r/ sZbtimer r r test_boot_time{ s
z&FreeBSDSpecificTestCase.test_boot_timez
no batteryc C s d d } t d } t d d | j d D } t j } t | d j d d } | d
} | j | j | | d k r | j | j t j
n | j | | j | d S) Nc S s8 t | d \ } } t | d \ } } d | | f S)N< z%d:%02d)divmod)Zsecsmr hr r r
secs2hours s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.secs2hoursz
acpiconf -i 0c S s6 g | ], } | j d d | j d d f q S) r r r[ )r )rX xr r r
<listcomp> s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.<listcomp>r zRemaining capacity:%r& zRemaining time:unknown) r dictr r) sensors_batteryr r' r. percentsecsleftZPOWER_TIME_UNLIMITED)r/ r r rc Zmetricsr Zremaining_timer r r test_sensors_battery s
z,FreeBSDSpecificTestCase.test_sensors_batteryc C s | j t j j t d | j t j j t d d k t j j } | d k rx | j t d d n | j | t d d d S)Nzhw.acpi.battery.lifezhw.acpi.acliner r zhw.acpi.battery.timer r[ )r. r) r r r Z
power_pluggedr )r/ r r r r #test_sensors_battery_against_sysctl s
z;FreeBSDSpecificTestCase.test_sensors_battery_against_sysctlz has batteryc C sL | j t # t d t d t d Wd QRX| j t j d S)Nzhw.acpi.battery.lifezhw.acpi.battery.timezhw.acpi.acline)Z assertRaisesrL r Z assertIsNoner) r )r/ r r r test_sensors_battery_no_battery s
z7FreeBSDSpecificTestCase.test_sensors_battery_no_batteryN)+rQ rR rS rU r$ r% staticmethodr^ r
ri rj rm rr ry r~ r r r r r r r
rV MUSE_AVAILABLEr r r r r r r r r r r r r r r r r r r r r r r rW sN
rW z OPENBSD onlyc @ s e Z d Z d d Z d S)OpenBSDSpecificTestCasec C sM t d } t j j | d } t j j t j } | j | | d S)Nz
kern.boottimez%a %b %d %H:%M:%S %Y)r datetimestrptimeZ
fromtimestampr) r r. )r/ r Zsys_btZ psutil_btr r r r s z&OpenBSDSpecificTestCase.test_boot_timeN)rQ rR rS r r r r r r s r z NETBSD onlyc @ s e Z d Z e 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 S)NetBSDSpecificTestCasec C se t d d ? } x5 | D]- } | j | r t | j d d Sq WWd QRXt d | d S)Nz
/proc/meminforbr i z
can't find %s)openr r r r )Zlook_forfr r r r
parse_meminfo s
#z$NetBSDSpecificTestCase.parse_meminfoc C s& | j t j j | j d d S)Nz MemTotal:)r. r) rE r7 r )r/ r r r test_vmem_total s z&NetBSDSpecificTestCase.test_vmem_totalc C s, | j t j j | j d d t d S)NzMemFree:r )r r) rE r9 r r )r/ r r r r s z%NetBSDSpecificTestCase.test_vmem_freec C s, | j t j j | j d d t d S)NzBuffers:r )r r) rE r r r )r/ r r r r s z(NetBSDSpecificTestCase.test_vmem_buffersc C s, | j t j j | j d d t d S)Nz
MemShared:r )r r) rE Zsharedr r )r/ r r r test_vmem_shared s z'NetBSDSpecificTestCase.test_vmem_sharedc C s, | j t j j | j d d t d S)Nz
SwapTotal:r )r r) r r7 r r )r/ r r r r s z)NetBSDSpecificTestCase.test_swapmem_totalc C s, | j t j j | j d d t d S)Nz SwapFree:r )r r) r r9 r r )r/ r r r r s z(NetBSDSpecificTestCase.test_swapmem_freec C s- t j } | j | j | j | j d S)N)r) r r. r8 r7 r9 )r/ Zsmemr r r r s z(NetBSDSpecificTestCase.test_swapmem_usedc C s t d d J } x@ | D], } | j d r t | j d } Pq Wt d Wd QRX| j t j j | d d d S)Nz
/proc/statr s intrr zcouldn't find liner i ) r r r r r r r) r r )r/ r r r r r r r s
z0NetBSDSpecificTestCase.test_cpu_stats_interruptsc C s t d d J } x@ | D], } | j d r t | j d } Pq Wt d Wd QRX| j t j j | d d d S)Nz
/proc/statr s ctxtr zcouldn't find liner i ) r r r r r r r) r r )r/ r r r r r r r s
z2NetBSDSpecificTestCase.test_cpu_stats_ctx_switchesN)rQ rR rS r r r r r r r r r r r r r r r r s r __main__)"rT r osrM r+ r) r r r r Z psutil.testsr r r r r
r r r
r sysconfr getuidr r r rV ZTestCaser! rW r r rQ __file__r r r r <module>
sJ K B