o
    ^                     @   sj   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddl	m
Z
 G d	d
 d
ejZdS )z"Define the resultset test package.    )absolute_import)division)print_function)unicode_literalsN)InfluxDBClientError)	ResultSetc                   @   s`   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd ZdS )TestResultSetz!Define the ResultSet test object.c                 C   sV   dddg dg dg dgddg dg d	gdgigi| _ t| j d d
 | _dS )z$Set up an instance of TestResultSet.resultsseriescpu_load_shorttimevaluehostregion)2015-01-29T21:51:28.968422294Z{Gz?server01us-west)r   ?server02r   )namecolumnsvaluesother_series)r   Q?r   r   r   N)Zquery_responser   rsself r   ?/usr/lib/python3/dist-packages/influxdb/tests/resultset_test.pysetUp   s    zTestResultSet.setUpc                 C   sP   ddddddddddg}|  |t| jd  |  |t| jjdd	 d
S )z/Test filtering by name in TestResultSet object.r   r   r   r   r   r   r   r   r   r   r   )ZmeasurementNassertEquallistr   Z
get_pointsr   Zexpectedr   r   r    test_filter_by_name/   s"   z!TestResultSet.test_filter_by_namec                 C   sX   ddddddddddg}|  |t| jddi  |  |t| jjddid d	S )
z,Test filter by tags in TestResultSet object.r   r   r   r   r"   r   r   )ZtagsNr#   r&   r   r   r    test_filter_by_tagsA   s&   z!TestResultSet.test_filter_by_tagsc              	   C   sd   |  t| jdddif dddddg |  t| jdddif dddddd	dd
ddg dS )z5Test filter by name and tags in TestResultSet object.r   r   r   r   r   r   r"   r   r   r   N)r$   r%   r   r   r   r   r    test_filter_by_name_and_tagsX   s,   z*TestResultSet.test_filter_by_name_and_tagsc                 C   s   |  | j ddg dS )z"Test keys in TestResultSet object.r   Nr   NN)r$   r   keysr   r   r   r    	test_keysp   s   zTestResultSet.test_keysc                 C   s   |  t| jd dS )z$Test length in TestResultSet object.   N)r$   lenr   r   r   r   r    test_lenz   s   zTestResultSet.test_lenc              
   C   s\   t | j }dd |D }| |dddddddd	d
ddgfddddddgfg dS )z#Test items in TestResultSet object.c                 S   s    g | ]}|d  t |d fqS )r      )r%   ).0itemr   r   r    
<listcomp>   s     z,TestResultSet.test_items.<locals>.<listcomp>r*   r   r   r   r   r   r   r   r+   r   N)r%   r   itemsr$   )r   r5   Zitems_listsr   r   r    
test_items   s6   zTestResultSet.test_itemsc                 C   s2   ddg}ddg}t ||}| |ddd dS )z1Test points from columns in TestResultSet object.col1col2r1   2)r7   r8   N)r   Zpoint_from_cols_valsZassertDictEqual)r   ZcolsZvalsZpointr   r   r    test_point_from_cols_vals   s   z'TestResultSet.test_point_from_cols_valsc              
   C   sx   t dg dg dg dgg ddgi}| | dg | t|d d	d
ddddd
ddddddddg dS )z7Test system query capabilities in TestResultSet object.r
   )another48h0m0s   F)default0r1   F)somename24h0m0s   T)r   durationreplicaNr>   )r   r   )r	   Nr	   r<   Fr=   r;   )rC   r>   rD   r   r?   r1   r>   rA   TrB   r@   N)r   r$   r,   r%   )r   r   r   r   r    test_system_query   s6   	
zTestResultSet.test_system_queryc                 C   s>   |  t tg dd W d   dS 1 sw   Y  dS )z-Test returning error in TestResultSet object.zBig error, many problems.)r
   errorN)ZassertRaisesr   r   r   r   r   r    test_resultset_error   s   
"z"TestResultSet.test_resultset_errorN)__name__
__module____qualname____doc__r!   r'   r(   r)   r-   r0   r6   r:   rE   rG   r   r   r   r    r      s    
r   )rK   Z
__future__r   r   r   r   ZunittestZinfluxdb.exceptionsr   Zinfluxdb.resultsetr   ZTestCaser   r   r   r   r    <module>   s   