U n5{i£@ã@s:dZddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZzdd lmZWnek rd Zd ZYnXd Zd ZdZdZdeZdZdeZGdd„dejƒZGdd„dƒZGdd„deejƒZGdd„deejƒZGdd„deejƒZGdd„dejƒZ Gdd „d ejƒZ!d S)!z4 Tests for L{twisted.words.protocols.jabber.client} é)Úsha1)ÚskipIf)Údefer)Úunittest)ÚclientÚerrorÚjidÚ xmlstream)ÚSASLInitiatingInitializer)Úutility)ÚsslN)TzSSL not available)FÚz//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úH/tmp/pip-unpacked-wheel-mnhg7r0q/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 rÚ assertRaisesrÚ StreamErrorrÚ assertEqualÚ condition)rÚexcrrrÚtestNotSupported/s z-CheckVersionInitializerTests.testNotSupportedN)Ú__name__Ú __module__Ú __qualname__rrr!rrrrr"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Ú authenticatorrrrrrEs    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)rÚeventÚhandlerÚdrrrÚwaitForLs  z$InitiatingInitializerHarness.waitForN)r"r#r$Ú__doc__rr/rrrrr%8s r%cs@eZdZdZ‡fdd„Zdd„Zdd„Zdd „Zd d „Z‡Z S) ÚIQAuthInitializerTestsz0 Tests for L{client.IQAuthInitializer}. cs2tƒ ¡t |j¡|_t d¡|j_d|j_ dS)Núuser@example.com/resourceÚsecret) ÚsuperrrÚIQAuthInitializerr rrÚJIDr(Úpasswordr©Ú __class__rrrds zIQAuthInitializerTests.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Úusernamer7Úresource© r Ú toResponseÚ addElementr=r/Ú IQ_AUTH_SETr'r*Úsend©ÚiqÚresponser.©Ú onAuthSetrrrÚ onAuthGetss      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. Úuserr3r?r;N) rÚstrr=r>r7r?r rAr'r*rD©rFrGrrrrIŠs  z7IQAuthInitializerTests.testPlainText..onAuthSet)r/Ú IQ_AUTH_GETrrrÚ gatherResults©rrJÚd1Úd2rrHrÚ testPlainTextjs    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. r;r<r>Údigestr?r@rErHrrrJªs      z4IQAuthInitializerTests.testDigest..onAuthGetcsbˆ dt|jjƒ¡ˆ tdƒ ¡t|jjƒ¡ˆ dt|jjƒ¡t  |d¡}ˆj j   |¡dS)rKrLs 12345secretr?r;N) rrMr=r>rÚ hexdigestrUr?r rAr'r*rDrNrrrrIÁs  z4IQAuthInitializerTests.testDigest..onAuthSetZ12345)r Zsidr/rOrrrrPrQrrHrÚ 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Ú StanzaErrorrAr'r*rDrNrrrrJàsz?IQAuthInitializerTests.testFailRequestFields..onAuthGet© r/rOrrÚ assertFailurerrZrrPrQrrrÚ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 |¡|Sr:r@rErHrrrJùs      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. rXNrYrNrrrrIsz6IQAuthInitializerTests.testFailAuth..onAuthSetr[rQrrHrÚ testFailAuthôs   z#IQAuthInitializerTests.testFailAuth) r"r#r$r0rrTrWr]r^Ú __classcell__rrr8rr1_s  6;r1cs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚBindInitializerTestsz. Tests for L{client.BindInitializer}. cs*tƒ ¡t |j¡|_t d¡|j_dS)Nr2) r4rrÚBindInitializerr rrr6r(rr8rrr+s zBindInitializerTests.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)Nr;Úbindrúuser@example.com/other resource)Úcontent)r rArBÚNS_BINDrbr'r*rDrNrrrÚonBind5s z.BindInitializerTests.testBasic..onBindcsˆ t d¡ˆjj¡dS)Nrc)rrr6r()r;rrrÚcb;sÿz*BindInitializerTests.testBasic..cb)r/Ú IQ_BIND_SETrÚstartr)rrP)rrfrgrRrSrrrÚ testBasic0s      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)NÚconflictrYrNrrrrfJsz0BindInitializerTests.testFailure..onBind) r/rhrrir\rrZrrP)rrfrRrSrrrÚ testFailureEs    z BindInitializerTests.testFailure)r"r#r$r0rrjrlr_rrr8rr`&s r`cs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚSessionInitializerTestsz1 Tests for L{client.SessionInitializer}. cstƒ ¡t |j¡|_dSr)r4rrÚSessionInitializerr rrr8rrrYs zSessionInitializerTests.setUpcs0‡fdd„}ˆ t|¡}ˆj ¡}t ||g¡S)zP Set up a stream, and act as if session establishment succeeds. cst |d¡}ˆjj |¡dS)Nr;)r rAr'r*rDrNrrrÚ onSessionbs z6SessionInitializerTests.testSuccess..onSession)r/ÚIQ_SESSION_SETrrirrP©rrorRrSrrrÚ testSuccess]s   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)NÚ forbiddenrYrNrrrroosz6SessionInitializerTests.testFailure..onSession) r/rprrir\rrZrrPrqrrrrljs    z#SessionInitializerTests.testFailure)r"r#r$r0rrrrlr_rrr8rrmTs  rmc@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}. r2r3Nú example.com)rr6Ú client_jidrZbasicClientFactoryÚ buildProtocolrr(Ú otherHostr7Ú initializersÚassertIsInstancer ÚTLSInitiatingInitializerr5Ú assertFalseÚrequired)rrÚtlsÚauthrrrÚ test_basic~s  z"BasicAuthenticatorTests.test_basicN)r"r#r$r0r€rrrrrtysrtc@s(eZdZdZdd„ZeeŽdd„ƒZdS)Ú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. r2r3Nru)rr6rvrÚXMPPClientFactoryrwrr(rxr7ryrzr r{r rarnÚ assertTruer}r|)rrrr~ÚsaslrbÚsessionrrrr€£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)rrr}ÚconfigurationForTLS©ZconfigsrrrËsz:XMPPAuthenticatorTests.test_tlsConfiguration..initr2r3)r‡Ú__init__r)TN)rr6rvr ZCertificateOptionsrr‚Úpatchr r{rwryrzZassertIs) rrr‡Úfactoryrrr~r„rbr…rrˆrÚtest_tlsConfigurationÄs ÿ z,XMPPAuthenticatorTests.test_tlsConfigurationN)r"r#r$r0r€rÚ skipWhenNoSSLrŒrrrrržs!r)"r0ÚhashlibrrrZtwisted.internetrZ twisted.trialZtwisted.words.protocols.jabberrrrr Z#twisted.words.protocols.jabber.saslr Ztwisted.words.xishr r Ú ImportErrorrrOrCrerhZ NS_SESSIONrpZTestCaserr%r1r`rmrtrrrrrÚs6       'H.%%