| Server IP : 209.38.156.173 / Your IP : 216.73.216.122 [ Web Server : Apache/2.4.52 (Ubuntu) System : Linux lakekumayuhotel 5.15.0-136-generic #147-Ubuntu SMP Sat Mar 15 15:53:30 UTC 2025 x86_64 User : root ( 0) PHP Version : 8.1.2-1ubuntu2.22 Disable Function : NONE Domains : 2 Domains MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib/python3/dist-packages/twisted/mail/test/__pycache__/ |
Upload File : |
o
�b \ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlmZ ddlmZ ddlmZmZ ddlmZ ddlZddlZddlZddlZddl Zddl Zddl Zddl Zddl Zddlm!Z!m"Z" ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z) dd l*m+Z+ dd
l,m-Z-m.Z.m/Z/m0Z0 dd l1m2Z2m3Z3 dd l m4Z4 dd
l5m6Z6 ddl7m8Z8m9Z9m:Z: ddl;m<Z< ddl=m>Z>m?Z? ddl@mAZA ddlBmCZC ddlDmEZEmFZFmGZG ddlHmIZI eeCdkd�G dd� deI��ZJeeCdkd�G dd� deI��ZKeeCdkd�G dd � d eI��ZLeeCdkd�G d d � d eI��ZMeeCdkd�G d d � d eI��ZNeeCdkd�G d!d"� d"eI��ZOeeCdkd�G d#d$� d$e"jPjQ��ZRG d%d&� d&�ZSeeCdkd�G d'd(� d(eIeS��ZTeeCdkd�G d)d*� d*eIeS��ZUeeCdkd�G d+d,� d,eI��ZVeeCdkd�G d-d.� d.eI��ZWeeCdkd�G d/d0� d0eI��ZXee"j"jY�G d1d2� d2��ZZeeCdkd�G d3d4� d4eI��Z[eeCdkd�G d5d6� d6eI��Z\G d7d8� d8e3j]�Z^eeCdkd�G d9d:� d:eI��Z_eeCdkd�G d;d<� d<eI��Z`G d=d>� d>�ZaeeCdkd�G d?d@� d@eI��ZbeeCdkd�G dAdB� dBeI��ZcddCl5mdZdmeZemfZf G dDdE� dEeejg�ZhdFdG� ZidHdI� ZjeeCdkd�G dJdK� dKeI��ZkeeCdkd�G dLdM� dMeI��ZlG dNdO� dO�ZmeeCdkd�G dPdQ� dQeI��ZnG dRdS� dS�ZoeeCdkd�G dTdU� dUeI��ZpG dVdW� dW�ZqG dXdY� dYe"jrjs�ZtG dZd[� d[e"jrju�ZvG d\d]� d]�ZweeCdkd�G d^d_� d_eI��ZxG d`da� da�ZyG dbdc� dc�ZzG ddde� de�Z{eeCdkd�G dfdg� dgeI��Z|dS )hz.
Tests for large portions of L{twisted.mail}.
� N)�md5)�skipIf)� Interface� implementer)� verifyClass)�cred�mail)�address�defer�
interfaces�protocol�reactor�task)�Deferred)�CannotListenError�DNSLookupError� ProcessDone�ProcessTerminated)�pop3�smtp)�_AttemptManager)�dns)� Record_CNAME� Record_MX�RRHeader)� DNSNameError)�failure�log)�FilePath)� platformType)�LineSendingProtocol�MemoryReactorClock�StringTransport)�TestCase�posixz twisted.mail only works on posixc @ sD e Zd Zeejdkd�dd� �Zdd� Zdd� Zd d
� Z d d � Z
d
S )�DomainWithDefaultsTests�� �not ported to Python 3c C s� dd� t d�D �}tj�|d�}| �t|�d� | �tt|��tt d��� | �t|�� �tt|��� t|�� �}|� � | �|dd� t d�D �� t|�
� �}|� � | �|tt dd��� |� � }|� � | �|dd� t d�D �� |� � }|� � | �|tt dd��� t d�D ] }| �|| |d � | �|�
|�|d � | �||v � q�|d = |d
= |d = | �t|�d � | �|d d� | �|d
d� | �|d d� |�d d
dd�� | �t|�d� | �|d d � | �|d d
� | �|d d� |�� | �t|�d� | �|�dd�d� | �|d d� | �|�� d� | �t|�d� |�� }| �|j|j� | �|j|j� d S )Nc S s i | ]}||d �qS ��
� ��.0�xr+ r+ �=/usr/lib/python3/dist-packages/twisted/mail/test/test_mail.py�
<dictcomp>@ � z7DomainWithDefaultsTests.testMethods.<locals>.<dictcomp>r* �Defaultc S � g | ]}||d f�qS r) r+ r, r+ r+ r/ �
<listcomp>I � z7DomainWithDefaultsTests.testMethods.<locals>.<listcomp>� c S r3 r) r+ r, r+ r+ r/ r4 Q r5 � � � � r+ �*)�a�b�cr< r= r> r �key�value)r? r@ )�ranger �DomainWithDefaultDict� assertEqual�len�list�iter�iterkeys� iteritems�sort�
itervalues�items�values�get�
assertTrue�update�clear�
setdefault�popitem�copy�domains�default)�self�drK rL r. �dcopyr+ r+ r/ � testMethods> sP z#DomainWithDefaultsTests.testMethodsc C sF t j �i d�}| �|jj||�� d|d< | �tddi�||�� dS )a�
Assert that the class name of a L{mail.mail.DomainWithDefaultDict}
instance and the string-formatted underlying domain dictionary both
appear in the string produced by the given string-returning function.
@type stringifier: one-argument callable
@param stringifier: either C{str} or C{repr}, to be used to get a
string to make assertions against.
r2 r@ r? N)r rB �assertIn� __class__�__name__�str)rV � stringifier�domainr+ r+ r/ �_stringificationTestv s
z,DomainWithDefaultsTests._stringificationTestc C � | � t� dS )z�
L{DomainWithDefaultDict.__str__} should return a string including
the class name and the domain mapping held by the instance.
N)r` r] �rV r+ r+ r/ �test_str� � z DomainWithDefaultsTests.test_strc C ra )z�
L{DomainWithDefaultDict.__repr__} should return a string including
the class name and the domain mapping held by the instance.
N)r` �reprrb r+ r+ r/ � test_repr� rd z!DomainWithDefaultsTests.test_reprc C sf t j �i d�}|�d� d}| �| jg�}| �dt|�� | �t|d d � | �||d d � dS ) z,
has_key is now deprecated.
r2 �anythingzotwisted.mail.mail.DomainWithDefaultDict.has_key was deprecated in Twisted 16.3.0. Use the `in` keyword instead.� r �category�messageN)r rB �has_key�
flushWarnings�test_has_keyDeprecationrC rD �DeprecationWarning)rV �sutrj �warningsr+ r+ r/ rm � s
�z/DomainWithDefaultsTests.test_has_keyDeprecationN) r\ �
__module__� __qualname__r �sys� version_inforY r` rc rf rm r+ r+ r+ r/ r% < s
7 r% c @ �, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� BounceTestsc C � t j �� | _d S �N)r � BounceDomainr_ rb r+ r+ r/ �setUp� � zBounceTests.setUpc C s | � tj| jjd� d S )Nzany user)� assertRaisesr � AddressErrorr_ �existsrb r+ r+ r/ �
testExists� s zBounceTests.testExistsc C s | � | j�dd�d� d S )Nzrandom q emailerr F)rC r_ � willRelayrb r+ r+ r/ � testRelay� s zBounceTests.testRelayc C s& | j �dd� | �tj| j jd� d S )N�bob�password)r_ �addUserr| r � SMTPBadRcptr~ rb r+ r+ r/ � testAddUser� s zBounceTests.testAddUserN)r\ rq rr rz r r� r� r+ r+ r+ r/ rv � s
rv c @ � e Zd ZdZdd� ZdS )�BounceWithSMTPServerTestsz_
Tests for L{twisted.mail.mail.BounceDomain} with
L{twisted.mail.smtp.SMTPServer}.
c C s� t j �� }t j �� }|�d|� t j�|�}|�d�}t j�|dd�}||_t � }|�
|� |� d� |� d� |� d� |� d� | � |j
� |�d� | �|�� �� �d�d d � dS )
zo
Incoming emails to a SMTP server with L{twisted.mail.mail.BounceDomain}
are rejected.
s foo.comNs HELO baz.nets MAIL FROM:<a@baz.net>s RCPT TO:<any@foo.com>s QUITs
���s( 550 Cannot receive for specified address)r � MailServicery � addDomain� protocols� SMTPFactory�
buildProtocol�SMTPDomainDelivery�deliveryr" �makeConnection� lineReceivedrN �
disconnecting�connectionLostrC r@ �strip�split)rV �servicer_ �factoryr � deliverer� transportr+ r+ r/ �
test_rejected� s&
�z'BounceWithSMTPServerTests.test_rejectedN)r\ rq rr �__doc__r� r+ r+ r+ r/ r� � s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zeej d kd
�d d � �Z
eej d kd
�d
d� �Z dS )�FileMessageTestsc C s6 d| _ d| _t| j d�| _tj�| j| j | j�| _d S )NzfileMessage.testFilezfinal.fileMessage.testFile�w)�name�final�open�fr � FileMessage�fprb r+ r+ r/ rz � s zFileMessageTests.setUpc C sn z| j �� W n ty Y nw zt�| j� W n ty" Y nw z t�| j� W d S ty6 Y d S w rx )r� �close�
BaseException�os�remover� r� rb r+ r+ r/ �tearDown� s � � �zFileMessageTests.tearDownc C s | j �� �| j�S rx )r� � eomReceived� addCallback� _cbFinalNamerb r+ r+ r/ �
testFinalName� � z FileMessageTests.testFinalNamec C s4 | � || j� | �| jj� | �tj�| j �� d S rx )
rC r� rN r� �closed� assertFalser� �pathr~ r� �rV �resultr+ r+ r/ r� � s z FileMessageTests._cbFinalNamer& r( c C sh d}|� � D ]}| j�|� q| j�� t| j��}| �|�� |� W d � d S 1 s-w Y d S )Nz"first line
second line
third line
)�
splitlinesr� r� r� r� r� rC �read)rV �contents�liner� r+ r+ r/ � testContents� s
"�z FileMessageTests.testContentsc C sT d}|� � D ]}| j�|� q| j�� | �tj�| j�� | �tj�| j �� d S )Nzfirst line
second line
)
r� r� r� r� r� r� r� r~ r� r� )rV r� r� r+ r+ r/ �testInterrupted s
z FileMessageTests.testInterruptedN) r\ rq rr rz r� r� r� r rs rt r� r� r+ r+ r+ r/ r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�MailServiceTestsc C rw rx )r r� r� rb r+ r+ r/ rz r{ zMailServiceTests.setUpc C s� | j �� }| �t|tj�� | �|�d�tj� | j � � }| �t|tj�� | �|�d�t
j � | j � � }| �t|tj�� | �|�d�t
j
� d S )N)� 127.0.0.1�90 )r� �getPOP3FactoryrN �
isinstancer �
ServerFactoryr� r �POP3�getSMTPFactoryr �SMTP�getESMTPFactory�ESMTP)rV r� r+ r+ r/ �
testFactories s
z MailServiceTests.testFactoriesc C sR t � }t � }|| jjd< || jjd< | �| j�d�|u � | �| j�� |u � d S )Nr_ � )�objectr� �portalsrN � lookupPortal�
defaultPortal)rV �o1�o2r+ r+ r/ � testPortals s z MailServiceTests.testPortalsN)r\ rq rr rz r� r� r+ r+ r+ r/ r�
s
r� c @ sh e Zd ZdZdd� Zdd� Zeejdkd�dd � �Z eejdkd�d
d � �Z
d d
� Z dd� Z dd� Z
dS )�StringListMailboxTestszc
Tests for L{StringListMailbox}, an in-memory only implementation of
L{pop3.IMailbox}.
c C sJ t j�g d��}| �|�d�d� | �|�d�d� | �|�d�d� dS )z�
L{StringListMailbox.listMessages} returns the length of the message at
the offset into the mailbox passed to it.
)�abc�abr< r r' rh r7 N�r �maildir�StringListMailboxrC � listMessages�rV �mailboxr+ r+ r/ �test_listOneMessage0 s z*StringListMailboxTests.test_listOneMessagec C s( t j�g d��}| �|�� g d�� dS )z�
L{StringListMailbox.listMessages} returns a list of the lengths of all
messages if not passed an index.
)r< r� r� )rh r' r7 Nr� r� r+ r+ r/ �test_listAllMessages: s z+StringListMailboxTests.test_listAllMessagesr& r( c C s* t j�ddg�}| �|�d��� d� dS )z�
L{StringListMailbox.getMessage} returns a file-like object from which
the contents of the message at the given offset into the mailbox can be
read.
�fooz
real contentsrh N)r r� r� rC �
getMessager� r� r+ r+ r/ �test_getMessageB s z&StringListMailboxTests.test_getMessagec C s, t j�ddg�}| �|�d�|�d�� dS )z�
L{StringListMailbox.getUidl} returns a unique identifier for the
message at the given offset into the mailbox.
r� �barr rh N)r r� r� �assertNotEqual�getUidlr� r+ r+ r/ � test_getUidlL s z#StringListMailboxTests.test_getUidlc C s@ t j�dg�}|�d� | �|�d�d� | �|�� dg� dS )z�
L{StringListMailbox.deleteMessage} marks a message for deletion causing
further requests for its length to return 0.
r� r N)r r� r� �
deleteMessagerC r� r� r+ r+ r/ �test_deleteMessageU s
z)StringListMailboxTests.test_deleteMessagec C sH t j�dg�}|�d� |�� | �|�d�d� | �|�� dg� dS )z�
L{StringListMailbox.undeleteMessages} causes any messages marked for
deletion to be returned to their original state.
r� r r' N)r r� r� r� �undeleteMessagesrC r� r� r+ r+ r/ �test_undeleteMessages_ s
z,StringListMailboxTests.test_undeleteMessagesc C sP t j�dg�}|�d� |�� |�� | �|�d�d� | �|�� dg� dS )zy
L{StringListMailbox.sync} causes any messages as marked for deletion to
be permanently deleted.
r� r N)r r� r� r� �syncr� rC r� r� r+ r+ r/ � test_syncj s
z StringListMailboxTests.test_syncN)r\ rq rr r� r� r� r rs rt r� r� r� r� r� r+ r+ r+ r/ r� ) s
r� c @ s0 e Zd ZdZdZdZdd� Zdd� Zdd� ZdS ) �&FailingMaildirMailboxAppendMessageTaskTc C s | j r
t�|||�S ttjd���NzFaked Permission Problem)�
_openstater� r� �OSError�errno�EPERM)rV �fn�attr�moder+ r+ r/ �osopen s z-FailingMaildirMailboxAppendMessageTask.osopenc C � | j r t�||�S ttjd��)NzFaked Space problem)� _writestater� �writer� r� �ENOSPC)rV �fh�datar+ r+ r/ �oswrite� � z.FailingMaildirMailboxAppendMessageTask.oswritec C r� r� )� _renamestater� �renamer� r� r� )rV �oldname�newnamer+ r+ r/ �osrename� r� z/FailingMaildirMailboxAppendMessageTask.osrenameN) r\ rq rr r� r� r� r� r� r� r+ r+ r+ r/ r� w s r� c @ r� )�_AppendTestMixinz�
Mixin for L{MaildirMailbox.appendMessage} test cases which defines a helper
for serially appending multiple messages to a mailbox.
c s: g �� ��fdd�}t �|� ��� }|��fdd�� |S )z�
Deliver the given messages one at a time. Delivery is serialized to
guarantee a predictable order in the mailbox (overlapped message deliver
makes no guarantees about which message which appear first).
c 3 s, � �D ]} � � | �}|��j� |V qd S rx )�
appendMessager� �append)�mrW ��mbox�messages�resultsr+ r/ r� � s �
�z0_AppendTestMixin._appendMessages.<locals>.appendc � � S rx r+ ��ignored)r r+ r/ �<lambda>� � z2_AppendTestMixin._appendMessages.<locals>.<lambda>)r � cooperate�whenDoner� )rV r r r� rW r+ r r/ �_appendMessages� s
z _AppendTestMixin._appendMessagesN)r\ rq rr r� r r+ r+ r+ r/ r� � s r� c @ sJ e Zd ZdZdd� Zdd� Zddd�Zeej d kd
�d d � �Z
d
d� Z dS )�MaildirAppendStringTests�O
Tests for L{MaildirMailbox.appendMessage} when invoked with a C{str}.
c C � | � � | _tj�| j� d S rx ��mktemprW r r� �initializeMaildirrb r+ r+ r/ rz � �
z MaildirAppendStringTests.setUpc C s |� d�}| �|t�S )N�TEST)r� �
assertFailure� Exception�rV r r rW r+ r+ r/ �_append� s
z MaildirAppendStringTests._appendNc C sz |dur| � t|jd|jj� ||j_|dur&| � t|jd|jj� ||j_|dur;| � t|jd|jj� ||j_dS dS )a�
Change the behavior of future C{rename}, C{write}, or C{open} calls made
by the mailbox C{mbox}.
@param rename: If not L{None}, a new value for the C{_renamestate}
attribute of the mailbox's append factory. The original value will
be restored at the end of the test.
@param write: Like C{rename}, but for the C{_writestate} attribute.
@param open: Like C{rename}, but for the C{_openstate} attribute.
Nr� r� r� )�
addCleanup�setattr�
AppendFactoryr� r� r� )rV r r r� r� r� r+ r+ r/ � _setState� s,
��� �z"MaildirAppendStringTests._setStater& r( c C sV t j�| j�}t|_| �|dd� tdd�D ��}|�| j dgd � |�| j
|� |S )��
L{MaildirMailbox.appendMessage} returns a L{Deferred} which fires when
the message has been added to the end of the mailbox.
c S s g | ]}d | �qS )�Xr+ �r- �ir+ r+ r/ r4 � s z8MaildirAppendStringTests.test_append.<locals>.<listcomp>rh � Nr* ) r r� �MaildirMailboxrW r� r r rA r� rC �
_cbTestAppend)rV r rW r+ r+ r/ � test_append� s z$MaildirAppendStringTests.test_appendc s� | � t� �� �d� | � � fdd�td�D �ttdd��� | jd� dd� | �d� �}|j| j� d dd
� |�| j� � |j| j� d dd � |�| j� � |j| j� d d � |S )
��
Check that the mailbox has the expected number (ten) of messages in it,
and that each has the expected contents, and that they are in the same
order as that in which they were appended.
r* c � g | ] }t � �|��� ��qS r+ �rD r� r� r �r r+ r/ r4 � � z:MaildirAppendStringTests._cbTestAppend.<locals>.<listcomp>rh r! NF)r� T)r� r� )r� r� )r� )rC rD r� rA rE r r r� r r+ r( r/ r# � s � z&MaildirAppendStringTests._cbTestAppend)NNN) r\ rq rr r� rz r r r rs rt r$ r# r+ r+ r+ r/ r
� s
#
r
c @ s8 e Zd ZdZdd� Zeejdkd�dd� �Zdd � Z d
S ) �MaildirAppendFileTestsr c C r rx r rb r+ r+ r/ rz r z MaildirAppendFileTests.setUpr& r( c C sx t j�| j�}g }tdd�D ] }t�� }|�d| � |�dd� |� |� | �
|j � q| � ||�}|�
| j|� |S )r rh r! r r )r r� r" rW rA �tempfile�
TemporaryFiler� �seekr� r r� r r� r# )rV r r r �temprW r+ r+ r/ r$ s
z"MaildirAppendFileTests.test_appendc s@ | � t� �� �d� | � � fdd�td�D �ttdd��� dS )r% r* c r& r+ r' r r( r+ r/ r4 $ r) z8MaildirAppendFileTests._cbTestAppend.<locals>.<listcomp>rh r! N)rC rD r� rA rE )rV r� r r+ r( r/ r# s �z$MaildirAppendFileTests._cbTestAppendN)
r\ rq rr r� rz r rs rt r$ r# r+ r+ r+ r/ r* � s
r* c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zeej d kd
�d d � �Z
d
S )� MaildirTestsc C r rx r rb r+ r+ r/ rz * r zMaildirTests.setUpc C � t �| j� d S rx )�shutil�rmtreerW rb r+ r+ r/ r� . r{ zMaildirTests.tearDownc C s� | j }tj�|d�}| �tj�|�otj�|�� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� | �tj�tj�|d��� d S )N�.Trash�new�cur�tmp)rW r� r� �joinrN r~ �isdir)rV rW �trashr+ r+ r/ �testInitializer1 s z MaildirTests.testInitializerc C sJ t �� }|�d� tj�|�}|�� }|�d� |�� }| �||k � dS )a
Each call to L{_MaildirNameGenerator.generate} returns a unique
string suitable for use as the basename of a new message file. The
names are ordered such that those generated earlier sort less than
those generated later.
g�������?N)r �Clock�advancer r� �_MaildirNameGenerator�generaterN )rV �clock� generator� firstName�
secondNamer+ r+ r/ �test_nameGeneratorD s
z MaildirTests.test_nameGeneratorr& r( c s� t jj� tjj�� �fdd�dD �}d}|D ]&}t� | j|�d��}|�d| � W d � n1 s3w Y |d }qtj� | j�}| �
|� � t t
dd��� | �
|� d�d � | �
|� d
�d � | �
|�d ��� d � | �
|�d��� d
� i }t
d�D ]}|�|�}| �||v � d||< qt j�|d
�\}}|�d
� | �
|� d
�d� | �t j�� | jdd|��� | �t j�� | j|d
��� |�� | �
|� d
�d � | �t j�� | jdd|��� | �t j�� | j|d
��� dS )zb
Exercise the methods of L{IMailbox} as implemented by
L{MaildirMailbox}.
c s& g | ]}t d �D ]}� |�� ��qqS )� )rA )r- r= r. ��j�nr+ r/ r4 ] s & z-MaildirTests.test_mailbox.<locals>.<listcomp>)r5 r4 rh r� r. Nr! r7 rD r9 �xxxxxxx�xxr* r r3 r5 )r� r� r7 r r� �_generateMaildirNamer� rW r� r" rC r� rE rA r� r� r� r� r� r� rN r~ r� ) rV �msgsr r� �fObj�mbrW �u�pr+ rE r/ � test_mailboxU s: �
"zMaildirTests.test_mailboxN) r\ rq rr rz r� r: rC r rs rt rP r+ r+ r+ r/ r/ ( s r/ c @ r� )�AbstractMaildirDomainTestszB
Tests for L{twisted.mail.maildir.AbstractMaildirDomain}.
c C s t tjjtjj� dS )zW
L{maildir.AbstractMaildirDomain} implements L{mail.IAliasableDomain}.
N)r r �IAliasableDomainr� �AbstractMaildirDomainrb r+ r+ r/ �test_interface� � z)AbstractMaildirDomainTests.test_interfaceN)r\ rq rr r� rT r+ r+ r+ r/ rQ � s rQ c @ s� e Zd ZdZdd� Zdd� Zeejdkd�dd � �Z d
d � Z
eejdkd�d d
� �Z eejdkd�dd� �Z eejdkd�dd� �Z
dS )�MaildirDirdbmDomainTestsz+
Tests for L{MaildirDirdbmDomain}.
c C s. | � � | _tj�� | _tj�| j| j�| _dS )zw
Create a temporary L{MaildirDirdbmDomain} and parent
L{MailService} before running each test.
N)r �Pr r� �Sr� �MaildirDirdbmDomain�Drb r+ r+ r/ rz � s
z MaildirDirdbmDomainTests.setUpc C s t �| j� dS )z_
Remove the temporary C{maildir} directory when the test has
finished.
N)r1 r2 rW rb r+ r+ r/ r� � s z!MaildirDirdbmDomainTests.tearDownr& r( c C sv d}|D ] \}}| j �||� q|D ]&\}}| �|| j jv � | �| j j| |� | �tj�tj�| j |��� qdS )z�
L{MaildirDirdbmDomain.addUser} accepts a user and password
argument. It stores those in a C{dbm} dictionary
attribute and creates a directory for each user.
))�user1�pwd1)�user2�pwd2)�user3�pwd3N)
rZ r� rN �dbmrC r� r� r~ r7 rW )rV �toAddrN rO r+ r+ r/ � test_addUser� s �z%MaildirDirdbmDomainTests.test_addUserc C sN | j �� }| �t|�d� | �tjj�|d �� | �tj j
|d j v � dS )z�
L{MaildirDirdbmDomain.getCredentialsCheckers} initializes and
returns one L{ICredentialsChecker} checker by default.
rh r N) rZ �getCredentialsCheckersrC rD rN r �checkers�ICredentialsChecker�
providedBy� credentials�IUsernamePassword�credentialInterfaces)rV �credsr+ r+ r/ �test_credentials� s
�z)MaildirDirdbmDomainTests.test_credentialsc C s� G dd� dt �}| j�dd� | �t| jjdd|� | j�ddtj�}| �t |�d� | �
|d tju � | �
tj� |d �� |d � dS )
a?
L{MaildirDirdbmDomain.requestAvatar} raises L{NotImplementedError}
unless it is supplied with an L{pop3.IMailbox} interface.
When called with an L{pop3.IMailbox}, it returns a 3-tuple
containing L{pop3.IMailbox}, an implementation of that interface
and a NOOP callable.
c @ s e Zd ZdS )z?MaildirDirdbmDomainTests.test_requestAvatar.<locals>.ISomethingN)r\ rq rr r+ r+ r+ r/ �
ISomething� s rm �userr� Nr' r rh r7 ) r rZ r� r| �NotImplementedError�
requestAvatarr �IMailboxrC rD rN rg )rV rm �tr+ r+ r/ �test_requestAvatar� s
�z+MaildirDirdbmDomainTests.test_requestAvatarc C sb | j �dd� | j �� d }tj�dd�}| �tjj|j |� tj�dd�}| �
|� |�d� dS )a
L{DirdbmDatabase.requestAvatarId} raises L{UnauthorizedLogin} if
supplied with invalid user credentials.
When called with valid credentials, L{requestAvatarId} returns
the username associated with the supplied credentials.
rn r� r �wrong passwordN) rZ r� rd r rh �UsernamePasswordr| �error�UnauthorizedLogin�requestAvatarIdrC )rV �databaserk r+ r+ r/ �test_requestAvatarId� s z-MaildirDirdbmDomainTests.test_requestAvatarIdc C sz | j �dd� | �| j �d�tj�| j jd�� d| j _| � | j �d�d� d| j _| �| j �d�tj�| j jd�� dS )a]
L{MaildirDirdbmDomain.userDirectory} is supplied with a user name
and returns the path to that user's maildir subdirectory.
Calling L{MaildirDirdbmDomain.userDirectory} with a
non-existent user returns the 'postmaster' directory if there
is a postmaster or returns L{None} if there is no postmaster.
rn r� F�nouserNT�
postmaster)
rZ r� rC �
userDirectoryr� r� r7 �rootr| �assertIdenticalrb r+ r+ r/ �test_userDirectory� s ��z+MaildirDirdbmDomainTests.test_userDirectoryN)r\ rq rr r� rz r� r rs rt rc rl rs rz r� r+ r+ r+ r/ rV � s
rV c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �StubAliasableDomainz>
Minimal testable implementation of IAliasableDomain.
Nc C � t � ��z{
No test coverage for invocations of this method on domain objects,
so we just won't implement it.
�ro �rV rn �memor+ r+ r/ r~ � zStubAliasableDomain.existsc C r� r� r� )rV rn r� r+ r+ r/ r� r� zStubAliasableDomain.addUserc C s g S )a!
This needs to succeed in order for other tests to complete
successfully, but we don't actually assert anything about its
behavior. Return an empty list. Sometime later we should return
something else and assert that a portal got set up properly.
r+ rb r+ r+ r/ rd � z*StubAliasableDomain.getCredentialsCheckersc C �
|| _ dS )zG
Just record the value so the test can check it later.
N)�
aliasGroup)rV �aliasesr+ r+ r/ �
setAliasGroup s
z!StubAliasableDomain.setAliasGrouprx )r\ rq rr r� r~ r� rd r� r+ r+ r+ r/ r� s
r� c @ st e Zd Zdd� Zdd� Zdd� Zeejdkd�d d
� �Z eejdkd�d d � �Z
d
d� Z dd� Z dd� Z
dd� ZdS )�ServiceDomainTestsc C sr t j �� | _t j�| jd �| _| j| j_d| j_d| j_| � � | _
t j � | j| j
�}|�
dd� | j�d|� d S )Nr �hostname� user� password� test.domain)r r� rX r� �DomainDeliveryBaserZ r� � protocolName�hostr �tmpdirr� rY r� r� �rV r_ r+ r+ r/ rz ( s
zServiceDomainTests.setUpc C r0 rx �r1 r2 r� rb r+ r+ r/ r� 4 r{ zServiceDomainTests.tearDownc C s4 t � }t� }|| j_| j�d|� | �|j|� dS )z�
Test that adding an IAliasableDomain to a mail service properly sets
up alias group references and such.
� example.comN)r� r� rX r� r� r r� )rV r� r_ r+ r+ r/ �testAddAliasableDomain7 s
z)ServiceDomainTests.testAddAliasableDomainr& r( c C sZ | j �dt�d�dg�}t�|�}tj�� }|� |�}| �
t |� � �d� | �
d|� d S )N)�
remotehostz123.232.101.234z<someguy@someplace>zuser@host.namerh �Received)rZ �receivedHeaderr �Address�io�BytesIO�email�parser�Parser�parserC rD rK rZ )rV �hdrr� � emailParserr r+ r+ r/ �testReceivedHeaderB s �
z%ServiceDomainTests.testReceivedHeaderc C s( t �ddd d�}t�| jj|��| j�S )Nzuser@test.domain�helo�wherever@whatever)r �Userr
�
maybeDeferredrZ �
validateTor� �
_cbValidateTo�rV rn r+ r+ r/ �testValidateToO s �z!ServiceDomainTests.testValidateToc C s | � t|�� d S rx )rN �callabler� r+ r+ r/ r� V r� z ServiceDomainTests._cbValidateToc C �* t �ddd d�}| �t�| jj|�t j�S )Nzresu@test.domainr� r� �r r� r r
r� rZ r� r� r� r+ r+ r/ �testValidateToBadUsernameY � �z,ServiceDomainTests.testValidateToBadUsernamec C r� )Nzuser@domain.testr� r� r� r� r+ r+ r/ �testValidateToBadDomain_ r� z*ServiceDomainTests.testValidateToBadDomainc C s� d}t �d�}| �| j�||�|u � d}t �d�}| �| j�||�|u � d}t �d�}| �| j�||�|u � | �t j| jjd |� d S )N)r� r� z<user@hostname>)r� �1.2.3.4z<>)r r� rN rZ � validateFromr| �
SMTPBadSender)rV r� �originr+ r+ r/ �testValidateFrome s
z#ServiceDomainTests.testValidateFromN)r\ rq rr rz r� r� r rs rt r� r� r� r� r� r� r+ r+ r+ r/ r� & s
r� c @ s� e Zd Zdd� Zdd� Zeejdkd�dd� �Zd d
� Z eejdkd�d d � �Z
eejdkd�d
d� �Z eejdkd�dd� �Z dd� Z
eejdkd�dd� �Zeejdkd�dd� �ZdS )�VirtualPOP3Testsc C s� | � � | _tj�� | _tj�| j| j�| _| j�dd� | j� d| j� t
j � | j�}t
|j| j�� � | | jjd< | jjd< tj�� | _| j| j_d| j_d S )Nr� r� r� r� z<unit test magic>)r r� r r� rX r� rY rZ r� r� r �portal�Portal�map�registerCheckerrd r� r� � VirtualPOP3rW r� �magic)rV r� r+ r+ r/ rz w s
zVirtualPOP3Tests.setUpc C r0 rx r� rb r+ r+ r/ r� � r{ zVirtualPOP3Tests.tearDownr& r( c C s* t | jjd ��� }| j�d|��| j�S )Nr� rn )r rW r� � hexdigest�authenticateUserAPOPr� �_cbAuthenticateAPOP�rV �respr+ r+ r/ �testAuthenticateAPOP� s �z%VirtualPOP3Tests.testAuthenticateAPOPc C �F | � t|�d� | � |d tj� | �tj�|d �� |d � d S �Nr' r rh r7 �rC rD r rq rN rg r� r+ r+ r/ r� � � z$VirtualPOP3Tests._cbAuthenticateAPOPc C s. t | jjd ��� }| �| j�d|�tjj�S )Nr� �resu) r rW r� r� r r� r rv rw r� r+ r+ r/ �!testAuthenticateIncorrectUserAPOP� s �z2VirtualPOP3Tests.testAuthenticateIncorrectUserAPOPc C s& t d��� }| �| j�d|�tjj�S )Nz wrong digestrn )r r� r rW r� r rv rw r� r+ r+ r/ �%testAuthenticateIncorrectResponseAPOP� s �z6VirtualPOP3Tests.testAuthenticateIncorrectResponseAPOPc C s | j �dd��| j�S )Nrn r� )rW �authenticateUserPASSr� �_cbAuthenticatePASSrb r+ r+ r/ �testAuthenticatePASS� s �z%VirtualPOP3Tests.testAuthenticatePASSc C r� r� r� r� r+ r+ r/ r� � r� z$VirtualPOP3Tests._cbAuthenticatePASSc C � | � | j�dd�tjj�S )Nr� r� �r rW r� r rv rw rb r+ r+ r/ �testAuthenticateBadUserPASS� � �z,VirtualPOP3Tests.testAuthenticateBadUserPASSc C r� )Nrn rt r� rb r+ r+ r/ � testAuthenticateBadPasswordPASS� r� z0VirtualPOP3Tests.testAuthenticateBadPasswordPASSN)r\ rq rr rz r� r rs rt r� r� r� r� r� r� r� r� r+ r+ r+ r/ r� u s"
r� c @ s e Zd Zdd� ZdS )�emptyc C s d S rx r+ rb r+ r+ r/ �__init__� s zempty.__init__N)r\ rq rr r� r+ r+ r+ r/ r� � s r� c @ s$ e Zd Zeejdkd�dd� �ZdS )�
RelayTestsr& r( c s� t j �� }t j�|�}t�d�t�ddd�g}t�ddd�t�ddd�g}|D ]'� t� }d |_d
|_ t� |_
t� |j
_ � fd d �|j
j _ | �
t|�|��� q'|D ]&� t� }d
|_t� |_
t� |j
_ � fdd �|j
j _ d|_ | �tj|j|� qQd S )Nz/var/run/mail-relay�TCPr� r� z 192.168.2.1�> r� i� z user@hostz tsoh@resuc r rx r+ r+ ��peerr+ r/ r � r z'RelayTests.testExists.<locals>.<lambda>z
some@placec r rx r+ r+ r� r+ r/ r � r z who@cares)r r� �relay� DomainQueuerr � UNIXAddress� IPv4Addressr� �orig�destr r� �getPeerrN r� r~ r| r r� )rV r� r_ �doRelay� dontRelayrn r+ r� r/ r � s2
� �
�zRelayTests.testExistsN)r\ rq rr r rs rt r r+ r+ r+ r/ r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � RelayerTestsc C s� | � � | _t�| j� g | _td�D ]J}tj�| jd|f �}t|d d��}t �
d|f d|f g|� W d � n1 s?w Y t|d d�}|� |� |� d d � | j�
|� qtj�� | _| j�| j� d S )
Nr* �body-%d�-H�wb�from-%d�to-%d�-Dr� r )r r� r� �mkdir� messageFilesrA r� r7 r� �pickle�dumpr� r- r� r r� � RelayerMixin�R� loadMessages)rV r r� r� r+ r+ r/ rz � s
�
zRelayerTests.setUpc C r0 rx r� rb r+ r+ r/ r� � r{ zRelayerTests.tearDownc C sP t d�D ]}| �| j�� d|f � | j�dd d d d � q| �| j�� d � d S )Nr* r� � )rA rC r� � getMailFrom�sentMail�rV r r+ r+ r/ � testMailFrom s zRelayerTests.testMailFromc C sR t d�D ]}| �| j�� d|f g� | j�dd d d d � q| �| j�� d � d S )Nr* r� r� )rA rC r� � getMailTor� r� r+ r+ r/ �
testMailTo s zRelayerTests.testMailToc C sd t d�D ]"}tj�| jd|f �}| �| j�� �� |� | j� dd d d d � q| �| j�� d � d S )Nr* r� r� )
rA r� r� r7 r� rC r� � getMailDatar� r� )rV r r� r+ r+ r/ � testMailData s
zRelayerTests.testMailDataN)r\ rq rr rz r� r� r� r� r+ r+ r+ r/ r� � s r� c @ ru )
�Managerc C s g | _ g | _g | _d S rx )�successr �donerb r+ r+ r/ r� �
zManager.__init__c C � | j �||f� d S rx )r� r� �rV r� rj r+ r+ r/ �
notifySuccess � zManager.notifySuccessc C r rx )r r� r r+ r+ r/ �
notifyFailure r zManager.notifyFailurec C � | j �|� d S rx )r� r� )rV r� r+ r+ r/ �
notifyDone r{ zManager.notifyDoneN)r\ rq rr r� r r r r+ r+ r+ r/ r� s
r� c @ ru )
�ManagedRelayerTestsc C sd t � | _ttddd��| _t� | _tj� | j�| _
| jd d � | j
_| jd d � | j
_ | j| j
_d S )Nr r6 r7 ) r� �managerrE rA r r� r� r � relaymanager�ManagedRelayerMixinr� �namesrb r+ r+ r/ rz & s zManagedRelayerTests.setUpc �D � j D ] }� j�dd d d d � q� �� jj� fdd�� j D �� d S )Nr� c � g | ]}� j |f�qS r+ �r� �r- r rb r+ r/ r4 4 r1 z>ManagedRelayerTests.testSuccessfulSentMail.<locals>.<listcomp>)r r� r� rC r r� r� r+ rb r/ �testSuccessfulSentMail/ �
�z*ManagedRelayerTests.testSuccessfulSentMailc r
)Ni& c r r+ r r rb r+ r/ r4 < r1 z:ManagedRelayerTests.testFailedSentMail.<locals>.<listcomp>)r r� r� rC r r r� r+ rb r/ �testFailedSentMail7 r z&ManagedRelayerTests.testFailedSentMailc C s, | j �t�t� �� | �| jj| jg� d S rx ) r� r� r �Failurer rC r r� r� rb r+ r+ r/ �testConnectionLost? s z&ManagedRelayerTests.testConnectionLostN)r\ rq rr rz r r r r+ r+ r+ r/ r $ s
r c @ sl e Zd Zdd� Zdd� Zeejdkd�dd� �Zeejdkd�d d
� �Z eejdkd�d d � �Z
d
d� Z dS )�DirectoryQueueTestsc C s� | � � | _t�| j� tj�| j�| _d| j_t d�D ]/}| j�
� \}}|� t � d|g|� W d � n1 s9w Y |�
d|f � |�� q| j�� d S )NF� �headers body: %d)r r� r� r� r r
�Queue�queue�noisyrA �createNewMessager� r� r� r� �
readDirectory)rV r �hdrF�msgFr+ r+ r/ rz F s
�
zDirectoryQueueTests.setUpc C r0 rx r� rb r+ r+ r/ r� T r{ z DirectoryQueueTests.tearDownr& r( c C s� | � | j�� � | �t| j�� �d� | j�� }| j�|d � | �t| j�� �d� | j�|d � | �t| j�� �d� d S )Nr r � )rN r �
hasWaitingrC rD �
getWaiting� setRelaying�
setWaiting)rV �waitingr+ r+ r/ � testWaitingW s
z DirectoryQueueTests.testWaitingc C s� | j �� D ]}| j �|� | �t| j �� �dt| j �� � � q| �| j �� � | j �� }| j �|d � | �t| j �� �d� | �t| j �� �d� d S )Nr r rh r ) r r" r# rC rD �
getRelayedr� r! r$ )rV r �relayedr+ r+ r/ � testRelayingc s �
z DirectoryQueueTests.testRelayingc C s~ | j �� d }| j �|� | j �|� | �t| j �� �d� | �t| j �� �d� | �|| j �� v � | �|| j �� v � d S )Nr r )r r" r# r� rC rD r' r� )rV �msgr+ r+ r/ �testDoner s z DirectoryQueueTests.testDonec C sV g }| j �� D ] }|�| j �|�� q|�� td�D ]
}| �|�d�d|g� qd S )Nr r r )r r" r� � getEnveloperI rA rC �pop)rV � envelopesr* r r+ r+ r/ � testEnvelope~ s �z DirectoryQueueTests.testEnvelopeN) r\ rq rr rz r� r rs rt r& r) r+ r/ r+ r+ r+ r/ r D s
r )�client�common�serverc @ � e Zd Zdd� Zddd�ZdS )�
TestAuthorityc C s t j�| � i | _d S rx )r1 � ResolverBaser� � addressesrb r+ r+ r/ r� � s
zTestAuthority.__init__Nc C sv || j v r0|tjkr0g }| j | D ]}t�|tjtjdt�d|��}|�|� qt�|g g f�S t� t
� t� |���S )N�< r )
r6 r �MXr �INr r� r
�succeed�failr r � DomainError)rV r� �cls�type�timeoutr r< r� r+ r+ r/ �_lookup� s zTestAuthority._lookuprx )r\ rq rr r� r@ r+ r+ r+ r/ r4 � � r4 c C s� t � | _t�| jg�}t�|�} tjd|dd�| _| j� � j}z tj
||dd�| _ W n t y9 | j�
� Y nw nqtjd|fgd�| _d S )Nrh r r� �� interface)�servers)r4 �authr2 �DNSServerFactoryr �DNSDatagramProtocolr
� listenTCP�port�getHost� listenUDP�udpPortr �
stopListeningr0 �Resolver�resolver)rV r� r �
portNumberr+ r+ r/ �setUpDNS� s
� �� rQ c C sZ g }|� t�| jj�� |� t�| jj�� z| jj�� W n t y' Y nw t�
|�S rx ) r� r
r� rI rM rL rO �
_parseCall�cancelr� � DeferredList)rV �dlr+ r+ r/ � tearDownDNS� s �
rV c @ s� e Zd ZdZdd� Zdd� Zdd� Zeej dkd �d
d � �Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zeej dkd �d*d+� �Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4S )5�MXTestsz6
Tests for L{mail.relaymanager.MXCalculator}.
c C s* t | � t�� | _tj�| j| j�| _d S rx ) rQ r r; r? r r
� MXCalculatorrO �mxrb r+ r+ r/ rz � s
z
MXTests.setUpc C s t | �S rx )rV rb r+ r+ r/ r� � s zMXTests.tearDownc C s | � tj�| j�jt� dS )zQ
L{MXCalculator}'s default clock is C{twisted.internet.reactor}.
N)r r r
rX rO r? r
rb r+ r+ r/ �test_defaultClock� s �zMXTests.test_defaultClockr& r( c C s" dg| j jd< | j�d��| j�S )N�the.email.test.domainr� )rE r6 rY �getMXr� �_cbSimpleSuccessrb r+ r+ r/ �testSimpleSuccess� s zMXTests.testSimpleSuccessc C s$ | � |jd� | � t|j�d� d S )Nr r[ )rC �
preferencer] r� �rV rY r+ r+ r/ r] � s zMXTests._cbSimpleSuccessc C s d| j _| �| j �d�t�S )NFr� )rY �fallbackToDomainr r\ �IOErrorrb r+ r+ r/ �testSimpleFailure� s zMXTests.testSimpleFailurec C s | � | j�d�t�S �Nr� )r rY r\ r rb r+ r+ r/ � testSimpleFailureWithFallback� r z%MXTests.testSimpleFailureWithFallbackc sF G ��fdd�d�}|� �j _�j ���}� �fdd�}|�|� |S )aZ
Issue an MX request for the given domain and arrange for it to be
responded to with the given records. Verify that the resulting mail
exchange is the indicated host.
@type domain: C{str}
@type records: C{list} of L{RRHeader}
@type correctMailExchange: C{str}
@rtype: L{Deferred}
c � e Zd Z� �fdd�ZdS )z,MXTests._exchangeTest.<locals>.DummyResolverc s& |� kr t ��g g f�S t �t� ��S rx )r
r: r; r )rV r� �r_ �recordsr+ r/ �lookupMailExchange� s z?MXTests._exchangeTest.<locals>.DummyResolver.lookupMailExchangeN�r\ rq rr ri r+ rg r+ r/ �
DummyResolver� � rk c s �� t| j�� � d S rx �rC r] r� )�record)�correctMailExchangerV r+ r/ �gotMailExchange� s z.MXTests._exchangeTest.<locals>.gotMailExchange)rY rO r\ r� )rV r_ rh ro rk rW rp r+ )ro r_ rh rV r/ �
_exchangeTest� s
zMXTests._exchangeTestc C sZ d}d}d}t |tjtd|�d�t |tjtd|�d�t |tjtd|�d�g}| �|||�S )zh
The MX record with the lowest preference is returned by
L{MXCalculator.getMX}.
r� �good.example.com�bad.example.comrh �r� r> �payloadr r7 )r r �TYPErq �rV r_ �good�badrh r+ r+ r/ �test_mailExchangePreference� s �z#MXTests.test_mailExchangePreferencec C sR d}d}d}t |tjtd|�d�t |tjtd|�d�g}| j�|� | �|||�S )z�
L{MXCalculator.getMX} returns the MX record with the lowest preference
which is not also marked as bad.
r� rr rs r rt rh �r r rv rY �markBadrq rw r+ r+ r/ �test_badExchangeExcluded
s � z MXTests.test_badExchangeExcludedc C s^ d}d}d}t |tjtd|�d�t |tjtd|�d�g}| j�|� | j�|� | �|||�S )z�
L{MXCalculator.getMX} returns the MX record with the lowest preference
if all the MX records in the response have been marked bad.
r� rs zworse.example.comr rt rh r{ )rV r_ ry �worserh r+ r+ r/ � test_fallbackForAllBadExchanges s � z'MXTests.test_fallbackForAllBadExchangesc C sb d}d}d}t |tjtd|�d�t |tjtd|�d�g}| j�|� | j�| jj� | �|||�S )z�
L{MXCalculator.getMX} returns the MX record with the lowest preference
if it was last marked bad longer than L{MXCalculator.timeOutBadMX}
seconds ago.
r� rr rs r rt rh ) r r rv rY r| r? r< � timeOutBadMXrq �rV r_ rx �
previouslyBadrh r+ r+ r/ �test_badExchangeExpires+ s �� z MXTests.test_badExchangeExpiresc C sn d}d}d}t |tjtd|�d�t |tjtd|�d�g}| j�|� | j�|� | j�| jj� | � |||�S )z�
L{MXCalculator.getMX} returns the MX record with the lowest preference
if it was marked good after it was marked bad.
r� rr rs r rt rh )
r r rv rY r| �markGoodr? r< r� rq r� r+ r+ r/ �test_goodExchangeUsed? s �� z MXTests.test_goodExchangeUsedc sJ d�d� G � �fdd�d�}|� | j _| j �� �}|�| jt�d�� |S )z�
If an MX lookup succeeds but the result set is empty,
L{MXCalculator.getMX} should try to look up an I{A} record for the
requested name and call back its returned Deferred with that
address.
r� z example.orgc s6 e Zd ZdZ� g g g fiZ� �iZdd� Zdd� ZdS )z9MXTests.test_successWithoutResults.<locals>.DummyResolverah
Fake resolver which will respond to an MX lookup with an empty
result set.
@ivar mx: A dictionary mapping hostnames to three-tuples of
results to be returned from I{MX} lookups.
@ivar a: A dictionary mapping hostnames to addresses to be
returned from I{A} lookups.
c S � t �| j| �S rx )r
r: rY r� r+ r+ r/ ri l r{ zLMXTests.test_successWithoutResults.<locals>.DummyResolver.lookupMailExchangec S r� rx )r
r: r< r� r+ r+ r/ �
getHostByNameo r{ zGMXTests.test_successWithoutResults.<locals>.DummyResolver.getHostByNameN)r\ rq rr r� rY r< ri r� r+ �r_ �ipr+ r/ rk ] s rk �r� �rY rO r\ r� rC r �rV rk rW r+ r� r/ �test_successWithoutResultsS s
z"MXTests.test_successWithoutResultsc C s< G dd� d�}|� | j _| j �d�}|�| jtdd�� |S )a'
Test that if the MX record lookup fails, fallback is enabled, and an A
record is available for the name, then the Deferred returned by
L{MXCalculator.getMX} ultimately fires with a Record_MX instance which
gives the address in the A record for the name.
c @ s e Zd ZdZdd� Zdd� ZdS )zAMXTests.test_failureWithSuccessfulFallback.<locals>.DummyResolverzw
Fake resolver which will fail an MX lookup but then succeed a
getHostByName call.
c S s t �t� �S rx )r
r; r r� r+ r+ r/ ri � s zTMXTests.test_failureWithSuccessfulFallback.<locals>.DummyResolver.lookupMailExchangec S s
t �d�S )Nr� )r
r: r� r+ r+ r/ r� � �
zOMXTests.test_failureWithSuccessfulFallback.<locals>.DummyResolver.getHostByNameN)r\ rq rr r� ri r� r+ r+ r+ r/ rk s rk r_ r� r� r� r� r+ r+ r/ �"test_failureWithSuccessfulFallbackw s
z*MXTests.test_failureWithSuccessfulFallbackc sP d� d�d�G � ��fdd�d�}|� | j _| j �� �}|�| jt�d�� |S )z�
If an MX lookup returns a single CNAME record as a result, MXCalculator
will perform an MX lookup for the canonical name indicated and return
the MX record which results.
�alias.example.com�canonical.example.com�mail.example.comc s e Zd ZdZ� ��fdd�ZdS )z;MXTests.test_cnameWithoutGlueRecords.<locals>.DummyResolverz�
Fake resolver which will return a CNAME for an MX lookup of a name
which is an alias and an MX for an MX lookup of the canonical name.
c sd |� krt �t|tjt��d�gg g f�S |�kr+t �t|tjtd��d�gg g f�S t �t|��S )Nrt r )r
r: r r rv r r; r r� ��alias� canonical�exchanger+ r/ ri � s2 ����
����zNMXTests.test_cnameWithoutGlueRecords.<locals>.DummyResolver.lookupMailExchangeN)r\ rq rr r� ri r+ r� r+ r/ rk � s rk r� r� r� r+ r� r/ � test_cnameWithoutGlueRecords� s
& z$MXTests.test_cnameWithoutGlueRecordsc sX G dd� d�}d� |� �j _�j �d� �}��|tjjj� � �fdd�}|�|� |S )z�
If L{MXCalculator.getMX} encounters a CNAME chain which is longer than
the length specified, the returned L{Deferred} should errback with
L{CanonicalNameChainTooLong}.
c @ s e Zd ZdZdZdd� ZdS )z.MXTests.test_cnameChain.<locals>.DummyResolverz
Fake resolver which generates a CNAME chain of infinite length in
response to MX lookups.
r c S s< | j d7 _ d| j f }t�t|tjt|�d�gg g f�S )Nrh �x-%d.example.comrt )� chainCounterr
r: r r rv )rV r_ r� r+ r+ r/ ri � s ����zAMXTests.test_cnameChain.<locals>.DummyResolver.lookupMailExchangeN)r\ rq rr r� r� ri r+ r+ r+ r/ rk � s rk r' r� c s: �� | jd td� d f �� �� �jjj� d � d S )Nr r� rh )rC �argsr rY rO r� )rv ��
cnameLimitrV r+ r/ �cbChainTooLong� s �z/MXTests.test_cnameChain.<locals>.cbChainTooLong) rY rO r\ r �twistedr r
�CanonicalNameChainTooLongr� )rV rk rW r� r+ r� r/ �test_cnameChain� s
zMXTests.test_cnameChainc sV g �d� d�d�G � ���fdd�d�}|� | j _| j �� �}|�| jt�d�� |S )z�
If an MX lookup returns a CNAME and the MX record for the CNAME, the
L{Deferred} returned by L{MXCalculator.getMX} should be called back
with the name from the MX record without further lookups being
attempted.
r� r� r� c s e Zd Z� ���fdd�ZdS )z8MXTests.test_cnameWithGlueRecords.<locals>.DummyResolverc sL |� ks�r g g g fS t �t� tjt��d�t�tjt�d�d�gg g f�S )Nrt r� )r
r: r r rv r r� �r� r� r� �lookedUpr+ r/ ri s$
��� ��zKMXTests.test_cnameWithGlueRecords.<locals>.DummyResolver.lookupMailExchangeNrj r+ r� r+ r/ rk � s rk r� r� r� r+ r� r/ �test_cnameWithGlueRecords� s
z!MXTests.test_cnameWithGlueRecordsc sH d� d�G � �fdd�d�}|� | j _| j �� �}| �|tjjj� |S )z�
If an MX lookup returns two CNAME records which point to each other,
the loop should be detected and the L{Deferred} returned by
L{MXCalculator.getMX} should be errbacked with L{CanonicalNameLoop}.
zcname1.example.comzcname2.example.comc rf )z<MXTests.test_cnameLoopWithGlueRecords.<locals>.DummyResolverc s4 t �t� tjt��d�t�tjt� �d�gg g f�S )Nrt )r
r: r r rv r� ��
firstAlias� secondAliasr+ r/ ri ' s ��� ��zOMXTests.test_cnameLoopWithGlueRecords.<locals>.DummyResolver.lookupMailExchangeNrj r+ r� r+ r/ rk & rl rk )rY rO r\ r r� r r
�CanonicalNameLoopr� r+ r� r/ � test_cnameLoopWithGlueRecords s
z%MXTests.test_cnameLoopWithGlueRecordsc C s$ g d�| j jd< | j�d��| j�S )N)zmx1.test.domainzmx2.test.domainzmx3.test.domainr� )rE r6 rY r\ r� � _cbManyRecordsSuccessfulLookuprb r+ r+ r/ �testManyRecords@ s �zMXTests.testManyRecordsc C sH | � t|j��dd�d dv � | j�t|j�� | j�d��| j|�S )N�.rh r )�mx1�mx2�mx3r� ) rN r] r� r� rY r| r\ r� � _cbManyRecordsDifferentResultr` r+ r+ r/ r� K s
�z&MXTests._cbManyRecordsSuccessfulLookupc C sB | � t|j�t|j�� | j�t|j�� | j�d��| j||�S rd )r� r] r� rY r| r\ r� �_cbManyRecordsLastResult)rV �nextMXrY r+ r+ r/ r� R s
�z%MXTests._cbManyRecordsDifferentResultc C sj | � t|j�t|j�� | � t|j�t|j�� | j�t|j�� | j�t|j�� | j�d��| j|�S rd ) r� r] r� rY r| r� r\ r� �"_cbManyRecordsRepeatSpecificResult)rV �lastMXrY r� r+ r+ r/ r� Z s �z MXTests._cbManyRecordsLastResultc C s | � t|j�t|j�� d S rx rm )rV �againMXr� r+ r+ r/ r� e s z*MXTests._cbManyRecordsRepeatSpecificResultN) r\ rq rr r� rz r� rZ r rs rt r^ r] rc re rq rz r} r r� r� r� r� r� r� r� r� r� r� r� r� r� r+ r+ r+ r/ rW � s8
$5.*#
rW c @ s` e Zd Ze�ed�du rdZdd� Zdd� Ze e
j dkd�d d
� �Z e e
j dkd�d d � �Z
dS )
�LiveFireExerciseTestsNz1UDP support is required to determining MX recordsc C s t | � g d�| _d S )N)� domainDir�insertionDomain�insertionQueue�destinationDomain�destinationQueue)rQ �tmpdirsrb r+ r+ r/ rz n s zLiveFireExerciseTests.setUpc C s* | j D ]
}tj�|�rt�|� qt| �S rx )r� r� r� r~ r1 r2 rV )rV rW r+ r+ r/ r� x s
�z LiveFireExerciseTests.tearDownr& r( c s� t j �� }|j�tj�� � t j�|d���� dd� |�
d�� |j d |j d< t |j d j��
� � |�t j�|�� |�� }tjd|dd��_tg d ��� t� �t�� }� fd
d �|_�fd d �|_t�d�j�� j |� ��fd
d�}�� |� �S )Nr� rn r� r� r� r r� rB )�
HELO mesonzMAIL FROM: <user@hostname>zRCPT TO: <user@test.domain>�DATA�This is the messager� �QUITc r rx r+ r+ �r0 r+ r/ r � r z9LiveFireExerciseTests.testLocalDelivery.<locals>.<lambda>c �
� � d �S rx ��callback�r� �r� r+ r/ r � �
c s> � � dd tj�d }|�d��� }��|�d�d� �j�� S �Nrn rh r r� ���) rp r rq r� r� r� �find�
smtpServerrM ��ignr r* )r_ rV r+ r/ �finished� s
z9LiveFireExerciseTests.testLocalDelivery.<locals>.finished) r r� �
smtpPortalr� r re �AllowAnonymousAccessr� rY r� r� r� r� rd �setQueuer� r� r� r
rH r� r r r �
ClientFactory�clientConnectionLost�
connectTCPrJ rI r� )rV r� r� r� r+ )r0 r_ r� rV r/ �testLocalDelivery~ s*
�
z'LiveFireExerciseTests.testLocalDeliveryc s� t j �� }|j�tj�� � t j�|d��|� d�� t
� d� |� t j
�d�� |j�t j�|�� t j
�|j��� jd7 _t j
��d��t j
��j��_dg�jjd< |� � }t j d|dd ��_ t j �� }|j�tj�� � t j�|d
���� d d � |� d�� t
� d
� |� t j
�d
�� t j
��d����!� |� � }t j d|dd ��_"�j"�#� j$�_%t&g d��� t'� �t(�)� }� fdd�|_(�fdd�|_*t �+d�j �#� j$|� ����fdd�}��,|� �S )Nr� zinsertion.domainr� )ztest.identity.hostnamerh r� zdestination.domainr rB r� rn r� r� )r� zMAIL FROM: <user@wherever>z"RCPT TO: <user@destination.domain>r� r� r� r� c r rx r+ r+ r� r+ r/ r � r z9LiveFireExerciseTests.testRelayDelivery.<locals>.<lambda>c r� rx r� r� r� r+ r/ r � r� c s& �� � }� ��fdd�}|�|� |S )Nc sT � � dd tj�d }|�d��� }��|�d�d� �j�� �j �� ��
� d S r� ) rp r rq r� r� r� r� � insServerrM �
destServer� stopServicer� )r_ �helperrV r+ r/ � delivered� s
zLLiveFireExerciseTests.testRelayDelivery.<locals>.finished.<locals>.delivered)�
checkStater� )r� r� r� )r_ r� r rV r+ r/ r� � s
z9LiveFireExerciseTests.testRelayDelivery.<locals>.finished)-r r� r� r� r re r� r� rY r� r� r� r� r
r rT �setDefaultDomainr� r� � SmartHostSMTPRelayingManagerr �fArgs�RelayStateHelperrX rO �mxcalcrE r6 r� r
rH r� r� � startServicer� rJ rI �PORTr r r r� r� r� r� )rV �insServr� �destServr� r+ )r0 r_ r� r� r rV r/ �testRelayDelivery� sJ
�
z'LiveFireExerciseTests.testRelayDelivery)r\ rq rr r � IReactorUDPr
�skiprz r� r rs rt r� r� r+ r+ r+ r/ r� i s
+r� c @ ru )
�LineBufferMessagec C s g | _ d| _d| _d S )NF)�lines�eom�lostrb r+ r+ r/ r� � r zLineBufferMessage.__init__c C r rx )r� r� )rV r� r+ r+ r/ r� � r{ z LineBufferMessage.lineReceivedc C s d| _ t�d�S )NTz
<Whatever>)r� r
r: rb r+ r+ r/ r� �
z LineBufferMessage.eomReceivedc C s
d| _ d S )NT)r� rb r+ r+ r/ r� r� z LineBufferMessage.connectionLostN)r\ rq rr r� r� r� r� r+ r+ r+ r/ r� � s
r� c @ sd e Zd Zg d�Zdd� Zeejdkd�dd� �Zdd � Z d
d � Z
eejdkd�d d
� �Z dd� Z dS )�
AliasTests�z
First linez Next liner� zAfter a blank linez Last linec C sx i }g d�}|D ] }t j�||dd � q| �|d ddg� | �|d dg� | �|d d g� | �|d
g d �� d S ) N)zuser: another@host
znextuser: |/bin/program
zuser: me@again
z"moreusers: :/etc/include/filename
z3multiuser: first@host, second@host,last@anotherhostr# rn z another@hostzme@again�nextuserz
|/bin/program� moreusersz:/etc/include/filename� multiuser)z
first@hostz second@hostzlast@anotherhost)r r� �handlerC )rV r� r� �lr+ r+ r/ �
testHandle s �zAliasTests.testHandler& r( c C s
dt � i}tjj|t�t�d��� �d�}| � t
|�d� |d }t |�}dD ] }| � |�
|�d� q'| � t
|�d� |d }t |�}d
D ] }| � |�
|�d� qE| � t
|�d� |d }t |�}| � |�
d �d� d
D ]}| � |�
|�dd| � ql| � t
|�d� d S )Nr� a� # Here's a comment
# woop another one
testuser: address1,address2, address3,
continuation@address, |/bin/process/this
usertwo:thisaddress,thataddress, lastaddress
lastuser: :/includable, /filename, |/program, address
)r� r' �testuser)�address1�address2�address3zcontinuation@addressz/bin/process/thisr� rD �usertwo)� thisaddress� thataddress� lastaddress�lastuserz /includable)z /filename�programr z %s not found)r� r r� �
loadAliasFiler� r� �textwrap�dedent�encoderC rD r] r� r� )rV rT r� �group�sr< r+ r+ r/ �testFileLoader" s6
�
��zAliasTests.testFileLoaderc C sD t � t � t � f}tj�|�}| jD ]}|�|� q|�� �| j|�S rx ) r� r r� � MultiWrapperr� r� r� r� �_cbMultiWrapper)rV rK r* �Lr+ r+ r/ �testMultiWrapperQ s
zAliasTests.testMultiWrapperc C s6 |D ]}| � |j� | �|j� | �| j|j� qd S rx )rN r� r� r� rC r� )rV r rK r r+ r+ r/ r Y s
�zAliasTests._cbMultiWrapperc C sH | � � }tj�|d d �}|�� }| jD ]}|�|� q|�� �| j |�S rx )
r r r� � FileAlias�createMessageReceiverr� r� r� r� �_cbTestFileAlias)rV �tmpfiler< r r� r+ r+ r/ �
testFileAlias_ s
zAliasTests.testFileAliasc C sL t |�� }|�� }W d � n1 sw Y | �dd� |D �| j� d S )Nc S � g | ]}|d d� �qS �Nr� r+ �r- r r+ r+ r/ r4 l r5 z/AliasTests._cbTestFileAlias.<locals>.<listcomp>�r� � readlinesrC r� )rV r r r� r� r+ r+ r/ r i �
� zAliasTests._cbTestFileAliasN)
r\ rq rr r� r� r rs rt r r r r
r r+ r+ r+ r/ r�
s
.
r� c @ s" e Zd ZdZdd� Zddd�ZdS )� DummyDomainz/
Test domain for L{AddressAliasTests}.
c C s
|| _ d S rx )r )rV r r+ r+ r/ r� t r� zDummyDomain.__init__Nc s |rt j�d��� �fdd�S )a
@returns: When a C{memo} is passed in this will raise a
L{smtp.SMTPBadRcpt} exception, otherwise a boolean
indicating if the C{user} and string version of
L{self.address} are equal or not.
@rtype: C{bool}
�hamc s �t � j�kS rx )r] r r+ r� r+ r/ r � s z$DummyDomain.exists.<locals>.<lambda>)r r r� r� r+ r� r/ r~ w s zDummyDomain.existsrx )r\ rq rr r� r� r~ r+ r+ r+ r/ r o s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�AddressAliasTestsz7
Tests for L{twisted.mail.alias.AddressAlias}.
c C s: t j�d�| _| jjt| j�i}t j�| j|| j�| _dS )z+
Setup an L{AddressAlias}.
zfoo@barN)r r r� r r_ r r� � AddressAlias)rV rT r+ r+ r/ rz � s zAddressAliasTests.setUpc C s | � | j�� � dS )z�
L{createMessageReceiever} calls C{exists()} on the domain object
which key matches the C{alias} passed to L{AddressAlias}.
N)rN r� r
rb r+ r+ r/ �test_createMessageReceiver� s z,AddressAliasTests.test_createMessageReceiverc C s | � t| j�d� dS )zP
The string presentation of L{AddressAlias} includes the alias.
z<Address foo@bar>N)rC r] r� rb r+ r+ r/ rc � rU zAddressAliasTests.test_strc C s | � | j�| jdi�d� dS )z�
L{resolve} will look for additional aliases when an C{aliasmap}
dictionary is passed, and returns L{None} if none were found.
r� N)rC r� �resolver rb r+ r+ r/ � test_resolve� s z AddressAliasTests.test_resolvec C s | � | j�i �d� dS )z�
L{resolve} returns L{None} when the alias could not be found in the
C{aliasmap} and no L{mail.smtp.User} with this alias exists either.
N)rC r� r rb r+ r+ r/ �test_resolveWithoutAliasmap� s z-AddressAliasTests.test_resolveWithoutAliasmapN) r\ rq rr r� rz r rc r r r+ r+ r+ r/ r � s r c @ s e Zd ZdgZdS )� DummyProcess�onEndN)r\ rq rr � __slots__r+ r+ r+ r/ r � s
r c @ r� )�MockProcessAliaszD
An alias processor that doesn't actually launch processes.
c C � dS )z(
Don't spawn a process.
Nr+ )rV �protor� r� r+ r+ r/ � spawnProcess� r z MockProcessAlias.spawnProcessN)r\ rq rr r� r" r+ r+ r+ r/ r � s r c @ s e Zd ZdZeZdS )�MockAliasGroupz3
An alias group using C{MockProcessAlias}.
N)r\ rq rr r� r �processAliasFactoryr+ r+ r+ r/ r# � s r# c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � StubProcessz�
Fake implementation of L{IProcessTransport}.
@ivar signals: A list of all the signals which have been sent to this fake
process.
c C s
g | _ d S rx )�signalsrb r+ r+ r/ r� � r� zStubProcess.__init__c C r )z8
No-op implementation of disconnection.
Nr+ rb r+ r+ r/ �loseConnection� r zStubProcess.loseConnectionc C s | j �|� dS )zL
Record a signal sent to this process for later inspection.
N)r& r� )rV �signalr+ r+ r/ �
signalProcess� s zStubProcess.signalProcessN)r\ rq rr r� r� r' r) r+ r+ r+ r/ r% � s
r% c @ s� e Zd ZdZe�ed�du rdZg d�Zdd� Z dd� Z
d d
� Z d d � Z e
ejd
kd�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Ze
ejd
kd�dd � �Ze
ejd
kd�d d � �ZdS ) �ProcessAliasTestsz%
Tests for alias resolution.
Nz IReactorProcess not supportedr� c C sB |d> dB }| � t�|�� | �t�|�|� | �t�|�� |S )a
Construct a status from the given exit code.
@type code: L{int} between 0 and 255 inclusive.
@param code: The exit status which the code will represent.
@rtype: L{int}
@return: A status integer for the given exit code.
� r )rN r� � WIFEXITEDrC � WEXITSTATUSr� � WIFSIGNALED)rV �code�statusr+ r+ r/ �
exitStatus� s
z ProcessAliasTests.exitStatusc C s: |}| � t�|�� | �t�|�|� | �t�|�� |S )a
Construct a status from the given signal.
@type signal: L{int} between 0 and 255 inclusive.
@param signal: The signal number which the status will represent.
@rtype: L{int}
@return: A status integer for the given signal.
)rN r� r. rC �WTERMSIGr� r, )rV r( r0 r+ r+ r/ � signalStatus� s
z ProcessAliasTests.signalStatusc C s t j| _dt _dS )zB
Replace L{smtp.DNSNAME} with a well-known value.
r� N)r �DNSNAMErb r+ r+ r/ rz s
zProcessAliasTests.setUpc C s | j t_ dS )z@
Restore the original value of L{smtp.DNSNAME}.
N)r4 r rb r+ r+ r/ r� s zProcessAliasTests.tearDownr& r( c sn t � �� �}|�d� t�|jd� tj�|jdd�}|� � }� j
D ]}|� |� q"� fdd�}|� � �
|�S )z�
Standard call to C{mail.alias.ProcessAlias}: check that the specified
script is called, and that the input is correctly transferred to it.
zX#!/bin/sh
rm -f process.alias.out
while read i; do
echo $i >> process.alias.out
donei� Nc sL t d�� }|�� }W d � n1 sw Y � �dd� |D �� j� d S )Nzprocess.alias.outc S r r r+ r r+ r+ r/ r4 ; r5 zPProcessAliasTests.test_processAlias.<locals>._cbProcessAlias.<locals>.<listcomp>r )r r� r� rb r+ r/ �_cbProcessAlias8 r z<ProcessAliasTests.test_processAlias.<locals>._cbProcessAlias)r r �
setContentr� �chmodr� r r� � ProcessAliasr
r� r� r� r� )rV �shr< r r� r5 r+ rb r/ �test_processAlias" s �
z#ProcessAliasTests.test_processAliasc st t �� }t� �tj�� � � ��� tj�� d|�}|�� }|� |j
� � ��fdd�}�� |tjj � |�
|� |S )a
If the alias child process does not exit within a particular period of
time, the L{Deferred} returned by L{MessageWrapper.eomReceived} should
fail with L{ProcessAliasTimeout} and send the I{KILL} signal to the
child process..
Nc s* �� �jdg� � �t��tj��� d S )N�KILL)rC r& � processEndedr r3 r( �SIGKILLr �r! rV r� r+ r/ �timedOutO s z<ProcessAliasTests.test_processAliasTimeout.<locals>.timedOut)r r; r% r r� �ProcessAliasProtocolr� �MessageWrapperr� r< �completionTimeoutr �ProcessAliasTimeoutr� )rV r
�receiverrW r? r+ r> r/ �test_processAliasTimeout? s
z*ProcessAliasTests.test_processAliasTimeoutc C sN t � }tj�� }|�|� tj�|dd�}|�t�t d��� | �
|� � t �S )z�
If the process associated with an L{mail.alias.MessageWrapper} exits
before I{eomReceived} is called, the L{Deferred} returned by
I{eomReceived} should fail.
Nr ) r% r r� r@ r� rA r< r r r r r� )rV r� r rD r+ r+ r/ � test_earlyProcessTerminationY �
z.ProcessAliasTests.test_earlyProcessTerminationc C sN t � }tj�� }|�|� tj�|dd�}|�t�t |��� | �
|� � t �S )z�
Verify that if the process associated with an
L{mail.alias.MessageWrapper} exits with the given status, the
L{Deferred} returned by I{eomReceived} fails with L{ProcessTerminated}.
N) r% r r� r@ r� rA r< r r r r r� )rV r0 r� r rD r+ r+ r/ �_terminationTestf rG z"ProcessAliasTests._terminationTestc C s | � | �d��S )z�
If the process associated with an L{mail.alias.MessageWrapper} exits
with a non-zero exit code, the L{Deferred} returned by I{eomReceived}
should fail.
rh )rH r1 rb r+ r+ r/ � test_errorProcessTerminations s z.ProcessAliasTests.test_errorProcessTerminationc C s | � | �tj��S )z�
If the process associated with an L{mail.alias.MessageWrapper} exits
because it received a signal, the L{Deferred} returned by
I{eomReceived} should fail.
)rH r3 r( �SIGHUPrb r+ r+ r/ � test_signalProcessTermination{ s z/ProcessAliasTests.test_signalProcessTerminationc
C sx i }dt |g d��i}tg d�|d�}tddg|d�}tj�d|d�}|�|||d �� |�|�}tt|j �}|�
� tttj�d
d d �tj� t � d �tj�
d
�g�}|�
� | �||� |�|�} tt| j �}
|
�
� tttj�dd d �tj�dd d �g�}|�
� | �|
|� |�|�} tt| j �} | �
� tttj�d
d d �tj� t � d �tj�
d
�g�}|�
� | �| |� d S )a�
Check that the C{resolve} method of alias processors produce the correct
set of objects:
- direct alias with L{mail.alias.AddressAlias} if a simple input is passed
- aliases in a file with L{mail.alias.FileWrapper} if an input in the format
'/file' is given
- aliases resulting of a process call wrapped by L{mail.alias.MessageWrapper}
if the format is '|process'
r� )r[ r] r_ )r[ �|echo�/file�alias1r] r_ �alias2�alias3�rN rO rP r[ N�echorM )�
TestDomainr# r r� r rO r r� r] �objsrI rA r � FileWrapperrC )
rV r� r_ �A1�A2�A3�res1�r1�expected�res2�r2�res3�r3r+ r+ r/ �test_aliasResolution� s\ ��
��
��
��z&ProcessAliasTests.test_aliasResolutionc
C s� i }dt |g �i}tj�d|d�}tj�d|d�}tj�d|d�}|�|||d�� | �|d �|�d� | �|d �|�d� | �|d �|�d� tddg|d�}||d< |�|�}tt |j
�}|� � tt tj� t
� d �g�} | � � | �|| � dS )
zN
Check that a cycle in alias resolution is correctly handled.
r� rO rN rP rQ NrL �alias4rR )rS r r� r rO rC r r# r� r] rT rI rA r )
rV r� r_ rV rW rX �A4�res�rr[ r+ r+ r/ �test_cyclicAlias� s"
z"ProcessAliasTests.test_cyclicAlias)r\ rq rr r� r �IReactorProcessr
r� r� r1 r3 rz r� r rs rt r: rE rF rH rI rK r` re r+ r+ r+ r/ r* � s(
@r* c @ r3 )rS c C s || _ || _d S rx )r� �users)rV r� rg r+ r+ r/ r� � r� zTestDomain.__init__Nc sn �j j��| jv r�fdd�S z| j� }W n ty" t����w |�| j|�� � r2� fdd�S t����)Nc s t j�� d d �S rx )r r� r r+ )rn r+ r/ r � s z#TestDomain.exists.<locals>.<lambda>c r rx r+ r+ )r� r+ r/ r � r )r� �localrg r� r� r r� r )rV rn r� r< r+ )r� rn r/ r~ � s
�
zTestDomain.existsrx )r\ rq rr r� r~ r+ r+ r+ r/ rS � rA rS c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
DummyQueuez�
A fake relay queue to use for testing.
This queue doesn't keep track of which messages are waiting to be relayed
or are in the process of being relayed.
@ivar directory: See L{__init__}.
c C r� )z�
@type directory: L{bytes}
@param directory: The pathname of the directory holding messages in the
queue.
N)� directory)rV rj r+ r+ r/ r� � s
zDummyQueue.__init__c C s8 t j�|�}t �| �|�d � t �| �|�d � dS )z�
Remove a message from the queue.
@type message: L{bytes}
@param message: The base filename of a message.
r� r� N)r� r� �basenamer� �getPath�rV rj r+ r+ r/ r� s zDummyQueue.donec C s t tj�| j|d �d�S )z�
Get the envelope file for a message in the queue.
@type message: L{bytes}
@param message: The base filename of a message.
@rtype: L{file}
@return: The envelope file for the message.
r� �rb)r� r� r� r7 rj rm r+ r+ r/ �getEnvelopeFile s
zDummyQueue.getEnvelopeFilec C s t j�| j|�S )z�
Return the full base pathname of a message in the queue.
@type message: L{bytes}
@param message: The base filename of a message.
@rtype: L{bytes}
@return: The full base pathname of the message.
)r� r� r7 rj rm r+ r+ r/ rl s
zDummyQueue.getPathc C sv t � � � dt| �� �}ttj�| j|d �d�}tj�| j|d �}tj�| j|d �}t|d�}|tj�|||�fS )z�
Create a new message in the queue.
@rtype: 2-L{tuple} of (E{1}) L{file}, (E{2}) L{FileMessage}
@return: The envelope file and a message receiver for a new message in
the queue.
�_r� r� z-Cr� ) �time�idr� r� r� r7 rj r r� )rV �fname�
headerFile� tempFilename�
finalFilename� messageFiler+ r+ r/ r ( s
�zDummyQueue.createNewMessagec C r )z�
Ignore the request to mark a message as waiting to be relayed.
@type message: L{bytes}
@param message: The base filename of a message.
Nr+ rm r+ r+ r/ r$ : r� zDummyQueue.setWaitingN)
r\ rq rr r� r� r� ro rl r r$ r+ r+ r+ r/ ri � s ri c @ r� )�!DummySmartHostSMTPRelayingManagera4
A fake smart host to use for testing.
@type managed: L{dict} of L{bytes} -> L{list} of
L{list} of L{bytes}
@ivar managed: A mapping of a string identifying a managed relayer to
filenames of messages the managed relayer is responsible for.
@ivar queue: See L{__init__}.
c C s i | _ || _dS )z�
Initialize the minimum necessary members of a smart host.
@type queue: L{DummyQueue}
@param queue: A queue that can be used for testing purposes.
N)�managedr )rV r r+ r+ r/ r� P s
z*DummySmartHostSMTPRelayingManager.__init__N)r\ rq rr r� r� r+ r+ r+ r/ rx D s rx c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
eejdkd�dd� �Zeejdkd�dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%�_AttemptManagerTestsa
Test the behavior of L{_AttemptManager}.
@type tmpdir: L{bytes}
@ivar tmpdir: The path to a temporary directory holding the message files.
@type reactor: L{MemoryReactorClock}
@ivar reactor: The reactor used for test purposes.
@type eventLog: L{None} or L{dict} of L{bytes} -> L{object}
@ivar eventLog: Information about the last informational log message
generated or none if no log message has been generated.
@type noisyAttemptMgr: L{_AttemptManager}
@ivar noisyAttemptMgr: An attempt manager which generates informational
log messages.
@type quietAttemptMgr: L{_AttemptManager}
@ivar quietAttemptMgr: An attempt manager which does not generate
informational log messages.
@type noisyMessage: L{bytes}
@ivar noisyMessage: The full base pathname of the message to be used with
the noisy attempt manager.
@type quietMessage: L{bytes}
@ivar quietMessage: The full base pathname of the message to be used with
the quiet.
c C s^ | � � | _t�| j� t� | _d| _t�| j � t
t t | j��d| j�| _
t
t t | j��d| j�| _d}d}tj�| j|�| _tj�| j|�| _t| jd d��� t| jd d��� |g| j
jjd< |g| jjjd < t| jd
d ��}t�d d
g|� W d � n1 s�w Y t| jd
d ��}t�ddg|� W d � dS 1 s�w Y dS )a
Set up a temporary directory for the queue, attempt managers with the
noisy flag on and off, message files for use with each attempt manager,
and a reactor. Also, register to be notified when log messages are
generated.
NTF� noisyMessage� quietMessager� r� � noisyRelayer� quietRelayerr� r� zfrom-noisy@domainzto-noisy@domainzfrom-quiet@domainzto-quiet@domain)r r� r� r� r! r
�eventLogr � addObserver� _logObserverr rx ri �noisyAttemptMgr�quietAttemptMgrr� r7 r{ r| r� r� r ry r� r� )rV �
noisyBaseName�
quietBaseName�enveloper+ r+ r/ rz { s:
� � �"�z_AttemptManagerTests.setUpc C s t �| j� t�| j� dS )zO
Unregister for log events and remove the temporary directory.
N)r �removeObserverr� r1 r2 r� rb r+ r+ r/ r� � s z _AttemptManagerTests.tearDownc C r� )z�
A log observer.
@type eventDict: L{dict} of L{bytes} -> L{object}
@param eventDict: Information about the last informational log message
generated.
N)r )rV � eventDictr+ r+ r/ r� � s
z!_AttemptManagerTests._logObserverc C s" t tt| j���}| �|j� dS )z�
When an attempt manager is created without the noisy parameter, the
noisy instance variable should default to true.
N)r rx ri r� rN r �rV �amr+ r+ r/ �test_initNoisyDefault� s z*_AttemptManagerTests.test_initNoisyDefaultc C � | � | jj� dS )z�
When an attempt manager is created with the noisy parameter set to
true, the noisy instance variable should be set to true.
N)rN r� r rb r+ r+ r/ �test_initNoisy� � z#_AttemptManagerTests.test_initNoisyc C r� )z�
When an attempt manager is created with the noisy parameter set to
false, the noisy instance variable should be set to false.
N)r� r� r rb r+ r+ r/ �test_initQuiet� r� z#_AttemptManagerTests.test_initQuietc C s$ t tt| j���}| �|jt� dS )z�
When an attempt manager is created without the reactor parameter, the
reactor instance variable should default to the global reactor.
N)r rx ri r� rC r
r� r+ r+ r/ �test_initReactorDefault� s z,_AttemptManagerTests.test_initReactorDefaultc C s | � | jj| j� dS )z�
When an attempt manager is created with a reactor provided, the
reactor instance variable should default to that reactor.
N)rC r� r
rb r+ r+ r/ �test_initReactor� s z%_AttemptManagerTests.test_initReactorc C � | j �d| j� | �| j� dS )zw
For an attempt manager with the noisy flag set, notifySuccess should
result in a log message.
r} N)r� r r{ rN r rb r+ r+ r/ �test_notifySuccessNoisy� � z,_AttemptManagerTests.test_notifySuccessNoisyc C r� )z|
For an attempt manager with the noisy flag not set, notifySuccess
should result in no log message.
r~ N)r� r r| r� r rb r+ r+ r/ �test_notifySuccessQuiet� r� z,_AttemptManagerTests.test_notifySuccessQuietr& r( c C r� )zw
For an attempt manager with the noisy flag set, notifyFailure should
result in a log message.
r} N)r� r r{ rN r rb r+ r+ r/ �test_notifyFailureNoisy� � z,_AttemptManagerTests.test_notifyFailureNoisyc C r� )z|
For an attempt manager with the noisy flag not set, notifyFailure
should result in no log message.
r~ N)r� r r| r� r rb r+ r+ r/ �test_notifyFailureQuiet� r� z,_AttemptManagerTests.test_notifyFailureQuietc C � | j �d� | �| j� dS )zt
For an attempt manager with the noisy flag set, notifyDone should
result in a log message.
r} N)r� r rN r rb r+ r+ r/ �test_notifyDoneNoisy
� z)_AttemptManagerTests.test_notifyDoneNoisyc C r� )zy
For an attempt manager with the noisy flag not set, notifyDone
should result in no log message.
r~ N)r� r r� r rb r+ r+ r/ �test_notifyDoneQuiet
r� z)_AttemptManagerTests.test_notifyDoneQuietc C �( | j �d� | �| j� | j�d� dS )z|
For an attempt manager with the noisy flag set, notifyNoConnection
should result in a log message.
r} r7 N)r� �notifyNoConnectionrN r r
r< rb r+ r+ r/ � test_notifyNoConnectionNoisy
� z1_AttemptManagerTests.test_notifyNoConnectionNoisyc C r� )z�
For an attempt manager with the noisy flag not set, notifyNoConnection
should result in no log message.
r~ r7 N)r� r� r� r r
r< rb r+ r+ r/ � test_notifyNoConnectionQuiet
r� z1_AttemptManagerTests.test_notifyNoConnectionQuietN)r\ rq rr r� rz r� r� r� r� r� r� r� r� r� r rs rt r� r� r� r� r� r� r+ r+ r+ r/ rz [ s( -
rz )}r� �
email.messager� � email.parserr� r� r� r� r1 r( rs r+ r� rq �hashlibr �unittestr �zope.interfacer r �zope.interface.verifyr �twisted.cred.checkersr� �twisted.cred.credentials�twisted.cred.portal�twisted.mail.alias�twisted.mail.mail�twisted.mail.maildir�twisted.mail.protocols�twisted.mail.relay�twisted.mail.relaymanagerr r �twisted.internetr r
r r r
r �twisted.internet.deferr �twisted.internet.errorr r r r � twisted.mailr r r �
twisted.namesr �twisted.names.dnsr r r �twisted.names.errorr �twisted.pythonr r �twisted.python.filepathr �twisted.python.runtimer �twisted.test.proto_helpersr r! r" �twisted.trial.unittestr# r% rv r� r� r� r� r� � _MaildirMailboxAppendMessageTaskr� r� r
r* r/ rQ rV rR r� r� r� r� r� r� r� r� r r r0 r1 r2 r5 r4 rQ rV rW r� r� r� r r r r� r8 r �
AliasGroupr# r% r* rS ri rx rz r+ r+ r+ r/ �<module> s� h & 0 M
� S ( X
s # NI %) D 2 d ) Q