
    5f                     h    d Z ddlZddlmZ ddlmZ  G d dej                        Z G d de      Zy)	z/Unit Tests for oslo.log with custom log handler    N)log)LogTestBasec                        e Zd Z fdZ xZS )CustomLogHandlerc                 :    d |_         t        t        |   |       y )N)argssuperr   emit)selfrecord	__class__s     L/usr/lib/python3/dist-packages/oslo_log/tests/unit/test_custom_loghandler.pyr
   zCustomLogHandler.emit   s    *62    )__name__
__module____qualname__r
   __classcell__r   s   @r   r   r      s    3 3r   r   c                   $     e Zd Z fdZd Z xZS )CustomLogHandlerTestCasec                    t         t        |           | j                  ddd       t	        j
                  d      | _        | j                  | j                  t               | j                  | j                  t        j                         y )Nz)HAS CONTEXT [%(request_id)s]: %(message)szNOCTXT: %(message)sz--DBG)logging_context_format_stringlogging_default_format_stringlogging_debug_format_suffix )handler)r	   r   setUpconfigr   	getLogger_add_handler_with_cleanupr   _set_log_level_with_cleanuploggingDEBUG)r   r   s    r   r   zCustomLogHandlerTestCase.setUp%   ss    &35 3@ 3H07	 	 	9
 ==$&&txx9I&J((7==Ar   c                     d}| j                   j                  |       | j                  d|z  | j                  j	                                y )NfoozNOCTXT: %s
)r   infoassertEqualstreamgetvalue)r   messages     r   test_logz!CustomLogHandlerTestCase.test_log0   s9    g'14;;3G3G3IJr   )r   r   r   r   r+   r   r   s   @r   r   r   $   s    	BKr   r   )	__doc__r"   oslo_logr   oslo_log.tests.unit.test_logr   StreamHandlerr   r    r   r   <module>r1      s5    6   43w,, 3K{ Kr   