a DfG@sddlZddlZddlZddlZddlmZejeje dZ ej dej e dddddddd Z Gd d d ejZGd d d eZdS)N)ComparisonTestCasez..2( )CRITICALERRORWARNINGINFODEBUGVERBOSEcsJeZdZdZfddZddZddZdd d Zd d Zd dZ Z S)MockLoggingHandlerz Mock logging handler to check for expected logs used by LoggingComparisonTestCase. Messages are available from an instance's ``messages`` dict, in order, indexed by a lowercase log level string (e.g., 'debug', 'info', etc.).cs:ggggggd|_ddddd|_tj|i|dS)Nr r r r rr zparam.param.warning()zparam.param.message()zparam.param.verbose()zparam.param.debug())r r r r )messages param_methodssuper__init__)selfargskwargs __class__b/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/holoviews/tests/utils.pyrszMockLoggingHandler.__init__cCs:|z"|j|j|W|n |0dS)z5Store a message to the instance's messages dictionaryN)acquirer levelnameappend getMessagerelease)rrecordrrremit$szMockLoggingHandler.emitcCs(|ggggggd|_|dS)Nr)rrr)rrrrreset,s zMockLoggingHandler.resetcCsdd|j|| dDS)z4Returns the last n lines captured at the given levelcSsg|] }t|qSr)str).0elrrr 4z+MockLoggingHandler.tail..N)r)rlevelnrrrtail2szMockLoggingHandler.tailcCsd}|j|dd}t|dkr:tdj|j|t|d|d|spt|j|j|t|dt|dn|j|dd S) zq Assert that the last line captured at the given level ends with a particular substring. z6 {method}: {last_line} does not end with: {substring}r#r*r$Missing {method} output: {substring}method substringr/ last_liner0N) r+lenAssertionErrorformatrreprendswithrpoprr)r0msgr2rrrassertEndsWith6s   z!MockLoggingHandler.assertEndsWithcCsd}|j|dd}t|dkr:tdj|j|t|d||dvrnt|j|j|t|dt|dn|j|dd S) zp Assert that the last line captured at the given level contains a particular substring. z5 {method}: {last_line} does not contain: {substring}r#r,rr-r.r1r3N)r+r4r5r6rr7rr9r:rrrassertContainsHs    z!MockLoggingHandler.assertContains)r#) __name__ __module__ __qualname____doc__rr!r"r+r<r= __classcell__rrrrrs  rcs,eZdZdZfddZfddZZS)LoggingComparisonTestCasez ComparisonTestCase with support for capturing param logging output. Subclasses must call super setUp to make the tests independent. Testing can then be done via the self.log_handler.tail and self.log_handler.assertEndsWith methods. cs>ttj}|j|_g|_tdd|_||jdS)Nr )r)) rsetUpparam parameterized get_loggerhandlersr log_handler addHandler)rlogrrrrDcs    zLoggingComparisonTestCase.setUpcs^ttj}|j|_|jj}|j| D]"\}}|D]}| t ||qBq6dS)N) rtearDownrErFrGrHrIrr"itemsrKLEVELS)rrKrr)Zmsgsr;rrrrLks   z"LoggingComparisonTestCase.tearDown)r>r?r@rArDrLrBrrrrrCZs rC)loggingossysrEZholoviews.element.comparisonrpathabspathsplit__file__cwdinsertjoinrNHandlerrrCrrrrs I