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clicksd      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   