o
    Ž¨Êh  ã                   @   sà   d dl Z d dlZd dlZd dlZ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 eƒ Zejdd„ ƒZejdd	„ ƒZejd
d„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZejdd„ ƒZdS )é    N)Ú	SparkPostÚSparkPostAPIException)Úioloopé   )ÚAsyncClientMockc                  C   s@   t jt jddddd� tdƒ} t ¡  | jj¡}|dksJ ‚d S )Nú.https://api.sparkpost.com/api/v1/transmissionséÈ   úapplication/jsonú{"results": "yay"}©ÚstatusÚcontent_typeÚbodyúfake-keyÚyay)	Ú	responsesÚaddÚPOSTr   r   ÚIOLoopÚrun_syncÚtransmissionÚsend)ÚspÚresults© r   úK/var/www/html/env/lib/python3.10/site-packages/test/tornado/test_tornado.pyÚtest_success_send   ó   ûr   c               	      s€  z¹t  d¡} t ¡ \}}t|dƒ�}| | ¡ W d   ƒ n1 s!w   Y  tjtjddddd� t	dƒ‰d	d
|dœ‰ ‡ ‡fdd„}t
 ¡  |¡}t tjd jj¡}t |d d d d ¡}| | d¡ksiJ ‚|dksoJ ‚d	d
t |  d¡¡ d¡dœ‰ ‡ ‡fdd„}t
 ¡  |¡}t tjd jj¡}t |d d d d ¡}| | d¡ks¬J ‚|dks²J ‚W t |¡ d S t |¡ w )NzHello 
World
Úwr   r   r	   r
   r   r   ztest.txtz
text/plain)ÚnameÚtypeÚfilenamec                      ó   ˆj jˆ gd�S ©N)Úattachments©r   r   r   ©Ú
attachmentr   r   r   r   6   ó   z0test_success_send_with_attachments.<locals>.sendr   Úcontentr$   ÚdataÚasciir   )r   r    r*   c                      r"   r#   r%   r   r&   r   r   r   I   r(   r   )ÚsixÚuÚtempfileÚmkstempÚopenÚwriter   r   r   r   r   r   r   ÚjsonÚloadsÚcallsÚrequestr   Úbase64Ú	b64decodeÚdecodeÚ	b64encodeÚencodeÚosÚunlink)Útest_contentÚ_Útemp_file_pathÚ	temp_filer   r   Úrequest_paramsr)   r   r&   r   Ú"test_success_send_with_attachments   sT   
ÿûýÿÿüÿrB   c                  C   ó`   t jt jddddd� t t¡� tdƒ} t ¡  	| j
j¡ W d   ƒ d S 1 s)w   Y  d S )Nr   iô  r	   zU
        {"errors": [{"message": "You failed", "description": "More Info"}]}
        r   r   )r   r   r   ÚpytestÚraisesr   r   r   r   r   r   r   ©r   r   r   r   Útest_fail_sendX   s   û	"þrG   c                     sL   t jt jddddd� tdƒ‰ ‡ fdd„} t ¡ j| d	d
�}|i ks$J ‚d S )Nú5https://api.sparkpost.com/api/v1/transmissions/foobarr   r	   z!{"results": {"transmission": {}}}r   r   c                      ó   ˆ j  d¡S ©NÚfoobar©r   Úgetr   rF   r   r   r   s   ó   ztest_success_get.<locals>.sendé   ©Útimeout)r   r   ÚGETr   r   r   r   )r   r   r   rF   r   Útest_success_geth   s   ûrS   c                     sl   t jt jddddd� t t¡� tdƒ‰ ‡ fdd„} t ¡ j	| d	d
� W d   ƒ d S 1 s/w   Y  d S )NrH   i”  r	   zW
        {"errors": [{"message": "cant find", "description": "where you go"}]}
        r   r   c                      rI   rJ   rL   r   rF   r   r   r   ‡   rN   ztest_fail_get.<locals>.sendrO   rP   )
r   r   rR   rD   rE   r   r   r   r   r   )r   r   rF   r   Útest_fail_gety   s   û	"ûrT   c                  C   s@   t jt jddddd� tdƒ} t ¡  | jj¡}|du sJ ‚d S )Nr   éÌ   r	   Ú r   r   T©	r   r   rR   r   r   r   r   r   Úlist©r   Úresponser   r   r   Útest_nocontent_getŒ   r   r[   c                  C   rC   )Nr   r   r	   z{"results":r   r   )r   r   rR   rD   rE   r   r   r   r   r   r   rX   rF   r   r   r   Útest_brokenjson_getš   s   û"þr\   c                  C   sD   t jt jddddd� tdƒ} t ¡  | jj¡}|ddiks J ‚d S )	Nr   r   r	   z{"ok": false}r   r   ÚokFrW   rY   r   r   r   Útest_noresults_get¨   s   ûr^   c                  C   s@   t jt jddddd� tdƒ} t ¡  | jj¡}|g ksJ ‚d S )Nr   r   r	   z{"results": []}r   r   rW   rY   r   r   r   Útest_success_list¶   r   r_   )r6   r2   r;   r.   rD   r,   Úsparkpost.tornador   r   Útornador   Úutilsr   r   Úactivater   rB   rG   rS   rT   r[   r\   r^   r_   r   r   r   r   Ú<module>   s8    

9





