← Back
Editing: 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" -]
Save File
Cancel