o bk2@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)smtp)mailmail) parseOptions)Failure) platformType) MemoryReactor)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)observerNcs.|_|_|_tj||j|j|jjdS)N)reactor) hostoptionsidentrsendmailsendertobody memoryReactor)rrrselfA/usr/lib/python3/dist-packages/twisted/mail/test/test_mailmail.pyr3sz$OptionsTests.setUp..sendmailr) routtwisted.loggerrrpatchrrrr)rrr 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)r!sysr assertRaises SystemExitr assertEqualargsrexcrrrtest_unspecifiedRecipientsAsz'OptionsTests.test_unspecifiedRecipientscC"|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'rr(r)r*rrrtest_listQueueInformationKsz&OptionsTests.test_listQueueInformationcCr-)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_stdioTransportSz 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_ignoreFullStopcCr-)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_copyAliasedSendergr2z#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:  ) rr!r%r&r'rr(codegetvaluer)rrsystemExitCodedatarrr test_versionps zOptionsTests.test_versioncC.td}|td|td}||jdS)zA The I{-odb} flag specifies background delivery. r6r$z-odbN)rr!r%r assertTrue backgroundrr$orrrtest_backgroundDelivery}z$OptionsTests.test_backgroundDeliverycCr<)zB The I{-odf} flags specifies foreground delivery. r6r$z-odfN)rr!r%r assertFalser>r?rrrtest_foregroundDeliveryrBz$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)rr!r%rr(lenrr?rrrtest_recipientsFromHeaderss z'OptionsTests.test_recipientsFromHeaderscC4td}|td|tgd}||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$)-F Larry rErKNrr!r%rr(rr?rrr test_setFroms zOptionsTests.test_setFromcCrI)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$)rJz"Groucho rEzinvaliduser4@example.comNrLr?rrr!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-oepr6argvr$rQN)rr!r%rrunr(rrgetloginrr= printErrors assertIsNonefailed)rrRr$rrrtest_runErrorsToStderrs z#OptionsTests.test_runErrorsToStderrcCstd}|td||}t}t}t|d}|d ||Wdn1s0wY|t d|d}|td|t | |j d | |j d | |j d dS) z Error messages for illegal UID value, illegal GID value, and illegal identity entry will be sent to stderr. r6r$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 N LOCAL_CFGrPrRz3Illegal UID in \[useraccess\] section: invaliduser1z3Illegal GID in \[groupaccess\] section: invalidgid1z,Illegal entry in \[identity\] section: funny)rr!r%mktemposgetuidgetgidopenwriteformatrrS assertRegexrr8)rr$filenamemyUidmyGidfrRrrrtest_readInvalidConfigs4    z#OptionsTests.test_readInvalidConfigcCsVddlm}|}t|d}||Wd||S1s"wY||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) loadConfigrYN)twisted.mail.scripts.mailmailrhr[r_r`)rconfigrhrcrfrrrgetConfigFromFiles    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:passwordhost1host2usernamepasswordz+ [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) rkr( smarthostdomain assertNotIn identities allowUIDsdenyUIDs allowGIDsdenyGIDs)rrjrrrtest_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)rbr(r8r)rr recipientrrrrr}s z-OptionsTests.test_senderror..sendmailrrN)r!rrOptionsrr r senderror)rroptsfailrrrtest_senderrorws  zOptionsTests.test_senderrorN)__name__ __module__ __qualname____doc__r rr#r,r0r1r3r4r;rArDrHrMrNrr rXrgrkrrrrrrr s8           6 Yr )rr\r%iorunittestrtwisted.copyrightrtwisted.internet.deferr twisted.mailrtwisted.mail.scriptsrrirtwisted.python.failurer twisted.python.runtimer twisted.test.proto_helpersr twisted.trial.unittestr r rrrrs