Your IP : 216.73.216.213
a
XC?h�. � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ G dd� d�ZG dd� dee�Z G d d
� d
ee�Z
G d d � d ee�Z e d
kr�e�
� dS )� N)�policy)�errors)�
TestEmailBasec @ s� e Zd ZejZdZejdd� �Zdd� Z dd� Z
e � d�Z
d d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �TestDefectsBaseFc c s
d V d S �N� ��self�defectr r �I/opt/alt/python39/lib64/python3.9/test/test_email/test_defect_handling.py� _raise_point s z TestDefectsBase._raise_pointc C s� t �d�}| �tj�� | �|�}W d � n1 s60 Y | jrJd S |�d�}| �t |d�� | �
t | � |��d� | �
| � |�d tj� d S )Na_ Subject: XX
From: xx@xx.dk
To: XX
Mime-version: 1.0
Content-type: multipart/mixed;
boundary="MS_Mac_OE_3071477847_720252_MIME_Part"
--MS_Mac_OE_3071477847_720252_MIME_Part
Content-type: multipart/alternative;
boundary="MS_Mac_OE_3071477847_720252_MIME_Part"
--MS_Mac_OE_3071477847_720252_MIME_Part
Content-type: text/plain; charset="ISO-8859-1"
Content-transfer-encoding: quoted-printable
text
--MS_Mac_OE_3071477847_720252_MIME_Part
Content-type: text/html; charset="ISO-8859-1"
Content-transfer-encoding: quoted-printable
<HTML></HTML>
--MS_Mac_OE_3071477847_720252_MIME_Part--
--MS_Mac_OE_3071477847_720252_MIME_Part
Content-type: image/gif; name="xx.gif";
Content-disposition: attachment
Content-transfer-encoding: base64
Some removed base64 encoded chars.
--MS_Mac_OE_3071477847_720252_MIME_Part--
r �defects� )�textwrap�dedentr r �StartBoundaryNotFoundDefect�_str_msg�raise_expected� get_payload�
assertTrue�hasattr� assertEqual�len� get_defects�assertIsInstance)r �source�msg�innerr r r � test_same_boundary_inner_outer s
%(
�z.TestDefectsBase.test_same_boundary_inner_outerc C s� t �d�}| �tj�� | �|�}W d � n1 s60 Y | jrJd S | �|�� t � | �
t | � |��d� | �| � |�d tj� | �| � |�d tj
� d S )Na� Date: Fri, 6 Apr 2001 09:23:06 -0800 (GMT-0800)
From: foobar
Subject: broken mail
MIME-Version: 1.0
Content-Type: multipart/report; report-type=delivery-status;
--JAB03225.986577786/zinfandel.lacita.com
One part
--JAB03225.986577786/zinfandel.lacita.com
Content-Type: message/delivery-status
Header: Another part
--JAB03225.986577786/zinfandel.lacita.com--
� r r )r r r r �NoBoundaryInMultipartDefectr r r r �strr r r �!MultipartInvariantViolationDefect�r r r r r r �test_multipart_no_boundaryA s
(
��z*TestDefectsBase.test_multipart_no_boundarya^ Date: Wed, 14 Nov 2007 12:56:23 GMT
From: foo@bar.invalid
To: foo@bar.invalid
Subject: Content-Transfer-Encoding: base64 and multipart
MIME-Version: 1.0
Content-Type: multipart/mixed;
boundary="===============3344438784458119861=="{}
--===============3344438784458119861==
Content-Type: text/plain
Test message
--===============3344438784458119861==
Content-Type: application/octet-stream
Content-Transfer-Encoding: base64
YWJj
--===============3344438784458119861==--
c C sz | � tj��" | �| j�d��}W d � n1 s40 Y | jrHd S | �t| � |��d� | �
| � |�d tj� d S )Nz"
Content-Transfer-Encoding: base64r r ) r r Z-InvalidMultipartContentTransferEncodingDefectr �
multipart_msg�formatr r r r r �r r r r r �test_multipart_invalid_cteu s ���"
�z*TestDefectsBase.test_multipart_invalid_ctec C s6 | j r
d S | �| j�d��}| �t| �|��d� d S )N� r �r r r% r& r r r r' r r r � test_multipart_no_cte_no_defect� s
z/TestDefectsBase.test_multipart_no_cte_no_defectc C sJ | j r
d S dD ]6}| �| j�d| ��}| �t| �|��dd| � qd S )N)Z7bitZ8bitZBINaryz
Content-Transfer-Encoding: r zcte=r* )r Zcter r r r �"test_multipart_valid_cte_no_defect� s
�z2TestDefectsBase.test_multipart_valid_cte_no_defectc C s� t �d�}| �tj�� | �|�}W d � n1 s60 Y | jrJd S | �t|d�� | � t
| � |��d� | � | � |�d tj� | � | � |�d tj
� d S )Na From: "Allison Dunlap" <xxx@example.com>
To: yyy@example.com
Subject: 64423
Date: Sun, 11 Jul 2004 16:09:27 -0300
MIME-Version: 1.0
Content-Type: multipart/alternative;
Blah blah blah
r
r r r )r r r r r r r r r r r r r r" r# r r r �test_lying_multipart� s
(
��z$TestDefectsBase.test_lying_multipartc C s� t �d�}| �tj�� | �|�}W d � n1 s60 Y | jrJd S |�d��d�}| �t | �
|��d� | � | �
|�d tj� d S )Na� Content-Type: multipart/mixed; boundary="AAA"
From: Mail Delivery Subsystem <xxx@example.com>
To: yyy@example.com
--AAA
Stuff
--AAA
Content-Type: message/rfc822
From: webmaster@python.org
To: zzz@example.com
Content-Type: multipart/mixed; boundary="BBB"
--BBB--
--AAA--
r r ) r r r r r r r r r r r r )r r ZouterZbadr r r �test_missing_start_boundary� s
(
�z+TestDefectsBase.test_missing_start_boundaryc C s� | � tj�� | �d�}W d � n1 s,0 Y | jr@d S | �|�� dg� | �|�� d� | �t| � |��d� | �
| � |�tjg� | �| � |�d j d� d S )Nz Line 1
Subject: test
body�Subject�bodyr r z Line 1
) r r Z#FirstHeaderLineIsContinuationDefectr r r �keysr r r �assertDefectsEqual�liner' r r r �&test_first_line_is_continuation_header� s (
�z6TestDefectsBase.test_first_line_is_continuation_headerc C s| | � tj�� | �d�}W d � n1 s,0 Y | jr@d S | �|�� dg� | �|�� d� | �| � |�tjg� d S )Nz&Subject: test
not a header
To: abc
b
r/ znot a header
To: abc
b
)
r r Z MissingHeaderBodySeparatorDefectr r r r1 r r2 r r' r r r �"test_missing_header_body_separator� s (
�z2TestDefectsBase.test_missing_header_body_separatorc C s| t �d�}| �|�}| �tj�� |jdd�}W d � n1 sB0 Y | jrVd S | �|d� | � | �
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk
T��decode� vi) r r r r r ZInvalidBase64PaddingDefectr r r r2 r �r r r Zpayloadr r r �"test_bad_padding_in_base64_payload� s
*
�z2TestDefectsBase.test_bad_padding_in_base64_payloadc C s| t �d�}| �|�}| �tj�� |jdd�}W d � n1 sB0 Y | jrVd S | �|d� | � | �
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk===
Tr6 r8 ) r r r r r Z InvalidBase64CharactersDefectr r r r2 r r9 r r r �$test_invalid_chars_in_base64_payload� s
*
�z4TestDefectsBase.test_invalid_chars_in_base64_payloadc C s| t �d�}| �|�}| �tj�� |jdd�}W d � n1 sB0 Y | jrVd S | �|d� | � | �
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
abcde
Tr6 s abcde) r r r r r ZInvalidBase64LengthDefectr r r r2 r r9 r r r �%test_invalid_length_of_base64_payload s
*
�z5TestDefectsBase.test_invalid_length_of_base64_payloadc C s� t �d�}| �tj�� | �|�}W d � n1 s60 Y | jrJd S | �t|� � �d� | �|� d�� � d� | �
| � |�tjg� d S )Na? To: 1@harrydomain4.com
Subject: Fwd: 1
MIME-Version: 1.0
Content-Type: multipart/alternative;
boundary="------------000101020201080900040301"
--------------000101020201080900040301
Content-Type: text/plain; charset=ISO-8859-1
Content-Transfer-Encoding: 7bit
Alternative 1
--------------000101020201080900040301
Content-Type: text/html; charset=ISO-8859-1
Content-Transfer-Encoding: 7bit
Alternative 2
r r zAlternative 2
) r r r r ZCloseBoundaryNotFoundDefectr r r r r r2 r r# r r r � test_missing_ending_boundary s
(
�z,TestDefectsBase.test_missing_ending_boundaryN)�__name__�
__module__� __qualname__r �defaultr �
contextlib�contextmanagerr r r$ r r r% r( r+ r, r- r. r4 r5 r: r; r< r= r r r r r s$
/
&
r c @ s e Zd Zdd� ZdS )�TestDefectDetectionc C s |j S r )r
�r �objr r r r 2 s z TestDefectDetection.get_defectsN)r> r? r@ r r r r r rD 0 s rD c @ s. e Zd ZG dd� dej�Zdd� Zdd� ZdS )�TestDefectCapturec @ s e Zd ZdZdd� ZdS )z TestDefectCapture.CapturePolicyNc C s | j �|� d S r )�captured�append)r rF r
r r r �register_defect: s z/TestDefectCapture.CapturePolicy.register_defect)r> r? r@ rH rJ r r r r �
CapturePolicy8 s rK c C s | j t� d�| _d S )N)rH )rK �listr )r r r r �setUp= s zTestDefectCapture.setUpc C s | j jS r )r rH rE r r r r @ s z TestDefectCapture.get_defectsN)r> r? r@ r Z EmailPolicyrK rM r r r r r rG 6 s rG c @ s0 e Zd ZejZejdd�ZdZejdd� �Z dS )�TestDefectRaisingT)Zraise_on_defectc c s4 | � |�� d V W d � n1 s&0 Y d S r )Z assertRaisesr r r r r J s z TestDefectRaising._raise_pointN)
r> r? r@ r r Zcloner rB rC r r r r r rN D s
rN �__main__)r ZunittestrB Zemailr r Ztest.test_emailr r rD rG rN r> �mainr r r r �<module> s )