a
��X]' � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl m Z m
Z
d dlmZ d dlZe�d�Ze�d�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.9/site-packages/systemd/test/test_journal.py�__init__ s zMockSender.__init__Nc K s� d| g}|d ur,t |d|�}|�d| � || u rJ| u rJd u rln n tjdd�d d d� \}}}|d ur�|�d| � |d ur�|�d �|�� |d ur�|�d
| � |�d d � |�� D �� | j�|� d S )
NzMESSAGE=�hex� MESSAGE_ID=� )�limitr � z
CODE_FILE=zCODE_LINE={:d}z
CODE_FUNC=c s s | ]\}}t ||�V qd S r r )�.0�key�valr r r � <genexpr>) � z"MockSender.send.<locals>.<genexpr>)�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 sL z
d V W n< t tfyF } z |j| kr0t�� � W Y d }~n
d }~0 0 d S r )�OSError�IOError�errno�pytest�skip)�code�er r r � skip_oserror, s
r. c c s* z
d V W n t y$ t�� Y n0 d S r )�
ValueErrorr* r+ r r r r �skip_valueerror5 s
r0 c C s| t jj} | tj�t jks J �| tj�t jks0J �| tjd �t jksHJ �| tjd �t jks`J �| tjd �t jksxJ �| tj�t jks�J �| tjd �t j ks�J �| tj
d �t j ks�J �| tj
�t j ks�J �| tj
d �t j ks�J �| tj d �t j k�sJ �| tj �t j k�sJ �| tj d �t j
k�s2J �| tj�t j
k�sHJ �| tj�t j
k�s^J �| tjd �t jk�sxJ �d S )N� )r �JournalHandlerZ map_priority�loggingZNOTSETZ LOG_DEBUG�DEBUGZLOG_INFO�INFOZ LOG_WARNINGZWARNZLOG_ERRZERRORZLOG_CRITZFATALZCRITICALZ LOG_ALERT)�pr r r �test_priorities< s" r7 c C sF ddi} t �t�� tjf i | �� W d � n1 s80 Y d S )Nz X r )r* �raisesr/ r r2 ��kwr r r �"test_journalhandler_init_exceptionQ s r; c C s" ddd�} t jtjfi | �� d S )Nr � )�X�X3)r r2 r3 r5 r9 r r r �test_journalhandler_initV s
r? c C s� t �dt jdddd d �} t� }dd|jd�}tjt jfi |��}|�| � t|j �dks\J �d|j d v snJ �d
|j d v s�J �d S ) N� test-logger�testpathr1 �testr r< )r= r> �sender_functionzX=3r zX3=4)
r3 � LogRecordr5 r r# r r2 �emit�lenr )�record�senderr: �handlerr r r �test_journalhandler_infoZ s
rJ c C sl t �dt jdddd d �} t� }tjt j|jd�}|�| � t|j �dksLJ �t
dd� |j d D ��shJ �d S ) Nr@ rA r1 rB )rC c s s | ]}|� d � V qdS )r N)�
startswith)r �mr r r r k r z4test_journalhandler_no_message_id.<locals>.<genexpr>r ) r3 rD r5 r r r2 r# rE rF r �all�rG rH rI r r r �!test_journalhandler_no_message_ide s
rO c C sj t �dt jdddd d �} t� }tjt j|jtd�}|�| � t |j
�dksNJ �dtj |j
d v sfJ �d S �Nr@ rA r1 rB �rC r r r ) r3 rD r5 r r r2 r# �TEST_MIDrE rF r r rN r r r �)test_journalhandler_message_id_on_handlerm s �
rS c C sl t �dt jdddd d �} t� }tjt j|jtjd�}|� | � t
|j �dksPJ �dtj |j d v shJ �d S rP ) r3 rD r5 r r r2 r# rR r rE rF r rN r r r �-test_journalhandler_message_id_on_handler_hexv s �
rT c C st t �dt jdddd d �} t| jd< t� }tjt j|jt d�}|�
| � t |j �dksXJ �dtj
|j d v spJ �d S � Nr@ rA r1 rB r rQ r r )r3 rD r5 � TEST_MID2�__dict__r r r2 r# rR rE rF r r rN r r r �)test_journalhandler_message_id_on_message s
�
rX c C sv t �dt jdddd d �} tj| jd< t� }tjt j|j t
d�}|� | � t |j
�dksZJ �dtj |j
d v srJ �d S rU )r3 rD r5 rV r rW r r r2 r# rR rE rF r rN r r r �-test_journalhandler_message_id_on_message_hex� s �
rY c C sR t �� } t �t j�}t �t j�}t �t j�}t �t jt jB t jB �}t �d�}d S )Nr )r �ReaderZ
LOCAL_ONLYZ RUNTIME_ONLY� SYSTEM_ONLY)�j1�j2�j3Zj4Zj5Zj6r r r �test_reader_init_flags� s r_ c C s� t �t�� t�tj� W d � n1 s,0 Y t� �" tj| jtjd�}W d � n1 sd0 Y t� �( tj| jtjtjB d�}W d � n1 s�0 Y tj| jtjtj B d�}d S )N)�path�flags)
r* r8 r/ r rZ ZOS_ROOTr0 �strpath� CURRENT_USERr[ )�tmpdirr\ r] r^ r r r �test_reader_os_root� s *�$
�$
�re c C s> t j| jd�}t jd| jd� t j| jd�}t j| jd� d S )N�r` r )r rZ rb )rd r\ r] r r r �test_reader_init_path� s rg c C s< t �t�� tjddd� W d � n1 s.0 Y d S )Nr ���rf �r* r8 r' r rZ r r r r � test_reader_init_path_invalid_fd� s rj c C s< t �t�� tjddd� W d � n1 s.0 Y d S )Nr rf ri r r r r �%test_reader_init_path_nondirectory_fd� s rk c C s� t �| jt j�}ttj�� tj|d�}W d � n1 s<0 Y t |�g ksVJ �t
� � tjtj |d�}W d � n1 s�0 Y t |�g ks�J �tjtj |d�}t |�g ks�J �d S �Nrf )
�os�openrb �O_RDONLYr. r) �ENOSYSr rZ �listr0 ZSYSTEMrc )rd �fdr\ r] r^ r r r �test_reader_init_path_fd� s *.rs c C s| t j| jd�}|� |jr J �W d � n1 s20 Y |jsFJ �t�t�� t|� W d � n1 sn0 Y d S rl )r rZ rb �closedr* r8 r' �next�rd �jr r r �test_reader_as_cm� s (
rx c C sP t j| jd�}|�* |�tj� |�tjj� W d � n1 sB0 Y d S rl )r rZ rb Zmessageid_matchr ZSD_MESSAGE_JOURNAL_STARTZSD_MESSAGE_JOURNAL_STOPr rv r r r �test_reader_messageid_match� s ry c C sT t j| jd�}|�. |�� |�t� |�tj� W d � n1 sF0 Y d S rl )r rZ rb Z this_bootrR r rv r r r �test_reader_this_boot� s
rz c C sT t j| jd�}|�. |�� |�t� |�tj� W d � n1 sF0 Y d S rl )r rZ rb Z this_machinerR r rv r r r �test_reader_this_machine� s
r{ c C s� t j| jd�}|�D ttj�� |�d�}W d � n1 s>0 Y W d � n1 s\0 Y t|t�stJ �|t� ks�J �d S )Nrf ZFOOBAR) r rZ rb r. r) rp Z query_unique�
isinstance�set�rd rw Zansr r r �test_reader_query_unique� s Fr c C s� t j| jd�}|�B ttj�� |�� }W d � n1 s<0 Y W d � n1 sZ0 Y t|t�srJ �|t� ks�J �d S rl ) r rZ rb r. r) rp Zenumerate_fieldsr| r} r~ r r r � test_reader_enumerate_fields� s Dr� c C st t j| jd�}|�B ttj�� |�� }W d � n1 s<0 Y W d � n1 sZ0 Y |du spJ �d S �Nrf F�r rZ rb r. r) rp Zhas_runtime_filesr~ r r r � test_reader_has_runtime_files� s
Dr� c C st t j| jd�}|�B ttj�� |�� }W d � n1 s<0 Y W d � n1 sZ0 Y |du spJ �d S r� r� r~ r r r � test_reader_has_persistent_files� s
Dr� c C sP ddd� i}t j| j|d�}|�dd�}|dks4J �|�dd�}|dksLJ �d S ) NZxxxc S s dS �N�yyyr ��argr r r �<lambda> r z(test_reader_converters.<locals>.<lambda>�r` �
converters� abcr� Zzzz� ��)r rZ rb Z_convert_field�rd r� rw r r r r �test_reader_converters� s r� c C s` dd� dd� d�}t j| j|d�}|�ddddgdd gd
��}|d dd d gdd gd
�ks\J �d S )
Nc S s dS r� r r� r r r r�
r z+test_reader_convert_entry.<locals>.<lambda>c S s dS )N�YYYr r� r r r r� r )�x1�x2r� r� r� s defs ��)r� Zy1r� Zy2r� r� )r rZ rb Z_convert_entryr� r r r �test_reader_convert_entry s ���r� c C sF t j| jd�}t�� }|�|� |�d� t�ddd�}|�|� d S )Nrf i90 i� � r< )r rZ rb �timeZ
seek_realtime�datetime)rd rw ZnowZlong_agor r r �test_seek_realtime s
r� c C sZ t tj�� t�d�} W d � n1 s*0 Y | �d�}|dv sJJ �td| d� d S )Nztest_journal.pyz message...
)� Nzprinted message...)�file)r. r) �ENOENTr �stream�write�print)r� �resr r r �test_journal_stream" s
(
r� )3Z
__future__r �
contextlibr� r) r3 rm r� Zuuid� tracebackr Zsystemdr r Zsystemd.journalr r* ZUUIDrR rV r �contextmanagerr. r0 r7 r; r? rJ rO rS rT rX rY r_ re rg rj rk rs rx ry rz r{ r r� r� r� r� r� r� r� r r r r �<module> sZ