← Back
Editing: test_mailmail.cpython-38.pyc
U W[�2 � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl mZ dd lmZ dd lmZ dd lmZ dd lmZ G d 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� e Zd ZdZe� Zdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zed kr�d e_d d � Zed kr�d e_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. c sf t � � _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)ZobserverNc s. | � _ |� _|� _tj| |j|j|j� jd�S )N)Zreactor) �host�options�identr �sendmail�sender�to�body� memoryReactor)r r r ��self� �A/usr/lib/python3/dist-packages/twisted/mail/test/test_mailmail.pyr 1 s �z$OptionsTests.setUp.<locals>.sendmailr ) r �outZtwisted.loggerr r �patchr r r r )r r r Z logObserverr r r r �setUp" s zOptionsTests.setUpc C s2 | � 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. �stdinz Subject: foo Hello, goodbye. )zNo recipients specified.N)r �sysr � assertRaises� SystemExitr � assertEqual�args�r �excr r r �test_unspecifiedRecipients; s �z'OptionsTests.test_unspecifiedRecipientsc C s" | � 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! r r" r# r$ r r r �test_listQueueInformationI s z&OptionsTests.test_listQueueInformationc C s" | � 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$ r r r �test_stdioTransportR s z OptionsTests.test_stdioTransportc C s@ | � 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$ r r r �test_ignoreFullStop\ s z OptionsTests.test_ignoreFullStopc C s" | � 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$ r r r �test_copyAliasedSenderh s z#OptionsTests.test_copyAliasedSenderc C sN t � }| �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: {} ) r r r r r! r r" �code�getvalue�formatr )r r ZsystemExitCode�datar r r � test_versionr s �zOptionsTests.test_versionc C s. t d�}| �td|� td�}| �|j� dS )zA The I{-odb} flag specifies background delivery. � r z-odbN)r r r r � assertTrue� background�r r �or r r �test_backgroundDelivery� s z$OptionsTests.test_backgroundDeliveryc C s. t d�}| �td|� td�}| �|j� dS )zB The I{-odf} flags specifies foreground delivery. r3 r z-odfN)r r r r Z assertFalser5 r6 r r r �test_foregroundDelivery� s z$OptionsTests.test_foregroundDeliveryc C s4 t d�}| �td|� td�}| �t|j�d� dS )zd The I{-t} flags specifies that recipients should be obtained from headers. zTo: Curly <invaliduser2@example.com> Cc: Larry <invaliduser1@example.com> Bcc: Moe <invaliduser3@example.com> Oh, a wise guy? r �-t� N)r r r r r" �lenr r6 r r r �test_recipientsFromHeaders� s �z'OptionsTests.test_recipientsFromHeadersc C s6 t d�}| �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 <invaliduser1@example.com>r: N�r r r r r" r r6 r r r � test_setFrom� s �zOptionsTests.test_setFromc C s6 t d�}| �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 <invaliduser4@example.com> From: Shemp <invaliduser4@example.com> r r> z"Groucho <invaliduser5@example.com>r: zinvaliduser4@example.comNr? r6 r r r �!test_overrideFromFlagByFromHeader� s �z.OptionsTests.test_overrideFromFlagByFromHeaderc C st d}t d�}| �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 �argvr rC N)r r r r �runr" r r �getloginr r4 Z printErrorsZ assertIsNoneZfailed)r rD r r r r �test_runErrorsToStderr� s z#OptionsTests.test_runErrorsToStderrZwin32zImailmail.run() does not work on win32 due to lack of support for getuid()c C s� t d�}| �td|� | �� }t�� }t�� }t|d��}|�d� ||�� W 5 Q R X | �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. r3 r �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_CFGrB rD z3Illegal UID in \[useraccess\] section: invaliduser1z3Illegal GID in \[groupaccess\] section: invalidgid1z,Illegal entry in \[identity\] section: funnyN)r r r �mktemp�os�getuid�getgid�open�writer0 r rE � assertRegexr r/ )r r �filenameZmyUidZmyGid�frD r r r �test_readInvalidConfig� s. � � � �z#OptionsTests.test_readInvalidConfigc C s< ddl m} | �� }t|d��}|�|� W 5 Q R X ||�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.mailmailrS rI rM rN )r �configrS rP rQ r r r �getConfigFromFile s z OptionsTests.getConfigFromFilec C sN | � d�}| �|jd� | � d�}| �|jd� | � d�}| �|jd� | �|jd� | � d�}| �d|j� | �|jd d 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) rV r" Z smarthostZdomainZ assertNotInZ identitiesZ allowUIDsZdenyUIDsZ allowGIDsZdenyGIDs)r rU r r r �test_loadConfig& s0 z OptionsTests.test_loadConfigc sB � 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. c s0 � � |d� � �|dg� � � |�� d� t� S )Nz postmaster@� testsender� ValueError)rO r" r/ r )r r Z recipientr r r r r q s z-OptionsTests.test_senderror.<locals>.sendmailr r` N)r r r ZOptionsr r ra Z senderror)r r ZoptsZfailr r r �test_senderrorl s zOptionsTests.test_senderrorN)�__name__� __module__� __qualname__�__doc__r r r r& r) r* r+ r, r2 r8 r9 r= r@ rA rG r �skiprR rV r_ rb r r r r r s0 �1Fr )rf rJ r Ztwisted.copyrightr Ztwisted.internet.deferr Z twisted.mailr Ztwisted.mail.scriptsr rT r Ztwisted.python.compatr Ztwisted.python.failurer Ztwisted.python.runtimer Ztwisted.test.proto_helpersr Ztwisted.trial.unittestr r r r r r �<module> s
Save File
Cancel