o
    Ž¨Êh–  ã                   @   s  zd dl mZ W n
   d dlmZ Y d dlZd dlZd dlmZ d dlmZ d dlmZ 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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 )"é    )ÚurlparseN)Ú	SparkPost)Ú	Templates)ÚSparkPostAPIExceptionc                  C   s*   t ddƒ} | jdd�}|d dksJ ‚d S )NÚuriÚkeyÚtest_id)Úidr	   )r   Ú_translate_keys)ÚtÚresults© r   úE/var/www/html/env/lib/python3.10/site-packages/test/test_templates.pyÚtest_translate_keys_with_id   s   
r   c                  C   s8   t jt jddddd� tdƒ} | j ¡ }|dksJ ‚d S )Nú*https://api.sparkpost.com/api/v1/templateséÈ   úapplication/jsonú{"results": "yay"}©ÚstatusÚcontent_typeÚbodyúfake-keyÚyay)Ú	responsesÚaddÚPOSTr   Ú	templatesÚcreate©Úspr   r   r   r   Útest_success_create   s   û
r!   c                  C   sX   t jt jddddd� t t¡� tdƒ} | j ¡  W d   ƒ d S 1 s%w   Y  d S )Nr   éô  r   úU
        {"errors": [{"message": "You failed", "description": "More Info"}]}
        r   r   )	r   r   r   ÚpytestÚraisesr   r   r   r   ©r    r   r   r   Útest_fail_create$   s   û	"þr'   c                  C   s>   t jt jddddd� tdƒ} | jjddd�}|d	ksJ ‚d S )
Nú1https://api.sparkpost.com/api/v1/templates/foobarr   r   r   r   r   Úfoobar©Únamer   )r   r   ÚPUTr   r   Úupdater   r   r   r   Útest_success_update4   s   ûr.   c                  C   s^   t jt jddddd� t t¡� tdƒ} | jjddd� W d   ƒ d S 1 s(w   Y  d S )	Nr(   r"   r   r#   r   r   r)   r*   )	r   r   r,   r$   r%   r   r   r   r-   r&   r   r   r   Útest_fail_updateC   s   û	"þr/   c                  C   s:   t jt jddddd� tdƒ} | j d¡}|dksJ ‚d S )	Nr(   r   r   r   r   r   r)   r   )r   r   ÚDELETEr   r   Údeleter   r   r   r   Útest_success_deleteS   s   ûr2   c                  C   óZ   t jt jddddd� t t¡� tdƒ} | j d¡ W d   ƒ d S 1 s&w   Y  d S )Nr(   r"   r   r#   r   r   r)   )	r   r   r0   r$   r%   r   r   r   r1   r&   r   r   r   Útest_fail_deleteb   ó   û	"þr4   c                  C   s:   t jt jddddd� tdƒ} | j d¡}|i ksJ ‚d S )Nr(   r   r   ú{"results": {}}r   r   r)   ©r   r   ÚGETr   r   Úgetr   r   r   r   Útest_success_getr   s   ûr:   c                  C   s>   t jt jdddddd� tdƒ} | j dd¡}|i ksJ ‚d S )	Nz<https://api.sparkpost.com/api/v1/templates/foobar?draft=trueTr   r   r6   ©Úmatch_querystringr   r   r   r   r)   r7   r   r   r   r   Útest_success_get_with_is_draft€   s   úr=   c                  C   r3   )Nr(   é”  r   úW
        {"errors": [{"message": "cant find", "description": "where you go"}]}
        r   r   r)   )	r   r   r8   r$   r%   r   r   r   r9   r&   r   r   r   Útest_fail_get�   r5   r@   c                  C   s8   t jt jddddd� tdƒ} | j ¡ }|g ksJ ‚d S )Nr   r   r   z{"results": []}r   r   )r   r   r8   r   r   Úlist)r    Úresponser   r   r   Útest_success_listŸ   s   û
rC   c                  C   sR   t jt jddddd� tdƒ} | j di ¡}t jd jjd	ks!J ‚|i ks'J ‚d S )
Nú9https://api.sparkpost.com/api/v1/templates/foobar/previewr   r   r6   r   r   r)   r   z{"substitution_data": {}})	r   r   r   r   r   ÚpreviewÚcallsÚrequestr   r   r   r   r   Útest_success_preview­   s   ûrH   c                  C   s`   t jt jdddddd� tdƒ} | j di d¡}tt jd	 jj	ƒ}|j
d
ks(J ‚|i ks.J ‚d S )NzDhttps://api.sparkpost.com/api/v1/templates/foobar/preview?draft=trueTr   r   r6   r;   r   r)   r   z
draft=true)r   r   r   r   r   rE   r   rF   rG   ÚurlÚquery)r    r   Úparsedr   r   r   Útest_success_preview_with_draft¼   s   úrL   c                  C   s\   t jt jddddd� t t¡� tdƒ} | j di ¡ W d   ƒ d S 1 s'w   Y  d S )NrD   r>   r   r?   r   r   r)   )	r   r   r   r$   r%   r   r   r   rE   r&   r   r   r   Útest_fail_previewÍ   s   û	"þrM   )Úurllib.parser   r$   r   Ú	sparkpostr   r   Úsparkpost.exceptionsr   r   Úactivater!   r'   r.   r/   r2   r4   r:   r=   r@   rC   rH   rL   rM   r   r   r   r   Ú<module>   sH    











