U  W[1ã@spdZddlmZmZddlmZddlmZddlm Z e  ¡sLddl m Z Gdd„deƒZ eƒ e  ¡¡d S) z& Tests for L{twisted.internet.stdio}. é)Úabsolute_importÚdivision)Úplatform)ÚReactorBuilder)ÚProtocol)Ú StandardIOc@s\eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z e   ¡rXdZdS)Ú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.setUpc s–| ¡‰G‡fdd„dtƒ}| ¡}t|dƒ}| d¡W5QRXt|dƒ0}|ƒ}t|| ¡|j ¡ˆd| ˆ¡W5QRX|  |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 ˆ ¡dS)Nóhello!)ÚdataÚstop)rr©ÚreactorrrÚ dataReceived(s zAStdioFilesTests.test_addReader..DataProtocol.dataReceivedN)Ú__name__Ú __module__Ú __qualname__rrrrrrÚ DataProtocol&srr rÚrb©ÚstdinÚstdoutrN) Ú buildReactorrr r ÚwriterÚfilenor Ú runReactorÚ assertEqualr)rrrÚfÚprotocolrrrÚtest_addReaders    þzStdioFilesTests.test_addReaderc s°| ¡‰G‡fdd„dtƒ}| ¡}t|dƒR}|ƒ}t|| ¡|j ¡ˆd|j d¡|j d¡|j  ¡|  ˆ¡W5QRXt|dƒ}|  |  ¡d¡W5QRXd S) 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ÚconnectionLostHszIStdioFilesTests.test_addWriter..DisconnectProtocol.connectionLostN)rrrr,rrrrÚDisconnectProtocolGsr-r ©r"r!róhellos, worldrs hello, worldN) r#rr r rr%r Ú transportr$ZloseConnectionr&r'Úread)rr-rr(r)rrrÚtest_addWriter@s  ÿ    zStdioFilesTests.test_addWriterc Cs| ¡}| |j|¡| ¡}t|dƒ ¡t|dƒP}ttƒ| ¡|j  ¡|d}|  |j |  ¡¡|j   ¡| |j |  ¡¡W5QRXdS)zq Removing a filesystem file reader from a reactor will make sure it is no longer polled. r rr N)r#rÚunbuildReactorr r r rrr%r ÚassertInÚ_readerÚ getReadersZ stopReadingÚ assertNotIn©rrrr(Ú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. r r.r/N)r#rr3r r r(rrr%r r0r$r4Ú_writerÚ getWritersZ stopWritingr7)rrr(r)r9rrrÚtest_removeWriterrs þ  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. r rr N)r#rr3r r r r(rrr%r Z removeAllr4r5r7r6)rrrr(r9ZremovedrrrÚtest_removeAllˆs ÿzStdioFilesTests.test_removeAllc Cst| ¡}| |j|¡| ¡}t|dƒ ¡t|dƒ4}ttƒ| ¡|j  ¡|d}|  |j |  ¡¡W5QRXdS)zW C{reactor.getReaders} includes descriptors that are filesystem files. r rr N) r#rr3r r r rrr%r r4r5r6r8rrrÚ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. r r.N)r#rr3r r r(rrr%r r7r;r<Z startWritingr4)rrr(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*r2r:r=r>r?r@rÚ isWindowsÚskiprrrrrs!rN)rAZ __future__rrZtwisted.python.runtimerZ#twisted.internet.test.reactormixinsrZtwisted.internet.protocolrrBZtwisted.internet.stdiorrÚglobalsÚupdateZmakeTestCaseClassesrrrrÚs    5