Your IP : 216.73.216.213
��Yf=, � @ s� d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z Gd d � d � Z Gd d � d e e � Z Gd d
� d
e e � Z
Gd d � d e e � Z e d
k r� e j
� d S)� N)�policy)�errors)�
TestEmailBasec @ s� e Z d Z e j Z d Z e j d d � � Z d d � Z d d � Z
e j d � Z
d d
� Z d d � Z d
d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d S) �TestDefectsBaseFc c s d Vd S)N� )�self�defectr r �I/opt/alt/python35/lib64/python3.5/test/test_email/test_defect_handling.py� _raise_point s z TestDefectsBase._raise_pointc C s� t j d � } | j t j � � | j | � } Wd QRX| j rE d S| j d � } | j t | d � � | j
t | j | � � d � | j
| j | � d t j � 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 j d � } | j t j � � | j | � } Wd QRX| j rE d S| j | j � t � | j
t | j | � � d � | j | j | � d t j � | j | j | � d t j
� 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 s� | j t j � � | j | j j d � � } Wd QRX| j rB d S| j t | j | � � d � | j
| j | � d t j � 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 sK | j r
d S| j | j j d � � } | j t | 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 sg | j r
d SxS d D]K } | j | j j d | � � } | j t | j | � � d d | � q Wd 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 j d � } | j t j � � | j | � } Wd QRX| j rE d S| j t | d � � | j t
| j | � � d � | j | j | � d t j � | j | j | � d t j
� 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 s� t j d � } | j t j � � | j | � } Wd QRX| j rE d S| j d � j d � } | j t | j
| � � d � | j | j
| � d t j � 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 t j � � | j d � } Wd QRX| j r6 d S| j | j � d g � | j | j � d � | j t | j | � � d � | j
| j | � t j g � | 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 s� | j t j � � | j d � } Wd QRX| j r6 d S| j | j � d g � | j | j � d � | j | j | � t j g � 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 s� t j d � } | j | � } | j t j � � | j d d � } Wd QRX| j rW d S| j | d � | j | j
| � t j g � d S)Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk
�decodeTs 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 s� t j d � } | j | � } | j t j � � | j d d � } Wd QRX| j rW d S| j | d � | j | j
| � t j g � d S)Nz� Subject: test
MIME-Version: 1.0
Content-Type: text/plain; charset="utf-8"
Content-Transfer-Encoding: base64
dmk===
r3 Ts 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 s� t j d � } | j t j � � | j | � } Wd QRX| j rE d S| j t | j � � d � | j | j d � j � d � | j
| j | � t j g � 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 r r r r r s" / &
r c @ s e Z d Z d d � Z d S)�TestDefectDetectionc C s | j S)N)r )r �objr r r r ! s z TestDefectDetection.get_defectsN)r8 r9 r: r r r r r r> s r> c @ sA e Z d Z Gd d � d e j � Z d d � Z d d � Z d S)�TestDefectCapturec @ s" e Z d Z d Z d d � Z d S)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)r8 r9 r: rA rC r r r r �
CapturePolicy' s rD c C s | j d t � � | _ d S)NrA )rD �listr )r r r r �setUp, s zTestDefectCapture.setUpc C s
| j j S)N)r rA )r r? r r r r / s z TestDefectCapture.get_defectsN)r8 r9 r: r Z EmailPolicyrD rF r r r r r r@ % s r@ c @ sF e Z d Z e j Z e j d d � Z d Z e j d d � � Z d S)�TestDefectRaisingZraise_on_defectTc c s | j | � �
d VWd QRXd S)N)Z assertRaises)r r r r r r
9 s z TestDefectRaising._raise_pointN)
r8 r9 r: r r Zcloner r< r= r
r r r r rG 3 s rG �__main__)r
Zunittestr< Zemailr r Ztest.test_emailr r r> r@ rG r8 �mainr r r r �<module> s �