o
    U¾hWM  ã                   @   sR   d dl Z d dlZd dlmZmZ d dlmZ d dlmZ dgZ	G dd„ deƒZ
dS )é    N)ÚmodelsÚutils)ÚMessage)ÚEmailMessageTestCaseÚTestMessageFlatteningc                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )r   c                 C   s@   |   d¡}|   d¡}dgt_| j |¡}| ¡ }|  ||¡ d S )Nú message_with_many_multiparts.emlú
text/plain)Ú_get_email_objectr   ÚTEXT_STORED_MIMETYPESÚmailboxÚprocess_incoming_messageÚget_email_objectÚassertEqual©ÚselfÚincoming_email_objectÚexpected_email_objectÚmsgÚactual_email_object© r   ú^/var/www/html/env/lib/python3.10/site-packages/django_mailbox/tests/test_message_flattening.pyÚ*test_quopri_message_is_properly_rehydrated   s   ÿÿþz@TestMessageFlattening.test_quopri_message_is_properly_rehydratedc                 C   s8   |   d¡}|   d¡}| j |¡}| ¡ }|  ||¡ d S )Nzmessage_with_attachment.eml)r	   r   r   r   r   r   r   r   r   Ú*test_base64_message_is_properly_rehydrated"   s   ÿÿþz@TestMessageFlattening.test_base64_message_is_properly_rehydratedc                 C   sH   |   d¡}|   d¡}tƒ }| ¡ |_| j |¡}| ¡ }|  ||¡ d S )Nz1message_with_defective_attachment_association.emlz8message_with_defective_attachment_association_result.eml)r	   r   Ú	as_stringÚbodyr   r   r   r   )r   r   r   Úmessager   r   r   r   r   Ú)test_message_handles_rehydration_problems5   s   ÿÿ
þz?TestMessageFlattening.test_message_handles_rehydration_problemsc                 C   sŒ   |   d¡}|   d¡}t ¡ }t d¡�}t |¡}d|d< dg|d< ||_| j 	|¡}W d   ƒ n1 s5w   Y  | 
¡ }|  ||¡ d S )Nr   z.message_with_many_multiparts_stripped_html.emlz!django_mailbox.utils.get_settingsTÚstrip_unallowed_mimetypesr   Úallowed_mimetypes)r	   r   Úget_settingsÚmockÚpatchÚcopyÚdeepcopyÚreturn_valuer   r   r   r   )r   r   r   Údefault_settingsr   Úalteredr   r   r   r   r   Ú#test_message_content_type_strippingJ   s&   ÿÿ

ù	þz9TestMessageFlattening.test_message_content_type_strippingc                 C   s0   |   d¡}| j |¡}d}|j}|  ||¡ d S )Nz!message_with_invalid_encoding.emlzÈWe offer loans to private individuals and corporate organizations at 2% interest rate. Interested serious applicants should apply via email with details of their requirements.

Warm Regards,
Loan Team)r	   r   r   Útextr   )r   r   r   Úexpected_textÚactual_textr   r   r   Ú(test_message_processing_unknown_encodingc   s   ÿÿz>TestMessageFlattening.test_message_processing_unknown_encodingN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r'   r+   r   r   r   r   r      s    )r"   r    Údjango_mailboxr   r   Údjango_mailbox.modelsr   Údjango_mailbox.tests.baser   Ú__all__r   r   r   r   r   Ú<module>   s    