o
    3i-                     @   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S )
TestModuleNc                 C   s   |  t|d t d S )Nr   )assertEqualtypestr)selfresult r	   f/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/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 )NF
testsecretzhttp://localhost:8000)hoston_error)r   r   posthogr   r	   r	   r
   setUp   s   zTestModule.setUpc                 C   s(   | j jddd}| | | j   d S )Nzpython module eventdistinct_id)r   )r   capturer   flushr   resr	   r	   r
   
test_track   s   
zTestModule.test_trackc                 C   s&   | j dd}| | | j   d S )N
previousIdr   )r   aliasr   r   r   r	   r	   r
   
test_alias   s   
zTestModule.test_aliasc                 C   s   | j   d S )N)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      s    r   )unittestr   r   TestCaser   r	   r	   r	   r
   <module>   s    