Your IP : 216.73.216.213
3
� f�. � @ 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�ej
� dS )� N)�policy)�errors)�
TestEmailBasec @ s� e Zd ZejZdZejdd� �Zdd� Z dd� Z
e j 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/python36/lib64/python3.6/test/test_email/test_defect_handling.py� _raise_point s z TestDefectsBase._raise_pointc C s� t jd�}| jtj�� | j|�}W d Q R X | jr6d S |jd�}| jt |d�� | j
t | j |��d� | j
| j |�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 jd�}| jtj�� | j|�}W d Q R X | jr6d S | j|j� t � | j
t | j |��d� | j| j |�d tj� | j| j |�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 sf | j tj�� | j| jjd��}W d Q R X | jr4d S | jt| j |��d� | j
| j |�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| jjd��}| jt| j|��d� d S )N� r )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 sN | j r
d S x>dD ]6}| j| jjd| ��}| jt| j|��dd| � qW d S )N�7bit�8bit�BINaryz
Content-Transfer-Encoding: r zcte=)r' r( r) )r r r" r# r r r )r Zcter r r r �"test_multipart_valid_cte_no_defect� s
z2TestDefectsBase.test_multipart_valid_cte_no_defectc C s� t jd�}| jtj�� | j|�}W d Q R X | jr6d S | jt|d�� | j t
| j |��d� | j | j |�d tj� | j | j |�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 r r �test_lying_multipart� s z$TestDefectsBase.test_lying_multipartc C sx t jd�}| jtj�� | j|�}W d Q R X | jr6d S |jd�jd�}| jt | j
|��d� | j | j
|�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� | j tj�� | jd�}W d Q R X | jr,d S | j|j� dg� | j|j� d� | jt| j |��d� | j
| j |�tjg� | j| j |�d j d� d S )Nz Line 1
Subject: test
body�SubjectZbodyr r z Line 1
) r
r Z#FirstHeaderLineIsContinuationDefectr r r �keysr r r �assertDefectsEqual�line)r r r r r �&test_first_line_is_continuation_header� s
z6TestDefectsBase.test_first_line_is_continuation_headerc
C sh | j tj�� | jd�}W d Q R X | jr,d S | j|j� dg� | j|j� d� | j| j |�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 r. r r/ r )r r r r r �"test_missing_header_body_separator� s z2TestDefectsBase.test_missing_header_body_separatorc
C sh t jd�}| j|�}| jtj�� |jdd�}W d Q R X | jrBd S | j|d� | j | j
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk
T)�decodes vi) r
r r r
r ZInvalidBase64PaddingDefectr r r r/ r )r r r �payloadr r r �"test_bad_padding_in_base64_payload� s
z2TestDefectsBase.test_bad_padding_in_base64_payloadc
C sh t jd�}| j|�}| jtj�� |jdd�}W d Q R X | jrBd S | j|d� | j | j
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk===
T)r3 s vi) r
r r r
r Z InvalidBase64CharactersDefectr r r r/ r )r r r r4 r r r �$test_invalid_chars_in_base64_payload� s
z4TestDefectsBase.test_invalid_chars_in_base64_payloadc
C sh t jd�}| j|�}| jtj�� |jdd�}W d Q R X | jrBd S | j|d� | j | j
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
abcde
T)r3 s abcde) r
r r r
r ZInvalidBase64LengthDefectr r r r/ r )r r r r4 r r r �%test_invalid_length_of_base64_payload s
z5TestDefectsBase.test_invalid_length_of_base64_payloadc
C sz t jd�}| jtj�� | j|�}W d Q R X | jr6d S | jt|j � �d� | j|j d�j � d� | j
| j |�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 r/ r )r 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, r1 r2 r5 r6 r7 r8 r r r r r s$ / &
r c @ s e Zd Zdd� ZdS )�TestDefectDetectionc C s |j S )N)r )r �objr r r r 2 s z TestDefectDetection.get_defectsN)r9 r: r; r r r r r r? 0 s r? c @ s. e Zd ZG dd� dej�Zdd� Zdd� ZdS )�TestDefectCapturec @ s e Zd ZdZdd� ZdS )z TestDefectCapture.CapturePolicyNc C s | j j|� d S )N)�captured�append)r r@ r r r r �register_defect: s z/TestDefectCapture.CapturePolicy.register_defect)r9 r: r; rB rD r r r r �
CapturePolicy8 s rE c C s | j t� d�| _d S )N)rB )rE �listr )r r r r �setUp= s zTestDefectCapture.setUpc C s | j jS )N)r rB )r r@ r r r r @ s z TestDefectCapture.get_defectsN)r9 r: r; r Z EmailPolicyrE rG r r r r r rA 6 s rA c @ s0 e Zd ZejZejdd�ZdZejdd� �Z dS )�TestDefectRaisingT)Zraise_on_defectc c s | j |�� d V W d Q R X d S )N)Z assertRaises)r r r r r r
J s z TestDefectRaising._raise_pointN)
r9 r: r; r r Zcloner r= r> r
r r r r rH D s rH �__main__)r
Zunittestr= Zemailr r Ztest.test_emailr r r? rA rH r9 �mainr r r r �<module> s )