U `[/@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dd l m Z dd l mZdd lmZdd lmZdd lmZddZGdddejZdS)z+ Test cases for L{twisted.logger._global}. )print_functionN)unittest)textFileLogObserver) LogPublisher)Logger) LogBeginner)MORE_THAN_ONCE_WARNING)LogLevel)nextLine)Failurecsjt|t|kr|||t|D]}t|Oq&fddfdd|D}|||dS)a Compare two sequences of log events, examining only the the keys which are present in both. @param test: a test case doing the comparison @type test: L{unittest.TestCase} @param actualEvents: A list of log events that were emitted by a logger. @type actualEvents: L{list} of L{dict} @param expectedEvents: A list of log events that were expected by a test. @type expected: L{list} of L{dict} cs,|}|D]}|kr||q|SN)copykeyspop)eventrkey) allMergedKeysA/usr/lib/python3/dist-packages/twisted/logger/test/test_global.pysimplify.s   zcompareEvents..simplifycsg|] }|qSrr.0r)rrr 5sz!compareEvents..N)len assertEqualsetr)ZtestZ actualEventsZexpectedEventsrZsimplifiedActualr)rrr compareEventss  rc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZdS)LogBeginnerTestsz# Tests for L{LogBeginner}. cCs^t|_t|_Gdddt}Gdddt}||_||_t|j|j|j|j|_ dS)Nc@seZdZeZeZdS)z&LogBeginnerTests.setUp..NotSysN)__name__ __module__ __qualname__objectstdoutstderrrrrrNotSysCsr%c@seZdZddZdddZdS)z+LogBeginnerTests.setUp..NotWarningscSs g|_dSr )warnings)selfrrr__init__Hsz4LogBeginnerTests.setUp..NotWarnings.__init__NcSs|j||||||fdS)a Emulate warnings.showwarning. @param message: A warning message to emit. @type message: L{str} @param category: A warning category to associate with C{message}. @type category: L{warnings.Warning} @param filename: A file name for the source code file issuing the warning. @type warning: L{str} @param lineno: A line number in the source file where the warning was issued. @type lineno: L{int} @param file: A file to write the warning message to. If L{None}, write to L{sys.stderr}. @type file: file-like object @param line: A line of source code to include with the warning message. If L{None}, attempt to read the line from C{filename} and C{lineno}. @type line: L{str} N)r&append)r'messagecategoryfilenamelinenofilelinerrr showwarningKsz7LogBeginnerTests.setUp..NotWarnings.showwarning)NN)rr r!r(r0rrrr NotWarningsGsr1) r publisherioStringIO errorStreamr" sysModulewarningsModulerbeginner)r'r%r1rrrsetUp?s 'zLogBeginnerTests.setUpcsftddd}ggfdd}fdd}|j||f||||g||gdS)z? Test that C{beginLoggingTo()} adds observers. rZfooZbarcs |Sr r)eevents1rrzBLogBeginnerTests.test_beginLoggingToAddObservers..cs |Sr r<r=events2rrrArBN)dictr8beginLoggingTor2rr'rZo1Zo2rr@rDrtest_beginLoggingToAddObserversvs    z0LogBeginnerTests.test_beginLoggingToAddObserverscsftddd}ggfdd}fdd}|||j||f||g||gdS)z\ Test that events are buffered until C{beginLoggingTo()} is called. r:rr;cs |Sr r<r=r?rrrArBzDLogBeginnerTests.test_beginLoggingToBufferedEvents..cs |Sr r<r=rCrrrArBN)rEr2r8rFrrGrrHr!test_beginLoggingToBufferedEventss    z2LogBeginnerTests.test_beginLoggingToBufferedEventscCs`t|dD]}|t|dq g}||jg|ttd|dtdd|DdS)a Verify that when more than C{limit} events are logged to L{LogBeginner}, only the last C{limit} are replayed by L{LogBeginner.beginLoggingTo}. @param limit: The maximum number of events the log beginner should buffer. @type limit: L{int} @param beginner: The L{LogBeginner} against which to verify. @type beginner: L{LogBeginner} @raise: C{self.failureException} if the wrong events are replayed by C{beginner}. @return: L{None} r:)countcss|]}|dVqdS)rKNrrrrr sz4LogBeginnerTests._bufferLimitTest..N)ranger2rErFr)rlist)r'limitr8rKZeventsrrr_bufferLimitTestsz!LogBeginnerTests._bufferLimitTestcCstj}|||jdS)z Up to C{LogBeginner._DEFAULT_BUFFER_SIZE} log events are buffered for replay by L{LogBeginner.beginLoggingTo}. N)rZ_DEFAULT_BUFFER_SIZErPr8)r'rOrrrtest_defaultBufferLimitsz(LogBeginnerTests.test_defaultBufferLimitcCs.d}t|j|j|j|j|d}|||dS)z The size of the L{LogBeginner} event buffer can be overridden with the C{initialBufferSize} initilizer argument. )ZinitialBufferSizeN)rr2r5r6r7rP)r'rOr8rrrtest_overrideBufferLimitsz)LogBeginnerTests.test_overrideBufferLimitc Csg}g}t}t|}|tddt\}}|j|j|g|tddt\}}|j|j|g|tddtt t j ||||d} t ||tddtdd| tddgt ||| tddg| } |d||| |d||| dS)z When invoked twice, L{LogBeginner.beginLoggingTo} will emit a log message warning the user that they previously began logging, and add the new log observers. Z prebuffer)rZ postbufferZpostwarn)Z log_formatZ log_levelZfileNowZlineNowZfileThenZlineThenz <{0}:{1}>N)r3r4rr2rEr r8rFr)r r warnrgetvalueassertInformat) r'r@rDZ fileHandleZ textObserverZ firstFilenameZ firstLineZsecondFilenameZ secondLinewarningoutputrrrtest_beginLoggingToTwicesH   z)LogBeginnerTests.test_beginLoggingToTwicecCs:t|jd}|d|jddd||jddS)zP Critical messages will be written as text to the error stream. Zobserverz ignore thisza critical {message}r*)r*za critical message N)rr2infocriticalrr5rUr'logrrrtest_criticalLoggings  z%LogBeginnerTests.test_criticalLoggingcCs8t|jd}|jd|d||jddS)z Once logging has begun with C{beginLoggingTo}, critical messages are no longer written to the output stream. r[rzanother critical messageN)rr2r8rFr]rr5rUr^rrrtest_criticalLoggingStopss   z*LogBeginnerTests.test_criticalLoggingStopscCsng}|j|jgtd|jjdt||tdddg|dd=td|jjdt||tdddgdS)z L{LogBeginner.beginLoggingTo} will re-direct the standard output and error streams by setting the C{stdio} and C{stderr} attributes on its sys module object. z Hello, world.r.r#)Z log_namespacelog_ioNz Error, world.r$) r8rFr)printr6r#rrEr$)r'xrrr%test_beginLoggingToRedirectStandardIO s   z6LogBeginnerTests.test_beginLoggingToRedirectStandardIOcCsD|jj}|jj}|jjddd||jj|||jj|dS)z L{LogBeginner.beginLoggingTo} will leave the existing stdout/stderr in place if it has been told not to replace them. rF)ZredirectStandardION)r6r#r$r8rFZassertIs)r'ZoldOutZoldErrrrrtest_beginLoggingToDontRedirect s z0LogBeginnerTests.test_beginLoggingToDontRedirectcCsttd}ttd}||j_||j_g}|j|jg| |jjj d| |jjj d|jj d|jj dt ||t ddt ddgdS) z When L{LogBeginner.beginLoggingTo} redirects stdout/stderr streams, the replacement streams will preserve the encoding of the replaced streams, to minimally disrupt any application relying on a specific encoding. z shift-JISbig5s s u李)rdu瑩N)r3 TextIOWrapperBytesIOr6r#r$r8rFr)rencodingwriterrE)r'ZweirdZweirderrrfrrr$test_beginLoggingToPreservesEncoding,sz5LogBeginnerTests.test_beginLoggingToPreservesEncodingc Cs|jdttdg}|j|jg|jdttdt}|jjdttd|d| |jj dttdddfdttd|dfgt ||t dtj d tjtdd gdS) z L{LogBeginner.beginLoggingTo} will redirect the warnings of its warnings module into the logging system. z a messager:zanother messagerz yet anotherrRrcN.)rXr+r,r-)r7r0DeprecationWarning__file__r8rFr)r3r4rr&rrEr r)r'rffrrrtest_warningsModuleEsPz$LogBeginnerTests.test_warningsModulecCsXttd}t|jd}|jd|d|j}|d||d||d|dS)zR The string resulting from a logged failure contains a traceback. z,this is not the behavior you are looking forr[z a failure)failure TracebackN)r Exceptionrr2rtr5rUrV)r'rrr_msgrrrtest_failuresAppendTracebacksjs     z.LogBeginnerTests.test_failuresAppendTracebacksN)rr r!__doc__r9rIrJrPrQrSrZr`rbrgrhrnrsrxrrrrr:s7  *   %r)ryZ __future__rr3Z twisted.trialrZ_filerZ _observerrZ_loggerrZ_globalrr Z_levelsr Ztest.test_stdlibr Ztwisted.python.failurer rZTestCaserrrrrs          !