CasperSecurity
o
`�wi�
� @ sR d dl Z d dlZddlmZ e�dddg�ZG dd� de j�ZG d d
� d
e�ZdS ) � N� )�_BaseTestCaseContext�_LoggingWatcher�records�outputc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_CapturingHandlerzM
A logging handler capturing all (raw and formatted) logging output.
c C s t j�| � tg g �| _d S �N)�logging�Handler�__init__r �watcher��self� r �$/usr/lib/python3.10/unittest/_log.pyr s z_CapturingHandler.__init__c C s d S r r r
r r r �flush s z_CapturingHandler.flushc C s* | j j�|� | �|�}| j j�|� d S r )r r �append�formatr )r �record�msgr r r �emit s
z_CapturingHandler.emitN)�__name__�
__module__� __qualname__�__doc__r r r r r r r r
s
r c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�_AssertLogsContextz6A context manager for assertLogs() and assertNoLogs() z"%(levelname)s:%(name)s:%(message)sc C s@ t �| |� || _|rtj�||�| _ntj| _d | _|| _ d S r )
r r � logger_namer � _nameToLevel�get�level�INFOr �no_logs)r � test_caser r r! r r r r ! s
z_AssertLogsContext.__init__c C s� t | jtj�r| j }| _n t�| j� }| _t�| j�}t� }|� | j
� |� |� |j | _ |j
d d � | _|j
| _|j| _|g|_
|� | j
� d|_| jrQd S |j S )NF)�
isinstancer r �Logger�logger� getLogger� Formatter�LOGGING_FORMATr �setLevelr � setFormatterr �handlers� old_handlers� old_level� propagate�
old_propagater! )r r% � formatter�handlerr r r � __enter__+ s"
z _AssertLogsContext.__enter__c C s� | j | j_| j| j_| j�| j� |d urdS | jr0t| j j
�dkr.| � d� | j j
�� d S d S t| j j
�dkrI| � d� t�| j�| jj�� d S d S )NFr zUnexpected logs found: {!r}z-no logs of level {} or higher triggered on {})r, r% r+ r/ r. r) r- r! �lenr r �
_raiseFailurer r r � getLevelNamer �name)r �exc_type� exc_value�tbr r r �__exit__? s&
��� ��z_AssertLogsContext.__exit__N)r r r r r( r r2 r: r r r r r s
r ) r � collections�caser �
namedtupler r
r r r r r r �<module> s �