o
    w+f                     @   sN   d dl Z d dlT zd dlZW n ey   d dlZY nw 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)selfglobal_stats r   G/var/www/html/venv/lib/python3.10/site-packages/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   	all_statsr   r   r   
test_Stats   s   
zUnitTests.test_Statsc                 C   `   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   
categories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"})CategoryStatsadd_categoryCategoryr   r   r   r   r    r!   r
   r   r   r   )r   category_statsr   r   r   test_categoryStats.      zUnitTests.test_categoryStatsc                 C   r$   )Nr   r%   r&   )r   subusers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"]})SubuserStatsadd_subuserSubuserr   r   r   r   r    r!   r
   r   r   r   )r   subuser_statsr   r   r   test_subuserStatsC   r.   zUnitTests.test_subuserStatsN)__name__
__module____qualname__r   r#   r-   r5   r   r   r   r   r      s
    r   )r   sendgrid.helpers.stats	unittest2unittestImportErrorTestCaser   r   r   r   r   <module>   s   