U
    ä«Fc  ã                   @   sP   d dl Z d dlT zd dlZW n ek
r8   d dlZY nX G dd„ dejƒZdS )é    N)Ú*c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
Ú	UnitTestsc                 C   s<   t dd�}|  tj| ¡ dd�d¡ |  tt|ƒtƒ¡ dS )zMinimum required for statsú
12-09-2017©Ú
start_dateT©Ú	sort_keysz{"start_date": "12-09-2017"}N)ÚStatsÚassertEqualÚjsonÚdumpsÚgetÚ
assertTrueÚ
isinstanceÚstr)ÚselfZglobal_stats© r   ú8/tmp/pip-unpacked-wheel-7j3043uk/test/unit/test_stats.pyÚtest_basicStats   s    
þüzUnitTests.test_basicStatsc                 C   sL   t dd�}d|_d|_d|_d|_d|_d|_|  tj	| 
¡ d	d
�d¡ d S )Nr   r   ú
12-10-2017ÚdayÚascÚclickséd   é   Tr   z¡{"aggregated_by": "day", "end_date": "12-10-2017", "limit": 100, "offset": 2, "sort_by_direction": "asc", "sort_by_metric": "clicks", "start_date": "12-09-2017"})r	   Úend_dateÚaggregated_byÚ_sort_by_directionÚsort_by_metricÚ_limitÚ_offsetr
   r   r   r   )r   Z	all_statsr   r   r   Ú
test_Stats   s    
þüzUnitTests.test_Statsc                 C   s`   t dddgd�}| tdƒ¡ d|_d|_d|_d	|_d
|_d|_|  	t
j| ¡ dd�d¡ d S )Nr   ÚfooÚbar)r   Ú
categoriesZwoor   r   r   r   r   r   Tr   zÆ{"aggregated_by": "day", "categories": ["foo", "bar", "woo"], "end_date": "12-10-2017", "limit": 100, "offset": 2, "sort_by_direction": "asc", "sort_by_metric": "clicks", "start_date": "12-09-2017"})ZCategoryStatsZadd_categoryZCategoryr   r   r   r   r   r    r
   r   r   r   )r   Zcategory_statsr   r   r   Útest_categoryStats.   s    þüzUnitTests.test_categoryStatsc                 C   s`   t dddgd�}| tdƒ¡ d|_d|_d|_d	|_d
|_d|_|  	t
j| ¡ dd�d¡ d S )Nr   r"   r#   )r   ZsubusersZblahr   r   r   r   r   r   Tr   zÅ{"aggregated_by": "day", "end_date": "12-10-2017", "limit": 100, "offset": 2, "sort_by_direction": "asc", "sort_by_metric": "clicks", "start_date": "12-09-2017", "subusers": ["foo", "bar", "blah"]})ZSubuserStatsZadd_subuserZSubuserr   r   r   r   r   r    r
   r   r   r   )r   Zsubuser_statsr   r   r   Útest_subuserStatsC   s    þüzUnitTests.test_subuserStatsN)Ú__name__Ú
__module__Ú__qualname__r   r!   r%   r&   r   r   r   r   r      s   r   )r   Zsendgrid.helpers.statsZ	unittest2ZunittestÚImportErrorZTestCaser   r   r   r   r   Ú<module>   s   