U n5{i1ã@s¨dZz ddlZWnek r(dZYnXddlmZddlmZddlmZddl m Z ddl m Z ddl mZmZGd d „d ƒZGd d „d e ƒZeƒ e ¡¡dS) z6 Tests for implementations of L{IReactorWin32Events}. éN)Ú verifyObject)ÚDeferred)ÚIReactorWin32Events)ÚReactorBuilder)ÚFailure)Ú getThreadIDÚ isInIOThreadc@sPeZdZdZdZdZZZdd„Zdd„Z dd „Z d d „Z d d „Z dd„Z dS)ÚListeneraÔ L{Listener} is an object that can be added to a L{IReactorWin32Events} reactor to receive callback notification when a Windows event is set. It records what thread its callback is invoked in and fires a Deferred. @ivar success: A flag which is set to C{True} when the event callback is called. @ivar logThreadID: The id of the thread in which the C{logPrefix} method is called. @ivar eventThreadID: The id of the thread in which the event callback is called. @ivar connLostThreadID: The id of the thread in which the C{connectionLost} method is called. @ivar _finished: The L{Deferred} which will be fired when the event callback is called. FNcCs ||_dS©N)Ú _finished)ÚselfÚfinished©rúJ/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/internet/test/test_win32events.pyÚ__init__/szListener.__init__cCs tƒ|_dS)Nr )rÚ logThreadID©r rrrÚ logPrefix2szListener.logPrefixcCsd|_tƒ|_|j d¡dS)NT)ÚsuccessrÚ eventThreadIDr ÚcallbackrrrrÚoccurred6szListener.occurredcCs tdƒ‚dS)Nz Some problem)Ú RuntimeErrorrrrrÚbrokenOccurred;szListener.brokenOccurredcCstdƒS)NzEntirely different problem)ÚEnvironmentErrorrrrrÚreturnValueOccurred>szListener.returnValueOccurredcCstƒ|_|j |¡dSr )rÚconnLostThreadIDr Zerrback)r ÚreasonrrrÚconnectionLostAszListener.connectionLost)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrrrrrrrrr s r c@sFeZdZdZegZdd„Zdd„Zdd„Zdd „Z d d „Z d d „Z dS)ÚWin32EventsTestsBuilderzD Builder defining tests relating to L{IReactorWin32Events}. cCs| ¡}tt|ƒdS)zn An instance of the reactor has all of the methods defined on L{IReactorWin32Events}. N)Ú buildReactorrr)r ÚreactorrrrÚtest_interfaceMsz&Win32EventsTestsBuilder.test_interfacecstƒ}| ¡‰t dddd¡}tƒ}| ‡fdd„¡t|ƒ}ˆ ||d¡ˆ tj |¡|  ˆ¡|  |j ¡|  ||j¡|  ||j¡dS)zš When an event which has been added to the reactor is set, the action associated with the event is invoked in the reactor thread. NFcsˆ ¡Sr ©Ústop©Zignored©r%rrÚ^óz7Win32EventsTestsBuilder.test_addEvent..r)rr$Ú win32eventÚ CreateEventrÚ addCallbackr ÚaddEventÚcallWhenRunningÚSetEventÚ runReactorÚ assertTruerÚ assertEqualrr)r ÚreactorThreadIDÚeventr Úlistenerrr*rÚ test_addEventUs  z%Win32EventsTestsBuilder.test_addEventcs†g‰‡‡fdd„}| ¡‰t dddd¡}tƒ}t|ƒ}| |¡ˆ ||d¡ˆ tj|¡|  ˆ¡|  |j ¡|  dgˆ¡dS)z{ Using L{IReactorWin32Events.addEvent} does not change which thread is reported as the I/O thread. csˆ tƒ¡ˆ ¡dSr )Úappendrr(r)©r%ÚresultsrrÚcheckns zAWin32EventsTestsBuilder.test_ioThreadDoesNotChange..checkNFrT) r$r-r.rr r/r0r1r2r3r4rr5)r r=r7r r8rr;rÚtest_ioThreadDoesNotChangegs   z2Win32EventsTestsBuilder.test_ioThreadDoesNotChangecsºtƒ}| ¡‰t dddd¡}g}tƒ}| |j¡| ‡fdd„¡t|ƒ}ˆ ||d¡ˆ  tj |¡|  ˆ¡|  |dt ¡|d t¡| ||j¡| dt| t¡ƒ¡dS)zÕ If the event handler raises an exception, the event is removed from the reactor and the handler's C{connectionLost} method is called in the I/O thread and the exception is logged. NFcsˆ ¡Sr r'r)r*rrr+Šr,zBWin32EventsTestsBuilder.test_disconnectedOnError..rré)rr$r-r.rÚaddBothr:r r0r1r2r3ÚassertIsInstancerÚtraprr5rÚlenZflushLoggedErrors©r r6r7Úresultr r8rr*rÚtest_disconnectedOnError}s  z0Win32EventsTestsBuilder.test_disconnectedOnErrorcs¤tƒ}| ¡‰t dddd¡}g}tƒ}| |j¡| ‡fdd„¡t|ƒ}ˆ ||d¡ˆ  tj |¡|  ˆ¡|  |dt ¡|d t¡| ||j¡dS)zµ If the event handler returns a value, the event is removed from the reactor and the handler's C{connectionLost} method is called in the I/O thread. NFcsˆ ¡Sr r'r)r*rrr+¤r,zFWin32EventsTestsBuilder.test_disconnectOnReturnValue..rr)rr$r-r.rr@r:r r0r1r2r3rArrBrr5rrDrr*rÚtest_disconnectOnReturnValue—s  z4Win32EventsTestsBuilder.test_disconnectOnReturnValuecCsZ| ¡}t dddd¡}tƒ}t|ƒ}| ||d¡| |j¡| |¡|  |j ¡dS)z¾ Event handlers added with L{IReactorWin32Events.addEvent} do not have C{connectionLost} called on them if they are still active when the reactor shuts down. NFr) r$r-r.rr r0r1r(r3Z assertIsNoner)r r%r7r r8rrrÚtest_notDisconnectedOnShutdown°s  z6Win32EventsTestsBuilder.test_notDisconnectedOnShutdownN) rr r!r"rZrequiredInterfacesr&r9r>rFrGrHrrrrr#Fsr#)r"r-Ú ImportErrorZzope.interface.verifyrZtwisted.internet.deferrZtwisted.internet.interfacesrZ#twisted.internet.test.reactormixinsrZtwisted.python.failurerZtwisted.python.threadablerrr r#ÚglobalsÚupdateZmakeTestCaseClassesrrrrÚs       0z