File: //lib/python3/dist-packages/twisted/mail/test/__pycache__/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