U
    >Afg                     @   s*   d dl Z d dlmZ G dd de jZdS )    N)Posthogc                   @   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 )
TestModuleNc                 C   s,   |  t|d t |  t|d t d S )Nr      )assertEqualtypebooldict)selfresult r   </tmp/pip-unpacked-wheel-39skdq2h/posthog/test/test_module.py_assert_enqueue_result	   s    z!TestModule._assert_enqueue_resultc                 C   s
   d| _ d S )NT)failedr	   r   r   r   r      s    zTestModule.failedc                 C   s   d| _ tdd| j d| _d S )NFZ
testsecretzhttp://localhost:8000)hostZon_error)r   r   posthogr   r   r   r   setUp   s    zTestModule.setUpc                 C   s   d | j _| t| j j d S N)r   Zapi_keyassertRaises	Exceptioncapturer   r   r   r   test_no_api_key   s    zTestModule.test_no_api_keyc                 C   s   d | j _| t| j j d S r   )r   r   r   r   r   r   r   r   r   test_no_host   s    zTestModule.test_no_hostc                 C   s&   | j dd}| | | j   d S )Ndistinct_idzpython module event)r   r   r   flushr	   resr   r   r   
test_track   s    
zTestModule.test_trackc                 C   s*   | j dddi}| | | j   d S )Nr   emailzuser@email.com)r   identifyr   r   r   r   r   r   test_identify!   s    
zTestModule.test_identifyc                 C   s&   | j dd}| | | j   d S )NZ
previousIdr   )r   aliasr   r   r   r   r   r   
test_alias&   s    
zTestModule.test_aliasc                 C   s   | j dd | j   d S )Nr   zhttps://posthog.com/contact)r   pager   r   r   r   r   	test_page+   s    zTestModule.test_pagec                 C   s   | j   d S r   )r   r   r   r   r   r   
test_flush/   s    zTestModule.test_flush)__name__
__module____qualname__r   r   r   r   r   r   r   r    r"   r$   r%   r   r   r   r   r      s   r   )Zunittestr   r   ZTestCaser   r   r   r   r   <module>   s   