U “ji2@sdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZGdddeZdS)zi Tests for L{twisted.mail.scripts.mailmail}, the implementation of the command line program I{mailmail}. N)StringIO)skipIf)version)Deferred) MemoryReactor)smtp)mailmail) parseOptions)Failure) platformType)TestCasec@seZdZdZeZddZddZddZdd Z d d Z d d Z ddZ ddZ ddZddZddZddZeedkdddZeedkdddZd 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||dd_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)LoggertextFileLogObserver_log)ZobserverNcs.|_|_|_tj||j|j|jjdS)N)Zreactor) hostoptionsidentrsendmailsendertobody memoryReactor)rrrselfC/tmp/pip-unpacked-wheel-59kc99xd/twisted/mail/test/test_mailmail.pyr3sz$OptionsTests.setUp..sendmailr) routZtwisted.loggerrrpatchrrrr)rrrZ logObserverrrrrsetUp#s  zOptionsTests.setUpcCs2|tdtd|ttg}||jddS)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)rsysr assertRaises SystemExitr assertEqualargsrexcrrrtest_unspecifiedRecipientsAsz'OptionsTests.test_unspecifiedRecipientscCs"|ttdg}||jddS)z The I{-bp} option for listing queue information is unsupported and if it is passed to L{parseOptions}, L{SystemExit} is raised. z-bpzUnsupported option.Nr"r#r r$r%r&rrrtest_listQueueInformationKsz&OptionsTests.test_listQueueInformationcCs"|ttdg}||jddS)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&rrrtest_stdioTransportSsz OptionsTests.test_stdioTransportcCs@|ttdg}||jd|ttdg}||jddS)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&rrrtest_ignoreFullStop\sz OptionsTests.test_ignoreFullStopcCs"|ttdg}||jddS)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&rrrtest_copyAliasedSendergsz#OptionsTests.test_copyAliasedSendercCsPt}|td||ttd}||jd|}||dt ddS)z~ The I{--version} option displays the version and raises L{SystemExit} with L{None} as the exit code. stdoutz --versionNzmailmail version:  ) rrr!r"r#r r$codegetvaluer)rrZsystemExitCodedatarrr test_versionps zOptionsTests.test_versioncCs.td}|td|td}||jdS)zA The I{-odb} flag specifies background delivery. r0r z-odbN)rrr!r assertTrue backgroundrr orrrtest_backgroundDelivery}sz$OptionsTests.test_backgroundDeliverycCs.td}|td|td}||jdS)zB The I{-odf} flags specifies foreground delivery. r0r z-odfN)rrr!r Z assertFalser6r7rrrtest_foregroundDeliverysz$OptionsTests.test_foregroundDeliverycCs4td}|td|td}|t|jddS)zd The I{-t} flags specifies that recipients should be obtained from headers. zTo: Curly Cc: Larry Bcc: Moe Oh, a wise guy? r -tN)rrr!r r$lenrr7rrrtest_recipientsFromHeaderss z'OptionsTests.test_recipientsFromHeaderscCs6td}|td|tdddg}||jddS)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;Nrrr!r r$rr7rrr test_setFromszOptionsTests.test_setFromcCs6td}|td|tdddg}||jddS)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 r r?z"Groucho r;zinvaliduser4@example.comNr@r7rrr!test_overrideFromFlagByFromHeaders z.OptionsTests.test_overrideFromFlagByFromHeaderwin32zImailmail.run() does not work on win32 due to lack of support for getuid()cCstd}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.pyinvaliduser2@example.comz-oepr0argvr rEN)rrr!rrunr$rrgetloginrr5Z printErrorsZ assertIsNonefailed)rrFr rrrtest_runErrorsToStderrs z#OptionsTests.test_runErrorsToStderrc Cstd}|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. r0r 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_CFGrDrFz3Illegal UID in \[useraccess\] section: invaliduser1z3Illegal GID in \[groupaccess\] section: invalidgid1z,Illegal entry in \[identity\] section: funnyN)rrr!mktemposgetuidgetgidopenwriteformatrrG assertRegexrr2)rr filenameZmyUidZmyGidfrFrrrtest_readInvalidConfigs8   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) loadConfigrK)twisted.mail.scripts.mailmailrWrLrPrQ)rconfigrWrTrUrrrgetConfigFromFiles   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 ddgdS)zj L{twisted.mail.scripts.mailmail.loadConfig} parses the config file for mailmail. z [addresses] smarthost=localhost localhostz' [addresses] default_domain=example.comz example.comz; [addresses] smarthost=localhost default_domain=example.comz1 [identity] host1=invalid host2=username:passwordZhost1Zhost2usernamepasswordz+ [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) rZr$Z smarthostdomainZ assertNotInZ identitiesZ allowUIDsZdenyUIDsZ allowGIDsZdenyGIDs)rrYrrrtest_loadConfigsXzOptionsTests.test_loadConfigcsBfdd}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|dtS)Nz postmaster@ testsender ValueError)rSr$r2r)rrZ recipientrrrrr}s z-OptionsTests.test_senderror..sendmailrrhN)rrrOptionsrr riZ senderror)rroptsZfailrrrtest_senderrorws   zOptionsTests.test_senderrorN)__name__ __module__ __qualname____doc__rrrr(r+r,r-r.r4r9r:r>rArBrr rJrVrZrgrlrrrrr s6           6Yr )rprMr!iorZunittestrZtwisted.copyrightrZtwisted.internet.deferrZtwisted.internet.testingrZ twisted.mailrZtwisted.mail.scriptsrrXr Ztwisted.python.failurer Ztwisted.python.runtimer Ztwisted.trial.unittestr r rrrrs