U  W[‚2ã@s dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZGd d„deƒZdS)zi Tests for L{twisted.mail.scripts.mailmail}, the implementation of the command line program I{mailmail}. éN)Úversion)ÚDeferred)Úsmtp)Úmailmail)Ú parseOptions)ÚNativeStringIO)ÚFailure)Ú platformType)Ú MemoryReactor)ÚTestCasec@sºeZdZdZeƒZdd„Zdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zedkrˆde_dd„Zedkržde_d d!„Zd"d#„Zd$d%„Zd&S)'Ú OptionsTestsz¸ Tests for L{parseOptions} which parses command line arguments and reads message text from stdin to produce an L{Options} instance which can be used to send a message. csftƒˆ_ddlm}m}|ˆjƒ}ˆ td||d¡dˆ_dˆ_dˆ_ ‡fdd„}ˆ td|¡dS) zz Override some things in mailmail, so that we capture C{stdout}, and do not call L{reactor.stop}. r)ÚtextFileLogObserverÚLoggerZ_log)ZobserverNcs.|ˆ_|ˆ_|ˆ_tj||j|j|jˆjdS)N)Zreactor) ÚhostÚoptionsÚidentrÚsendmailÚsenderÚtoÚbodyÚ memoryReactor)rrr©Úself©úA/usr/lib/python3/dist-packages/twisted/mail/test/test_mailmail.pyr1sÿz$OptionsTests.setUp..sendmailr) rÚoutZtwisted.loggerr rÚpatchrrrr)rr rZ logObserverrrrrÚsetUp"s  zOptionsTests.setUpcCs2| tdtdƒ¡| ttg¡}| |jd¡dS)zÒ If no recipients are given in the argument list and there is no recipient header in the message text, L{parseOptions} raises L{SystemExit} with a string describing the problem. ÚstdinzSubject: foo Hello, goodbye. )zNo recipients specified.N)rÚsysrÚ assertRaisesÚ SystemExitrÚ assertEqualÚargs©rÚexcrrrÚtest_unspecifiedRecipients;s  ÿz'OptionsTests.test_unspecifiedRecipientscCs"| ttdg¡}| |jd¡dS)z™ The I{-bp} option for listing queue information is unsupported and if it is passed to L{parseOptions}, L{SystemExit} is raised. z-bp©zUnsupported option.N©r r!rr"r#r$rrrÚtest_listQueueInformationIsz&OptionsTests.test_listQueueInformationcCs"| ttdg¡}| |jd¡dS)z´ The I{-bs} option for using stdin and stdout as the SMTP transport is unsupported and if it is passed to L{parseOptions}, L{SystemExit} is raised. z-bsr'Nr(r$rrrÚtest_stdioTransportRsz OptionsTests.test_stdioTransportcCs@| ttdg¡}| |jd¡| ttdg¡}| |jd¡dS)z» The I{-i} and I{-oi} options for ignoring C{"."} by itself on a line are unsupported and if either is passed to L{parseOptions}, L{SystemExit} is raised. z-ir'z-oiNr(r$rrrÚtest_ignoreFullStop\sz OptionsTests.test_ignoreFullStopcCs"| ttdg¡}| |jd¡dS)z¿ The I{-om} option for copying the sender if they appear in an alias expansion is unsupported and if it is passed to L{parseOptions}, L{SystemExit} is raised. z-omr'Nr(r$rrrÚtest_copyAliasedSenderhsz#OptionsTests.test_copyAliasedSendercCsNtƒ}| td|¡| ttd¡}| |jd¡| ¡}| |d  t ¡¡dS)z~ The I{--version} option displays the version and raises L{SystemExit} with L{None} as the exit code. Ústdoutz --versionNzmailmail version: {} ) rrrr r!rr"ÚcodeÚgetvalueÚformatr)rrZsystemExitCodeÚdatarrrÚ test_versionrsÿzOptionsTests.test_versioncCs.tdƒ}| td|¡tdƒ}| |j¡dS)zA The I{-odb} flag specifies background delivery. Ú rz-odbN)rrrrÚ assertTrueÚ background©rrÚorrrÚtest_backgroundDeliverysz$OptionsTests.test_backgroundDeliverycCs.tdƒ}| td|¡tdƒ}| |j¡dS)zB The I{-odf} flags specifies foreground delivery. r3rz-odfN)rrrrZ assertFalser5r6rrrÚtest_foregroundDelivery‹sz$OptionsTests.test_foregroundDeliverycCs4tdƒ}| td|¡tdƒ}| t|jƒd¡dS)zd The I{-t} flags specifies that recipients should be obtained from headers. zTo: Curly Cc: Larry Bcc: Moe Oh, a wise guy? rú-téN)rrrrr"Úlenrr6rrrÚtest_recipientsFromHeaders•s ÿz'OptionsTests.test_recipientsFromHeaderscCs6tdƒ}| td|¡tdddgƒ}| |jd¡dS)zw When a message has no I{From:} header, a I{From:} value can be specified with the I{-F} flag. z3To: invaliduser2@example.com Subject: A wise guy? rú-Fz Larry r:N©rrrrr"rr6rrrÚ test_setFrom¥s ÿzOptionsTests.test_setFromcCs6tdƒ}| td|¡tdddgƒ}| |jd¡dS)z‘ The I{-F} flag specifies the From: value. However, I{-F} flag is overriden by the value of From: in the e-mail header. zLTo: Curly From: Shemp rr>z"Groucho r:zinvaliduser4@example.comNr?r6rrrÚ!test_overrideFromFlagByFromHeader²s ÿz.OptionsTests.test_overrideFromFlagByFromHeadercCstd}tdƒ}| td|¡| td|¡t ¡| |jjt ¡¡| |jj dg¡|  |jj ¡|  tj ¡dS)z½ Call L{mailmail.run}, and specify I{-oep} to print errors to stderr. The sender, to, and printErrors options should be set and there should be no failure. ©ztest_mailmail.pyúinvaliduser2@example.comz-oepr3ÚargvrrCN)rrrrÚrunr"rrÚgetloginrr4Z printErrorsZ assertIsNoneZfailed)rrDrrrrÚtest_runErrorsToStderr¿sz#OptionsTests.test_runErrorsToStderrZwin32zImailmail.run() does not work on win32 due to lack of support for getuid()c Cs¸tdƒ}| td|¡| ¡}t ¡}t ¡}t|dƒ}| d  ||¡¡W5QRX| t d|¡d}| td|¡t   ¡|  |j  ¡d¡|  |j  ¡d ¡|  |j  ¡d ¡d S) z… Error messages for illegal UID value, illegal GID value, and illegal identity entry will be sent to stderr. r3rÚwa[useraccess] allow=invaliduser2,invaliduser1 deny=invaliduser3,invaliduser4,{} order=allow,deny [groupaccess] allow=invalidgid1,invalidgid2 deny=invalidgid1,invalidgid2,{} order=deny,allow [identity] localhost=funny [addresses] smarthost=localhost default_domain=example.com Z LOCAL_CFGrBrDz3Illegal UID in \[useraccess\] section: invaliduser1z3Illegal GID in \[groupaccess\] section: invalidgid1z,Illegal entry in \[identity\] section: funnyN)rrrÚmktempÚosÚgetuidÚgetgidÚopenÚwriter0rrEÚ assertRegexrr/)rrÚfilenameZmyUidZmyGidÚfrDrrrÚtest_readInvalidConfig×s.  ô ÿ ÿ ÿz#OptionsTests.test_readInvalidConfigc Cs<ddlm}| ¡}t|dƒ}| |¡W5QRX||ƒS)aª Read a mailmail configuration file. The mailmail script checks the twisted.mail.scripts.mailmail.GLOBAL_CFG variable and then the twisted.mail.scripts.mailmail.LOCAL_CFG variable for the path to its config file. @param config: path to config file @type config: L{str} @return: A parsed config. @rtype: L{twisted.mail.scripts.mailmail.Configuration} r)Ú loadConfigrH)Útwisted.mail.scripts.mailmailrSrIrMrN)rÚconfigrSrPrQrrrÚgetConfigFromFile s   zOptionsTests.getConfigFromFilecCsN| d¡}| |jd¡| d¡}| |jd¡| d¡}| |jd¡| |jd¡| d¡}| d|j¡| |jdd d g¡| d ¡}| |jd g¡| d ¡}| |jd dg¡| d¡}| |jd dg¡| |jddg¡| d¡}| |jdg¡| d¡}| |j dg¡| d¡}| |jddg¡| |j ddg¡dS)zj L{twisted.mail.scripts.mailmail.loadConfig} parses the config file for mailmail. z [addresses] smarthost=localhostZ localhostz' [addresses] default_domain=example.comz example.comz; [addresses] smarthost=localhost default_domain=example.comz1 [identity] host1=invalid host2=username:passwordZhost1Zhost2ZusernameZpasswordz+ [useraccess] allow=invalid1,35 order=allowé#z# [useraccess] deny=35,36 order=denyé$z/ [useraccess] allow=35,36 deny=37,38 order=denyé%é&z( [groupaccess] allow=gid1,41 order=allowé)z! [groupaccess] deny=41 order=denyz6 [groupaccess] allow=41,42 deny=43,44 order=allow,denyé*é+é,N) rVr"Z smarthostZdomainZ assertNotInZ identitiesZ allowUIDsZdenyUIDsZ allowGIDsZdenyGIDs)rrUrrrÚtest_loadConfig&s0          zOptionsTests.test_loadConfigcsB‡fdd„}ˆ td|¡t ¡}d|_ttƒƒ}t ||¡dS)zš L{twisted.mail.scripts.mailmail.senderror} sends mail back to the sender if an error occurs while sending mail to the recipient. cs0ˆ |d¡ˆ |dg¡ˆ | ¡d¡tƒS)Nz postmaster@Ú testsenderÚ ValueError)rOr"r/r)rrZ recipientrrrrrqs z-OptionsTests.test_senderror..sendmailrr`N)rrrZOptionsrrraZ senderror)rrZoptsZfailrrrÚtest_senderrorls   zOptionsTests.test_senderrorN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrr&r)r*r+r,r2r8r9r=r@rArGr ÚskiprRrVr_rbrrrrr s0        ÿ1Fr )rfrJrZtwisted.copyrightrZtwisted.internet.deferrZ twisted.mailrZtwisted.mail.scriptsrrTrZtwisted.python.compatrZtwisted.python.failurerZtwisted.python.runtimer Ztwisted.test.proto_helpersr Ztwisted.trial.unittestr r rrrrÚs