o ¼>h£@ã@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 Ú AuthenticatorÚ XmlStreamrÚCheckVersionInitializerÚinit)ÚselfÚaÚxs©rú„/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/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 ÚXmlPipeÚpipeÚsinkr ÚConnectAuthenticatorÚ 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}. )rÚDeferredÚ addCallbackr+ÚsourceÚaddOnetimeObserverÚcallback)rÚeventÚhandlerÚdrrrÚwaitForLs z$InitiatingInitializerHarness.waitForN)r%r&r'Ú__doc__rr7rrrrr(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Úusernamer?Úresource© r Ú toResponseÚ addElementrFr7Ú IQ_AUTH_SETr+r1Úsend©ÚiqÚresponser6©Ú 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. Úuserr;rHrDN) r!ÚstrrFrGr?rHr rJr+r1rM©rOrPrrrrRŠs  z7IQAuthInitializerTests.testPlainText..onAuthSet)r7Ú IQ_AUTH_GETrrrÚ gatherResults©rrSÚd1Úd2rrQrÚ 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. crB)zÀ Called when the initializer sent a query for authentication methods. The response informs the client that digest authentication is supported. rDrErGÚdigestrHrIrNrQrrrSªrTz4IQAuthInitializerTests.testDigest..onAuthGetcsbˆ dt|jjƒ¡ˆ tdƒ ¡t|jjƒ¡ˆ dt|jjƒ¡t  |d¡}ˆj j   |¡dS)rUrVs 12345secretrHrDN) r!rWrFrGrÚ hexdigestr_rHr rJr+r1rMrXrrrrRÁs  z4IQAuthInitializerTests.testDigest..onAuthSetÚ12345)r Úsidr7rYrrrrZr[rrQrÚ 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Ú StanzaErrorrJr+r1rMrXrrrrSàsz?IQAuthInitializerTests.testFailRequestFields..onAuthGet© r7rYrrÚ assertFailurerrhrrZr[rrrÚtestFailRequestFieldsÛs  z,IQAuthInitializerTests.testFailRequestFieldscsL‡‡fdd„}‡fdd„‰ˆ t|¡}ˆj ¡}ˆ |tj¡t ||g¡S)z; Test initializer failure to authenticate. crBrCrIrNrQrrrSùrTz6IQAuthInitializerTests.testFailAuth..onAuthGetcre)z Called when the initializer sent the authentication request. The server checks the credentials and responds with a not-authorized stanza error. rfNrgrXrrrrRsz6IQAuthInitializerTests.testFailAuth..onAuthSetrir[rrQrÚ testFailAuthôs   z#IQAuthInitializerTests.testFailAuth) r%r&r'r8rr^rcrkrlÚ __classcell__rrr@rr9_s 6;r9có0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚBindInitializerTestsz. Tests for L{client.BindInitializer}. cs*tƒ ¡t |j¡|_t d¡|j_dS)Nr:) r<rrÚBindInitializerr rrr>r.rr@rrr+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)NrDÚbindrúuser@example.com/other resource)Úcontent)r rJrKÚNS_BINDrqr+r1rMrXrrrÚonBind5s z.BindInitializerTests.testBasic..onBindcsˆ t d¡ˆjj¡dS)Nrr)r!rr>r.)rDrrrÚcb;sÿz*BindInitializerTests.testBasic..cb)r7Ú IQ_BIND_SETrÚstartr0rrZ)rrurvr\r]rrrÚ testBasic0s     zBindInitializerTests.testBasiccrd)zH Set up a stream, and act as if resource binding fails. có"t d¡ |¡}ˆjj |¡dS)NÚconflictrgrXrrrruJóz0BindInitializerTests.testFailure..onBind) r7rwrrxrjrrhrrZ)rrur\r]rrrÚ testFailureEó   z BindInitializerTests.testFailure)r%r&r'r8rryr}rmrrr@rro&s  rocrn)ÚSessionInitializerTestsz1 Tests for L{client.SessionInitializer}. cstƒ ¡t |j¡|_dSr)r<rrÚSessionInitializerr rrr@rrrYs zSessionInitializerTests.setUpcs0‡fdd„}ˆ t|¡}ˆj ¡}t ||g¡S)zP Set up a stream, and act as if session establishment succeeds. cst |d¡}ˆjj |¡dS)NrD)r rJr+r1rMrXrrrÚ onSessionbs z6SessionInitializerTests.testSuccess..onSession)r7ÚIQ_SESSION_SETrrxrrZ©rrr\r]rrrÚ testSuccess]s   z#SessionInitializerTests.testSuccesscrd)zM Set up a stream, and act as if session establishment fails. crz)NÚ forbiddenrgrXrrrror|z6SessionInitializerTests.testFailure..onSession) r7r‚rrxrjrrhrrZrƒrrrr}jr~z#SessionInitializerTests.testFailure)r%r&r'r8rr„r}rmrrr@rrTs   rc@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)rr>Ú client_jidrÚbasicClientFactoryÚ buildProtocolr!r.Ú otherHostr?Ú initializersÚassertIsInstancer ÚTLSInitiatingInitializerr=Ú assertFalseÚrequired)rrÚtlsÚauthrrrÚ test_basic~s  z"BasicAuthenticatorTests.test_basicN)r%r&r'r8r“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. r:r;Nr‡)rr>rˆrÚXMPPClientFactoryrŠr!r.r‹r?rŒrr rŽr rpr€Ú assertTruerr)rrrr‘ÚsaslrqÚ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©ÚconfigsrrrËsz:XMPPAuthenticatorTests.test_tlsConfiguration..initr:r;)ršÚ__init__r)TN)rr>rˆr ÚCertificateOptionsrr•Úpatchr rŽrŠrŒrÚassertIs) rrršÚfactoryrrr‘r—rqr˜rr›rÚtest_tlsConfigurationÄs ÿ z,XMPPAuthenticatorTests.test_tlsConfigurationN)r%r&r'r8r“rÚ skipWhenNoSSLr¢rrrrr”žs !r”)"r8ÚhashlibrrrÚtwisted.internetrÚ twisted.trialÚtwisted.words.protocols.jabberrrrr Ú#twisted.words.protocols.jabber.saslr Útwisted.words.xishr r Ú ImportErrorr£rYrLrtrwÚ NS_SESSIONr‚ÚTestCaserr(r9rorr†r”rrrrÚs:       þ'H.%%