
    i_                         d dl Zd dlmc mZ d dlZd dlmZ d dl	Z	e	j                  j                  d d       d dlmZ  ee      Zd Zd Zy)    N)
TestClientz!/home/jang/Projects/ai-researcher)appc                     t         j                  d      } | j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}| j                  } |       }d
di}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}x}}y	)u   测试健康检查端点z/health   ==z3%(py2)s
{%(py2)s = %(py0)s.status_code
} == %(py5)sresponsepy0py2py5assert %(py7)spy7Nstatusok)zC%(py4)s
{%(py4)s = %(py2)s
{%(py2)s = %(py0)s.json
}()
} == %(py7)s)r   r   py4r   zassert %(py9)spy9clientgetstatus_code
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationjson)	r
   @py_assert1@py_assert4@py_assert3@py_format6@py_format8@py_assert6@py_assert5@py_format10s	            9/home/jang/Projects/ai-researcher/tests/unit/test_main.pytest_health_endpointr+      s	   zz)$H&3&3&&&&3&&&&&&8&&&8&&&&&&3&&&&&&&==.=?.x..?.....?.......8...8...=...?...........    c                  r   t         j                  d      } | j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}| j                         }d
}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }	dd|	iz  }t        t        j                  |            d	x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }	dd|	iz  }t        t        j                  |            d	x}}y	)u   测试API版本信息z/api/v1/r   r   r	   r
   r   r   r   Nversion)in)z%(py1)s in %(py3)sdata)py1py3zassert %(py5)sr   namer   )
r
   r"   r#   r$   r%   r&   r0   @py_assert0@py_assert2@py_format4s
             r*   test_api_versionr7      s>   zz*%H&3&3&&&&3&&&&&&8&&&8&&&&&&3&&&&&&&==?D9996T>6T6TTr,   )builtinsr   _pytest.assertion.rewrite	assertionrewriter   pytestfastapi.testclientr   syspathinsertsrc.mainr   r   r+   r7    r,   r*   <module>rC      s;       ) 
 6 7 	C/r,   