o Óó·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…dkrmdZdZndZdZGdd„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 ú]C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/test/test_ident.pyÚsetUpszClassParserTests.setUpcCó6t ¡}|jj |ddf¡|j d¡| |tj¡S)zX 'UNKNOWN-ERROR' error should map to the L{ident.IdentError} exception. é{éÈz 123, 456 : ERROR : UNKNOWN-ERRORN) rÚDeferredr ÚqueriesÚappendÚ lineReceivedÚ assertFailurerÚ IdentError©r Údr r rÚtest_indentErroró z!ClassParserTests.test_indentErrorcCr)zN 'NO-USER' error should map to the L{ident.NoUser} exception. éêrz234, 456 : ERROR : NO-USERN) rrr rrrrrÚNoUserrr r rÚtest_noUSerError(rz!ClassParserTests.test_noUSerErrorcCr)zX 'INVALID-PORT' error should map to the L{ident.InvalidPort} exception. iYé7z 345, 567 : ERROR : INVALID-PORTN) rrr rrrrrÚ InvalidPortrr r rÚtest_invalidPortError1rz&ClassParserTests.test_invalidPortErrorcCr)zV 'HIDDEN-USER' error should map to the L{ident.HiddenUser} exception. r iz567, 789 : ERROR : HIDDEN-USERN) rrr rrrrrÚ HiddenUserrr r rÚtest_hiddenUserError:rz%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°N) rrr rrZconnectionLostrZFailurerZConnectionLostrrrrr r rÚtest_lostConnectionCsz$ClassParserTests.test_lostConnectionN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr"r$r%r r r rr s   r c@óeZdZdd„ZdS)ÚTestIdentServercCs|jS©N)Ú resultValue©r Z serverAddressZ clientAddressr r rÚlookupOszTestIdentServer.lookupN©r&r'r(r/r r r rr+Nó r+c@r*)ÚTestErrorIdentServercCs| ¡‚r,)Ú exceptionTyper.r r rr/TszTestErrorIdentServer.lookupNr0r r r rr2Sr1r2c@s eZdZdS)Ú NewExceptionN)r&r'r(r r r rr4Xsr4c@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]%}|dd…=| d|f¡| d|f¡|  |d|fd|fg¡qrdS)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)r2ÚmakeConnectionrrÚsendLinerrr3rÚ assertEqualrr!r#r4ZflushLoggedErrorsÚ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+r;rrr<r-rr=)r r?r@r r rÚ testSuccess…s  zServerParserTests.testSuccessN)r&r'r(rBrDr r r rr5\s (r5z=Lr6ó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ÚProcServerMixinr=ZdottedQuadFromHexStringÚ_addr1©r r?r r rÚtestDottedQuadFromHexString£sz*ProcMixinTests.testDottedQuadFromHexStringcCs"t ¡}| | td¡d¡dS)Nz:0277)rGiw)rrHr=Z unpackAddressrIrJr r rÚtestUnpackAddress§sz ProcMixinTests.testUnpackAddresscCs t ¡}| | |j¡d¡dS)N)©rGé©ú1.2.3.4éúr)rrHr=Z parseLineÚlinerJr r rÚtestLineParser«s ÿzProcMixinTests.testLineParsercsTg‰t ¡}‡fdd„|_‡fdd„|_ˆ | dd¡|jdf¡ˆ ˆdg¡dS)Ncó tˆjgƒSr,©ÚiterrRr r r rÚ´ó z4ProcMixinTests.testExistingAddress..csˆ |¡dfdS)NÚrootr6)r)Úuid)Úusernamer rrWµsrMrOrYr)rrHÚentriesZ getUsernamer=r/Z SYSTEM_NAMErJr )r r[rÚtestExistingAddress±sÿz"ProcMixinTests.testExistingAddresscsVt ¡}‡fdd„|_ˆ tj|jdd¡ˆ tj|jdd¡ˆ tj|jdd¡dS)NcrTr,rUr r r rrW½rXz7ProcMixinTests.testNonExistingAddress..©rGérOrM)z1.2.3.5rQ)rPiû)rrHr\Ú assertRaisesrr/rJr 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. N)rrÚ sampleFile)ÚargsÚkwargs©Z open_callsr r rÚ mocked_openÈs z8ProcMixinTests.testLookupProcNetTcp..mocked_openÚopenr^rO)z /proc/net/tcpN)ÚpatchÚbuiltinsrrHr`rr/r=)r rfr?r rerÚtestLookupProcNetTcpÂs z#ProcMixinTests.testLookupProcNetTcpN) r&r'r(rIÚ_addr2rRrbrKrLrSr]rarjr r r rrF˜süÿÿ rF)r)riÚstructÚiorZtwisted.internetrrZtwisted.internet.testingrZtwisted.protocolsrZtwisted.pythonrZ twisted.trialrZTestCaser Z IdentServerr+r2Ú RuntimeErrorr4r5ÚpackrIrkrFr r r rÚs(     :4