Your IP : 216.73.216.213
3
� fR< � @ 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�ej� dS )� N)�headerregistryc C s | j � }|j|� |S )N)�copy�update)Z
base_defaultsZ differences�defaults� r �@/opt/alt/python36/lib64/python3.6/test/test_email/test_policy.py�
make_defaults s
r c @ s� e Zd ZdZddddddd�Zej� Zejddej j
j dej j
j dej
jd �� ej j
� 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 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'd.d/� Z(G d0d1� d1�Z)d2d3� Z*d4d5� Z+d6d7� Z,d8d9� Z-d:d;� Z.d<d=� Z/dS )>�PolicyAPITestsT�N �
Z8bitFN)�max_line_length�linesepZcte_type�raise_on_defect� mangle_from_�message_factoryZlong)�utf8r �header_factoryZ
refold_source�content_managerr r r
z
)r
r )r
r r r c C sf x`| j j� D ]R\}}xH|j� D ]<\}}| j||d�� | jt||�|dj|�� W d Q R X q W q W d S )N)�policy�attrz2change {} docs/docstrings if defaults have changed)�policies�items�subTest� assertEqual�getattr�format)�selfr �expectedr �valuer r r �
test_defaultsB s z PolicyAPITests.test_defaultsc C s� xz| j j� D ]l\}}xbt|�D ]V}| j||d��> |jd�sRtttjj |�t
j �rVw n| j ||dj
|�� W d Q R X q W q W d S )N)r r �_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_coveredJ s
z*PolicyAPITests.test_all_attributes_coveredc C sL | j t��}tjj� W d Q R X t|j�}d}x|D ]}| j||� q4W d S )N�fold� fold_binary�header_fetch_parse�header_source_parse�header_store_parse)r* r+ r, r- r. )� assertRaises� TypeErrorr$ r �Policy�strZ exceptionr( )r �cm�msgZabstract_methods�methodr r r �test_abcV s
zPolicyAPITests.test_abcc
C sp xj| j j� D ]\\}}x4|D ],}| jt|d �� t||d � W d Q R X qW | jtd�� d |_W d Q R X q W d S )Nz .*read-onlyzno attribute.*foo)r r �assertRaisesRegex�AttributeError�setattr�foo)r r r r r r r �test_policy_is_immutableb s
z'PolicyAPITests.test_policy_is_immutablec C sT xN| j j� D ]@\}}dd� |D �}|jf |�}x |D ]}| jt||�� q4W q W d S )Nc S s i | ]
}d |�qS )Nr )�.0r r r r �
<dictcomp>n s 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_clonedj s
z0PolicyAPITests.test_set_policy_attrs_when_clonedc C sT xN| j D ]D}| jt�� |d d� W d Q R X | jt�� |d d� W d Q R X qW d S )N)Z this_keyword_should_not_be_valid)Znewtline)r r/ r0 )r r? r r r �*test_reject_non_policy_keyword_when_calleds s
z9PolicyAPITests.test_reject_non_policy_keyword_when_calledc C s� | j j� }tjjjdd�}tjjjdd�}|| }|jdd� x&|j� D ]\}}| jt ||�|� qHW || }|jdd� x&|j� D ]\}}| jt ||�|� q�W |tjj }x&|j� D ]\}}| jt ||�|� q�W d S )N�d )r �2 )
�policy_defaultsr r$ r �defaultr> r r r r )r r Zp1Zp2Zaddedr r r r r �test_policy_additionz s
z#PolicyAPITests.test_policy_additionc C sh G dd� d�}|� }t � }tjj� }|j||� | j|j|g� t � }|j||� | j|j||g� d S )Nc @ s e Zd Zdd� ZdS )z2PolicyAPITests.test_register_defect.<locals>.Dummyc S s
g | _ d S )N)�defects)r r r r �__init__� s z;PolicyAPITests.test_register_defect.<locals>.Dummy.__init__N)�__name__�
__module__� __qualname__rH r r r r �Dummy� s rL )�objectr$ r r% �register_defectr rG )r rL �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 )N)rG )r r r r rH � s z PolicyAPITests.MyObj.__init__N)rI rJ rK rH r r r r �MyObj� s rS c @ s e Zd ZdS )zPolicyAPITests.MyDefectN)rI rJ rK r r r r �MyDefect� s rT c
C s@ | j � }| jd�}| j| jd�� tjjj||� W d Q R X d S )Nzthe telly is broken)rS rT r7 r$ r �strict�
handle_defect)r r: rP r r r �#test_handle_defect_raises_on_strict� s
z2PolicyAPITests.test_handle_defect_raises_on_strictc C sb | j � }| jd�}tjjj||� | j|j|g� | jd�}tjjj||� | j|j||g� d S )N�one�two)rS rT r$ r rE rV r rG )r r: �defect1rQ 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 )NrG )�superrH )r �args�kw)� __class__r r rH � s z PolicyAPITests.MyPolicy.__init__c C s | j j|� d S )N)rG �append)r rO rP r r r rN � s z'PolicyAPITests.MyPolicy.register_defect)rI rJ rK rG rH rN �
__classcell__r r )r_ r �MyPolicy� s rb c
C sD | j � }| jd�}| j| jd�� | jdd�j||� W d Q R X d S )Nzthe telly is brokenT)r )rS rT r7 rb rV )r r: rP r r r �,test_overridden_register_defect_still_raises� s
z;PolicyAPITests.test_overridden_register_defect_still_raisesc C s~ | j � }| jd�}| j� }|j||� | j|j|g� | j|jg � | jd�}|j||� | j|j||g� | j|jg � d S )NrX rY )rS rT rb rV r rG )r r: rZ Z my_policyrQ r r r �%test_overridden_register_defect_works� s
z4PolicyAPITests.test_overridden_register_defect_worksc C s> t jjjdd�}| j|jd� | j|tj� | j|tj � d S )NZTest�test)
r$ r rE 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)rI rJ rK r rh �parser r r r �Foo� s rl c C sx t jj� }t jj� }|jjd| j� |jdd�}| j|| j� | j|tj � |jdd�}| j|| j� | j|tj � d S )Nr: re )
r$ r r% r � map_to_typerl rg ZassertNotIsInstancer rh )r �policy1�policy2ri r r r �$test_each_Policy_gets_unique_factory� s
z3PolicyAPITests.test_each_Policy_gets_unique_factoryc C sZ t jj� }|j� }|jjd| j� |jdd�}| j|| j� |jdd�}| j|| j� d S )Nr: re )r$ r r% r> r rm rl rg )r rn ro ri r r r �test_clone_copies_factory� s
z(PolicyAPITests.test_clone_copies_factoryc C sH t jj� }|j}|t jj }| j|j|� t jj| }| j|j|� d S )N)r$ r r% r rU r )r ZmypolicyZ myfactory� newpolicyr r r �"test_new_factory_overrides_default� s
z1PolicyAPITests.test_new_factory_overrides_defaultc C s: t jjt jj }| j|jt jjj� | j|jddi� d S )Nr T)r$ r rE rU r r r% �__dict__)r rr r r r �6test_adding_default_policies_preserves_default_factory� s zEPolicyAPITests.test_adding_default_policies_preserves_default_factoryc C s4 t jjjdd�}|jdd�}| j|ddd � d S ) N� )r �Subjectu ą� z
Subject:
z =?utf-8?q?=C4=85?=
u ąąąąąąąąąąąą)r$ r rE r> 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 sd d}xZdD ]R}| j |d��< tjjj|d�}| jtjj�� |jd|� W d Q R X W d Q R X q
W d S ) Nz;Melt away the pounds with this one simple trick! 1234567890� � � )�maxlen)r rw )rz r{ r| ) r r$ r rE r> r/ �errorsZHeaderParseErrorr* )r Zsubjectr} r r r r �test_short_maxlen_error� s
z&PolicyAPITests.test_short_maxlen_error)0rI rJ rK Z longMessageZcompat32_defaultsr rD r r$ r r% r r �messageZ EmailMessageZ
new_policyZcompat32r rE ZSMTPZSMTPUTF8ZHTTPrU r r r) r6 r; r@ rA rF rR rS � ExceptionrT rW r[ rb rc rd rj rl rp rq rs ru ry r r r r r r sl
r c @ s e Zd ZdS )�
TestExceptionN)rI rJ rK r r r r r� 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 )Nre )r� )r r] r^ r r r � badmethod s z(TestPolicyPropagation.MyPolicy.badmethodN) rI rJ rK r� r* r+ Zheader_fetch_parserr- r. r r r r rb s rb c C s, | j td�� tjd| jd� W d Q R X d S )Nz^test$zSubject: test
)r )r7 r� r$ �message_from_stringrb )r r r r �test_message_from_string s z.TestPolicyPropagation.test_message_from_stringc C s, | j td�� tjd| jd� W d Q R X d S )Nz^test$s Subject: test
)r )r7 r� r$ Zmessage_from_bytesrb )r r r r �test_message_from_bytes$ s z-TestPolicyPropagation.test_message_from_bytesc C s6 t jd�}| jtd�� tj|| jd� W d Q R X d S )NzSubject: test
z^test$)r )�io�StringIOr7 r� r$ Zmessage_from_filerb )r �fr r r �test_message_from_file) s
z,TestPolicyPropagation.test_message_from_filec C s6 t jd�}| jtd�� tj|| jd� W d Q R X d S )Ns Subject: test
z^test$)r )r� �BytesIOr7 r� r$ Zmessage_from_binary_filerb )r r� r r r � test_message_from_binary_file. s
z3TestPolicyPropagation.test_message_from_binary_filec C s6 t jj| jd�}| jtd�� |jd� W d Q R X d S )N)r z^test$zSubject: test
)r$ �parserZParserrb r7 r� Zparsestr)r �pr r r � test_parser5 s z!TestPolicyPropagation.test_parserc C s6 t jj| jd�}| jtd�� |jd� W d Q R X d S )N)r z^test$s Subject: test
)r$ r� Z BytesParserrb r7 r� Z
parsebytes)r r� r r r �test_bytes_parser: s z'TestPolicyPropagation.test_bytes_parser�Subject: test
Nc C s* |d krt jjj� n|| _t j|| jd�S )N)r )r$ r rE r> r� )r �sourcer r r r � _make_msgC s z TestPolicyPropagation._make_msgc C s | j � }| j|j| j� d S )N)r� �assertIsr )r r4 r r r �(test_parser_propagates_policy_to_messageG s z>TestPolicyPropagation.test_parser_propagates_policy_to_messagec C s6 | j tjd��}x |j� D ]}| j|j| j� qW d 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 r4 �partr r r �-test_parser_propagates_policy_to_sub_messagesK s zCTestPolicyPropagation.test_parser_propagates_policy_to_sub_messagesc C sL | j dtjjjdd�d�}tj� }tjj|�}|j |� | j
|j � d� d S )NzSubject: test
To: foo
�X)r
)r zSubject: testXTo: fooXX) r� r$ r rE r> r� r� � generator� GeneratorZflattenr �getvalue)r r4 �s�gr r r �+test_message_policy_propagates_to_generator^ s
zATestPolicyPropagation.test_message_policy_propagates_to_generatorc C s. | j dtjjjdd�d�}| j|j� d� d S )NzSubject: test
To: foo
r� )r
)r zSubject: testXTo: fooXX)r� r$ r rE r> r Z as_string)r r4 r r r �%test_message_policy_used_by_as_stringf s z;TestPolicyPropagation.test_message_policy_used_by_as_string)r� N)rI rJ rK r$ r r1 rb 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 jj� }| jt|jdd� d S )NZFromzspam
egg@foo.py)r$ r r% r/ �
ValueErrorr. )r �instancer r r �(test_header_store_parse_rejects_newlinesn s
z=TestConcretePolicies.test_header_store_parse_rejects_newlinesN)rI rJ rK r� r r r r r� l s r� �__main__)r� r& r� ZunittestZ email.errorsr$ Z email.policyZ email.parserZemail.generatorZ
email.messager r ZTestCaser r� r� r� r� rI �mainr r r r �<module> s$ X