U n5{i<ã@sødZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZGd d „d ejƒZGd d „d e jƒZGd d„de jƒZGdd„deƒZGdd„dejƒZe dd¡dd…dkrÚdZdZndZdZGdd„dejƒZdS)z0 Test cases for twisted.protocols.ident module. éN)ÚStringIO)ÚdeferÚerror)ÚStringTransport)Úident)Úfailure)Úunittestc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚClassParserTestsz* Test parsing of ident responses. cCst ¡|_dS)z7 Create an ident client used in tests. N)rZ IdentClientÚclient©Úself©r ú;/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/test/test_ident.pyÚsetUpszClassParserTests.setUpcCs6t ¡}|jj |ddf¡|j d¡| |tj¡S)zX 'UNKNOWN-ERROR' error should map to the L{ident.IdentError} exception. é{éÈz 123, 456 : ERROR : UNKNOWN-ERROR) rÚDeferredr ÚqueriesÚappendÚ lineReceivedÚ assertFailurerÚ IdentError©r Údr r rÚtest_indentErrors z!ClassParserTests.test_indentErrorcCs6t ¡}|jj |ddf¡|j d¡| |tj¡S)zN 'NO-USER' error should map to the L{ident.NoUser} exception. éêrz234, 456 : ERROR : NO-USER) rrr rrrrrÚNoUserrr r rÚtest_noUSerError(s z!ClassParserTests.test_noUSerErrorcCs6t ¡}|jj |ddf¡|j d¡| |tj¡S)zX 'INVALID-PORT' error should map to the L{ident.InvalidPort} exception. iYé7z 345, 567 : ERROR : INVALID-PORT) rrr rrrrrÚ InvalidPortrr r rÚtest_invalidPortError1s z&ClassParserTests.test_invalidPortErrorcCs6t ¡}|jj |ddf¡|j d¡| |tj¡S)zV 'HIDDEN-USER' error should map to the L{ident.HiddenUser} exception. riz567, 789 : ERROR : HIDDEN-USER) rrr rrrrrÚ HiddenUserrr r rÚtest_hiddenUserError:s z%ClassParserTests.test_hiddenUserErrorcCs@t ¡}|jj |ddf¡|j t t  ¡¡¡|  |t j ¡S)zy A pending query which failed because of a ConnectionLost should receive an L{ident.IdentError}. iýi°) rrr rrZconnectionLostrZFailurerZConnectionLostrrrrr r rÚtest_lostConnectionCsz$ClassParserTests.test_lostConnectionN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr r"r#r r r rr s    r c@seZdZdd„ZdS)ÚTestIdentServercCs|jS©N)Ú resultValue©r Z serverAddressZ clientAddressr r rÚlookupOszTestIdentServer.lookupN©r$r%r&r,r r r rr(Nsr(c@seZdZdd„ZdS)ÚTestErrorIdentServercCs | ¡‚dSr))Ú exceptionTyper+r r rr,TszTestErrorIdentServer.lookupNr-r r r rr.Ssr.c@s eZdZdS)Ú NewExceptionN)r$r%r&r r r rr0Xsr0c@seZdZdd„Zdd„ZdS)ÚServerParserTestscCs4tƒ}| tƒ¡g}|j|_tj|_| d¡|  |dd¡tj |_| d¡|  |dd¡tj |_| d¡|  |dd ¡tj |_| d ¡|  |d d ¡t |_| d ¡|  |dd¡| t ¡}|  t|ƒd¡dD]J}|dd…=| d|f¡| d|f¡|  |d|fd|fg¡qädS)Nz123, 345rz 123, 345 : ERROR : UNKNOWN-ERRORz432, 210éz432, 210 : ERROR : NO-USERz987, 654éz987, 654 : ERROR : INVALID-PORTz756, 827éz756, 827 : ERROR : HIDDEN-USERz987, 789éz 987, 789 : ERROR : UNKNOWN-ERROR)éÿÿÿÿriiz%d, 5z5, %dz%d, 5 : ERROR : INVALID-PORTz5, %d : ERROR : INVALID-PORT)r.ÚmakeConnectionrrÚsendLinerrr/rÚ assertEqualrrr!r0ZflushLoggedErrorsÚlen)r ÚpÚLZerrsÚportr r rÚ testErrors]s>        þþzServerParserTests.testErrorscCsBtƒ}| tƒ¡g}|j|_d|_| d¡| |dd¡dS)N)ZSYSÚUSERz123, 456rz123, 456 : USERID : SYS : USER)r(r7rrr8r*rr9)r r;r<r r rÚ testSuccess…s  zServerParserTests.testSuccessN)r$r%r&r>r@r r r rr1\s(r1z=Lr2óZ0100007FZ04030201Z7F000001Z01020304c@sPeZdZdeefZdeZdd„Zdd„Zdd„Z d d „Z d d „Z d d„Z dS)ÚProcMixinTestszj4: %s:0019 %s:02FA 0A 00000000:00000000 00:00000000 00000000 0 0 10927 1 f72a5b80 3000 0 0 2 -1zd sl local_address rem_address st tx_queue rx_queue tr tm->when retrnsmt uid timeout inode cCst ¡}| | t¡d¡dS)Nú 127.0.0.1)rÚProcServerMixinr9ZdottedQuadFromHexStringÚ_addr1©r r;r r rÚtestDottedQuadFromHexString£sz*ProcMixinTests.testDottedQuadFromHexStringcCs"t ¡}| | td¡d¡dS)Nz:0277)rCiw)rrDr9Z unpackAddressrErFr r rÚtestUnpackAddress§sz ProcMixinTests.testUnpackAddresscCs t ¡}| | |j¡d¡dS)N)©rCé©ú1.2.3.4éúr)rrDr9Z parseLineÚlinerFr r rÚtestLineParser«s  ÿzProcMixinTests.testLineParsercsTg‰t ¡}‡fdd„|_‡fdd„|_ˆ | dd¡|jdf¡ˆ ˆdg¡dS)Ncs tˆjgƒSr)©ÚiterrNr r r rÚ´óz4ProcMixinTests.testExistingAddress..csˆ |¡dfdS)NÚrootr2)r)Úuid)Úusernamer rrRµrSrIrKrTr)rrDÚentriesZ getUsernamer9r,Z SYSTEM_NAMErFr )r rVrÚtestExistingAddress±s ÿz"ProcMixinTests.testExistingAddresscsVt ¡}‡fdd„|_ˆ tj|jdd¡ˆ tj|jdd¡ˆ tj|jdd¡dS)Ncs tˆjgƒSr)rPr r r rrR½rSz7ProcMixinTests.testNonExistingAddress..©rCérKrI)z1.2.3.5rM)rLiû)rrDrWÚ assertRaisesrr,rFr r rÚtestNonExistingAddress»s z%ProcMixinTests.testNonExistingAddresscsRg‰‡‡fdd„}ˆ td|¡t ¡}ˆ tj|jdd¡ˆ difgˆ¡dS)zS L{ident.ProcServerMixin.lookup} uses the Linux TCP process table. csˆ ||f¡tˆjƒS)z[ Mock for the open call to prevent actually opening /proc/net/tcp. )rrÚ sampleFile)ÚargsÚkwargs©Z open_callsr r rÚ mocked_openÈsz8ProcMixinTests.testLookupProcNetTcp..mocked_openÚopenrYrK)z /proc/net/tcpN)ÚpatchÚbuiltinsrrDr[rr,r9)r rar;r r`rÚtestLookupProcNetTcpÂs z#ProcMixinTests.testLookupProcNetTcpN) r$r%r&rEÚ_addr2rNr]rGrHrOrXr\rer r r rrB˜süÿÿ rB)r'rdÚstructÚiorZtwisted.internetrrZtwisted.internet.testingrZtwisted.protocolsrZtwisted.pythonrZ twisted.trialrZTestCaser Z IdentServerr(r.Ú RuntimeErrorr0r1ÚpackrErfrBr r r rÚs&     :4