File: //usr/local/lib/python3.7/test/test_email/__pycache__/test_parser.cpython-37.opt-2.pyc
B
��g� � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z G dd� de �Z
G dd� d�Z d d
� Z G d d � d e e �Z
d
d� Zdd� ZG dd� de e �Zedkr�e�� dS )� N)�Message� EmailMessage)�default)�
TestEmailBasec @ s: e Zd ZG dd� de�Zejjdd�Zdd� Z dd� Z
d S )
�TestCustomMessagec s e Zd Z� fdd�Z� ZS )zTestCustomMessage.MyMessagec s || _ t� �� d S )N)� check_policy�super�__init__)�self�policy)� __class__� �7/usr/local/lib/python3.7/test/test_email/test_parser.pyr s z$TestCustomMessage.MyMessage.__init__)�__name__�
__module__� __qualname__r �
__classcell__r
r
)r r � MyMessage s r Zboo)�linesepc C s6 t jd| j| jd�}| �|| j� | �|j| j� d S )NzSubject: bogus
msg
)r )�email�message_from_stringr �MyPolicy�assertIsInstance�assertIsr )r
�msgr
r
r �7test_custom_message_gets_policy_if_possible_from_string s
zITestCustomMessage.test_custom_message_gets_policy_if_possible_from_stringc C s@ t �d�}tj|| j| jd�}| �|| j� | �|j| j� d S )NzSubject: bogus
msg
)r ) �io�StringIOr �message_from_filer r r r r )r
Z source_filer r
r
r �5test_custom_message_gets_policy_if_possible_from_file s
zGTestCustomMessage.test_custom_message_gets_policy_if_possible_from_fileN) r r r r r r r �cloner r r r
r
r
r r s r c @ s4 e Zd Zdd� ZG dd� de�Zdd� Zdd� Zd S )
�TestParserBasec C sl xf| j D ]\}| j|jd��D |dtd�}| �|�� dddddd d
d d g � | �|�� d
� W d Q R X qW d S )N)�parseru� Next-Line: not
broken
Null: not broken
Vertical-Tab: not broken
Form-Feed: not broken
File-Separator: not broken
Group-Separator: not broken
Record-Separator: not broken
Line-Separator: not
broken
Paragraph-Separator: not
broken
)r )z Next-Lineu not
broken)ZNullz
not broken)z Vertical-Tabz
not broken)z Form-Feedz
not broken)zFile-Separatorz
not broken)zGroup-Separatorz
not broken)zRecord-Separatorz
not broken)zLine-Separatoru not
broken)zParagraph-Separatoru not
broken� )�parsers�subTestr r Z assertEqual�itemsZ get_payload)r
r" r r
r
r �test_only_split_on_cr_lf& s
z'TestParserBase.test_only_split_on_cr_lfc @ s e Zd ZdS )zTestParserBase.MyMessageN)r r r r
r
r
r r F s r c
C sV xP| j D ]F}| j|jd��. tj| jd�}|d|d�}| �|| j� W d Q R X qW d S )N)r" )�message_factoryz
To: foo
test)r )r$ r% r r r r r )r
r" r r r
r
r �%test_custom_message_factory_on_policyI s
z4TestParserBase.test_custom_message_factory_on_policyc
C sd x^| j D ]T}| j|jd��< tj| jd�}|dt|d�}| �|| j� | �|t� W d Q R X qW d S )N)r" )r( z
To: foo
test)r ) r$ r% r r r r r ZassertNotIsInstancer )r
r" r r r
r
r �!test_factory_arg_overrides_policyP s z0TestParserBase.test_factory_arg_overrides_policyN)r r r r' r r r) r* r
r
r
r r! $ s r! c O s t �| �}tj|f|�|�S )N)r r r r )�s�args�kw�fr
r
r r [ s
r c @ s e Zd ZejefZdS )�
TestParserN)r r r r r r r$ r
r
r
r r/ _ s r/ c O s t j| �� f|�|�S )N)r �message_from_bytes�encode)r+ r, r- r
r
r r0 b s r0 c O s t �| �� �}tj|f|�|�S )N)r �BytesIOr1 r �message_from_binary_file)r+ r, r- r. r
r
r r3 e s r3 c @ s e Zd ZeefZdS )�TestBytesParserN)r r r r0 r3 r$ r
r
r
r r4 i s r4 �__main__)r r ZunittestZ
email.messager r Z email.policyr Ztest.test_emailr r r! r r/ r0 r3 r4 r �mainr
r
r
r �<module> s 7