File: //usr/lib/python3/dist-packages/twisted/logger/test/__pycache__/test_flatten.cpython-310.pyc
o
�b�% � @ s� d Z ddlZddlmZ ddlmZmZmZ zddlm Z W n e
y) dZ Y nw ddl m Z ddl
mZmZmZmZ dd lmZ dd
lmZ G d d � d e j�ZdS )
z+
Test cases for L{twisted.logger._format}.
� N)�count)�Any�Callable�Optional)�tzset)�unittest� )� KeyFlattener�
aFormatter� extractField� flattenEvent)� formatEvent)�LogEventc @ s� e Zd ZdZd"dd�Zd"dd�Zd"dd �Zd"d
d �Z d#d ee de fd
d�Z
d"dd�Z d"dd�Z d"dd�Z
dd� fdee ge f ddfdd�Zd"dd�Zd"d d �Zd"d d �Zd"d d!�ZdS )$�FlatFormattingTestsz9
Tests for flattened event formatting functions.
�returnNc sr t � � G dd� d�}td� fdd�|� dddd �}t|� t|�}|d
= |d = t�t�|��}| �t|�d � d
S )a)
L{flattenEvent} will "flatten" an event so that, if scrubbed of all but
serializable objects, it will preserve all necessary data to be
formatted once serialized. When presented with an event thusly
flattened, L{formatEvent} will produce the same output.
c @ s e Zd ZdZdS )z;FlatFormattingTests.test_formatFlatEvent.<locals>.Ephemeral�valueN)�__name__�
__module__� __qualname__� attribute� r r �B/usr/lib/python3/dist-packages/twisted/logger/test/test_flatten.py� Ephemeral% s r z�callable: {callme()} attribute: {object.attribute} numrepr: {number!r} numstr: {number!s} strrepr: {string!r} unistr: {unistr!s}c s t � �S �N)�nextr ��counterr r �<lambda>1 s z:FlatFormattingTests.test_formatFlatEvent.<locals>.<lambda>� �hello� ö)�
log_format�callme�object�number�string�unistrr" r# uM callable: 0 attribute: value numrepr: 7 numstr: 7 strrepr: 'hello' unistr: öN)r �dictr �json�loads�dumps� assertEqualr
)�selfr �event1�event2�event3r r r �test_formatFlatEvent s&
��z(FlatFormattingTests.test_formatFlatEventc C s< t ddd�}t|� t�t�|��}| �t|��d�� dS )zH
If the format string is invalid, an error is produced.
zstrrepr: {string!X}r )r! r% zUnable to format eventN)r' r r( r) r* �
assertTruer
�
startswith)r, r- r. r r r � test_formatFlatEventBadFormatJ s �z1FlatFormattingTests.test_formatFlatEventBadFormatc C sD G dd� d�}|� }t d|d�}t|� |�� | �t|�d� dS )z�
L{formatEvent} will prefer the stored C{str()} or C{repr()} value for
an object, in case the other version.
c @ s, e Zd ZdZdZd dd�Zdefdd�ZdS )
zPFlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistablez.
Unpersitable object.
Fr Nc S s
d| _ dS )z0
Self destruct.
TN��
destructed�r, r r r � selfDestructe s
z]FlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistable.selfDestructc S s | j rdS dS )Nzpost-serialization garbagezun-persistabler4 r6 r r r �__repr__k s zYFlatFormattingTests.test_formatFlatEventWithMutatedFields.<locals>.Unpersistable.__repr__�r N)r r r �__doc__r5 r7 �strr8 r r r r �
Unpersistable^ s
r< z unpersistable: {unpersistable})r! �
unpersistablez unpersistable: un-persistableN)r' r r7 r+ r
)r, r< �upr- r r r �%test_formatFlatEventWithMutatedFieldsX s z9FlatFormattingTests.test_formatFlatEventWithMutatedFieldsc C s dt dt fdd�}z
| �|d�d� W n ty � w | �|d�d� | �|d �d
� | �|d �d � | �|d
�d� | �|d�d� | �|d�d� | �|d�d� | �|d�d� | �|d�d� t� }t�d�\\}}}}|duswJ �| �|�|||�d� | �|�|||�d � dS ) zh
Test that L{KeyFlattener.flatKey} returns the expected keys for format
fields.
�formatr c S s> t �| �D ]\}}}}|d usJ �t� �|||� S J d��)NFz*Unable to derive key from format: {format})r
�parser �flatKey)r@ � literalText� fieldName�
formatSpec�
conversionr r r �
keyFromFormat s � z=FlatFormattingTests.test_keyFlattening.<locals>.keyFromFormatz{}z!:z{foo}zfoo!:z{foo!s}zfoo!s:z{foo!r}zfoo!r:z{foo:%s}zfoo!:%sz{foo:!}zfoo!:!z{foo::}zfoo!::z
{foo!s:%s}zfoo!s:%sz {foo!s:!}zfoo!s:!z {foo!s::}zfoo!s::z{x}Nzx!:zx!:/2)r; r+ �
ValueErrorr r
rA rB )r, rG �
sameFlattenerrC rD rE rF r r r �test_keyFlatteningy s< �� ��z&FlatFormattingTests.test_keyFlattening�eventc sJ |du rt � � G � fdd�d�}td|� d�}t|� | �t|�d� |S )z�
The same format field used twice in one event is rendered twice.
@param event: An event to flatten. If L{None}, create a new event.
@return: C{event} or the event created.
Nc s" e Zd ZdZdef� fdd� ZdS )zJFlatFormattingTests._test_formatFlatEvent_fieldNamesSame.<locals>.CountStrz&
Hack
r c s t t� ��S r )r; r r6 r r r �__str__� s zRFlatFormattingTests._test_formatFlatEvent_fieldNamesSame.<locals>.CountStr.__str__N)r r r r: r; rL r r r r �CountStr� s rM z{x} {x}�r! �xz0 1)r r' r r+ r
)r, rK rM r r r �$_test_formatFlatEvent_fieldNamesSame� s �z8FlatFormattingTests._test_formatFlatEvent_fieldNamesSamec C s | � � dS )zR
The same format field used twice in one event is rendered twice.
N�rP r6 r r r �"test_formatFlatEventFieldNamesSame� s z6FlatFormattingTests.test_formatFlatEventFieldNamesSamec C s | � � }| � |� dS )zb
The same event flattened twice gives the same (already rendered)
result.
NrQ )r, rK r r r �'test_formatFlatEventFieldNamesSameAgain� s z;FlatFormattingTests.test_formatFlatEventFieldNamesSameAgainc C s, t ddd�}t|� t|�}| �|d� dS )zs
L{formatEvent} will handle a flattened event with tailing text after
a replacement field.
ztest {x} trailingr rN ztest value trailingN)r' r r
r+ )r, rK �resultr r r � test_formatEventFlatTrailingText� s �z4FlatFormattingTests.test_formatEventFlatTrailingTextc C s | S r r )rO r r r r � s z FlatFormattingTests.<lambda>� flattenFirstc s� G dd� d�� G � fdd�d�}t d|� d�}||��dtdtf�fd d
� }| �|d �d � | �|d
�d� | �|d�d� dS )z�
L{extractField} will extract a field used in the format string.
@param flattenFirst: callable to flatten an event
c @ s e Zd Zdefdd�ZdS )z=FlatFormattingTests.test_extractField.<locals>.ObjectWithReprr c S s dS )N�reprr r6 r r r r8 � s zFFlatFormattingTests.test_extractField.<locals>.ObjectWithRepr.__repr__N)r r r r; r8 r r r r �ObjectWithRepr� s rX c s$ e Zd Zd� fdd� Zddd�ZdS )z8FlatFormattingTests.test_extractField.<locals>.Somethingr Nc s d| _ � � | _d S )Nr )r$ r# r6 �rX r r �__init__� s zAFlatFormattingTests.test_extractField.<locals>.Something.__init__c S s t d��)Nz
Just in case.)�NotImplementedErrorr6 r r r � __getstate__ s zEFlatFormattingTests.test_extractField.<locals>.Something.__getstate__r9 )r r r rZ r\ r rY r r � Something� s r] z%{something.number} {something.object})r! � something�fieldr c s
t | � �S r )r )r_ )� flattenedr r �extract
s
z6FlatFormattingTests.test_extractField.<locals>.extractzsomething.numberr zsomething.number!s�7zsomething.object!srW N)r' r; r r+ )r, rV r] rK ra r )rX r` r �test_extractField� s �z%FlatFormattingTests.test_extractFieldc C s dt dt fdd�}| �|� dS )zi
L{extractField} behaves identically if the event is explicitly
flattened first.
rK r c S s t | � | S r )r )rK r r r r` s zDFlatFormattingTests.test_extractFieldFlattenFirst.<locals>.flattenedN)r rc )r, r` r r r � test_extractFieldFlattenFirst s z1FlatFormattingTests.test_extractFieldFlattenFirstc C s( ddd�}t |� | �|ddd�� dS )zT
L{flattenEvent} will do nothing to an event with no format string.
�b� )�a�cN�r r+ �r, �
inputEventr r r � test_flattenEventWithoutFormat s
z2FlatFormattingTests.test_flattenEventWithoutFormatc C s, dddd�}t |� | �|dddd�� dS )zz
L{flattenEvent} will do nothing to an event with a format string that
contains no format fields.
re rf zsimple message�rg rh r! Nri rj r r r � test_flattenEventWithInertFormat( � ��z4FlatFormattingTests.test_flattenEventWithInertFormatc C s, dddd�}t |� | �|dddd�� dS )zb
L{flattenEvent} will do nothing to an event with log_format set to
None.
re rf Nrm ri rj r r r � test_flattenEventWithNoneFormat8 ro z3FlatFormattingTests.test_flattenEventWithNoneFormatr9 r )r r r r: r0 r3 r? rJ r r rP rR rS rU r rc rd rl rn rp r r r r r s2
.
!>��
�
� �
�
#
r )r: r( � itertoolsr �typingr r r �timer � ImportError�
twisted.trialr �_flattenr r
r r �_formatr
� _interfacesr �TestCaser r r r r �<module> s �