3
��X]' � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl m Z m
Z
d dlmZ d dlZejd�Zejd�ZG dd� d�Zejd d
� �Zejd d � �Zd
d� Zdd� Zdd� Zdd� Zdd� Z dd� Z dd� Z dd � 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/d0� Z)d1d2� Z*d3d4� Z+d5d6� Z,d7d8� Z-d9d:� Z.d;d<� Z/d=d>� Z0d?d@� Z1dAdB� Z2dS )C� )�print_functionN)�journal�id128)�
_make_lineZ 8441372f8dca4ca98694a6091fd8519fZ 8441370000000000000000001fd85000c @ s e Zd Zdd� Zddd�ZdS )�
MockSenderc C s
g | _ d S )N)�buf)�self� r �$/usr/lib64/python3.6/test_journal.py�__init__ s zMockSender.__init__Nc K s� d| g}|d k r,t |d|�}|jd| � || koH| koHd kn rltjdd�d d d� \}}}|d k r�|jd| � |d k r�|jd j|�� |d k r�|jd
| � |jd d � |j� D �� | jj|� d S )
NzMESSAGE=�hexz MESSAGE_ID=� )�limitr � z
CODE_FILE=zCODE_LINE={:d}z
CODE_FUNC=c s s | ]\}}t ||�V qd S )N)r )�.0�key�valr r r
� ) s z"MockSender.send..)�getattr�append�
_traceback�
extract_stack�format�extend�itemsr ) r ZMESSAGE�
MESSAGE_IDZ CODE_FILEZ CODE_LINEZ CODE_FUNC�kwargs�args�idr r r
�send s
" zMockSender.send)NNNN)�__name__�
__module__� __qualname__r r r r r r
r s r c c sJ y
d V W n: t tfk
rD } z|j| kr2tj� � W Y d d }~X nX d S )N)�OSError�IOError�errno�pytest�skip)�code�er r r
� skip_oserror, s
r* c c s, y
d V W n t k
r& tj� Y nX d S )N)�
ValueErrorr& r' r r r r
�skip_valueerror5 s
r, c C s| t jj} | tj�t jks t�| tj�t jks0t�| tjd �t jksHt�| tjd �t jks`t�| tj d �t jksxt�| tj �t jks�t�| tj d �t j
ks�t�| tj d �t j
ks�t�| tj �t j
ks�t�| tj d �t j ks�t�| tj
d �t j k�st�| tj
�t j k�st�| tj
d �t jk�s2t�| tj�t jk�sHt�| tj�t jk�s^t�| tjd �t jk�sxt�d S )N� )r �JournalHandlerZ map_priority�loggingZNOTSETZ LOG_DEBUG�AssertionError�DEBUGZLOG_INFO�INFOZ LOG_WARNINGZWARNZLOG_ERRZERRORZLOG_CRITZFATALZCRITICALZ LOG_ALERT)�pr r r
�test_priorities< s" r4 c
C s. ddi} t jt�� tjf | � W d Q R X d S )Nz X r )r& �raisesr+ r r. )�kwr r r
�"test_journalhandler_init_exceptionQ s r7 c C s ddd�} t jtjf| � d S )Nr � )�X�X3)r r. r/ r2 )r6 r r r
�test_journalhandler_initV s
r; c C s� t jdt jdddd d �} t� }dd|jd�}tjt jf|�}|j| � t|j �dksXt
�d|j d ksjt
�d
|j d ks|t
�d S ) Nz test-logger�testpathr- �testr r8 )r9 r: �sender_functionzX=3r zX3=4) r/ � LogRecordr2 r r r r. �emit�lenr r0 )�record�senderr6 �handlerr r r
�test_journalhandler_infoZ s
rE c C sl t jdt jdddd d �} t� }tjt j|jd�}|j| � t|j �dksLt
�t dd� |j d D ��sht
�d S ) Nz test-loggerr< r- r= )r> c s s | ]}|j d � V qdS )z MESSAGE_ID=N)�
startswith)r �mr r r
r k s z4test_journalhandler_no_message_id..r ) r/ r? r2 r r r. r r@ rA r r0 �all)rB rC rD r r r
�!test_journalhandler_no_message_ide s
rI c C sj t jdt jdddd d �} t� }tjt j|jtd�}|j| � t |j
�dksNt �dtj |j
d ksft �d S )Nz test-loggerr< r- r= )r> r z MESSAGE_ID=r )
r/ r? r2 r r r. r �TEST_MIDr@ rA r r0 r )rB rC rD r r r
�)test_journalhandler_message_id_on_handlerm s
rK c C sl t jdt jdddd d �} t� }tjt j|jtjd�}|j | � t
|j �dksPt �dtj |j d ksht �d S )Nz test-loggerr< r- r= )r> r z MESSAGE_ID=r )
r/ r? r2 r r r. r rJ r r@ rA r r0 )rB rC rD r r r
�-test_journalhandler_message_id_on_handler_hexv s
rL c C st t jdt jdddd d �} t| jd< t� }tjt j|jt d�}|j
| � t |j �dksXt
�dtj |j d kspt
�d S ) Nz test-loggerr< r- r= r )r> r z MESSAGE_ID=r )r/ r? r2 � TEST_MID2�__dict__r r r. r rJ r@ rA r r0 r )rB rC rD r r r
�)test_journalhandler_message_id_on_message s
rO c C sv t jdt jdddd d �} tj| jd< t� }tjt j|j t
d�}|j | � t |j
�dksZt�dtj |j
d ksrt�d S ) Nz test-loggerr< r- r= r )r> r z MESSAGE_ID=r )r/ r? r2 rM r rN r r r. r rJ r@ rA r r0 )rB rC rD r r r
�-test_journalhandler_message_id_on_message_hex� s
rP c C sR t j� } t jt j�}t jt j�}t jt j�}t jt jt jB t jB �}t jd�}d S )Nr )r �ReaderZ
LOCAL_ONLYZ RUNTIME_ONLY� SYSTEM_ONLY)�j1�j2�j3Zj4Zj5Zj6r r r
�test_reader_init_flags� s rV c C s� t jt�� tjtj� W d Q R X t� � tj| jtjd�}W d Q R X t� � tj| jtjtjB d�}W d Q R X tj| jtjtj B d�}d S )N)�path�flags)
r& r5 r+ r rQ ZOS_ROOTr, �strpath� CURRENT_USERrR )�tmpdirrS rT rU r r r
�test_reader_os_root� s r\ c C s> t j| jd�}t jd| jd� t j| jd�}t j| jd� d S )N)rW r )r rQ rY )r[ rS rT r r r
�test_reader_init_path� s r] c C s( t jt�� tjddd� W d Q R X d S )Nr r- )rW �����)r& r5 r# r rQ r r r r
� test_reader_init_path_invalid_fd� s r_ c C s( t jt�� tjddd� W d Q R X d S )Nr )rW )r& r5 r# r rQ r r r r
�%test_reader_init_path_nondirectory_fd� s r` c C s� t j| jt j�}ttj�� tj|d�}W d Q R X t |�g ksBt
�t � � tjtj |d�}W d Q R X t |�g kstt
�tjtj
|d�}t |�g ks�t
�d S )N)rW )�os�openrY �O_RDONLYr* r% �ENOSYSr rQ �listr0 r, ZSYSTEMrZ )r[ �fdrS rT rU r r r
�test_reader_init_path_fd� s rg c C sV t j| jd�}|� |j s t�W d Q R X |js4t�tjt�� t|� W d Q R X d S )N)rW ) r rQ rY �closedr0 r& r5 r# �next)r[ �jr r r
�test_reader_as_cm� s
rk c C s< t j| jd�}|� |jtj� |jtjj� W d Q R X d S )N)rW )r rQ rY Zmessageid_matchr ZSD_MESSAGE_JOURNAL_STARTZSD_MESSAGE_JOURNAL_STOPr )r[ rj r r r
�test_reader_messageid_match� s rl c C s@ t j| jd�}|�$ |j� |jt� |jtj� W d Q R X d S )N)rW )r rQ rY Z this_bootrJ r )r[ rj r r r
�test_reader_this_boot� s
rm c C s@ t j| jd�}|�$ |j� |jt� |jtj� W d Q R X d S )N)rW )r rQ rY Z this_machinerJ r )r[ rj r r r
�test_reader_this_machine� s
rn c C s^ t j| jd�}|�& ttj�� |jd�}W d Q R X W d Q R X t|t�sLt �|t� ksZt �d S )N)rW ZFOOBAR)
r rQ rY r* r% rd Z query_unique�
isinstance�setr0 )r[ rj �ansr r r
�test_reader_query_unique� s rr c C s\ t j| jd�}|�$ ttj�� |j� }W d Q R X W d Q R X t|t�sJt �|t� ksXt �d S )N)rW )
r rQ rY r* r% rd Zenumerate_fieldsro rp r0 )r[ rj rq r r r
� test_reader_enumerate_fields� s rs c C sL t j| jd�}|�$ ttj�� |j� }W d Q R X W d Q R X |dksHt�d S )N)rW F)r rQ rY r* r% rd �has_runtime_filesr0 )r[ rj rq r r r
� test_reader_has_runtime_files� s
ru c C sL t j| jd�}|�$ ttj�� |j� }W d Q R X W d Q R X |dksHt�d S )N)rW F)r rQ rY r* r% rd rt r0 )r[ rj rq r r r
� test_reader_has_persistent_files� s
rv c C sP ddd� i}t j| j|d�}|jdd�}|dks4t�|jdd�}|dksLt�d S ) NZxxxc S s dS )N�yyyr )�argr r r
� s z(test_reader_converters..)rW �
converterss abcrw Zzzzs ��)r rQ rY Z_convert_fieldr0 )r[ rz rj r r r r
�test_reader_converters� s r{ c C s` dd� dd� d�}t j| j|d�}|jddddgdd gd
��}|d dd d gdd gd
�ks\t�d S )
Nc S s dS )Nrw r )rx r r r
ry
s z+test_reader_convert_entry..c S s dS )N�YYYr )rx r r r
ry s )�x1�x2)rW rz s abcs ��s defs ��)r} Zy1r~ Zy2rw r| )r rQ rY Z_convert_entryr0 )r[ rz rj r r r r
�test_reader_convert_entry s r c C sF t j| jd�}tj� }|j|� |jd� tjddd�}|j|� d S )N)rW i90 i� � r8 )r rQ rY �timeZ
seek_realtime�datetime)r[ rj ZnowZlong_agor r r
�test_seek_realtime s
r� c C sF t tj�� tjd�} W d Q R X | jd�}|dks6t�td| d� d S )Nztest_journal.pyz message...
� zprinted message...)�file)r� N)r* r% �ENOENTr �stream�writer0 �print)r� �resr r r
�test_journal_stream" s
r� )3Z
__future__r �
contextlibr� r% r/ ra r� Zuuid� tracebackr Zsystemdr r Zsystemd.journalr r& ZUUIDrJ rM r �contextmanagerr* r, r4 r7 r; rE rI rK rL rO rP rV r\ r] r_ r` rg rk rl rm rn rr rs ru rv r{ r r� r� r r r r
� sV