
    g
                         d dl Z d dlZd dlmZmZ d dlZd dlZd dlmZmZm	Z	 d dl
mZ  G d dej                        Zej                  j                  dg d      d	        Zy)
    N)datedatetime)DatetimeSerializer
batch_postdetermine_server_host)TEST_API_KEYc                   6    e Zd Zd Zd Zd Zd Zd Zd Zd Z	y)	TestRequestsc                 h    t        t        ddddg      }| j                  |j                  d       y )Ndistinct_idpython eventtrackr   eventtypebatch   r   r   assertEqualstatus_codeselfress     N/var/www/openai/venv/lib/python3.12/site-packages/posthog/test/test_request.pytest_valid_requestzTestRequests.test_valid_request   s.    mVdnu.v-wx#.    c                 @    | j                  t        t        dddd       y )N
testsecrethttps://t.posthog.comFz[{]assertRaises	Exceptionr   r   s    r   test_invalid_request_errorz'TestRequests.test_invalid_request_error   s    )Z?VX]_der   c                 @    | j                  t        t        ddg        y )Nr   t.posthog.com/r   r!   r$   s    r   test_invalid_hostzTestRequests.test_invalid_host   s    )Z?OWYZr   c           
          dt        ddddddd      i}t        j                  |t        	      }| j	                  |d
       y )Ncreatedi                 i clsz){"created": "2012-03-04T05:06:07.891011"})r   jsondumpsr   r   )r   dataresults      r   test_datetime_serializationz(TestRequests.test_datetime_serialization   s?    8D!Q1a@AD&89!LMr   c                     t        j                         }d|i}t        j                  |t              }d|j                         z  }| j                  ||       y )Nr*   r0   z{"created": "%s"})r   todayr2   r3   r   	isoformatr   )r   r8   r4   r5   expecteds        r   test_date_serializationz$TestRequests.test_date_serialization   sI    

5!D&89&)::*r   c                 j    t        t        ddddgd      }| j                  |j                  d       y )Nr   r   r   r      r   timeoutr   r   r   s     r   test_should_not_timeoutz$TestRequests.test_should_not_timeout#   s5    ah!i jtv
 	#.r   c                     | j                  t        j                        5  t        dddddgd       d d d        y # 1 sw Y   y xY w)Nkeyr   r   r   r   g-C6?r>   )r"   requestsReadTimeoutr   r$   s    r   test_should_timeoutz TestRequests.test_should_timeout)   s=    x334mn^efgqw 544s	   =AN)
__name__
__module____qualname__r   r%   r(   r6   r;   r@   rE    r   r   r
   r
      s(    /f[N
+/r   r
   zhost, expected))r    r    )https://t.posthog.com/rJ   )t.posthog.comrK   )r'   r'   )#https://us.posthog.com.rg.proxy.comrL   )app.posthog.comrM   )eu.posthog.comrN   )zhttps://app.posthog.comhttps://us.i.posthog.com)zhttps://eu.posthog.comhttps://eu.i.posthog.com)zhttps://us.posthog.comrO   )zhttps://app.posthog.com/rO   )zhttps://eu.posthog.com/rP   )zhttps://us.posthog.com/rO   )NrO   c                 $    t        |       |k(  sJ y )N)r   )hostr:   s     r   test_routing_to_custom_hostrS   0   s    ( !&(222r   )r2   unittestr   r   pytestrC   posthog.requestr   r   r   posthog.test.test_utilsr   TestCaser
   markparametrizerS   rI   r   r   <module>r[      s[      #   Q Q 0!8$$ !H &3'&3r   