File: //usr/local/lib/python3.10/test/test_email/__pycache__/test_defect_handling.cpython-310.pyc
o
�i�. � @ 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
krJe�
� dS 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 �A/usr/local/lib/python3.10/test/test_email/test_defect_handling.py� _raise_point s �
z TestDefectsBase._raise_pointc C s� t �d�}| �tj��
| �|�}W d � n1 sw Y | jr%d 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 sw Y | jr%d 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 sw Y | jr$d 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 rd 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 rd S dD ]}| �| 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 sw Y | jr%d 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 sw Y | jr%d 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 sw 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 sw 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 �| t �d�}| �|�}| �tj�� |jdd�}W d � n1 s!w Y | jr+d 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 �payloadr r r �"test_bad_padding_in_base64_payload� �
�
�z2TestDefectsBase.test_bad_padding_in_base64_payloadc C r6 )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk===
Tr7 r9 ) r r r r r Z InvalidBase64CharactersDefectr r r r2 r r: r r r �$test_invalid_chars_in_base64_payload� r= z4TestDefectsBase.test_invalid_chars_in_base64_payloadc C r6 )Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
abcde
Tr7 s abcde) r r r r r ZInvalidBase64LengthDefectr r r r2 r r: r r r �%test_invalid_length_of_base64_payload r= z5TestDefectsBase.test_invalid_length_of_base64_payloadc C s� t �d�}| �tj��
| �|�}W d � n1 sw Y | jr%d 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)rA rB rC r r r r r rG 0 s rG 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 rI r
r r r �register_defect: s z/TestDefectCapture.CapturePolicy.register_defect)rA rB rC rK rM r r r r �
CapturePolicy8 s rN c C s | j t� d�| _d S )N)rK )rN �listr )r r r r �setUp= s zTestDefectCapture.setUpc C s | j jS r )r rK rH r r r r @ s z TestDefectCapture.get_defectsN)rA rB rC r Z EmailPolicyrN rP r r r r r rJ 6 s rJ c @ s0 e Zd ZejZejdd�ZdZejdd� �Z dS )�TestDefectRaisingT)Zraise_on_defectc c s8 � | � |�� d V W d � d S 1 sw Y d S r )Z assertRaisesr r r r r J s � "�z TestDefectRaising._raise_pointN)
rA rB rC r r Zcloner rE rF r r r r r rQ D s rQ �__main__)r ZunittestrE Zemailr r Ztest.test_emailr r rG rJ rQ rA �mainr r r r �<module> s ) �