o b@sUdZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl m Z mZddlZddlZddlZddlZddlmZddlmZmZddlmZdd lmZdd lmZmZdd lmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.ddl/m0Z0e e1e2d<z ddl3m4Z4m5Z5Wn e6ydZ7YnwdZ7ej89e!sdZ7ddZ:ee$j;GdddZdZ?Gdd d Z@Gd!d"d"e@e$jAZBGd#d$d$e@e$jCZDGd%d&d&ZEGd'd(d(e(jFZGGd)d*d*e0eEZHGd+d,d,ZIGd-d.d.ZJGd/d0d0eJe$jKZLGd1d2d2eJe$jMZNGd3d4d4ZOGd5d6d6eOe0ZPGd7d8d8eOe0ZQeejRjGd9d:d:ZSee$jTGd;d<d<ZUGd=d>d>ZVGd?d@d@e0eEZWGdAdBdBe0ZXGdCdDdDeDZYGdEdFdFe0eEZZGdGdHdHe0Z[GdIdJdJe0eEZ\GdKdLdLe j]Z^GdMdNdNe0Z_GdOdPdPe0Z`eeGdQdRdRZaGdSdTdTZbGdUdVdVe0ZcGdWdXdXe0ZdGdYdZdZe0ZeGd[d\d\e0ZfGd]d^d^e0ZgGd_d`d`e0ZhGdadbdbe.ZiGdcdddde0ZjdS)ez* Test cases for twisted.mail.smtp module. N)BytesIO)AnyListOptionalTupleType)directlyProvides implementer)cred)AllowAnonymousAccessICredentialsChecker) IAnonymous)UnauthorizedLogin)IRealmPortal)addressdefererror interfacesprotocolreactortask)smtp)LOGINCredentials)basicloopback)LineLog) MemoryReactorStringTransport)TestCasesslSkip)ClientTLSContextServerTLSContextzOpenSSL not presentzReactor doesn't support SSLcOdSN)spameggsr%r%=/usr/lib/python3/dist-packages/twisted/mail/test/test_smtp.py spameater/r)c@0eZdZdZddZddZddZdd Zd S) BrokenMessagez L{BrokenMessage} is an L{IMessage} which raises an unexpected exception from its C{eomReceived} method. This is useful for creating a server which can be used to test client retry behavior. cCr#r$r%selfuserr%r%r(__init__;r*zBrokenMessage.__init__cCr#r$r%r.liner%r%r( lineReceived>r*zBrokenMessage.lineReceivedcCtd)Nz"Some problem, delivery is failing.) RuntimeErrorr.r%r%r( eomReceivedAzBrokenMessage.eomReceivedcCr#r$r%r6r%r%r(connectionLostDr*zBrokenMessage.connectionLostN)__name__ __module__ __qualname____doc__r0r3r7r9r%r%r%r(r,3s  r,c@(eZdZdZddZddZddZdS) DummyMessagez L{BrokenMessage} is an L{IMessage} which saves the message delivered to it to its domain object. @ivar domain: A L{DummyDomain} which will be used to store the message once it is received. cC||_||_g|_dSr$)domainr/buffer)r.rAr/r%r%r(r0Q zDummyMessage.__init__cCs td|s|j|dSdS)Ns/Received: From yyy.com \(\[.*\]\) by localhost;)rematchrBappendr1r%r%r(r3Vs zDummyMessage.lineReceivedcCs>d|jd}|jj|jjj|t }| d|S)N saved) joinrBrAmessagesr/destlocalrFrDeferredcallback)r.messagedeferredr%r%r(r7[s  zDummyMessage.eomReceivedN)r:r;r<r=r0r3r7r%r%r%r(r?Hs  r?c@ eZdZdZddZddZdS) DummyDomainzf L{DummyDomain} is an L{IDomain} which keeps track of messages delivered to it in memory. cCsi|_|D]}g|j|<qdSr$)rJ)r.namesnamer%r%r(r0is zDummyDomain.__init__cs2jjjvrtfddSttS)Ncs tSr$r?r%r-r%r(p z$DummyDomain.exists..)rKrLrJrsucceedfailr SMTPBadRcptr-r%r-r(existsnszDummyDomain.existsN)r:r;r<r=r0r[r%r%r%r(rRcs rRsSubject: hello Goodbye c@s>eZdZdddZddZddZdd Zd d Zd d ZdS)MyClientNcCs:|dur ddgttf}|d|_|d|_|d|_dS)Nzmoshez@foo.barr)rmail_sender _recipient_datar. messageInfor%r%r(r0|s   zMyClient.__init__cC|jSr$)r`r6r%r%r( getMailFromzMyClient.getMailFromcCrer$)rar6r%r%r( getMailTorgzMyClient.getMailTocCrer$)rbr6r%r%r( getMailDatargzMyClient.getMailDatacC ||_dSr$)_error)r.excr%r%r( sendError zMyClient.sendErrorcCsd|_d|_d|_dSr$)r`rarb)r.coderespnumOk addresseslogr%r%r(sentMails zMyClient.sentMailr$) r:r;r<r0rfrhrirmrtr%r%r%r(r\{s  r\c@seZdZdddZdS) MySMTPClientNcCstj|dt||dSNsfoo.baz)r SMTPClientr0r\rcr%r%r(r0szMySMTPClient.__init__r$r:r;r<r0r%r%r%r(ruruc@seZdZdddZdS) MyESMTPClientNcCs tj|||dt|dSrv)r ESMTPClientr0r\)r.secretcontextFactoryr%r%r(r0szMyESMTPClient.__init__)r{Nrxr%r%r%r(rzryrzc@eZdZddZdS) LoopbackMixincCs t||Sr$)r loopbackTCP)r.serverclientr%r%r(r zLoopbackMixin.loopbackN)r:r;r<rr%r%r%r(r rc@s$eZdZgdZddZddZdS)FakeSMTPServer) 220 hellos250 nice to meet you 250 greatrs354 go on, ladcCs4g|_|jdd|_|j||jdSr$)rB clientDatareversesendLinepopr6r%r%r(connectionMades zFakeSMTPServer.connectionMadecCsr|j||dkr|jd|jn|dkr!|jdn |dkr*|j|jr7||jdSdS)NsQUITs221 see ya around .s 250 gotcha RSET)rBrF transportwriteloseConnectionrrrr1r%r%r(r3s    zFakeSMTPServer.lineReceivedN)r:r;r<rrr3r%r%r%r(rs rc@sHeZdZdZddZgdZddZddZd d Zd d Z d dZ dS)SMTPClientTestsz' Tests for L{smtp.SMTPClient}. cCsjg}t}|j|_|t|d|||dtj | |dj | t |dddS)z L{smtp.SMTPClient.timeoutConnection} calls the C{sendError} hook with a fatal L{SMTPTimeoutError} with the current line log. rrsHTimeout waiting for SMTP server response <<< 220 hello >>> HELO foo.baz N)rurFrmmakeConnectionrr3timeoutConnectionassertIsInstancerSMTPTimeoutError assertTrueisFatal assertEqualbytes)r.errorsrr%r%r(test_timeoutConnections   z&SMTPClientTests.test_timeoutConnection) s HELO foo.bazsMAIL FROM:sRCPT TO:sDATAsSubject: hellor{sGoodbyerrcs0t}t|}|fdd|S)z L{smtp.SMTPClient} sends I{HELO}, I{MAIL FROM}, I{RCPT TO}, and I{DATA} commands based on the return values of its C{getMailFrom}, C{getMailTo}, and C{getMailData} methods. csjjSr$)rrBexpected_outputxr.rr%r(rVsz/SMTPClientTests.test_messages..)rurr addCallback)r.rdr%rr( test_messagess  zSMTPClientTests.test_messagescCsdtddgtdf}t}|||d||jd||j|j | |j t dS)z If there is an error while producing the message body to the connection, the C{sendError} callback is invoked. zalice@example.comzbob@example.comfoo(220 Ok 250 Ok 250 Ok 250 Ok 354 Ok N) rurrr dataReceivedassertNotIdenticalproducer assertFalse streaming stopProducingrrk Exceptionr.rrr%r%r(test_transferErrors   z"SMTPClientTests.test_transferErrorcCPtd}t}|||tjdddd||d||j dS)z If L{smtp.SMTPClient.sendError} is called with an L{SMTPClientError} which is fatal, it disconnects its transport without writing anything more to it. N{fooTrr{) rrwrrrmSMTPClientErrorrvaluer disconnectingrr%r%r(test_sendFatalError  z#SMTPClientTests.test_sendFatalErrorcCr)z If L{smtp.SMTPClient.sendError} is called with an L{SMTPClientError} which is not fatal, it sends C{"QUIT"} and waits for the server to close the connection. NrrFrQUIT ) rrwrrrmrrrrrrr%r%r(test_sendNonFatalErrorrz&SMTPClientTests.test_sendNonFatalErrorcCsHtd}t}|||td||d||j dS)z If L{smtp.SMTPClient.sendError} is called with an exception which is not an L{SMTPClientError}, it disconnects its transport without writing anything more to it. Nrr{) rrwrrrmrrrrrrr%r%r(test_sendOtherError,s  z#SMTPClientTests.test_sendOtherErrorN) r:r;r<r=rrrrrrrr%r%r%r(rs   rc@$eZdZddZddZddZdS)DummySMTPMessagecCr@r$)rusersrB)r.rrr%r%r(r0;rCzDummySMTPMessage.__init__cCs|j|dSr$)rBrFr1r%r%r(r3@szDummySMTPMessage.lineReceivedcCsrd|jd}|jdjdt|jdj}}g}|jD] }|t|q||||f|jjt |<t dS)NrGrrH) rIrBrhelororigrFrrOtuplerrX)r.rOrorigin recipientsr/r%r%r(r7Cs"  zDummySMTPMessage.eomReceivedN)r:r;r<r0r3r7r%r%r%r(r:s rc@s,eZdZddZddZddZddZd S) DummyProtocCs|j|i|_dSr$)dummyMixinBaserrOr6r%r%r(rNs  zDummyProto.connectionMadecGr#r$r%)r&r%r%r(receivedHeaderRr*zDummyProto.receivedHeadercstd_fddS)Ncs tgSr$)rr%r-r%r(rVW z'DummyProto.validateTo..)SimpleDeliverydeliveryr-r%r-r( validateToUs zDummyProto.validateTocC|Sr$r%r.rrr%r%r( validateFromYr*zDummyProto.validateFromN)r:r;r<rrrrr%r%r%r(rMs  rc@eZdZejZdS) DummySMTPN)r:r;r<rSMTPrr%r%r%r(r] rc@r) DummyESMTPN)r:r;r<rESMTPrr%r%r%r(rarrc @seZdZUdZeeejed<dZ eee j ed<dddgddgdfddd d gdd d gd fdd d gddgdfgZ gdZ eeeeeefed<e D]6\ZZZZZZe deddddfeD]Ze deddddfqce ddedeeeefffqLddZdS)AnotherTestCaseN serverClass clientClassfoo.comsmoshez@foo.comsmoshez@bar.coms(From: Moshe To: Moshe Hi, how are you? s tttt@rrr.comsuuu@ooosyyy@eeesSubject: pass ..rrrr.. s@this,@is,@ignored:foo@bar.coms@ignore,@this,@too:bar@foo.coms foo@bar.coms bar@foo.coms Subject: apa To: foo 123 . 456 ))r{s220.* $NN)sHELO foo.com 250.* $NN)sRSET rNNdatas MAIL FROM: s250.* s RCPT TO:.fooFactoryrN)r:r;r<rAr%r%r%r( fooFactorysrNs354rr]r s. r^)rrfactoryrrrrclearrDrEAssertionError splitlineslistrrOr setTimeout)r.rarsendexpectmsg msgexpectrr2rpmsgdatarecipexpectedr%r%r( test_buffers:          zAnotherTestCase.test_buffer)r:r;r<rrrrProtocol__annotations__rrrwrJrrrrrhelo_from_to_realfromrealtorrFrcptrr%r%r%r(resL   , rc@eZdZeZeZdS)AnotherESMTPTestsN)r:r;r<rrrzrr%r%r%r(rrc@r)AnotherSMTPTestsN)r:r;r<rrrurr%r%r%r(rrrc@s4eZdZddiZejjejjfZddZ ddZ dS) DummyCheckertestuser testpasswordcCs"t|j|j|j|j|jSr$)r maybeDeferred checkPasswordrusernamer_cbCheckr. credentialsr%r%r(requestAvatarIds  zDummyChecker.requestAvatarIdcCs|r|Stjr$)r rr)r.resultrr%r%r(rs zDummyChecker._cbCheckN) r:r;r<rr rIUsernamePasswordIUsernameHashedPasswordcredentialInterfacesrrr%r%r%r(rs rc@r+) rz[ L{SimpleDelivery} is a message delivery factory with no interesting behavior. cCrjr$_messageFactory)r.messageFactoryr%r%r(r0rnzSimpleDelivery.__init__cCr#r$r%r.rrrr%r%r(rr*zSimpleDelivery.receivedHeadercCrr$r%rr%r%r(rr*zSimpleDelivery.validateFromcsfddS)Ncs Sr$rr%r-r%r(rVrWz+SimpleDelivery.validateTo..r%r-r%r-r(rszSimpleDelivery.validateToN)r:r;r<r=r0rrrr%r%r%r(rs  rc@r) DummyRealmcGstjtdddfS)NcSr#r$r%r%r%r%r(rVz*DummyRealm.requestAvatar..)rIMessageDeliveryr)r.avatarIdmindrr%r%r( requestAvatarzDummyRealm.requestAvatarN)r:r;r<rr%r%r%r(r rr c@r) AuthTestscstt}tj|}|ttdtjji|_t d}t d}| | |}|fdd|S)z L{ESMTPClient} can authenticate using the I{CRAM-MD5} SASL mechanism. @see: U{http://tools.ietf.org/html/rfc2195} sCRAM-MD5rrcsjdS)Nr])r authenticatedrrr%r(rVsz,AuthTests.test_crammd5Auth..)r r portalrregisterCheckerrrrCramMD5CredentialsrzrCramMD5ClientAuthenticatorregisterAuthenticatorrrr.realmprcAuthrr%rr(test_crammd5Auth s     zAuthTests.test_crammd5Authcspt}tj|}|ttdti|_td}t d}| | |}| fdd|S)z L{ESMTPClient} can authenticate using the I{LOGIN} SASL mechanism. @see: U{http://sepp.oetiker.ch/sasl-2.1.19-ds/draft-murchison-sasl-login-00.txt} LOGINrrc jSr$rrrrr%r(rV1rz*AuthTests.test_loginAuth..)r r rrrrrrrzrLOGINAuthenticatorrrrrr%rr(test_loginAuths      zAuthTests.test_loginAuthcsrt}tj|}|ttdtji|_t d}t d}| | |}| fdd|S)a When communicating with a server which implements the I{LOGIN} SASL mechanism using C{"Username:"} as the challenge (rather than C{"User Name\0"}), L{ESMTPClient} can still authenticate successfully using the I{LOGIN} mechanism. rrrcrr$rrrr%r(rVGrz8AuthTests.test_loginAgainstWeirdServer..)r r rrrrrrrrzr rrrrr%rr(test_loginAgainstWeirdServer4s     z&AuthTests.test_loginAgainstWeirdServerN)r:r;r<rr!r"r%r%r%r(r s rc@s<eZdZddZddZddZddZd d Zd d Zd S)SMTPHelperTestscCs6i}tdD]}td}|||vd||<qdS)Nitestcase)ranger messageidr)r.rimr%r%r( testMessageIDLs    zSMTPHelperTests.testMessageIDcCsRddgddgtddgddgtddgg}|D] \}}|t||qdS)Nsuser@host.namess"User Name" ssomeguy@someplacesr{s<>)rAddressr quoteaddr)r.casescer%r%r( testQuoteAddrSs   zSMTPHelperTests.testQuoteAddrcCs$tdddd}|t|ddS)Ns user@hostshelo.host.namez user@host)rUserrstr)r.ur%r%r(testUser_szSMTPHelperTests.testUsercCsrgd}|D]0\}}|t||t|f||d||t||t|f||d|qdS)N)) Hello worlds Hello+20world)z Hello+worlds Hello+2Bworld)zs+00+01+02+03+04+05)ze=mc2@example.comse+3Dmc2@example.comxtext)rr xtext_encodelenencode xtext_decodedecode)r.r,caserr%r%r(testXtextEncodingcs z!SMTPHelperTests.testXtextEncodingcCDd}|t|d|dt|f||dd|ddS)z Specifying an error policy to C{unicode.encode} with the I{xtext} codec should produce the same result as not specifying the error policy. r4strictr5N)rrr6r8r7)r.textr%r%r(test_encodeWithErrorsqs z%SMTPHelperTests.test_encodeWithErrorscCr=)zO Similar to L{test_encodeWithErrors}, but for C{bytes.decode}. s Hello worldr>r5N)rrr9r:r7)r.rr%r%r(test_decodeWithErrors}s z%SMTPHelperTests.test_decodeWithErrorsN) r:r;r<r)r/r3r<r@rAr%r%r%r(r#Ks  r#c@eZdZdZddZdS)NoticeTLSClientFcCst|||d|_dSNT)rzesmtpState_starttlstls)r.rorpr%r%r(rEs z#NoticeTLSClient.esmtpState_starttlsN)r:r;r<rFrEr%r%r%r(rC rCc@seZdZeZddZdS)TLSTestscsBt}t}t|dt|dfdd}|S)N)r~cs jdjddSrD)rrF startedTLSignoredrr.rr%r(checkszTLSTests.testTLS..check)r!r"rCrrr)r. clientCTX serverCTXrMr%rLr(testTLSs   zTLSTests.testTLSN)r:r;r<r skiprPr%r%r%r(rHrGrHc@r)EmptyLineTestscCspt}t}|||d|d|}|t |d| |d d||dddS)z If L{smtp.SMTP} receives an empty line, it responds with a 500 error response code and a message about a syntax error. r{Nr^rs220r]s500 Error: bad syntax) rrrrr3rrrrr7r startswith)r.protoroutr%r%r(test_emptyLineSyntaxErrors    z(EmptyLineTests.test_emptyLineSyntaxErrorN)r:r;r<rVr%r%r%r(rRrrRc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) TimeoutTestszE Check that SMTP client factories correctly use the timeout. cstt|tddd}j|_t}||||_fdd} |t j  |} d d|S)zR Connect the clientFactory, and check the timeout on the request. TCP example.netcsddS)N?)rseconds)ignclockr.r%r(rMrz(TimeoutTests._timeoutTest..checkg?g?)rClock buildProtocolr IPv4Address callLaterrrr assertFailurerrradvance)r.onDone clientFactoryrtrMrr%r^r( _timeoutTests    zTimeoutTests._timeoutTestcC.t}tjddtd|ddd}|||S)z Test timeout for L{smtp.SMTPSenderFactory}: the response L{Deferred} should be errback with a L{smtp.SMTPTimeoutError}. source@addresssrecipient@address Message bodyrr[retriestimeoutrrMrSMTPSenderFactoryrrir.rfrgr%r%r(test_SMTPClientRecipientBytess z*TimeoutTests.test_SMTPClientRecipientBytescCrj)z- Use a L{unicode} recipient. rkrecipient@addressrlrr[rmrprrr%r%r(test_SMTPClientRecipientUnicode z,TimeoutTests.test_SMTPClientRecipientUnicodecCrj)z. Use a L{list} of recipients. rk)zrecipient1@addresssrecipient2@addressrlrr[rmrprrr%r%r(test_SMTPClientRecipientListrvz)TimeoutTests.test_SMTPClientRecipientListc Cs2t}tjddddtd|ddd}|||S) z Test timeout for L{smtp.ESMTPSenderFactory}: the response L{Deferred} should be errback with a L{smtp.SMTPTimeoutError}. rpasswordrkrtrlrr[rm)rrMrESMTPSenderFactoryrrirrr%r%r(test_ESMTPClients zTimeoutTests.test_ESMTPClientcCsGddd}g}t}||jtjdd|d|ddd}d |_t}| t d d d }|j |_ t }|||d ||jd||j|d||g|j|d||g|j|d||g||ddS)z The timeout is not allowed to expire after the server has accepted a DATA command and the client is actively sending data to it. c@s"eZdZdZddZdddZdS)z.SlowFilez A file-like which returns one byte from each read call until the specified number of bytes have been returned. cSrjr$_size)r.sizer%r%r(r0rnzETimeoutTests.test_resetTimeoutWhileSending..SlowFile.__init__NcSs|jr |jd8_dSdS)Nr]xr{r{)r.maxr%r%r(readszATimeoutTests.test_resetTimeoutWhileSending..SlowFile.readr$)r:r;r<r=r0rr%r%r%r(SlowFilesrrkrtr]rrrm example.orgrXrYrZrNr^s250 Ok s]HELO example.org MAIL FROM: RCPT TO: DATA x . RSET )rrM addErrbackrFrrqrArr`rarrbrcrrrrrrrrerresumeProducingr)r.rfailedrfrgr_rrr%r%r(test_resetTimeoutWhileSendingsJ            z*TimeoutTests.test_resetTimeoutWhileSendingN) r:r;r<r=rirsrurwrzrr%r%r%r(rWs rWc@rQ)(MultipleDeliveryFactorySMTPServerFactorya L{MultipleDeliveryFactorySMTPServerFactory} creates SMTP server protocol instances with message delivery factory objects supplied to it. Each factory is used for one connection and then discarded. Factories are used in the order they are supplied. cCrjr$)_messageFactories)r.messageFactoriesr%r%r(r0qrnz1MultipleDeliveryFactorySMTPServerFactory.__init__cCs$tj||}t|jd|_|S)Nr)r ServerFactoryrarrrr)r.addrrr%r%r(ratsz6MultipleDeliveryFactorySMTPServerFactory.buildProtocolN)r:r;r<r=r0rar%r%r%r(ris rc@rQ)SMTPSenderFactoryTestsz. Tests for L{smtp.SMTPSenderFactory}. cC^t}t}tddtd|}|dd|}|d||t d| |j ddS)zt L{smtp.SMTPSenderFactory} removes the current protocol when the client connection is lost. rkrtmessage localhostrZNBye.) rrrMrrqr connectTCPraclientConnectionLostrConnectionDonercurrentProtocolr.r sentDeferredrg connectorr%r%r(2test_removeCurrentProtocolWhenClientConnectionLost  zISMTPSenderFactoryTests.test_removeCurrentProtocolWhenClientConnectionLostcCr)zv L{smtp.SMTPSenderFactory} removes the current protocol when the client connection is failed. rkrtrrrZNr) rrrMrrqrrraclientConnectionFailedrrrrrr%r%r(4test_removeCurrentProtocolWhenClientConnectionFailedrzKSMTPSenderFactoryTests.test_removeCurrentProtocolWhenClientConnectionFailedN)r:r;r<r=rrr%r%r%r(rzs rc@rB)SMTPSenderFactoryRetryTestszD Tests for the retry behavior of L{smtp.SMTPSenderFactory}. c sddtgGfdddtj}ttfddg}||_tjd|dd }|} |j t }t d d t|}d |_t|j|j|} |jfd d}|||S)z If the protocol created by L{SMTPSenderFactory} loses its connection before receiving confirmation of message delivery, it reconnects and tries to deliver the message again. salicessome message textcseZdZdZfddZdS)zHSMTPSenderFactoryRetryTests.test_retryAfterDisconnect..CleanSMTPz An SMTP subclass which ensures that its transport will be disconnected before the test ends. cs|jtj||dSr$) addCleanuprrrr) innerSelfrr6r%r(rs zWSMTPSenderFactoryRetryTests.test_retryAfterDisconnect..CleanSMTP.makeConnectionN)r:r;r<r=rr%r6r%r( CleanSMTPsrcs t|Sr$rU)r/)rAr%r(rVrWzGSMTPSenderFactoryRetryTests.test_retryAfterDisconnect..r 127.0.0.1) interfacesbob@example.orgs @example.comrcs6jddgittddS)z Verify that the message was successfully delivered and flush the error which caused the first attempt to fail. rGr]N)rrJr7flushLoggedErrorsr5rJrArO recipientr.r%r(cbSentszESMTPSenderFactoryRetryTests.test_retryAfterDisconnect..cbSent)rRrrrr,rr listenTCPgetHostr stopListeningrrMrqrrArhostport disconnectr) r.r serverFactory serverPort serverHostrrgclientConnectorrr%rr(test_retryAfterDisconnects4      z5SMTPSenderFactoryRetryTests.test_retryAfterDisconnectN)r:r;r<r=rr%r%r%r(r rc@rQ)SingletonRealmz Trivial realm implementation which is constructed with an interface and an avatar and returns that avatar when asked for that interface. cCs||_||_dSr$ravatar)r.rrr%r%r(r0s zSingletonRealm.__init__cGs,|D]}||jur||jddfSqdS)NcSr#r$r%r%r%r%r(rVr z.SingletonRealm.requestAvatar..r)r.r rrifacer%r%r(rs  zSingletonRealm.requestAvatarN)r:r;r<r=r0rr%r%r%r(rs rc@r>) NotImplementedDeliveryz Non-implementation of L{smtp.IMessageDelivery} which only has methods which raise L{NotImplementedError}. Subclassed by various tests to provide the particular behavior being tested. cCr4Nz3This oughtn't be called in the course of this test.NotImplementedErrorrr%r%r(rr8z#NotImplementedDelivery.validateFromcCr4rrr-r%r%r(rr8z!NotImplementedDelivery.validateTocCr4rrr r%r%r(rr8z%NotImplementedDelivery.receivedHeaderN)r:r;r<r=rrrr%r%r%r(rs  rc@sfeZdZdZejfddZddZddZdd Z d d Z d d Z ddZ e eddZddZdS)SMTPServerTestsza Test various behaviors of L{twisted.mail.smtp.SMTP} and L{twisted.mail.smtp.ESMTP}. cCs>|}d|_t}|||t|d|dS)zb Test that the specified hostname shows up in the SMTP server's greeting. s example.comN)rrrr9rrassertInr)r.rsrhr%r%r(testSMTPGreetingHosts  z$SMTPServerTests.testSMTPGreetingHostcC:t}t}|||t|d|dS)z Test that the string "ESMTP" does not appear in the SMTP server's greeting since that string strongly suggests the presence of support for various SMTP extensions which are not supported by L{smtp.SMTP}. ESMTPN) rrrrr9rr assertNotInrr.rrhr%r%r(testSMTPGreetingNotExtended  z+SMTPServerTests.testSMTPGreetingNotExtendedcCs|tjdS)zS Similar to testSMTPGreetingHost, but for the L{smtp.ESMTP} class. N)rrrr6r%r%r(testESMTPGreetingHostsz%SMTPServerTests.testESMTPGreetingHostcCr)z Test that the string "ESMTP" does appear in the ESMTP server's greeting since L{smtp.ESMTP} does support the SMTP extensions which that advertises to the client. rN) rrrrr9rrrrrr%r%r(testESMTPGreetingExtended$rz)SMTPServerTests.testESMTPGreetingExtendedcCsDt}t}|||d|t|d| dS)zN Sending an unimplemented command is responded to with a 500. s DOAGOODTHINGs500 Command not implementedN) rrrrr3r9rrrrrr%r%r(test_SMTPUnknownCommand0s   z'SMTPServerTests.test_SMTPUnknownCommandcCGdddt}ttj|}t|tg}t}||_t}| || d| | d| t ||ddS)z Test that a C{MAIL FROM} command with an acceptable address is responded to with the correct success code. c@rB)zDSMTPServerTests.test_acceptSenderAddress..AcceptanceDeliveryzM Delivery object which accepts all senders as valid. cSrr$r%rr%r%r(rFr*zQSMTPServerTests.test_acceptSenderAddress..AcceptanceDelivery.validateFromNr:r;r<r=rr%r%r%r(AcceptanceDeliveryArrHELO example.com MAIL FROM: s250 Sender address accepted Nrrrr rr rrrrrrr9rConnectionLostrr)r.rrrrTtransr%r%r(test_acceptSenderAddress;s   z(SMTPServerTests.test_acceptSenderAddresscCr)z Test that a C{MAIL FROM} command with an address rejected by a L{smtp.IMessageDelivery} instance is responded to with the correct error code. c@rB)zMSMTPServerTests.test_deliveryRejectedSenderAddress..RejectionDeliveryzO Delivery object which rejects all senders as invalid. cSs t|r$)r SMTPBadSenderrr%r%r(rjrnzZSMTPServerTests.test_deliveryRejectedSenderAddress..RejectionDelivery.validateFromNrr%r%r%r(RejectionDeliveryerrrrV550 Cannot receive from specified address : Sender not acceptable Nr)r.rrrrTrr%r%r("test_deliveryRejectedSenderAddress^s   z2SMTPServerTests.test_deliveryRejectedSenderAddresscCsGddd}ttjt}t||g}t}||_t}||| d| | d| t ||ddS)z Test that a C{MAIL FROM} command with an address rejected by an L{smtp.SMTP} instance's portal is responded to with the correct error code. c@seZdZdZefZddZdS)zQSMTPServerTests.test_portalRejectedSenderAddress..DisallowAnonymousAccesszZ Checker for L{IAnonymous} which rejects authentication attempts. cSs ttSr$)rrYrrr%r%r(rrzaSMTPServerTests.test_portalRejectedSenderAddress..DisallowAnonymousAccess.requestAvatarIdN)r:r;r<r=r rrr%r%r%r(DisallowAnonymousAccesss rrrrNrrr rrrrrrrrr9rrrr)r.rrrrTrr%r%r( test_portalRejectedSenderAddresss    z0SMTPServerTests.test_portalRejectedSenderAddresscCstttjt}t|g}t}||_t}||| d| | d| t ||ddS)z Test that a C{MAIL FROM} command issued without first authenticating when a portal has been configured to disallow anonymous logins is responded to with the correct error code. rrsd550 Cannot receive from specified address : Unauthenticated senders not allowed Nr)r.rrrTrr%r%r("test_portalRejectedAnonymousSenders    z2SMTPServerTests.test_portalRejectedAnonymousSenderN)r:r;r<r=rrrrrrrrrr r rrr%r%r%r(rs    #(  *rc@sxeZdZddZ dddZddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZdS)ESMTPAuthenticationTestscCs0|j|j||||jdS)z Assert that when the given bytes are delivered to the ESMTP server instance, it responds with the indicated lines. @type bytes: str @type response: list of str N)rrrrrrr)r.rresponser%r%r(assertServerResponses  z-ESMTPAuthenticationTests.assertServerResponseusernamepasswordcCs|\}}}}||g|tjjj|||j||| || t j || t j ||t j dddf|dg|jdS)a Assert that a login attempt has been made, that the credentials and interfaces passed to it are correct, and that when the login request is satisfied, a successful response is sent by the ESMTP server instance. @param loginArgs: A C{list} previously passed to L{portalFactory}. @param username: The login user. @param password: The login password. NcSr#r$r%r%r%r%r(rVr zDESMTPAuthenticationTests.assertServerAuthenticated..s235 Authentication successful.)rrrtwistedr rr providedByrrrrIMessageDeliveryFactoryr rNrrr)r. loginArgsrrxrrrrr%r%r(assertServerAuthenticateds z2ESMTPAuthenticationTests.assertServerAuthenticatedcCs@tdti|_d|j_ttdddd|_|j |jdS)zI Create an ESMTP instance attached to a StringTransport. rs localhostrXri90) peerAddressN) rrrrrrrrbrrr6r%r%r(setUps  zESMTPAuthenticationTests.setUpcCs|jtdS)zT Disconnect the ESMTP instance to clean up its timeout DelayedCall. N)rr9rrr6r%r%r(tearDownsz!ESMTPAuthenticationTests.tearDowncsGfddd}|S)NcseZdZfddZdS)z;ESMTPAuthenticationTests.portalFactory..DummyPortalcst}||||f|Sr$)rrMrF)r.rrrr loginListr%r(login szAESMTPAuthenticationTests.portalFactory..DummyPortal.loginN)r:r;r<rr%rr%r( DummyPortal srr%)r.rrr%rr( portalFactory sz&ESMTPAuthenticationTests.portalFactorycCsX|j|jd|j}||dd||dd|t|ddS)zV Test that AUTH is advertised to clients which issue an EHLO command. EHLO rs/250-localhost Hello 127.0.0.1, nice to meet your]s250 AUTH LOGINr^N)rrrrrrrr7)r. responseLinesr%r%r('test_authenticationCapabilityAdvertiseds  z@ESMTPAuthenticationTests.test_authenticationCapabilityAdvertisedcCsg}|||j_|jd|j|ddtd g|tdddtd g|td dg| |d S) zJ Test that the LOGIN authentication mechanism can be used r AUTH LOGIN 334 User Namerr PasswordrN rrrrrrrbase64 b64encodestriprr.rr%r%r(test_plainAuthentication!s   z1ESMTPAuthenticationTests.test_plainAuthenticationcCsg}|||j_|jd|j|ddtd g|tdddtd g|dg|j |dd d S) zM Test that giving an empty password for plain auth succeeds. rrrrrrrr{)rxNrrr%r%r(%test_plainAuthenticationEmptyPassword8s    z>ESMTPAuthenticationTests.test_plainAuthenticationEmptyPasswordcCs~g}|||j_|jd|j|dtd ddtd g|td dg| |dS) z The response to the first challenge may be included on the AUTH command line. Test that this is also supported. r AUTH LOGIN rrrrrNrrr%r%r('test_plainAuthenticationInitialResponseNs  z@ESMTPAuthenticationTests.test_plainAuthenticationInitialResponsecCs<g}|||j_|jd|jd|ddgdS)zW Test that a challenge/response sequence can be aborted by the client. rrs* s501 Authentication abortedN)rrrrrrr%r%r(test_abortAuthenticationbs   z1ESMTPAuthenticationTests.test_abortAuthenticationcCsHg}|||j_|jd|jd|ddg||gdS)zz Test that a response which is not properly Base64 encoded results in the appropriate error code. rrsx +501 Syntax error in parameters or argumentsNrrrrrrrr%r%r(!test_invalidBase64EncodedResponsens  z:ESMTPAuthenticationTests.test_invalidBase64EncodedResponsecCs<g}|||j_|jd|ddg||gdS)z Like L{test_invalidBase64EncodedResponse} but for the case of an initial response included with the C{AUTH} command. rsAUTH LOGIN x rNrrr%r%r((test_invalidBase64EncodedInitialResponses zAESMTPAuthenticationTests.test_invalidBase64EncodedInitialResponsecCsg}|||j_|jd|j|dtd ddtd g|td dg| \}}}}| t d| d |j| t|t d d S) z If the L{Deferred} returned by L{Portal.login} fires with an exception of any type other than L{UnauthorizedLogin}, the exception is logged and the client is informed that the authentication attempt has failed. rrrrrrrzSomething wrong with the servers9451 Requested action aborted: local error in processing r]N)rrrrrrrrrrrerrbackr5rrr7r)r.rrrrrr%r%r(test_unexpectedLoginFailures   z4ESMTPAuthenticationTests.test_unexpectedLoginFailureN)rr)r:r;r<rrrrrrrrrrrrrr%r%r%r(rs      rc@r>) SMTPClientErrorTestsz, Tests for L{smtp.SMTPClientError}. cC tdd}|t|ddS)z The string representation of a L{SMTPClientError} instance includes the response code and response string. rz some textz 123 some textNrrrr1r.errr%r%r(test_str zSMTPClientErrorTests.test_strcCr)z If the response code supplied to L{SMTPClientError} is negative, it is excluded from the string representation. sfoo barzfoo barNrrr%r%r(test_strWithNegativeCoderz-SMTPClientErrorTests.test_strWithNegativeCodecCsDtd}|d|dtjdd|d}|t|ddS) z If a line log is supplied to L{SMTPClientError}, its contents are included in the string representation of the exception instance. stestlogs secondlinedz test error)rsz"100 test error testlog secondline N)rrFrrr1r)r.rsrr%r%r(test_strWithLogs   z$SMTPClientErrorTests.test_strWithLogN)r:r;r<r=rr r r%r%r%r(rs  rc@rB)SenderMixinSentMailTestsz Tests for L{smtp.SenderMixin.sentMail}, used in particular by L{smtp.SMTPSenderFactory} and L{smtp.ESMTPSenderFactory}. csrttj}|fddtjddtd|ddd}|t d d d }d d g}| ddd||j |S)z L{smtp.SenderMixin.sentMail} adds only the addresses with failing SMTP response codes to the log passed to the factory's errback. cs|jdS)Ns'bob@example.com: 199 Error in sending. )rrs)r.r6r%r(rVszFSenderMixinSentMailTests.test_onlyLogFailedAddresses..rkrtrlrr[rmrXrYrZ)salice@example.comsNo errors here!)sbob@example.comsError in sending.rs Test responser]) rdrrMrSMTPDeliveryErrorrrqrrarrbrtrs)r.rfrgrrrr%r6r(test_onlyLogFailedAddressess(  z4SenderMixinSentMailTests.test_onlyLogFailedAddressesN)r:r;r<r=rr%r%r%r(r s r c@s8eZdZdZddZddZddZdd Zd d Zd S) ESMTPDowngradeTestCasezU Tests for the ESMTP -> SMTP downgrade functionality in L{smtp.ESMTPClient}. cCstddd|_dS)Nrr)rr|clientProtocolr6r%r%r(rrzESMTPDowngradeTestCase.setUpcCs^t}d|j_d|j_d|j_|j||jd||jd|d| dS)z If both authentication and transport security are not required, and it is asked for, it will fall back to allowing HELO. FT220 localhost 500 not an esmtp server HELO testuser N rrrequireAuthenticationrequireTransportSecurity heloFallbackrrrrrr.rr%r%r( test_requireHELOFallbackOperates   z7ESMTPDowngradeTestCase.test_requireHELOFallbackOperatescCs^t}d|j_d|j_d|j_|j||jd||jd|d| dS)z If authentication is required, and HELO fallback is on, HELO fallback must not be honoured, as authentication requires EHLO to succeed. TFrrrNrrr%r%r(!test_requireAuthFailsHELOFallback rz8ESMTPDowngradeTestCase.test_requireAuthFailsHELOFallbackcCs^t}d|j_d|j_d|j_|j||jd||jd|d| dS)z If TLS is required and the connection is insecure, HELO fallback must not be honoured, as STARTTLS requires EHLO to succeed. FTrrrNrrr%r%r( test_requireTLSFailsHELOFallbackrz7ESMTPDowngradeTestCase.test_requireTLSFailsHELOFallbackcCsjt}t|tjd|j_d|j_d|j_|j||j d| |j d| d| dS)z If TLS is provided at the transport level, we can honour the HELO fallback if we're set to require TLS. FTrrrN) rrr ISSLTransportrrrrrrrrrrr%r%r(/test_requireTLSAndHELOFallbackSucceedsIfOverTLS-s    zFESMTPDowngradeTestCase.test_requireTLSAndHELOFallbackSucceedsIfOverTLSN) r:r;r<r=rrrrr!r%r%r%r(rs rc@sdeZdZdZeZdZdZddZddZ dd Z d d Z d d Z ddZ ddZddZddZdS) SSLTestCasezD Tests for the TLS negotiation done by L{smtp.ESMTPClient}. s3220 localhost NO UCE NO UBE NO RELAY PROBES ESMTP s1250-localhost Hello 127.0.0.1, nice to meet you cCs*tdtd|_d|j_dd|j_dS)NrrTcSsdS)Nztest@example.orgr%r%r%r%r(rVNr z#SSLTestCase.setUp..)rr|r!rrrfr6r%r%r(rIs zSSLTestCase.setUpcCsZt}t|tj|j||j|j||j|j || d| dS)a Verify that when L{smtp.ESMTPClient} connects to a server over a transport providing L{ISSLTransport}, C{requireTransportSecurity} is C{True}, and it is presented with the given capabilities, it will try to send its mail and not first attempt to negotiate TLS using the I{STARTTLS} protocol action. @param capabilities: Bytes to include in the test server's capability response. These must be formatted exactly as required by the protocol, including a line which ends the capability response. @type param: L{bytes} @raise: C{self.failureException} if the behavior of C{self.clientProtocol} is not as described. sMAIL FROM: N) rrrr rrrSERVER_GREETINGr EHLO_RESPONSErr)r. capabilitiesrr%r%r($_requireTransportSecurityOverSSLTestPs  z0SSLTestCase._requireTransportSecurityOverSSLTestcC|ddS)z When C{requireTransportSecurity} is C{True} and the client is connected over an SSL transport, mail may be delivered. 250 AUTH LOGIN Nr&r6r%r%r($test_requireTransportSecurityOverSSLoz0SSLTestCase.test_requireTransportSecurityOverSSLcCNt}|j||j|j||j|jd|d|dS)z When C{requireTransportSecurity} is C{True} and the client is connected over a non-SSL transport, if the server offers the I{STARTTLS} extension, it is used before mail is delivered. 250-AUTH LOGIN 250 STARTTLS s STARTTLS N rrrrr#rr$rrrr%r%r('test_requireTransportSecurityTLSOfferedvs z3SSLTestCase.test_requireTransportSecurityTLSOfferedcCr')z When C{requireTransportSecurity} is C{True} and the client is connected over an SSL transport, if the server offers the I{STARTTLS} extension, it is not used before mail is delivered. r-Nr)r6r%r%r(.test_requireTransportSecurityTLSOfferedOverSSLsz:SSLTestCase.test_requireTransportSecurityTLSOfferedOverSSLcCr,)z When C{requireTransportSecurity} is C{True} and the client is connected over a non-SSL transport, if the server does not offer the I{STARTTLS} extension, mail is not delivered. r(rNr.rr%r%r(*test_requireTransportSecurityTLSNotOffereds  z6SSLTestCase.test_requireTransportSecurityTLSNotOfferedcCsV|jj}~|j|jgd}|t|d||ddt||ddddS)< L{smtp.ESMTPClient.tlsMode} is deprecated. offendingFunctionsr]rcategoryrOStlsMode attribute of twisted.mail.smtp.ESMTPClient is deprecated since Twisted 13.0N)rtlsMode flushWarnings%test_esmtpClientTlsModeDeprecationGetrr7assertIdenticalDeprecationWarning)r.val warningsShownr%r%r(r9s z1SSLTestCase.test_esmtpClientTlsModeDeprecationGetcstfdddS)z L{smtp.ESMTPClient.__getattr__} raises an attribute error for other attribute names which do not exist. csjjSr$)r doesNotExistr%r6r%r(rVszQSSLTestCase.test_esmtpClientTlsModeDeprecationGetAttributeError..N) assertRaisesAttributeErrorr6r%r6r(3test_esmtpClientTlsModeDeprecationGetAttributeErrorsz?SSLTestCase.test_esmtpClientTlsModeDeprecationGetAttributeErrorcCsTd|j_|j|jgd}|t|d||ddt||ddddS) r2Fr3r]rr5rOr6N)rr7r8%test_esmtpClientTlsModeDeprecationSetrr7r:r;)r.r=r%r%r(rBs z1SSLTestCase.test_esmtpClientTlsModeDeprecationSetN)r:r;r<r=r rQr#r$rr&r*r/r0r1r9rArBr%r%r%r(r"?s  r"c@seZdZdZdZddZdS)AbortableStringTransportzK A version of L{StringTransport} that supports C{abortConnection}. FcCsd|_|dS)z A testable version of the C{ITCPTransport.abortConnection} method. Since this is a special case of closing the connection, C{loseConnection} is also called. TN)abortingrr6r%r%r(abortConnections z(AbortableStringTransport.abortConnectionN)r:r;r<r=rDrEr%r%r%r(rCs rCc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS) SendmailTestsz2 Tests for L{twisted.mail.smtp.sendmail}. cCs(ttj\}}}}|t|ddS)z{ The default C{reactor} parameter of L{twisted.mail.smtp.sendmail} is L{twisted.internet.reactor}. r^N)inspect getargspecrsendmailrr)r.argsvarArgskeywordsdefaultsr%r%r("test_defaultReactorIsGlobalReactorsz0SendmailTests.test_defaultReactorIsGlobalReactorc Cslt}tjdddd|||ddd |jdd}||jd||jd||jd ||jd d S) zk L{twisted.mail.smtp.sendmail} creates the ESMTP factory with the ESMTP arguments. rrkrtrT)rrrxrrrr^rbarN) rrrI tcpClientsr_requireTransportSecurity_requireAuthenticationrrx)r.rrxrrr%r%r(_honorsESMTPArgumentss" z#SendmailTests._honorsESMTPArgumentscC|jdddS)zw L{twisted.mail.smtp.sendmail} should accept C{username} and C{password} which are L{unicode}. rbarrrxrSr6r%r%r(&test_honorsESMTPArgumentsUnicodeUserPWr+z4SendmailTests.test_honorsESMTPArgumentsUnicodeUserPWcCrT)zu L{twisted.mail.smtp.sendmail} should accept C{username} and C{password} which are L{bytes}. rrOrVrWr6r%r%r($test_honorsESMTPArgumentsBytesUserPWr+z2SendmailTests.test_honorsESMTPArgumentsBytesUserPWcCsBt}td}tjddd||d|jdd}||j|dS) z| L{twisted.mail.smtp.sendmail} will pass through the message untouched if it is a file-like object. sFile!rrkrtrrr^N)rrrrIrPassertIsfile)r.r messageFilerr%r%r(test_messageFilePassthroughsz)SendmailTests.test_messageFilePassthroughcCsLt}tjdddd|d|jdd}|j}|d||ddS) z L{twisted.mail.smtp.sendmail} will turn non-file-like objects (eg. strings) into file-like objects before sending. rrkrtrrZrr^N)rrrIrPr\seekrr)r.rrr]r%r%r(test_messageStringMadeFile(s z(SendmailTests.test_messageStringMadeFilecCs<t}tjdddd|dd|jdd}||jd d S) zk L{twisted.mail.smtp.sendmail} passes through the sender domain name, if provided. rrkrtrr)rsenderDomainNamerr^rN)rrrIrPrrA)r.rrr%r%r(test_senderDomainName:sz#SendmailTests.test_senderDomainNamecCsPt}tjdddd|d}|||jdjd||}|t j dS) z When a user cancels L{twisted.mail.smtp.sendmail} before the connection is made, the connection is closed by L{twisted.internet.interfaces.IConnector.disconnect}. rrkrtrrZrTN) rrrIcancelr connectors _disconnectedfailureResultOftraprCancelledError)r.rrfailurer%r%r(test_cancelBeforeConnectionMadeKs z-SendmailTests.test_cancelBeforeConnectionMadecCst}t}tjdddd|d}|jdd}|d}|||||j d ||j d | |}| t jdS) z When a user cancels L{twisted.mail.smtp.sendmail} after the connection is made, the connection is closed by L{twisted.internet.interfaces.ITransport.abortConnection}. rrkrtrrZrr^NT)rrCrrIrPrarrcrrDrrfrgrrh)r.rrrrrrir%r%r(test_cancelAfterConnectionMade^s"   z,SendmailTests.test_cancelAfterConnectionMadeN) r:r;r<r=rNrSrXrYr^r`rbrjrkr%r%r%r(rFs rF)kr=rrGrDiortypingrrrrrzope.interfacerr twisted.cred.checkersrtwisted.cred.credentialstwisted.cred.errortwisted.cred.portalr r r r rrrtwisted.internetrrrrrrr twisted.mailrtwisted.mail._credrtwisted.protocolsrrtwisted.python.utilrtwisted.test.proto_helpersrrtwisted.trial.unittestrr1rtwisted.test.ssl_helpersr!r" ImportErrorr IReactorSSLrr)IMessager,r?rRr_r\rwrur|rzr LineReceiverrrrrrrrrrrrcheckersrr rr rr#rCrHrRrWrrrrrrrrrr rr"rCrFr%r%r%r(s    $       um B=7$@R\!(J