File: //usr/local/lib/python3.7/test/test_email/__pycache__/test_defect_handling.cpython-37.pyc
B
��g�. � @ 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 �@/usr/local/lib/python3.7/test/test_email/test_defect_handling.py� _raise_point s z TestDefectsBase._raise_pointc C s� t �d�}| �tj�� | �|�}W d Q R X | jr6d 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 Q R X | jr6d 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 sf | � tj�� | �| j�d��}W d Q R X | jr4d 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 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�d| ��}| �t| �|��dd| � qW d S )N)Z7bitZ8bitZBINaryz
Content-Transfer-Encoding: r zcte=)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 �d�}| �tj�� | �|�}W d Q R X | jr6d 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 r r �test_lying_multipart� s z$TestDefectsBase.test_lying_multipartc C sx t �d�}| �tj�� | �|�}W d Q R X | jr6d 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 Q R X | jr,d S | �|�� dg� | �|�� d� | �t| � |��d� | �
| � |�tjg� | �| � |�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 | � tj�� | �d�}W d Q R X | 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 r+ r r, r )r r r r r �"test_missing_header_body_separator� s z2TestDefectsBase.test_missing_header_body_separatorc C sh t �d�}| �|�}| �tj�� |jdd�}W d Q R X | jrBd 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)�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 �d�}| �|�}| �tj�� |jdd�}W d Q R X | jrBd 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)r0 s vi) r
r r r
r Z InvalidBase64CharactersDefectr r r r, r )r r r r1 r r r �$test_invalid_chars_in_base64_payload� s
z4TestDefectsBase.test_invalid_chars_in_base64_payloadc C sh t �d�}| �|�}| �tj�� |jdd�}W d Q R X | jrBd S | �|d� | � | �
|�tjg� d S )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
abcde
T)r0 s abcde) r
r r r
r ZInvalidBase64LengthDefectr r r r, r )r r r r1 r r r �%test_invalid_length_of_base64_payload s
z5TestDefectsBase.test_invalid_length_of_base64_payloadc C sz t �d�}| �tj�� | �|�}W d Q R X | jr6d 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 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) r. r/ r2 r3 r4 r5 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)r6 r7 r8 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 �|� d S )N)�captured�append)r r= r r r r �register_defect: s z/TestDefectCapture.CapturePolicy.register_defect)r6 r7 r8 r? rA r r r r �
CapturePolicy8 s rB c C s | j t� d�| _d S )N)r? )rB �listr )r r r r �setUp= s zTestDefectCapture.setUpc C s | j jS )N)r r? )r r= r r r r @ s z TestDefectCapture.get_defectsN)r6 r7 r8 r Z EmailPolicyrB rD r r r r r r> 6 s r> c @ s0 e Zd ZejZejdd�ZdZejdd� �Z dS )�TestDefectRaisingT)Zraise_on_defectc c s | � |�� d V W d Q R X d S )N)Z assertRaises)r r r r r r
J s z TestDefectRaising._raise_pointN)
r6 r7 r8 r r Zcloner r: r; r
r r r r rE D s rE �__main__)r
Zunittestr: Zemailr r Ztest.test_emailr r r< r> rE r6 �mainr r r r �<module> s )