U sÍ@gâAã@sLdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZddlmZdd lmZzdd lmZWnek r d Zd ZYnXd Zd ZdZdZdeZdZdeZGdd„de jƒZGdd„de ƒZ!Gdd„de!e jƒZ"Gdd„de!e jƒZ#Gdd„de!e jƒZ$Gdd„de jƒZ%Gdd „d e jƒZ&d S)!z4 Tests for L{twisted.words.protocols.jabber.client} é)Úabsolute_importÚdivision)Úsha1)Údefer)Úunicode)Úunittest)ÚclientÚerrorÚjidÚ xmlstream)ÚSASLInitiatingInitializer)Úutility)ÚsslNzSSL 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@s$eZdZdd„Zdd„Zdd„ZdS)ÚCheckVersionInitializerTestscCs"t ¡}t |¡}t |¡|_dS©N)r Z AuthenticatorZ XmlStreamrZCheckVersionInitializerÚinit)ÚselfÚaÚxs©rúF/usr/lib/python3/dist-packages/twisted/words/test/test_jabberclient.pyÚsetUp#s z"CheckVersionInitializerTests.setUpcCsd|jj_|j ¡dS)z3 Test supported version number 1.0 )érN)rr ÚversionÚ initialize©rrrrÚ testSupported)s z*CheckVersionInitializerTests.testSupportedcCs.d|jj_| tj|jj¡}| d|j¡dS)zK Test unsupported version number 0.0, and check exception. )rrzunsupported-versionN) rr rZ assertRaisesr Z StreamErrorrÚ assertEqualZ condition)rÚexcrrrÚtestNotSupported1s z-CheckVersionInitializerTests.testNotSupportedN)Ú__name__Ú __module__Ú __qualname__rrrrrrrr"src@s eZdZdZdd„Zdd„ZdS)Ú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. cCs4g|_t ¡|_|jj|_t d¡|_|j|j_dS)Nz example.org)Úoutputr ZXmlPipeÚpipeZsinkr ZConnectAuthenticatorÚ authenticatorrrrrrHs    z"InitiatingInitializerHarness.setUpcCs(t ¡}| |¡|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}. )rZDeferredÚ addCallbackr%ÚsourceZaddOnetimeObserverÚcallback)rZeventZhandlerÚdrrrÚwaitForPs  z$InitiatingInitializerHarness.waitForN)r r!r"Ú__doc__rr+rrrrr#;s r#cs@eZdZdZ‡fdd„Zdd„Zdd„Zdd „Zd d „Z‡Z S) ÚIQAuthInitializerTestsz0 Tests for L{client.IQAuthInitializer}. cs6tt|ƒ ¡t |j¡|_t d¡|j _d|j _ dS)Núuser@example.com/resourceÚsecret) Úsuperr-rrÚIQAuthInitializerr rr ÚJIDr&Úpasswordr©Ú __class__rrriszIQAuthInitializerTests.setUpcs>‡‡fdd„}‡fdd„‰ˆ t|¡}ˆj ¡}t ||g¡S)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. csXt |d¡}| d¡|j d¡|j d¡|j d¡ˆ tˆ¡}ˆjj |¡|S©zÄ Called when the initializer sent a query for authentication methods. The response informs the client that plain-text authentication is supported. Úresult©zjabber:iq:authÚqueryÚusernamer3Úresource© r Ú toResponseÚ addElementr9r+Ú IQ_AUTH_SETr%r(Úsend©ÚiqÚresponser*©Ú onAuthSetrrrÚ onAuthGetys      z7IQAuthInitializerTests.testPlainText..onAuthGetcsZˆ dt|jjƒ¡ˆ dt|jjƒ¡ˆ dt|jjƒ¡t |d¡}ˆjj   |¡dS)úÇ Called when the initializer sent the authentication request. The server checks the credentials and responds with an empty result signalling success. Úuserr/r;r7N) rrr9r:r3r;r r=r%r(r@©rBrCrrrrEs  z7IQAuthInitializerTests.testPlainText..onAuthSet)r+Ú IQ_AUTH_GETrrrÚ gatherResults©rrFÚd1Úd2rrDrÚ testPlainTextps    z$IQAuthInitializerTests.testPlainTextcsF‡‡fdd„}‡fdd„‰dˆj_ˆ t|¡}ˆj ¡}t ||g¡S)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. csXt |d¡}| d¡|j d¡|j d¡|j d¡ˆ tˆ¡}ˆjj |¡|S)zÀ Called when the initializer sent a query for authentication methods. The response informs the client that digest authentication is supported. r7r8r:Údigestr;r<rArDrrrF±s      z4IQAuthInitializerTests.testDigest..onAuthGetcsbˆ dt|jjƒ¡ˆ tdƒ ¡t|jjƒ¡ˆ dt|jjƒ¡t  |d¡}ˆj j   |¡dS)rGrHs 12345secretr;r7N) rrr9r:rZ hexdigestrPr;r r=r%r(r@rIrrrrEÈs ÿ z4IQAuthInitializerTests.testDigest..onAuthSetZ12345)r Zsidr+rJrrrrKrLrrDrÚ testDigest§s    z!IQAuthInitializerTests.testDigestcs>‡fdd„}ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)zT Test initializer failure of request for fields for authentication. cs"t d¡ |¡}ˆ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-authorizedN©r Ú StanzaErrorr=r%r(r@rIrrrrFèsz?IQAuthInitializerTests.testFailRequestFields..onAuthGet© r+rJrrÚ assertFailurer rTrrKrLrrrÚtestFailRequestFieldsäs   z,IQAuthInitializerTests.testFailRequestFieldscsL‡‡fdd„}‡fdd„‰ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)z; Test initializer failure to authenticate. csXt |d¡}| d¡|j d¡|j d¡|j d¡ˆ tˆ¡}ˆjj |¡|Sr6r<rArDrrrFs      z6IQAuthInitializerTests.testFailAuth..onAuthGetcs"t d¡ |¡}ˆjj |¡dS)z Called when the initializer sent the authentication request. The server checks the credentials and responds with a not-authorized stanza error. rRNrSrIrrrrEsz6IQAuthInitializerTests.testFailAuth..onAuthSetrUrLrrDrÚ testFailAuthýs   z#IQAuthInitializerTests.testFailAuth) r r!r"r,rrOrQrWrXÚ __classcell__rrr4rr-ds  7=r-cs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚBindInitializerTestsz. Tests for L{client.BindInitializer}. cs.tt|ƒ ¡t |j¡|_t d¡|j _dS)Nr.) r0rZrrÚBindInitializerr rr r2r&rr4rrr5szBindInitializerTests.setUpcsF‡fdd„}‡fdd„}ˆ t|¡}ˆj ¡}| |¡t ||g¡S)zK Set up a stream, and act as if resource binding succeeds. cs<t |d¡}| tdf¡|jjdddˆjj |¡dS)Nr7Úbindr úuser@example.com/other resource)Zcontent)r r=r>ÚNS_BINDr\r%r(r@rIrrrÚonBind?s  ÿz.BindInitializerTests.testBasic..onBindcsˆ t d¡ˆjj¡dS)Nr])rr r2r&)r7rrrÚcbFs ÿz*BindInitializerTests.testBasic..cb)r+Ú IQ_BIND_SETrÚstartr'rrK)rr_r`rMrNrrrÚ testBasic;s      zBindInitializerTests.testBasiccs>‡fdd„}ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)zH Set up a stream, and act as if resource binding fails. cs"t d¡ |¡}ˆjj |¡dS)NZconflictrSrIrrrr_Tsz0BindInitializerTests.testFailure..onBind) r+rarrbrVr rTrrK)rr_rMrNrrrÚ testFailurePs    z BindInitializerTests.testFailure)r r!r"r,rrcrdrYrrr4rrZ0s rZcs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚSessionInitializerTestsz1 Tests for L{client.SessionInitializer}. cs tt|ƒ ¡t |j¡|_dSr)r0rerrÚSessionInitializerr rrr4rrrdszSessionInitializerTests.setUpcs0‡fdd„}ˆ t|¡}ˆj ¡}t ||g¡S)zP Set up a stream, and act as if session establishment succeeds. cst |d¡}ˆjj |¡dS)Nr7)r r=r%r(r@rIrrrÚ onSessionns z6SessionInitializerTests.testSuccess..onSession)r+ÚIQ_SESSION_SETrrbrrK©rrgrMrNrrrÚ testSuccessis   z#SessionInitializerTests.testSuccesscs>‡fdd„}ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)zM Set up a stream, and act as if session establishment fails. cs"t d¡ |¡}ˆjj |¡dS)NZ forbiddenrSrIrrrrg{sz6SessionInitializerTests.testFailure..onSession) r+rhrrbrVr rTrrKrirrrrdws    z#SessionInitializerTests.testFailure)r r!r"r,rrjrdrYrrr4rre_s rec@seZdZdZdd„ZdS)ÚBasicAuthenticatorTestszB Test for both BasicAuthenticator and basicClientFactory. cCsˆt d¡|_t |jd¡ d¡}| d|jj¡| |j|jj¡| d|jj ¡|j \}}|  |t j ¡|  |tj¡| |j¡dS)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}. r.r/Nú example.com)r r2Ú client_jidrZbasicClientFactoryÚ buildProtocolrr&Ú otherHostr3Ú initializersÚassertIsInstancer ÚTLSInitiatingInitializerr1Ú assertFalseÚrequired)rrÚtlsZauthrrrÚ test_basic‹s ÿÿ z"BasicAuthenticatorTests.test_basicN)r r!r"r,rvrrrrrk†srkc@s&eZdZdZdd„Zdd„Zee_dS)ÚXMPPAuthenticatorTestsz@ Test for both XMPPAuthenticator and XMPPClientFactory. cCsÌt d¡|_t |jd¡ d¡}| d|jj¡| |j|jj¡| d|jj ¡|j \}}}}}|  |t j ¡|  |t¡|  |tj¡|  |tj¡| |j¡| |j¡| |j¡| |j¡dS)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. r.r/Nrl)r r2rmrÚXMPPClientFactoryrnrr&ror3rprqr rrr r[rfZ assertTruertrs)rrrruÚsaslr\Úsessionrrrrv²s$ ÿÿ    z!XMPPAuthenticatorTests.test_basicc s„g‰d ‡fdd„ }t d¡|_t ¡}tj|jd|d}| tj d|¡|  d¡}|j \}}}}} |  |tj ¡|  |ˆd ¡dS) zG A TLS configuration is passed to the TLS initializer. TNcsˆ |¡dSr)Úappend)rrrtÚconfigurationForTLS©ZconfigsrrrÛsz:XMPPAuthenticatorTests.test_tlsConfiguration..initr.r/)r|Ú__init__r)TN)r r2rmrZCertificateOptionsrrxZpatchr rrrnrprqZassertIs) rrr|Úfactoryrrruryr\rzrr}rÚtest_tlsConfigurationÕs þ z,XMPPAuthenticatorTests.test_tlsConfigurationN)r r!r"r,rvr€Ú skipWhenNoSSLÚskiprrrrrw­s#rw)'r,Z __future__rrZhashlibrZtwisted.internetrZtwisted.python.compatrZ twisted.trialrZtwisted.words.protocols.jabberrr r r Z#twisted.words.protocols.jabber.saslr Ztwisted.words.xishr rÚ ImportErrorrrJr?r^raZ NS_SESSIONrhZTestCaserÚobjectr#r-rZrerkrwrrrrÚs8       )M/''