
    _i                         d dl Zd dlmc mZ d dlZd dlZd dlZej                  j                  d d       d dlmZ d Zd Zy)    Nz!/home/jang/Projects/ai-researcher)Settingsc                  l   t               } | 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	}||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}||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}||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}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }	t        t        j                  |	            dx}x}}y)u   测试配置默认值u   AI研究员==)z0%(py2)s
{%(py2)s = %(py0)s.APP_NAME
} == %(py5)ssettingspy0py2py5assert %(py7)spy7Nz0.1.0)z3%(py2)s
{%(py2)s = %(py0)s.APP_VERSION
} == %(py5)sg333333?)zI%(py2)s
{%(py2)s = %(py0)s.QUERY_ROUTER_COMPLEXITY_THRESHOLD
} == %(py5)si  )zI%(py2)s
{%(py2)s = %(py0)s.QUERY_ROUTER_MAX_SYNC_DURATION_MS
} == %(py5)sdata_sourcesg      ?)z%(py1)s == %(py4)s)py1py4zassert %(py6)spy6)r   APP_NAME
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationAPP_VERSION!QUERY_ROUTER_COMPLEXITY_THRESHOLD!QUERY_ROUTER_MAX_SYNC_DURATION_MSQUERY_ROUTER_WEIGHTS)
r   @py_assert1@py_assert4@py_assert3@py_format6@py_format8@py_assert0@py_assert2@py_format5@py_format7s
             ;/home/jang/Projects/ai-researcher/tests/unit/test_config.pytest_settings_defaultsr)      s)   zH------------8---8-------------*7*7****7******8***8******7*******55<<5<<<<5<<<<<<8<<<8<<<5<<<<<<<<<<55==5====5======8===8===5==========((8@D@8D@@@@8D@@@8@@@D@@@@@@@    c                  L   t         j                  j                  d      } t         j                  j                  d      }t         j                  j                  d      }	 dt         j                  d<   dt         j                  d<   dt         j                  d<   ddlm} |j                          t               }|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}||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}||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}}| r| t         j                  d<   n#dt         j                  v rt         j                  d= |r|t         j                  d<   |r|t         j                  d<   ydt         j                  v rt         j                  d= yy# | r| t         j                  d<   n#dt         j                  v rt         j                  d= |r|t         j                  d<   |r|t         j                  d<   w dt         j                  v rt         j                  d= w w xY w)u!   测试配置从环境变量加载DATABASE_URL	REDIS_URLOPENAI_API_KEYz%postgresql://test:test@localhost/testzredis://localhost:6379z
sk-test123r   )get_settingsr   )z4%(py2)s
{%(py2)s = %(py0)s.DATABASE_URL
} == %(py5)sr   r   r   r   N)z1%(py2)s
{%(py2)s = %(py0)s.REDIS_URL
} == %(py5)s)z6%(py2)s
{%(py2)s = %(py0)s.OPENAI_API_KEY
} == %(py5)s)osenvironget
src.configr/   cache_clearr   r,   r   r   r   r   r   r   r   r   r-   r.   )
original_dboriginal_redisoriginal_keyr/   r   r   r    r!   r"   r#   s
             r(   test_settings_from_envr8      s    **..0KZZ^^K0N::>>"23L-%L

>"":

;'3

#$ 	,  ":$$O(OO$(OOOOO$(OOOOOOOxOOOxOOO$OOO(OOOOOOOO!!=%==!%=====!%=======x===x===!===%========&&6,6&,6666&,666666x666x666&666,6666666 )4BJJ~&rzz)

>*&4BJJ{#+7BJJ'(+

+, , )4BJJ~&rzz)

>*&4BJJ{#+7BJJ'(+

+, ,s   J1N B
P#)builtinsr   _pytest.assertion.rewrite	assertionrewriter   r0   pytestsyspathinsertr3   r   r)   r8    r*   r(   <module>rB      s5      	   6 7 A"-r*   