o Óó·i£@ã@s8dZddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZzdd lmZWn eyGd Zd ZYnwd 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újC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-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 rÚ assertRaisesrÚ StreamErrorrÚ assertEqualÚ condition)rÚexcrrrÚtestNotSupported/s z-CheckVersionInitializerTests.testNotSupportedN)Ú__name__Ú __module__Ú __qualname__rrr!rrrrr"s rc@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}. N)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. cóXt |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ÚresourceN© r Ú toResponseÚ addElementr>r/Ú IQ_AUTH_SETr'r*Úsend©ÚiqÚresponser.©Ú onAuthSetrrrÚ onAuthGetsó     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 rBr'r*rE©rGrHrrrrJŠs  z7IQAuthInitializerTests.testPlainText..onAuthSetN)r/Ú IQ_AUTH_GETrrrÚ gatherResults©rrKÚd1Úd2rrIrÚ 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. cr:)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@NrArFrIrrrKªrLz4IQAuthInitializerTests.testDigest..onAuthGetcsbˆ dt|jjƒ¡ˆ tdƒ ¡t|jjƒ¡ˆ dt|jjƒ¡t  |d¡}ˆj j   |¡dS)rMrNs 12345secretr@r<N) rrOr>r?rÚ hexdigestrWr@r rBr'r*rErPrrrrJÁs  z4IQAuthInitializerTests.testDigest..onAuthSetZ12345N)r Zsidr/rQrrrrRrSrrIrÚ testDigest s    z!IQAuthInitializerTests.testDigestcó>‡fdd„}ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)zT Test initializer failure of request for fields for authentication. có"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Ú StanzaErrorrBr'r*rErPrrrrKàsz?IQAuthInitializerTests.testFailRequestFields..onAuthGetN© r/rQrrÚ assertFailurerr^rrRrSrrrÚtestFailRequestFieldsÛs  z,IQAuthInitializerTests.testFailRequestFieldscsL‡‡fdd„}‡fdd„‰ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)z; Test initializer failure to authenticate. cr:r;rArFrIrrrKùrLz6IQAuthInitializerTests.testFailAuth..onAuthGetcr[)z Called when the initializer sent the authentication request. The server checks the credentials and responds with a not-authorized stanza error. r\Nr]rPrrrrJsz6IQAuthInitializerTests.testFailAuth..onAuthSetNr_rSrrIrÚ testFailAuthôs   z#IQAuthInitializerTests.testFailAuth) r"r#r$r0rrVrYrarbÚ __classcell__rrr8rr1_s 6;r1có0eZdZdZ‡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 rBrCÚNS_BINDrgr'r*rErPrrrÚonBind5s z.BindInitializerTests.testBasic..onBindcsˆ t d¡ˆjj¡dS)Nrh)rrr6r()r<rrrÚcb;sÿz*BindInitializerTests.testBasic..cbN)r/Ú IQ_BIND_SETrÚstartr)rrR)rrkrlrTrUrrrÚ testBasic0s     zBindInitializerTests.testBasiccrZ)zH Set up a stream, and act as if resource binding fails. có"t d¡ |¡}ˆjj |¡dS)NÚconflictr]rPrrrrkJóz0BindInitializerTests.testFailure..onBindN) r/rmrrnr`rr^rrR)rrkrTrUrrrÚ testFailureEó   z BindInitializerTests.testFailure)r"r#r$r0rrorsrcrrr8rre&s  recrd)Ú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 rBr'r*rErPrrrÚ onSessionbs z6SessionInitializerTests.testSuccess..onSessionN)r/ÚIQ_SESSION_SETrrnrrR©rrwrTrUrrrÚ testSuccess]s   z#SessionInitializerTests.testSuccesscrZ)zM Set up a stream, and act as if session establishment fails. crp)NÚ forbiddenr]rPrrrrworrz6SessionInitializerTests.testFailure..onSessionN) r/rxrrnr`rr^rrRryrrrrsjrtz#SessionInitializerTests.testFailure)r"r#r$r0rrzrsrcrrr8rruTs   ruc@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ˆrrrrr|ys r|c@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. r2r3Nr})rr6r~rÚXMPPClientFactoryrrr(r€r7rr‚r rƒr rfrvÚ assertTruer…r„)rrrr†ÚsaslrgÚ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)rr6r~r ZCertificateOptionsrrŠÚpatchr rƒrrr‚ZassertIs) rrrÚfactoryrrr†rŒrgrrrrÚ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 Ú ImportErrorr•rQrDrjrmZ NS_SESSIONrxZTestCaserr%r1rerur|r‰rrrrÚs:       þ'H.%%