File: //proc/self/root/lib/python3/dist-packages/twisted/logger/test/__pycache__/test_json.cpython-38.pyc
U
��W[�H � @ s� d Z ddlmZmZ ddlmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZmZmZm Z m Z ddl m Z dd� Z!G dd� de
�Z"G dd� de
�Z#G dd� de
�Z$dS )z$
Tests for L{twisted.logger._json}.
� )�StringIO�BytesIO)� verifyObject�BrokenMethodImplementation)�unicode)�TestCase)�Failure� )� ILogObserver� LogPublisher)� formatEvent)�LogLevel)� extractField)�globalLogPublisher)� eventAsJSON�
eventFromJSON�jsonFileLogObserver�eventsFromJSONLogFile�log)�Loggerc C s" | � |t� | �|�d�d� |S )a�
Assert a few things about the result of L{eventAsJSON}, then return it.
@param testCase: The L{TestCase} with which to perform the assertions.
@type testCase: L{TestCase}
@param savedJSON: The result of L{eventAsJSON}.
@type savedJSON: L{unicode} (we hope)
@return: C{savedJSON}
@rtype: L{unicode}
@raise AssertionError: If any of the preconditions fail.
�
r )�assertIsInstancer � assertEqual�count)ZtestCaseZ savedJSON� r �?/usr/lib/python3/dist-packages/twisted/logger/test/test_json.py�savedJSONInvariants s r c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
SaveLoadTestsz2
Tests for loading and saving log events.
c C s t | t|��S )z�
Serialize some an events, assert some things about it, and return the
JSON.
@param event: An event.
@type event: L{dict}
@return: JSON.
)r r )�self�eventr r r �savedEventJSON8 s
z SaveLoadTests.savedEventJSONc C s | � t| �i ��i � dS )zX
Saving and loading an empty dictionary results in an empty dictionary.
N�r r r �r r r r �test_simpleSaveLoadE s z!SaveLoadTests.test_simpleSaveLoadc C s& | � t| �ddd���ddd�� dS )a
Saving and loading a dictionary with some simple values in it results
in those same simple values in the output; according to JSON's rules,
though, all dictionary keys must be L{unicode} and any non-L{unicode}
keys will be converted.
r �4)� �3��1r& Nr! r" r r r �
test_saveLoadL s �zSaveLoadTests.test_saveLoadc C s, | � t| �dt� d���dddid�� dS )zx
Saving and loading an object which cannot be represented in JSON will
result in a placeholder.
r r' �
unpersistableTN�r r r �objectr" r r r �test_saveUnPersistableY s �z$SaveLoadTests.test_saveUnPersistablec C s, | � t| �dt� d���dddid�� dS )zD
Non-ASCII keys and values can be saved and loaded.
u 䌡)u ሴr& r* TNr+ r" r r r �test_saveNonASCIId s
� �z SaveLoadTests.test_saveNonASCIIc C sZ dd� }d|t d��i}ttk r.|�ddi� | �t| �|��d|t d���d�i� dS ) zx
Any L{bytes} objects will be saved as if they are latin-1 so they can
be faithfully re-loaded.
c S s$ t tkrd�tt| ��S t | �S d S )N� )�bytes�str�join�map�chr��xr r r �asbytesu s z-SaveLoadTests.test_saveBytes.<locals>.asbytesZhello� s skippedZokay�charmapN)�ranger0 r1 �updater r r �decode)r r7 �
inputEventr r r �test_saveBytesp s �z SaveLoadTests.test_saveBytesc C s@ G dd� dt �}d|d�d�}t| �|��}| �t|�d� dS )a
Saving and loading an object which cannot be represented in JSON, but
has a string representation which I{can} be saved as JSON, will result
in the same string formatting; any extractable fields will retain their
data types.
c @ s e Zd Zdd� Zdd� ZdS )z@SaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprablec S s
|| _ d S �N��value)r rA r r r �__init__� s zISaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprable.__init__c S s dS )NZreprabler r" r r r �__repr__� s zISaveLoadTests.test_saveUnPersistableThenFormat.<locals>.Reprable.__repr__N)�__name__�
__module__� __qualname__rB rC r r r r �Reprable� s rG z{object} {object.value}� ��
log_formatr, z
reprable 7N)r, r r r r )r rG r= Z outputEventr r r � test_saveUnPersistableThenFormat� s �z.SaveLoadTests.test_saveUnPersistableThenFormatc C sb G dd� dt �}td|� d�}t| �|��}| �td|�d� | �ttd|� | �ttd|� dS ) zs
L{extractField} can extract fields from an object that's been saved and
loaded from JSON.
c @ s e Zd Zdd� ZdS )z8SaveLoadTests.test_extractingFieldsPostLoad.<locals>.Objc S s
d| _ d S )N�Y r@ r" r r r rB � s zASaveLoadTests.test_extractingFieldsPostLoad.<locals>.Obj.__init__N)rD rE rF rB r r r r �Obj� s rM z{object.value}rI z object.valuerL r, N)r, �dictr r r r � assertRaises�KeyError)r rM r= � loadedEventr r r � test_extractingFieldsPostLoad� s z+SaveLoadTests.test_extractingFieldsPostLoadc C s� g }t |jd�}z dd W n& tk
rB t� }|�d|� Y nX ddl}|�� d dk rd|�� | �t |�d� t
| � |d ��d }| � |t� | �
|�t�� | � |�� t� dS )zl
Round-tripping a failure through L{eventAsJSON} preserves its class and
structure.
��observerr% r za message about failureN� log_failure)r �append�ZeroDivisionErrorr �failure�sys�exc_infoZ exc_clearr �lenr r r Z
assertTrueZcheckZ getTracebackr1 )r �eventsr �frY Zloadedr r r � test_failureStructurePreserved� s z,SaveLoadTests.test_failureStructurePreservedc C s0 t tjd�}t| �|��}| �|d tj� dS )zo
It's important that the C{log_level} key remain a
L{constantly.NamedConstant} object.
�� log_levelr` N)rN r
�warnr r ZassertIs)r r= rQ r r r �test_saveLoadLevel� s z SaveLoadTests.test_saveLoadLevelc C s t d�}| �|tdd�� dS )z�
If a saved bit of JSON (let's say, from a future version of Twisted)
were to persist a different log_level, it will resolve as None.
zZ{"log_level": {"name": "other", "__class_uuid__": "02E59486-F24D-46AD-8224-3ACDF2A5732A"}}Nr_ )r r rN )r rQ r r r �test_saveLoadUnknownLevel� s �z'SaveLoadTests.test_saveLoadUnknownLevelN)rD rE rF �__doc__r r# r) r- r. r> rK rR r^ rb rc r r r r r 3 s
r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�FileLogObserverTestsz+
Tests for L{jsonFileLogObserver}.
c C sZ t � �J}t|�}ztt|� W n, tk
rJ } z| �|� W 5 d}~X Y nX W 5 Q R X dS )zh
A L{FileLogObserver} returned by L{jsonFileLogObserver} is an
L{ILogObserver}.
N)r r r r
r Zfail)r �
fileHandlerT �er r r �test_interface� s z#FileLogObserverTests.test_interfacec K sV |� dd�}t� �:}t|f|�}tdd�}||� | �|�� d�|�� W 5 Q R X dS )a�
Asserts that an observer created by L{jsonFileLogObserver} with the
given arguments writes events serialized as JSON text, using the given
record separator.
@param recordSeparator: A record separator.
@type recordSeparator: L{unicode}
@param kwargs: Keyword arguments to pass to L{jsonFileLogObserver}.
@type kwargs: L{dict}
�recordSeparator� r% r5 z{0}{{"x": 1}}
N)�getr r rN r �getvalue�format)r �kwargsri rf rT r r r r �assertObserverWritesJSON� s
�z-FileLogObserverTests.assertObserverWritesJSONc C s | � � dS )z�
A L{FileLogObserver} created by L{jsonFileLogObserver} writes events
serialzed as JSON text to a file when it observes events.
By default, the record separator is C{u"\x1e"}.
N�ro r" r r r �(test_observeWritesDefaultRecordSeparator s z=FileLogObserverTests.test_observeWritesDefaultRecordSeparatorc C s | j dd� dS )z�
A L{FileLogObserver} created by L{jsonFileLogObserver} writes events
serialzed as JSON text to a file when it observes events.
This test sets the record separator to C{u""}.
� �ri Nrp r" r r r �&test_observeWritesEmptyRecordSeparator s z;FileLogObserverTests.test_observeWritesEmptyRecordSeparatorc s� t � }t� }g }|�|j� |�t|�� t|d�}z dd W n |�d� Y nX t |�� �}tt |��}� fdd�}||� ||� dS )z�
A L{FileLogObserver} created by L{jsonFileLogObserver} writes failures
serialized as JSON text to a file when it observes events.
rS r% r zfailed as expectedc s` � � t| �d� | \}� �d|� |d }� �|t� |�� }� � |jjj� d�t
� d�� d S )Nr% rU �co) r r[ ZassertInr r ZgetTracebackObject�tb_frame�f_code� co_filename�rstrip�__file__)Z logEventsZ failureEventZ
failureObjectZtracebackObjectr" r r � checkEvents, s �z@FileLogObserverTests.test_failureFormatting.<locals>.checkEventsN)
r r � addObserverrV r r rX rl �listr )r �ioZ publisherZloggedZlogger�readerZ deserializedr{ r r" r �test_failureFormatting s
z+FileLogObserverTests.test_failureFormattingN) rD rE rF rd rh ro rq rt r� r r r r re � s
re c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd S )!�LogFileReaderTestsz-
Tests for L{eventsFromJSONLogFile}.
c s& g � _ � fdd�}|� _t�|� d S )Nc s& | d t jkr"d| kr"� j�| � d S )NZ
log_namespace�record)�jsonLog� namespace� errorEventsrV )r r" r r rT D s
��z*LogFileReaderTests.setUp.<locals>.observer)r� � logObserverr r| )r rT r r" r �setUpA s zLogFileReaderTests.setUpc C s t �| j� d S r? )r ZremoveObserverr� r" r r r �tearDownP s zLogFileReaderTests.tearDownc K sF t |f|�}| �t|�ddi� | �t|�ddi� | �tt|� dS )a
Test that L{eventsFromJSONLogFile} reads two pre-defined events from a
file: C{{u"x": 1}} and C{{u"y": 2}}.
@param fileHandle: The file to read from.
@param kwargs: Keyword arguments to pass to L{eventsFromJSONLogFile}.
r6 r% �yr N)r r �nextrO �
StopIteration)r rf rn r\ r r r � _readEventsT s z LogFileReaderTests._readEventsc C s4 t d��"}| �|� | �t| j�d� W 5 Q R X dS )z�
L{eventsFromJSONLogFile} reads events from a file and automatically
detects use of C{u"\x1e"} as the record separator.
� {"x": 1}
{"y": 2}
r N�r r� r r[ r� �r rf r r r �&test_readEventsAutoWithRecordSeparatord s �
z9LogFileReaderTests.test_readEventsAutoWithRecordSeparatorc C s4 t d��"}| �|� | �t| j�d� W 5 Q R X dS )z�
L{eventsFromJSONLogFile} reads events from a file and automatically
detects use of C{u""} as the record separator.
z{"x": 1}
{"y": 2}
r Nr� r� r r r �'test_readEventsAutoEmptyRecordSeparatorq s �
z:LogFileReaderTests.test_readEventsAutoEmptyRecordSeparatorc C s8 t d��&}| j|dd� | �t| j�d� W 5 Q R X dS )z{
L{eventsFromJSONLogFile} reads events from a file and is told to use
a specific record separator.
z{"x": 1}
{"y": 2}
�rs r Nr� r� r r r �&test_readEventsExplicitRecordSeparator~ s �z9LogFileReaderTests.test_readEventsExplicitRecordSeparatorc C s8 t d��&}| j|dd� | �t| j�d� W 5 Q R X dS )zM
L{eventsFromJSONLogFile} handles buffering a partial event.
r� r% �Z
bufferSizer Nr� r� r r r � test_readEventsPartialBuffer� s �z/LogFileReaderTests.test_readEventsPartialBufferc C s� t d��n}t|�}| �t|�ddi� | �tt|� | �t| j�d� | �| jd d d� | �| jd d d � W 5 Q R X d
S ) zF
If the JSON text for a record is truncated, skip it.
z {"x": 1 {"y": 2}
r� r r% r rJ z0Unable to read truncated JSON record: {record!r}r� s {"x": 1N�r r r r� rO r� r[ r� �r rf r\ r r r �test_readTruncated� s � �z%LogFileReaderTests.test_readTruncatedc C sT t d��B}t|�}| �t|�ddi� | �tt|� | �t| j�d� W 5 Q R X dS )zg
If the file being read from vends L{unicode}, strings decode from JSON
as-is.
u {"currency": "€"}
�currency� €r Nr� r� r r r �test_readUnicode� s
z#LogFileReaderTests.test_readUnicodec C sT t d��B}t|�}| �t|�ddi� | �tt|� | �t| j�d� W 5 Q R X dS )zh
If the file being read from vends L{bytes}, strings decode from JSON as
UTF-8.
s {"currency": "€"}
r� r� r N�r r r r� rO r� r[ r� r� r r r �test_readUTF8Bytes� s
z%LogFileReaderTests.test_readUTF8Bytesc C sX t d��F}t|dd�}| �t|�ddi� | �tt|� | �t| j�d� W 5 Q R X dS )z�
If the JSON text for a record is truncated in the middle of a two-byte
Unicode codepoint, we don't want to see a codec exception and the
stream is read properly when the additional data arrives.
s {"x": "€"}
� r� r6 r� r Nr� r� r r r �test_readTruncatedUTF8Bytes� s
z.LogFileReaderTests.test_readTruncatedUTF8Bytesc C s� t d��n}t|�}| �t|�ddi� | �tt|� | �t| j�d� | �| jd d d� | �| jd d d � W 5 Q R X d
S ) zh
If the JSON text for a record contains invalid UTF-8 text, ignore that
record.
s {"x": "�"}
{"y": 2}
r� r r% r rJ z2Unable to decode UTF-8 for JSON record: {record!r}r� s {"x": "�"}
Nr� r� r r r �test_readInvalidUTF8Bytes� s � � �z,LogFileReaderTests.test_readInvalidUTF8Bytesc C s� t d��n}t|�}| �t|�ddi� | �tt|� | �t| j�d� | �| jd d d� | �| jd d d � W 5 Q R X d
S ) zD
If the JSON text for a record is invalid, skip it.
z {"x": }
{"y": 2}
r� r r% r rJ �&Unable to read JSON record: {record!r}r� s {"x": }
Nr� r� r r r �test_readInvalidJSON� s � �z'LogFileReaderTests.test_readInvalidJSONc C sl t d��Z}t|�}| �tt|� | �t| j�d� | �| jd d d� | �| jd d d� W 5 Q R X dS ) zI
Multiple events without a record separator are skipped.
z {"x": 1}
{"y": 2}
r% r rJ r� r� s {"x": 1}
{"y": 2}
N)r r rO r� r� r r[ r� r� r r r �test_readUnseparated
s � � �z'LogFileReaderTests.test_readUnseparatedc C sf t dd�}t� �L}t|�}||� |�d� t|�}| �t|�|f� | �t| j�d� W 5 Q R X dS )z�
Data written by a L{FileLogObserver} returned by L{jsonFileLogObserver}
and read by L{eventsFromJSONLogFile} is reconstructed properly.
r% r5 r N) rN r r �seekr r �tupler[ r� )r r rf rT r\ r r r �test_roundTrip% s
z!LogFileReaderTests.test_roundTripN)rD rE rF rd r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� < s
r� N)%rd r~ r r Zzope.interface.verifyr r Ztwisted.python.compatr Ztwisted.trial.unittestr Ztwisted.python.failurer Z _observerr
r Z_formatr Z_levelsr
Z_flattenr Z_globalr Z_jsonr r r r r r� Z_loggerr r r re r� r r r r �<module> s" -]