Your IP : 216.73.216.213
a
XC?h�A � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dlm
Z
dd� Z G dd� dej �Z
G dd� de�ZG d d
� d
ej �ZG d d � d ej �Zed
kr�e�� dS )� N)�headerregistryc C s | � � }|�|� |S �N)�copy�update)Z
base_defaultsZ differences�defaults� r �@/opt/alt/python39/lib64/python3.9/test/test_email/test_policy.py�
make_defaults s
r c @ s� e Zd ZdZdddddddd�Ze�� Ze�ddej j
j dej j
j dej
jd �� ej �
� Zej jeei �ej jeei �ej jeed
d i�ej jeed dd ��ej jeed dd
��ej jeeddi�eeei �iZej ee d< dd� Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z d d!� Z G d"d#� d#�Z!G d$d%� d%e"�Z#d&d'� Z$d(d)� Z%G d*d+� d+ej j
�Z&d,d-� Z'd.d/� Z(d0d1� Z)G d2d3� d3�Z*d4d5� Z+d6d7� Z,d8d9� Z-d:d;� Z.d<d=� Z/d>d?� Z0d@dA� Z1dS )B�PolicyAPITestsT�N �
Z8bitFN)�max_line_length�linesepZcte_type�raise_on_defect� mangle_from_�message_factory�verify_generated_headersZlong)�utf8r �header_factoryZ
refold_source�content_managerr r r z
)r r )r r
r r c
C sr | j �� D ]b\}}|�� D ]P\}}| j||d��* | �t||�|d�|�� W d � q1 s`0 Y qq
d S )N��policy�attrz2change {} docs/docstrings if defaults have changed)�policies�items�subTest� assertEqual�getattr�format)�selfr �expectedr �valuer r r �
test_defaultsC s ��z PolicyAPITests.test_defaultsc
C s� | j �� D ]�\}}t|�D ]v}| j||d��T |�d�sNtttjj |�t
j �r^W d � qn| � ||d�
|�� W d � q1 s�0 Y qq
d S )Nr �_z{} is not fully tested)r r �dirr �
startswith�
isinstancer �emailr � EmailPolicy�types� FunctionType�assertInr )r r r r r r r �test_all_attributes_coveredK s
���z*PolicyAPITests.test_all_attributes_coveredc C s\ | � t��}tj�� W d � n1 s*0 Y t|j�}d}|D ]}| �||� qFd S )N)�fold� fold_binaryZheader_fetch_parse�header_source_parse�header_store_parse)� assertRaises� TypeErrorr' r �Policy�strZ exceptionr+ )r �cm�msgZabstract_methods�methodr r r �test_abcW s (
zPolicyAPITests.test_abcc
C s� | j �� D ]�\}}|D ]@}| �t|d �� t||d � W d � q1 sL0 Y q| �td�� d |_W d � q
1 s�0 Y q
d S )Nz .*read-onlyzno attribute.*foo)r r �assertRaisesRegex�AttributeError�setattr�foo)r r r r r r r �test_policy_is_immutablec s ,z'PolicyAPITests.test_policy_is_immutablec C sP | j �� D ]@\}}dd� |D �}|jf i |��}|D ]}| �t||�� q4q
d S )Nc S s i | ]
}|d �qS r r )�.0r r r r �
<dictcomp>o � zDPolicyAPITests.test_set_policy_attrs_when_cloned.<locals>.<dictcomp>)r r �cloneZ assertIsNoner )r � policyclassr Z testattrdictr r r r r �!test_set_policy_attrs_when_clonedk s
z0PolicyAPITests.test_set_policy_attrs_when_clonedc C sx | j D ]l}| �t�� |d d� W d � n1 s40 Y | �t�� |d d� W d � q1 sh0 Y qd S )N)Z this_keyword_should_not_be_valid)Znewtline)r r1 r2 )r rB r r r �*test_reject_non_policy_keyword_when_calledt s
( z9PolicyAPITests.test_reject_non_policy_keyword_when_calledc C s� | j �� }tjjjdd�}tjjjdd�}|| }|jdd� |�� D ]\}}| �t ||�|� qF|| }|jdd� |�� D ]\}}| �t ||�|� q~|tjj }|�� D ]\}}| �t ||�|� q�d S )N�d �r
�2 )
�policy_defaultsr r' r �defaultrA r r r r )r r �p1�p2Zaddedr r! r r r �test_policy_addition{ s
z#PolicyAPITests.test_policy_additionc C sj d}t j�� }d|d< t jjjdd�}t jjjd d�}| �|�d|d �|� | �|�d|d �|� d S )Nz Subject: =?utf-8?q?=C3=A1?=
� á�Subjectr rF )r' �message� EmailMessager rI rA r r- )r r r6 rJ rK r r r � test_fold_zero_max_line_length� s
z-PolicyAPITests.test_fold_zero_max_line_lengthc C sh G dd� d�}|� }t � }tj�� }|�||� | �|j|g� t � }|�||� | �|j||g� d S )Nc @ s e Zd Zdd� ZdS )z2PolicyAPITests.test_register_defect.<locals>.Dummyc S s
g | _ d S r ��defects�r r r r �__init__� s z;PolicyAPITests.test_register_defect.<locals>.Dummy.__init__N��__name__�
__module__� __qualname__rU r r r r �Dummy� s rZ )�objectr' r r( �register_defectr rS )r rZ �obj�defectr �defect2r r r �test_register_defect� s
z#PolicyAPITests.test_register_defectc @ s e Zd Zdd� ZdS )zPolicyAPITests.MyObjc C s
g | _ d S r rR rT r r r rU � s z PolicyAPITests.MyObj.__init__NrV r r r r �MyObj� s ra c @ s e Zd ZdS )zPolicyAPITests.MyDefectN�rW rX rY r r r r �MyDefect� s rc c C sT | � � }| �d�}| �| jd�� tjj�||� W d � n1 sF0 Y d S )N�the telly is broken)ra rc r9 r' r �strict�
handle_defect�r r< r^ r r r �#test_handle_defect_raises_on_strict� s
z2PolicyAPITests.test_handle_defect_raises_on_strictc C sb | � � }| �d�}tjj�||� | �|j|g� | �d�}tjj�||� | �|j||g� d S �NZoneZtwo)ra rc r' r rI rf r rS )r r< �defect1r_ r r r �#test_handle_defect_registers_defect� s
z2PolicyAPITests.test_handle_defect_registers_defectc s( e Zd ZdZ� fdd�Zdd� Z� ZS )zPolicyAPITests.MyPolicyNc s t � j|dg i|�� d S )NrS )�superrU �r �args�kw�� __class__r r rU � s z PolicyAPITests.MyPolicy.__init__c C s | j �|� d S r )rS �append)r r] r^ r r r r\ � s z'PolicyAPITests.MyPolicy.register_defect)rW rX rY rS rU r\ �
__classcell__r r rp r �MyPolicy� s rt c C sX | � � }| �d�}| �| jd��$ | jdd��||� W d � n1 sJ0 Y d S )Nrd T)r )ra rc r9 rt rf rg r r r �,test_overridden_register_defect_still_raises� s
z;PolicyAPITests.test_overridden_register_defect_still_raisesc C s~ | � � }| �d�}| �� }|�||� | �|j|g� | �|jg � | �d�}|�||� | �|j||g� | �|jg � d S ri )ra rc rt rf r rS )r r< rj Z my_policyr_ r r r �%test_overridden_register_defect_works� s
z4PolicyAPITests.test_overridden_register_defect_worksc C s> t jj�dd�}| �|jd� | �|tj� | �|tj � d S )NZTest�test)
r' r rI r r �name�assertIsInstancer �UnstructuredHeaderZ
BaseHeader)r �hr r r �test_default_header_factory� s z*PolicyAPITests.test_default_header_factoryc @ s e Zd ZejjZdS )zPolicyAPITests.FooN)rW rX rY r rz �parser r r r �Foo� s r~ c C sx t j�� }t j�� }|j�d| j� |�dd�}| �|| j� | �|tj � |�dd�}| �|| j� | �|tj � d S �Nr< rw )
r' r r( r � map_to_typer~ ry ZassertNotIsInstancer rz �r Zpolicy1Zpolicy2r{ r r r �$test_each_Policy_gets_unique_factory� s
z3PolicyAPITests.test_each_Policy_gets_unique_factoryc C sZ t j�� }|�� }|j�d| j� |�dd�}| �|| j� |�dd�}| �|| j� d S r )r' r r( rA r r� r~ ry r� r r r �test_clone_copies_factory� s
z(PolicyAPITests.test_clone_copies_factoryc C sH t j�� }|j}|t jj }| �|j|� t jj| }| �|j|� d S r )r' r r( r re r )r ZmypolicyZ myfactory� newpolicyr r r �"test_new_factory_overrides_default� s
z1PolicyAPITests.test_new_factory_overrides_defaultc C s: t jjt jj }| �|jt jjj� | �|jddi� d S )Nr T)r' r rI re r r r( �__dict__)r r� r r r �6test_adding_default_policies_preserves_default_factory� s
�zEPolicyAPITests.test_adding_default_policies_preserves_default_factoryc C s, t jjjdd�}|�dd�}| �|d� d S )N� rF rN u ąąąąąąąąąąąąz�Subject:
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
=?utf-8?q?=C4=85?=
)r' r rI rA r- r )r r Zactualr r r �*test_non_ascii_chars_do_not_cause_inf_loop� s �z9PolicyAPITests.test_non_ascii_chars_do_not_cause_inf_loopc
C s� d}dD ]z}| j |d��Z tjjj|d�}| �tjj�� |�d|� W d � n1 sZ0 Y W d � q1 sx0 Y qd S )Nz;Melt away the pounds with this one simple trick! 1234567890)� � � )�maxlenrF rN ) r r' r rI rA r1 �errorsZHeaderParseErrorr- )r Zsubjectr� r r r r �test_short_maxlen_error s z&PolicyAPITests.test_short_maxlen_errorc C s� t jjjdd�}dD ]v}| j|d��V t jd|d�}G dd� dt�}|d = ||�|d < | �|�� |� d
�� W d � q1 s�0 Y qd S ) NF)r )z Header: Value
Bad: Injection
zHeader: NoNewLine)�textzHeader: Value
Body�r c @ s e Zd ZdZdd� ZdS )zCPolicyAPITests.test_verify_generated_headers.<locals>.LiteralHeader�Headerc [ s | S r r )r �kwargsr r r r- ' s zHPolicyAPITests.test_verify_generated_headers.<locals>.LiteralHeader.foldN)rW rX rY rx r- r r r r �
LiteralHeader% s r� r� z
Body) r' r rI rA r �message_from_stringr4 r � as_string)r r r� rO r� r r r � test_verify_generated_headers s � �z,PolicyAPITests.test_verify_generated_headers)2rW rX rY Z longMessageZcompat32_defaultsr rH r r' r r( r r rO rP Z
new_policyZcompat32r rI ZSMTPZSMTPUTF8ZHTTPre r r" r, r8 r= rC rD rL rQ r` ra � Exceptionrc rh rk rt ru rv r| r~ r� r� r� r� r� r� r� r r r r r
s� � �
�
��
��
�
�
r
c @ s e Zd ZdS )�
TestExceptionNrb r r r r r� 7 s r� c @ sz e Zd ZG dd� dejj�Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z ddd�Z
dd� Zdd� Zdd� Zdd� ZdS ) �TestPolicyPropagationc @ s( e Zd Zdd� Ze Z ZZe ZZdS )z TestPolicyPropagation.MyPolicyc O s t d��d S )Nrw )r� rm r r r � badmethod@ s z(TestPolicyPropagation.MyPolicy.badmethodN) rW rX rY r� r- r. Zheader_fetch_parserr/ r0 r r r r rt ? s rt c C s@ | � td�� tjd| jd� W d � n1 s20 Y d S )N�^test$�Subject: test
r� )r9 r� r' r� rt rT r r r �test_message_from_stringE s �z.TestPolicyPropagation.test_message_from_stringc C s@ | � td�� tjd| jd� W d � n1 s20 Y d S )Nr� � Subject: test
r� )r9 r� r' Zmessage_from_bytesrt rT r r r �test_message_from_bytesJ s �z-TestPolicyPropagation.test_message_from_bytesc C sJ t �d�}| �td�� tj|| jd� W d � n1 s<0 Y d S )Nr� r� r� )�io�StringIOr9 r� r' Zmessage_from_filert �r �fr r r �test_message_from_fileO s
z,TestPolicyPropagation.test_message_from_filec C sJ t �d�}| �td�� tj|| jd� W d � n1 s<0 Y d S )Nr� r� r� )r� �BytesIOr9 r� r' Zmessage_from_binary_filert r� r r r � test_message_from_binary_fileT s
z3TestPolicyPropagation.test_message_from_binary_filec C sJ t jj| jd�}| �td�� |�d� W d � n1 s<0 Y d S )Nr� r� r� )r' �parserZParserrt r9 r� Zparsestr�r �pr r r � test_parser[ s z!TestPolicyPropagation.test_parserc C sJ t jj| jd�}| �td�� |�d� W d � n1 s<0 Y d S )Nr� r� r� )r' r� Z BytesParserrt r9 r� Z
parsebytesr� r r r �test_bytes_parser` s z'TestPolicyPropagation.test_bytes_parserr� Nc C s* |d u rt jj�� n|| _t j|| jd�S )Nr� )r' r rI rA r� )r �sourcer r r r � _make_msgi s z TestPolicyPropagation._make_msgc C s | � � }| �|j| j� d S r )r� �assertIsr �r r6 r r r �(test_parser_propagates_policy_to_messagem s z>TestPolicyPropagation.test_parser_propagates_policy_to_messagec C s2 | � t�d��}|�� D ]}| �|j| j� qd S )Na+ Subject: mime test
MIME-Version: 1.0
Content-Type: multipart/mixed, boundary="XXX"
--XXX
Content-Type: text/plain
test
--XXX
Content-Type: text/plain
test2
--XXX--
)r� �textwrap�dedent�walkr� r )r r6 �partr r r �-test_parser_propagates_policy_to_sub_messagesq s zCTestPolicyPropagation.test_parser_propagates_policy_to_sub_messagesc C sL | j dtjjjdd�d�}t�� }tj�|�}|� |� | �
|� � d� d S �NzSubject: test
To: foo
�X)r r� zSubject: testXTo: fooXX) r� r' r rI rA r� r� � generator� GeneratorZflattenr �getvalue)r r6 �s�gr r r �+test_message_policy_propagates_to_generator� s �
zATestPolicyPropagation.test_message_policy_propagates_to_generatorc C s. | j dtjjjdd�d�}| �|�� d� d S r� )r� r' r rI rA r r� r� r r r �%test_message_policy_used_by_as_string� s �z;TestPolicyPropagation.test_message_policy_used_by_as_string)r� N)rW rX rY r' r r3 rt r� r� r� r� r� r� r� r� r� r� r� r r r r r� : s
r� c @ s e Zd Zdd� ZdS )�TestConcretePoliciesc C s t j�� }| �t|jdd� d S )NZFromzspam
egg@foo.py)r' r r( r1 �
ValueErrorr0 )r �instancer r r �(test_header_store_parse_rejects_newlines� s
�z=TestConcretePolicies.test_header_store_parse_rejects_newlinesN)rW rX rY r� r r r r r� � s r� �__main__)r� r) r� ZunittestZ email.errorsr' Z email.policyZ email.parserZemail.generatorZ
email.messager r ZTestCaser
r� r� r� r� rW �mainr r r r �<module> s$ (X