Yf@dZddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZ dd lmZd Zn#e$rd Zd ZYnwxYwd ZdZdZdezZdZdezZGddejZGddZGddeejZGddeejZGddeejZGddejZ Gdd ejZ!d S)!z4 Tests for L{twisted.words.protocols.jabber.client} )sha1)skipIf)defer)unittest)clienterrorjid xmlstream)SASLInitiatingInitializer)utility)ssl)FN)TzSSL not availablez//iq[@type="get"]/query[@xmlns="jabber:iq:auth"]z//iq[@type="set"]/query[@xmlns="jabber:iq:auth"]z urn:ietf:params:xml:ns:xmpp-bindz"/iq[@type="set"]/bind[@xmlns="%s"]z#urn:ietf:params:xml:ns:xmpp-sessionz%/iq[@type="set"]/session[@xmlns="%s"]c eZdZdZdZdZdS)CheckVersionInitializerTestsctj}tj|}tj||_dSN)r Authenticator XmlStreamrCheckVersionInitializerinit)selfaxss V/var/www/html/env/lib/python3.11/site-packages/twisted/words/test/test_jabberclient.pysetUpz"CheckVersionInitializerTests.setUp#s6  # % %   # #2266 cZd|jj_|jdS)z3 Test supported version number 1.0 )rN)rr version initializers r testSupportedz*CheckVersionInitializerTests.testSupported(s+'- # rcd|jj_|tj|jj}|d|jdS)zK Test unsupported version number 0.0, and check exception. )rrzunsupported-versionN) rr r assertRaisesr StreamErrorr assertEqual condition)rexcs rtestNotSupportedz-CheckVersionInitializerTests.testNotSupported/sM'- # 1493GHH . >>>>>rN)__name__ __module__ __qualname__rr"r)rrrr"sA777 ?????rrceZdZdZdZdZdS)InitiatingInitializerHarnessa Testing harness for interacting with XML stream initializers. This sets up an L{utility.XmlPipe} to create a communication channel between the initializer and the stubbed receiving entity. It features a sink and source side that both act similarly to a real L{xmlstream.XmlStream}. The sink is augmented with an authenticator to which initializers can be added. The harness also provides some utility methods to work with event observers and deferreds. cg|_tj|_|jj|_t jd|_|j|j_dS)Nz example.org)outputr XmlPipepipesinkr ConnectAuthenticator authenticatorr!s rrz"InitiatingInitializerHarness.setUpEsI O%% &;MJJ'+'9$$$rctj}|||jj||j|S)a Observe an output event, returning a deferred. The returned deferred will be fired when the given event has been observed on the source end of the L{XmlPipe} tied to the protocol under test. The handler is added as the first callback. @param event: The event to be observed. See L{utility.EventDispatcher.addOnetimeObserver}. @param handler: The handler to be called with the observed event object. @rtype: L{defer.Deferred}. )rDeferred addCallbackr3sourceaddOnetimeObservercallback)reventhandlerds rwaitForz$InitiatingInitializerHarness.waitForLsE N   g ++E1:>>>rN)r*r+r,__doc__rr@r-rrr/r/8s<  :::rr/c:eZdZdZfdZdZdZdZdZxZ S)IQAuthInitializerTestsz0 Tests for L{client.IQAuthInitializer}. cttj|j|_t jd|j_d|j_ dS)Nuser@example.com/resourcesecret) superrrIQAuthInitializerr rr JIDr6passwordr __class__s rrzIQAuthInitializerTests.setUpdsO  ,T^<< !$)D!E!E&.###rcfd}fdt|}j}t j||gS)z Test plain-text authentication. Act as a server supporting plain-text authentication and expect the C{password} field to be filled with the password. Then act as if authentication succeeds. cltj|d}|d|jd|jd|jdt }jj||Sz Called when the initializer sent a query for authentication methods. The response informs the client that plain-text authentication is supported. resultzjabber:iq:authqueryusernamerJresource r toResponse addElementrRr@ IQ_AUTH_SETr3r:sendiqresponser? onAuthSetrs r onAuthGetz7IQAuthInitializerTests.testPlainText..onAuthGets!+B99H    ; < < < N % %j 1 1 1 N % %j 1 1 1 N % %j 1 1 1 [)44A I  ! !( + + +Hrc~dt|jjdt|jjdt|jjt j|d}jj |dS) Called when the initializer sent the authentication request. The server checks the credentials and responds with an empty result signalling success. userrFrTrPN) r&strrRrSrJrTr rVr3r:rYr[r\rs rr]z7IQAuthInitializerTests.testPlainText..onAuthSets   VS):%;%; < < <   Xs28+<'='= > > >   ZRX->)?)? @ @ @!+B99H I  ! !( + + + + +r)r@ IQ_AUTH_GETrr r gatherResultsrr^d1d2r]s` @r testPlainTextz$IQAuthInitializerTests.testPlainTextjsz      . , , , , , \\+y 1 1Y ! ! # #"B8,,,rcfd}fddj_t|}j}t j||gS)a" Test digest authentication. Act as a server supporting digest authentication and expect the C{digest} field to be filled with a sha1 digest of the concatenated stream session identifier and password. Then act as if authentication succeeds. cltj|d}|d|jd|jd|jdt }jj||S)z Called when the initializer sent a query for authentication methods. The response informs the client that digest authentication is supported. rPrQrSdigestrTrUrZs rr^z4IQAuthInitializerTests.testDigest..onAuthGets!+B99H    ; < < < N % %j 1 1 1 N % %h / / / N % %j 1 1 1 [)44A I  ! !( + + +Hrcdt|jjt dt|jjdt|jjtj |d}j j |dS)rarbs 12345secretrTrPN) r&rcrRrSr hexdigestrmrTr rVr3r:rYrds rr]z4IQAuthInitializerTests.testDigest..onAuthSets   VS):%;%; < < <   T.11;;==s28??S?S T T T   ZRX->)?)? @ @ @!+B99H I  ! !( + + + + +r12345)r sidr@rerr rrfrgs` @r testDigestz!IQAuthInitializerTests.testDigests      . , , , , , %\\+y 1 1Y ! ! # #"B8,,,rcfd}t|}j}|t jtj||gS)zT Test initializer failure of request for fields for authentication. ctjd|}jj|dS)z Called when the initializer sent a query for authentication methods. The server responds that the client is not authorized to authenticate. not-authorizedNr StanzaErrorrVr3r:rYrds rr^z?IQAuthInitializerTests.testFailRequestFields..onAuthGetsC ()9::EEbIIH I  ! !( + + + + +r r@rerr assertFailurerrwrrf)rr^rhris` rtestFailRequestFieldsz,IQAuthInitializerTests.testFailRequestFieldssv  , , , , ,\\+y 1 1Y ! ! # # 2u0111"B8,,,rcfd}fdt|}j}|t jtj||gS)z; Test initializer failure to authenticate. cltj|d}|d|jd|jd|jdt }jj||SrOrUrZs rr^z6IQAuthInitializerTests.testFailAuth..onAuthGetr_rctjd|}jj|dS)z Called when the initializer sent the authentication request. The server checks the credentials and responds with a not-authorized stanza error. ruNrvrds rr]z6IQAuthInitializerTests.testFailAuth..onAuthSetsC()9::EEbIIH I  ! !( + + + + +rrxrgs` @r testFailAuthz#IQAuthInitializerTests.testFailAuths       . , , , , ,\\+y 1 1Y ! ! # # 2u0111"B8,,,r) r*r+r,rArrjrrrzr~ __classcell__rLs@rrCrC_s///// 4-4-4-l9-9-9-v---2/-/-/-/-/-/-/-rrCc.eZdZdZfdZdZdZxZS)BindInitializerTestsz. Tests for L{client.BindInitializer}. cttj|j|_t jd|j_dS)NrE) rGrrBindInitializerr rr rIr6rKs rrzBindInitializerTests.setUp+sD  *4>:: !$)D!E!Ercfd}fd}t|}j}||t j||gS)zK Set up a stream, and act as if resource binding succeeds. ctj|d}|tdf|jddjj|dS)NrPbindr user@example.com/other resource)content)r rVrWNS_BINDrr3r:rYrds ronBindz.BindInitializerTests.testBasic..onBind5si +B99H   & 1 2 2 2 M $ $U4U $ V V V I  ! !( + + + + +rcltjdjjdS)Nr)r&r rIr6)rPrs rcbz*BindInitializerTests.testBasic..cb;s;   9::D.onBindJs@(44??CCH I  ! !( + + + + +r) r@rrrryrrwrrf)rrrhris` r testFailurez BindInitializerTests.testFailureEsp  , , , , ,\\+v . . Y__   2u0111"B8,,,r)r*r+r,rArrrrrs@rrr&seFFFFF ---* - - - - - - -rrc.eZdZdZfdZdZdZxZS)SessionInitializerTestsz1 Tests for L{client.SessionInitializer}. cttj|j|_dSr)rGrrSessionInitializerr rrKs rrzSessionInitializerTests.setUpYs,  -dn== rcfd}t|}j}t j||gS)zP Set up a stream, and act as if session establishment succeeds. cptj|d}jj|dS)NrP)r rVr3r:rYrds r onSessionz6SessionInitializerTests.testSuccess..onSessionbs4 +B99H I  ! !( + + + + +r)r@IQ_SESSION_SETrrrrfrrrhris` r testSuccessz#SessionInitializerTests.testSuccess]sW  , , , , ,\\.) 4 4 Y__  "B8,,,rcfd}t|}j}|t jtj||gS)zM Set up a stream, and act as if session establishment fails. ctjd|}jj|dS)N forbiddenrvrds rrz6SessionInitializerTests.testFailure..onSessionos@(55@@DDH I  ! !( + + + + +r) r@rrrryrrwrrfrs` rrz#SessionInitializerTests.testFailurejsp  , , , , ,\\.) 4 4 Y__   2u0111"B8,,,r)r*r+r,rArrrrrs@rrrTs`>>>>> - - - - - - - - - -rrceZdZdZdZdS)BasicAuthenticatorTestszB Test for both BasicAuthenticator and basicClientFactory. c$tjd|_tj|jdd}|d|jj||j|jj|d|jj |j \}}| |tj | |tj||jdS)a Authenticator and stream are properly constructed by the factory. The L{xmlstream.XmlStream} protocol created by the factory has the new L{client.BasicAuthenticator} instance in its C{authenticator} attribute. It is set up with C{jid} and C{password} as passed to the factory, C{otherHost} taken from the client JID. The stream futher has two initializers, for TLS and authentication, of which the first has its C{required} attribute set to C{True}. rErFN example.com)r rI client_jidrbasicClientFactory buildProtocolr&r6 otherHostrJ initializersassertIsInstancer TLSInitiatingInitializerrH assertFalserequired)rrtlsauths r test_basicz"BasicAuthenticatorTests.test_basic~s'"=>>  &t A A O OPT U U (8(BCCC "*:*>??? 2#3#<===O T c9#EFFF dF$<=== &&&&&rN)r*r+r,rArr-rrrrys-'''''rrc4eZdZdZdZeedZdS)XMPPAuthenticatorTestsz@ Test for both XMPPAuthenticator and XMPPClientFactory. c<tjd|_tj|jdd}|d|jj||j|jj|d|jj |j \}}}}}| |tj | |t| |tj| |tj||j||j||j||jdS)z Test basic operations. Setup an XMPPClientFactory, which sets up an XMPPAuthenticator, and let it produce a protocol instance. Then inspect the instance variables of the authenticator and XML stream objects. rErFNr)r rIrrXMPPClientFactoryrr&r6rrJrrr rr rr assertTruerr)rrrrsaslrsessions rrz!XMPPAuthenticatorTests.test_basicsa'"=>>  %dox @ @ N Nt T T (8(BCCC "*:*>??? 2#3#<===-/O)dD' c9#EFFF d$=>>> dF$:;;; gv'@AAA  %%%  &&&  &&& )*****rc g d fd }tjd|_tj}t j|jd|}|tj d|| d}|j \}}}}} | |tj | | ddS) zG A TLS configuration is passed to the TLS initializer. TNc2|dSr)append)rrrconfigurationForTLSconfigss rrz:XMPPAuthenticatorTests.test_tlsConfiguration..inits NN. / / / / /rrErF)r__init__r)TN)r rIrr CertificateOptionsrrpatchr rrrrassertIs) rrrfactoryrrrrrrrs @rtest_tlsConfigurationz,XMPPAuthenticatorTests.test_tlsConfigurations  0 0 0 0 0 0'"=>> "466* OX;N    95z4HHH  " "4 ( (-/O)dD' c9#EFFF )71:66666rN)r*r+r,rArr skipWhenNoSSLrr-rrrrsQ+++B V]77777rr)"rAhashlibrrrtwisted.internetr twisted.trialtwisted.words.protocols.jabberrrr r #twisted.words.protocols.jabber.saslr twisted.words.xishr r r ImportErrorrerXrr NS_SESSIONrTestCaserr/rCrrrrr-rrrsw """"""""""""HHHHHHHHHHHHIIIIII&&&&&& $$$$$$  MM 000 C/MMM0 @ ? ,2W< 2 8:E?????8#4???,$$$$$$$$ND-D-D-D-D-98;LD-D-D-N+-+-+-+-+-79J+-+-+-\"-"-"-"-"-:H