← Back
Editing: test_bsd.cpython-37.pyc
B �=�_�P � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z dd l m Z dd l mZ dd l mZ dd l mZ dd l mZ ddl mZ ddl mZ er�ddlmZ e� Ze�� dkr�ed�r�dndZndZdZdd� Zdd� Ze�e d�G dd� de ��Z e�e d�G dd � d e ��Z e�e d�G d d � d e ��Z e�e d �G d d!� d!e ��Z e�e d"�G d#d$� d$e ��Z e!d%k�r�dd&l"m#Z# e#e$� dS )'z$Tests specific to all BSD platforms.� N)�BSD)�FREEBSD)�NETBSD)�OPENBSD)�spawn_testproc)� HAS_BATTERY)�PsutilTestCase)�retry_on_failure)�sh)�TOLERANCE_SYS_MEM)� terminate)�unittest)�which)� getpagesize�museTFc C sh t d| �}tr(||�d�d d� }n ts0trF||�d�d d� }yt|�S tk rb |S X dS )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)�cmdline�result� r �H/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_bsd.py�sysctl- s r c C s@ t d�}x&|�d�D ]}|�| �rP qW td��t|�� d �S )z+Thin wrapper around 'muse' cmdline utility.r � zline not foundr )r �split� startswithr r )�field�out�liner r r r < s zBSD onlyc @ s� e Zd ZdZedd� �Zedd� �Ze�e d�dd� �Z d d � Z e�e d � d �d d� �Z e�e d � d �dd� �Zdd� ZdS )� BSDTestCasez)Generic tests common to all BSD variants.c C s t � j| _d S )N)r �pid)�clsr r r � setUpClassP s zBSDTestCase.setUpClassc C s t | j� d S )N)r r# )r$ r r r � tearDownClassT s zBSDTestCase.tearDownClassz -o lstart doesn't work on NETBSDc C sP t d| j �}|�dd��� }t�| j��� }t�dt� |��}| � ||� d S )Nzps -o lstart -p %sZSTARTED� z%a %b %e %H:%M:%S %Y) r r# �replace�strip�psutil�ProcessZ create_time�time�strftime� localtime� assertEqual)�self�outputZstart_psZ start_psutilr r r �test_process_create_timeX s z$BSDTestCase.test_process_create_timec C s� dd� }x�t jdd�D ]�}t �|j�}||j�\}}}}| �|j|� | �|j|� t|j| �dkrz| � d|j|f � t|j | �dkr| � d|j |f � qW d S )Nc S s� t d| ��� }|�d�}|�d� |�d�}|�� d d� \}}}}|dkrRd}t|�d }t|�d }t|�d }||||fS )Nz df -k "%s"r r � Znoner' i )r r) r �popr )�pathr �linesr! �dev�total�used�freer r r �dfd s z"BSDTestCase.test_disks.<locals>.dfF)�alli � zpsutil=%s, df=%s) r* Zdisk_partitions� disk_usage� mountpointr/ �devicer8 �absr: Zfailr9 )r0 r; �part�usager7 r8 r9 r: r r r � test_disksa s zBSDTestCase.test_disksr zsysctl cmd not availablec C s t d�}| �tjdd�|� d S )Nzhw.ncpuT)Zlogical)r r/ r* � cpu_count)r0 �systr r r �test_cpu_count_logical| s z"BSDTestCase.test_cpu_count_logicalc C s t d�}| �|t�� j� d S )Nz hw.physmem)r r/ r* �virtual_memoryr8 )r0 �numr r r �test_virtual_memory_total� s z%BSDTestCase.test_virtual_memory_totalc C s� xzt �� �� D ]j\}}ytd| �}W n tk r: Y qX | j|jd|k|d� d|kr| �|jtt � d|�d �� qW d S )Nz ifconfig %s�RUNNING)�msg�mtuz mtu (\d+)r ) r* Z net_if_stats�itemsr � RuntimeErrorr/ �isuprL r �re�findall)r0 �nameZstatsr r r r �test_net_if_stats� s z BSDTestCase.test_net_if_statsN)�__name__� __module__� __qualname__�__doc__� classmethodr% r& r �skipIfr r2 rC r rF rI rS r r r r r" L s r" z FREEBSD onlyc @ sf e Zd Zedd� �Zedd� �Ze� dd� �Zdd� Zd d � Z d d � Z e� d d� �Z e� dd� �Z dS )�FreeBSDPsutilTestCasec C s t � j| _d S )N)r r# )r$ r r r r% � s z FreeBSDPsutilTestCase.setUpClassc C s t | j� d S )N)r r# )r$ r r r r& � s z#FreeBSDPsutilTestCase.tearDownClassc C s� t d| j �}t�| j�jdd�}|�d�dd � }x||r�|�� }|�� }|d d� \}}}} } |�� } | �d||f | j� | �t | �| j � | j � d�s6| �|d | j � q6W d S ) Nzprocstat -v %sF)Zgroupedr r � z%s-%s�[� ) r r# r* r+ Z memory_mapsr r4 r/ �addrr Zrssr5 r ) r0 r �mapsr6 r! �fields�_�start�stopZperms�res�mapr r r �test_memory_maps� s z&FreeBSDPsutilTestCase.test_memory_mapsc C s< t d| j �}| �t�| j��� |�d�d �� d � d S )Nzprocstat -b %sr r ���)r r# r/ r* r+ Zexer )r0 r r r r �test_exe� s z FreeBSDPsutilTestCase.test_exec C sL t d| j �}| �d�t�| j��� �d�|�d�d �� dd � �� d S )Nzprocstat -c %s� r r r )r r# r/ �joinr* r+ r r )r0 r r r r � test_cmdline� s z"FreeBSDPsutilTestCase.test_cmdlinec C s� t d| j �}|�d�d �� dd� \}}}}}}t�| j�}|�� } |�� } | �| jt |�� | �| j t |�� | �| j t |�� | �| jt |�� | �| j t |�� | �| j t |�� d S )Nzprocstat -s %sr r r � ) r r# r r* r+ �uids�gidsr/ �realr � effective�saved) r0 r ZeuidZruidZsuidZegidZrgidZsgid�prm rn r r r �test_uids_gids� s & z$FreeBSDPsutilTestCase.test_uids_gidsc C s� g }t d| j �}t�| j�}x�|�d�D ]�}|�� �� }d|krtt|�� d �}|�� j }| � ||� |� d � q*d|kr*t|�� d �}|�� j }| � ||� |� d � q*W t |�dkr�td��d S )Nzprocstat -r %sr z voluntary contextrg z involuntary contextr z)couldn't find lines match in procstat out)r r# r* r+ r �lowerr) r Znum_ctx_switches� voluntaryr/ �append� involuntary�lenrN )r0 �testedr rr r! � pstat_value� psutil_valuer r r �test_ctx_switches� s" z'FreeBSDPsutilTestCase.test_ctx_switchesc C s� g }t d| j �}t�| j�}x�|�d�D ]�}|�� �� }d|kr�td|�� d �d�d �}|�� j }| � ||� |� d � q*d|kr*td|�� d �d�d �}|�� j }| � ||� |� d � q*W t |�dkr�td ��d S ) Nzprocstat -r %sr z user timez0.rg �.z system timer z)couldn't find lines match in procstat out)r r# r* r+ r rt r) �floatZ cpu_times�userr/ rv �systemrx rN )r0 ry r rr r! rz r{ r r r �test_cpu_times� s" z$FreeBSDPsutilTestCase.test_cpu_timesN) rT rU rV rX r% r& r rf rh rk rs r| r� r r r r rZ � s rZ c @ s� e Zd Zedd� �Zdd� Ze� dd� �Ze� dd� �Ze� d d � �Z e� d d � �Z e� d d� �Z e� dd� �Z e �e d�dd� �Ze �e d�e� dd� ��Ze �e d�e� dd� ��Ze �e d�e� dd� ��Ze �e d�e� dd� ��Ze �e d�e� d d � ��Ze �e d�e� d d � ��Zd d!� Zd"d#� Zd$d%� Ze� d&d'� �Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z e �e d0�d1d2� �Z e �e d0�d3d4� �Z!e �e d5�d6d7� �Z"d8d9� Z#d:S );�FreeBSDSystemTestCasec C sR t d��� d } t�d| �}|s,td| ��dd� |dd� D �\}}}|||fS ) Nz swapinfo -krg z\s+zCan't parse swapinfo: %sc s s | ]}t |�d V qdS )i N)r )�.0rr r r r � <genexpr>� s z7FreeBSDSystemTestCase.parse_swapinfo.<locals>.<genexpr>r r3 )r � splitlinesrP r r )r1 �partsr8 r9 r: r r r �parse_swapinfo� s z$FreeBSDSystemTestCase.parse_swapinfoc C s� d}yt t|��}W n tk r2 | �d� Y nX | �t�� j|� d}t|�}t |�� d �d�d �}t |�� d �d�d �}| �t�� j |� | �t�� j |� d S )Nzdev.cpu.0.freqz#frequencies not supported by kernelzdev.cpu.0.freq_levelsr �/rg ) r r rN �skipTestr/ r* Zcpu_freq�currentr �max�min)r0 �sensor� sysctl_resultZmax_freqZmin_freqr r r �!test_cpu_frequency_against_sysctl s z7FreeBSDSystemTestCase.test_cpu_frequency_against_sysctlc C s& t d�t }| jt�� j|td� d S )Nzvm.stats.vm.v_active_count)�delta)r �PAGESIZE�assertAlmostEqualr* rG �activer )r0 rE r r r �test_vmem_active s z&FreeBSDSystemTestCase.test_vmem_activec C s& t d�t }| jt�� j|td� d S )Nz vm.stats.vm.v_inactive_count)r� )r r� r� r* rG �inactiver )r0 rE r r r �test_vmem_inactive s z(FreeBSDSystemTestCase.test_vmem_inactivec C s& t d�t }| jt�� j|td� d S )Nzvm.stats.vm.v_wire_count)r� )r r� r� r* rG �wiredr )r0 rE r r r �test_vmem_wired$ s z%FreeBSDSystemTestCase.test_vmem_wiredc C s& t d�t }| jt�� j|td� d S )Nzvm.stats.vm.v_cache_count)r� )r r� r� r* rG �cachedr )r0 rE r r r �test_vmem_cached* s z&FreeBSDSystemTestCase.test_vmem_cachedc C s& t d�t }| jt�� j|td� d S )Nzvm.stats.vm.v_free_count)r� )r r� r� r* rG r: r )r0 rE r r r �test_vmem_free0 s z$FreeBSDSystemTestCase.test_vmem_freec C s" t d�}| jt�� j|td� d S )Nz vfs.bufspace)r� )r r� r* rG �buffersr )r0 rE r r r �test_vmem_buffers6 s z'FreeBSDSystemTestCase.test_vmem_bufferszmuse not installedc C s t d�}| �t�� j|� d S )NZTotal)r r/ r* rG r8 )r0 rH r r r �test_muse_vmem_total> s z*FreeBSDSystemTestCase.test_muse_vmem_totalc C s" t d�}| jt�� j|td� d S )NZActive)r� )r r� r* rG r� r )r0 rH r r r �test_muse_vmem_activeC s z+FreeBSDSystemTestCase.test_muse_vmem_activec C s" t d�}| jt�� j|td� d S )NZInactive)r� )r r� r* rG r� r )r0 rH r r r �test_muse_vmem_inactiveJ s z-FreeBSDSystemTestCase.test_muse_vmem_inactivec C s" t d�}| jt�� j|td� d S )NZWired)r� )r r� r* rG r� r )r0 rH r r r �test_muse_vmem_wiredQ s z*FreeBSDSystemTestCase.test_muse_vmem_wiredc C s" t d�}| jt�� j|td� d S )N�Cache)r� )r r� r* rG r� r )r0 rH r r r �test_muse_vmem_cachedX s z+FreeBSDSystemTestCase.test_muse_vmem_cachedc C s" t d�}| jt�� j|td� d S )NZFree)r� )r r� r* rG r: r )r0 rH r r r �test_muse_vmem_free_ s z)FreeBSDSystemTestCase.test_muse_vmem_freec C s" t d�}| jt�� j|td� d S )NZBuffer)r� )r r� r* rG r� r )r0 rH r r r �test_muse_vmem_buffersf s z,FreeBSDSystemTestCase.test_muse_vmem_buffersc C s | j t�� jtd�dd� d S )Nzvm.stats.sys.v_swtchi� )r� )r� r* � cpu_stats� ctx_switchesr )r0 r r r �test_cpu_stats_ctx_switchesm s z1FreeBSDSystemTestCase.test_cpu_stats_ctx_switchesc C s | j t�� jtd�dd� d S )Nzvm.stats.sys.v_intri� )r� )r� r* r� � interruptsr )r0 r r r �test_cpu_stats_interruptsq s z/FreeBSDSystemTestCase.test_cpu_stats_interruptsc C s | j t�� jtd�dd� d S )Nzvm.stats.sys.v_softi� )r� )r� r* r� �soft_interruptsr )r0 r r r � test_cpu_stats_soft_interruptsu s z4FreeBSDSystemTestCase.test_cpu_stats_soft_interruptsc C s | j t�� jtd�dd� d S )Nzvm.stats.sys.v_syscalli@ )r� )r� r* r� �syscallsr )r0 r r r �test_cpu_stats_syscallsy s z-FreeBSDSystemTestCase.test_cpu_stats_syscallsc C s( | � � \}}}| jt�� j|td� d S )N)r� )r� r� r* � swap_memoryr: r )r0 r8 r9 r: r r r �test_swapmem_free� s z'FreeBSDSystemTestCase.test_swapmem_freec C s( | � � \}}}| jt�� j|td� d S )N)r� )r� r� r* r� r9 r )r0 r8 r9 r: r r r �test_swapmem_used� s z'FreeBSDSystemTestCase.test_swapmem_usedc C s( | � � \}}}| jt�� j|td� d S )N)r� )r� r� r* r� r8 r )r0 r8 r9 r: r r r �test_swapmem_total� s z(FreeBSDSystemTestCase.test_swapmem_totalc C sL t d�}||�d�d d � }|d |�d�� }t|�}| �|t�� � d S )Nzsysctl kern.boottimez sec = � �,)r r r r/ r* � boot_time)r0 �sZbtimer r r �test_boot_time� s z$FreeBSDSystemTestCase.test_boot_timez no batteryc C s� dd� }t d�}tdd� |�d�D ��}t�� }t|d �dd ��}|d }| �|j|� |d krt| �|j tj � n| �||j �|� d S ) Nc S s( t | d�\}}t |d�\}}d||f S )N�< z%d:%02d)�divmod)Zsecs�mr� �hr r r � secs2hours� s z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.secs2hoursz acpiconf -i 0c S s( g | ] }|� d �d |� d �d f�qS )� r rg )r )r� �xr r r � <listcomp>� s z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.<listcomp>r zRemaining capacity:�%r' zRemaining time:�unknown) r �dictr r* �sensors_batteryr r( r/ �percent�secsleft�POWER_TIME_UNLIMITED)r0 r� r r` �metricsr� Zremaining_timer r r �test_sensors_battery� s z*FreeBSDSystemTestCase.test_sensors_batteryc C sl | � t�� jtd�� | � t�� jtd�dk� t�� j}|dk rT| � td�d� n| � |td�d � d S )Nzhw.acpi.battery.lifezhw.acpi.acliner r zhw.acpi.battery.timerg r� )r/ r* r� r� r � power_pluggedr� )r0 r� r r r �#test_sensors_battery_against_sysctl� s z9FreeBSDSystemTestCase.test_sensors_battery_against_sysctlz has batteryc C s@ | � t�� td� td� td� W d Q R X | �t�� � d S )Nzhw.acpi.battery.lifezhw.acpi.battery.timezhw.acpi.acline)Z assertRaisesrN r Z assertIsNoner* r� )r0 r r r � test_sensors_battery_no_battery� s z5FreeBSDSystemTestCase.test_sensors_battery_no_batteryc C s� t �d�}x�t|�D ]�}d| }y ttt|�d d� ��}W n tk rZ | �d� Y nX | jt � � d | j |dd� d| }ttt|�d d� ��}| � t � � d | j |� qW d S ) NTzdev.cpu.%s.temperaturerg z$temperatures not supported by kernelZcoretempr] )r� zdev.cpu.%s.coretemp.tjmax) r* rD �ranger r~ r rN r� r� Zsensors_temperaturesr� r/ �high)r0 Znum_cpusZcpur� r� r r r �(test_sensors_temperatures_against_sysctl� s z>FreeBSDSystemTestCase.test_sensors_temperatures_against_sysctlN)$rT rU rV � staticmethodr� r� r r� r� r� r� r� r� r rY �MUSE_AVAILABLEr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r r r r r� � sB r� z OPENBSD onlyc @ s e Zd Zdd� ZdS )�OpenBSDTestCasec C s6 t d�}tj�|d�}tj�t�� �}| �||� d S )Nz kern.boottimez%a %b %d %H:%M:%S %Y)r �datetime�strptime� fromtimestampr* r� r/ )r0 r� Zsys_btZ psutil_btr r r r� � s z OpenBSDTestCase.test_boot_timeN)rT rU rV r� r r r r r� � s r� z NETBSD onlyc @ s` e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dS )�NetBSDTestCasec C sR t dd��2}x*|D ]"}|�| �rt|�� d �d S qW W d Q R X td| ��d S )Nz /proc/meminfo�rtr i z can't find %s)�openr r r r )Zlook_for�fr! r r r � parse_meminfo� s "z NetBSDTestCase.parse_meminfoc C s | � t�� j| �d�� d S )Nz MemTotal:)r/ r* rG r8 r� )r0 r r r �test_vmem_total� s z NetBSDTestCase.test_vmem_totalc C s | j t�� j| �d�td� d S )NzMemFree:)r� )r� r* rG r: r� r )r0 r r r r� s z NetBSDTestCase.test_vmem_freec C s | j t�� j| �d�td� d S )NzBuffers:)r� )r� r* rG r� r� r )r0 r r r r� s z NetBSDTestCase.test_vmem_buffersc C s | j t�� j| �d�td� d S )Nz MemShared:)r� )r� r* rG �sharedr� r )r0 r r r �test_vmem_shared s z NetBSDTestCase.test_vmem_sharedc C s | j t�� j| �d�td� d S )Nz SwapTotal:)r� )r� r* r� r8 r� r )r0 r r r r� s z!NetBSDTestCase.test_swapmem_totalc C s | j t�� j| �d�td� d S )Nz SwapFree:)r� )r� r* r� r: r� r )r0 r r r r� s z NetBSDTestCase.test_swapmem_freec C s" t �� }| �|j|j|j � d S )N)r* r� r/ r9 r8 r: )r0 �smemr r r r� s z NetBSDTestCase.test_swapmem_usedc C sb t dd��8}x0|D ] }|�d�rt|�� d �}P qW td��W d Q R X | jt�� j|dd� d S )Nz /proc/stat�rbs intrr zcouldn't find linei� )r� ) r� r r r r r� r* r� r� )r0 r� r! r� r r r r� s z(NetBSDTestCase.test_cpu_stats_interruptsc C sb t dd��8}x0|D ] }|�d�rt|�� d �}P qW td��W d Q R X | jt�� j|dd� d S )Nz /proc/statr� s ctxtr zcouldn't find linei� )r� ) r� r r r r r� r* r� r� )r0 r� r! r� r r r r� * s z*NetBSDTestCase.test_cpu_stats_ctx_switchesN)rT rU rV r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� �__main__)� run_from_name)%rW r� �osrP r, r* r r r r Z psutil.testsr r r r r r r r r Zpsutil._psutil_posixr r� �getuidr� r r rY r" rZ r� r� r� rT Zpsutil.tests.runnerr� �__file__r r r r �<module> sP KY rB
Save File
Cancel