o Y@sddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZGdddeZdd ZGd d d eZdS) N)settings)TestCase)modelsutils)MailboxMessagec@s eZdZdS)EmailIntegrationTimeoutN)__name__ __module__ __qualname__r r K/var/www/html/env/lib/python3.10/site-packages/django_mailbox/tests/base.pyrsrcCsJttjtjtd|d }|WdS1swYdSNmessagesrbopenospathjoindirname__file__read)namefr r r get_email_as_text $rcspeZdZddgZfddZdddZdd Zd d Zdd d ZddZ ddZ fddZ fddZ Z S)EmailMessageTestCasez MIME-VersionzContent-Transfer-Encodingcst}|d|_|d|_|d|_tjjdd|_t j d|_ t j d|_ t j d|_d |_d |_d t_|jt_d t_|j t_|j t_d t_tt|dS)Nallowed_mimetypesstrip_unallowed_mimetypestext_stored_mimetypeszfrom@example.com) from_email EMAIL_ACCOUNTEMAIL_PASSWORDEMAIL_SMTP_SERVERznobody@nowhere.comi,z+django.core.mail.backends.smtp.EmailBackendiKT)r get_settings_ALLOWED_MIMETYPES_STRIP_UNALLOWED_MIMETYPES_TEXT_STORED_MIMETYPESrobjectscreatemailboxrenvironget test_account test_passwordtest_smtp_servertest_from_emailmaximum_wait_secondsr EMAIL_BACKEND EMAIL_HOST EMAIL_PORTEMAIL_HOST_USEREMAIL_HOST_PASSWORD EMAIL_USE_TLSsuperrsetUp)self dm_settings __class__r r r:$s$  zEmailMessageTestCase.setUpNcCsBt|j} t|krt|j|}|r|Stdq)NT)timer2rr+ get_new_mailsleep)r;r+ condition maximum_waitrr r r _get_new_messages>s   z&EmailMessageTestCase._get_new_messagescCsJttjtjtd|d }|WdS1swYdSrr)r;rrr r r _get_email_as_textHrz'EmailMessageTestCase._get_email_as_textcCs$||}tjr t|St|SN)rFsixPY3emailmessage_from_bytesmessage_from_string)r;rcopyr r r _get_email_objectSs   z&EmailMessageTestCase._get_email_objectcCsd|dkr|dd}|dd}|dddd}|dddd}||kr0dSdS) a Check if headers are (close enough to) identical. * This is particularly tricky because Python 2.6, Python 2.7 and Python 3 each handle header strings slightly differently. This should mash away all of the differences, though. * This also has a small loophole in that when re-writing e-mail payload encodings, we re-build the Content-Type header, so if the header was originally unquoted, it will be quoted when rehydrating the e-mail message. z content-type"z  z FT)lowerreplaceupper)r;leftrightheaderr r r _headers_identicalZs z'EmailMessageTestCase._headers_identicalcCs|D]1\}}||s||jvrq||std||j||||ds5td|t|t||fq|D]1\}}||sH||jvrHq:||sRtd||j||||dsktd|t|t||fq:||krz||||r|}|}t|t|kr|||t t|D] }| ||||qdS|dus|dur|dur|jdur||||dur|jdur|||dSdSdS| | kr|||dSdS)NzExtra header '%s')rWzHeader '%s' unequal: %s %s) itemsALLOWED_EXTRA_HEADERSAssertionErrorrXrepr is_multipart_raise_mismatched get_payloadlenrangecompare_email_objectsstrip)r;rUrVkeyvalue left_payloadsright_payloadsnr r r rbssj           z*EmailMessageTestCase.compare_email_objectscCstd||f)Nz$Message payloads do not match: %s %s)r[ as_stringr;rUrVr r r r^s z&EmailMessageTestCase._raise_mismatchedcs,t|tjjstt|||S|||SrG) isinstancerJmessagerr9r assertEqualrbrjr=r r rms z EmailMessageTestCase.assertEqualcsLtjD]}|q|jt_|jt_|j t_ |j t t |dSrG)rr)alldeleter&rALLOWED_MIMETYPESr'STRIP_UNALLOWED_MIMETYPESr(TEXT_STORED_MIMETYPESr+r9rtearDown)r;rlr=r r rss  zEmailMessageTestCase.tearDownrG)r r r rZr:rErFrNrXrbr^rmrs __classcell__r r r=r rs    3 r)rJos.pathrr@rH django.confr django.testrdjango_mailboxrrdjango_mailbox.modelsrr Exceptionrrrr r r r s