o
    3Ih}                     @   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   X/home/air/sanwanet/gpt-api/venv/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      d | j _| t| j j d S N)r   api_keyassertRaises	Exceptioncapturer   r   r   r   test_no_api_key      zTestModule.test_no_api_keyc                 C   r   r   )r   r   r   r   r   r   r   r   r   test_no_host   r   zTestModule.test_no_hostc                 C   &   | j dd}| | | j   d S )Ndistinct_idzpython module event)r   r   r   flushr	   resr   r   r   
test_track      
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   r   )N
previousIdr   )r   aliasr   r    r!   r   r   r   
test_alias(   r$   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_flush1   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   )unittestr   r   TestCaser   r   r   r   r   <module>   s    