
    Sj
                     (   d dl Z d dlZd dlZd dlZd dlmZ  e        dedefdZ ej                  d      d        Z	dZ
 e j                  d	      xs*  e j                  d
      xs  e j                  d      xs dZ e j                  d      ZerN e j                  d      xs dZ edei      Z
e
j                  dd      e j                   d<   ee j                   d<    ej                  d      d        Z ej                  d      d        Zy)    N)load_dotenvidentity_requestreturnc                 \   t        j                  d      }|st        d      t        t        j                  d      xs d      }t        j                  d      xs d|| j	                  dd      d	| }t        j                  |d
z   |      }|j                          |j                         S )N	IDMUX_URLz?IDMUX_URL is not set. E2E tests must use idmux for credentials.GITHUB_RUN_NUMBERr   GITHUB_REF_NAMElocalconcurrencyd   )refName	runNumberr   /)json)	osgetenvEnvironmentErrorintgetrequestspostraise_for_statusr   )r   	idmux_url
run_numberpayloadresps        V/root/.hermes/venv/lib/python3.12/site-packages/firecrawl/__tests__/e2e/v2/conftest.py_idmuxr   	   s    		+&I`aaRYY238q9J99./:7'++M3? 	G ==Sw7D99;    session)scopec                      t        j                  d      xs0 t        j                  d      xs t        j                  d      xs dS )NTEST_URLFIRECRAWL_API_URLAPI_URLhttps://api.firecrawl.dev)r   r    r   r   api_urlr(      sB     			* 	'99()	'99Y	' '	r   r#   r$   r%   r&   r   PYTEST_RUN_NAMEzpy-e2enameapiKey API_KEYc                      t         xsJ t        j                  d      xs dt        j                  d      xs t        j                  d      xs ddS )Nr-   r,   TEST_TEAM_IDTEAM_ID)r+   teamId)	_IDENTITYr   r   r'   r   r   api_identityr3   3   sW      B299Y#7#=2SaIb  JAfhfofopyfz  JA  A  B  Br   T)autousec           
      T   t        | dd       }|2t        |d      s&	 ddlm}  ||j	                  dd      |      |_        t        | dd       }|:t        |d	      s-	 ddlm} t        |d	 ||j	                  dd      |             y y y # t        $ r Y Uw xY w# t        $ r Y y w xY w)
Ninstanceclientr   )	Firecrawlr+   r,   )api_keyr(   module	firecrawl)getattrhasattrr;   r8   r   r7   	Exceptionsetattr)requestr3   r(   instr8   mods         r   _inject_clientrC   7   s     7J-Dh 7	+#L,<,<Xr,JT[\DK '8T
*C
wsK8	+Ci8H8HSU8V`g&hi  9	  		  		s#   %B +B 	BB	B'&B')r   r   pytestr   dotenvr   dictr   fixturer(   r2   r   _API_URL
_IDMUX_URLrun_namer   environr3   rC   r'   r   r   <module>rL      s5   	     T d  i  ! 	BIIj #ryy$%#ryy# #	 	 RYY{#
ryy*+7xH)*I%MM(B7BJJy$BJJyi B !B  r   