o ¯bÅã@s`dZddlmZddlmZddlmZe ¡sddlm Z Gdd„deƒZ e ƒ  e   ¡¡dS) z& Tests for L{twisted.internet.stdio}. é)ÚProtocol)ÚReactorBuilder)Úplatform)Ú StandardIOc@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z e   ¡r.dZdSdS)ÚStdioFilesTestszI L{StandardIO} supports reading and writing to filesystem files. cCs4| ¡}t|dƒ ¡t|dƒ|_| |jj¡dS)NÚwbzrb+)ÚmktempÚopenÚcloseÚ extraFileÚ addCleanup)ÚselfÚpath©rúB/usr/lib/python3/dist-packages/twisted/internet/test/test_stdio.pyÚsetUps zStdioFilesTests.setUpcs¾| ¡‰G‡fdd„dtƒ}| ¡}t|dƒ }| d¡Wdƒn1s'wYt|dƒ}|ƒ}t|| ¡|j ¡ˆd| ˆ¡Wdƒn1sQwY|  |j d¡dS)zc Adding a filesystem file reader to a reactor will make sure it is polled. cseZdZdZ‡fdd„ZdS)z4StdioFilesTests.test_addReader..DataProtocolócs(|j|7_|jdkrˆ ¡dSdS)Nóhello!)ÚdataÚstop)r r©ÚreactorrrÚ dataReceived&s  ÿzAStdioFilesTests.test_addReader..DataProtocol.dataReceivedN)Ú__name__Ú __module__Ú __qualname__rrrrrrÚ DataProtocol#srrrNÚrb©ÚstdinÚstdoutr) Ú buildReactorrrr ÚwriterÚfilenor Ú runReactorÚ assertEqualr)r rrÚfÚprotocolrrrÚtest_addReaders"  ÿ ü ÷ zStdioFilesTests.test_addReadercsÚ| ¡‰G‡fdd„dtƒ}| ¡}t|dƒ.}|ƒ}t|| ¡|j ¡ˆd|j d¡|j d¡|j  ¡|  ˆ¡Wdƒn1sHwYt|dƒ}|  |  ¡d ¡WdƒdS1sfwYdS) zc Adding a filesystem file writer to a reactor will make sure it is polled. cseZdZ‡fdd„ZdS)z:StdioFilesTests.test_addWriter..DisconnectProtocolcs ˆ ¡dS)N)r)r ÚreasonrrrÚconnectionLostHs zIStdioFilesTests.test_addWriter..DisconnectProtocol.connectionLostN)rrrr*rrrrÚDisconnectProtocolGsr+r©r rróhellos, worldNrs hello, world) r!rrr rr#r Ú transportr"ÚloseConnectionr$r%Úread)r r+rr&r'rrrÚtest_addWriter@s& ü    ò "ÿzStdioFilesTests.test_addWritercCs¦| ¡}| |j|¡| ¡}t|dƒ ¡t|dƒ.}ttƒ| ¡|j  ¡|d}|  |j |  ¡¡|j   ¡| |j |  ¡¡WdƒdS1sLwYdS)zq Removing a filesystem file reader from a reactor will make sure it is no longer polled. rrrN)r!r ÚunbuildReactorrr r rrr#r ÚassertInÚ_readerÚ getReadersÚ stopReadingÚ assertNotIn©r rrr&ÚstdiorrrÚtest_removeReader`s ü "öz!StdioFilesTests.test_removeReadercCsˆ| ¡}| |j|¡t| ¡dƒ|_}tƒ}t|| ¡|j  ¡|d}|j   d¡|  |j | ¡¡|j  ¡| |j | ¡¡dS)zq Removing a filesystem file writer from a reactor will make sure it is no longer polled. rr,r-N)r!r r2r rr&rrr#r r.r"r3Ú_writerÚ getWritersÚ stopWritingr7)r rr&r'r9rrrÚtest_removeWriterwsÿ  z!StdioFilesTests.test_removeWritercCs„| ¡}| |j|¡| ¡}t|dƒ ¡t|dƒ|_}ttƒ|  ¡|j   ¡|d}|  ¡}|  |j |¡| |j | ¡¡dS)zk Calling C{removeAll} on a reactor includes descriptors that are filesystem files. rrrN)r!r r2rr r r&rrr#r Ú removeAllr3r4r7r5)r rrr&r9ÚremovedrrrÚtest_removeAllŒsüzStdioFilesTests.test_removeAllcCsŠ| ¡}| |j|¡| ¡}t|dƒ ¡t|dƒ }ttƒ| ¡|j  ¡|d}|  |j |  ¡¡WdƒdS1s>wYdS)zW C{reactor.getReaders} includes descriptors that are filesystem files. rrrN) r!r r2rr r rrr#r r3r4r5r8rrrÚtest_getReaders¦s ü"øzStdioFilesTests.test_getReaderscCsx| ¡}| |j|¡t| ¡dƒ|_}ttƒ| ¡|j  ¡|d}|  |j |  ¡¡|j   ¡| |j |  ¡¡dS)zW C{reactor.getWriters} includes descriptors that are filesystem files. rr,N)r!r r2r rr&rrr#r r7r;r<Ú startWritingr3)r rr&r9rrrÚtest_getWriters»sü zStdioFilesTests.test_getWriterszoStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.N)rrrÚ__doc__rr(r1r:r>rArBrDrÚ isWindowsÚskiprrrrrs$ ÿÿrN)rEÚtwisted.internet.protocolrÚ#twisted.internet.test.reactormixinsrÚtwisted.python.runtimerrFÚtwisted.internet.stdiorrÚglobalsÚupdateÚmakeTestCaseClassesrrrrÚs    G