Melika Kheirieh commited on
Commit
d224abb
·
1 Parent(s): c4b6bf9

fix(tests): disable API key auth in conftest to fix 401 errors

Browse files
Files changed (1) hide show
  1. tests/conftest.py +18 -12
tests/conftest.py CHANGED
@@ -1,18 +1,24 @@
1
  import os
 
2
  from dotenv import load_dotenv
 
 
3
 
4
- ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
5
- ENV_PATH = os.path.join(ROOT_DIR, ".env")
6
- load_dotenv(dotenv_path=ENV_PATH)
 
7
 
 
 
 
 
 
8
 
9
- # --- add: make tests independent of real OPENAI_* in CI ---
10
- def _ensure_openai_env_for_tests():
11
- # map PROXY_* -> OPENAI_* or set a harmless dummy
12
- if not os.getenv("OPENAI_API_KEY"):
13
- os.environ["OPENAI_API_KEY"] = os.getenv("PROXY_API_KEY", "DUMMY_TEST_KEY")
14
- if not os.getenv("OPENAI_BASE_URL") and os.getenv("PROXY_BASE_URL"):
15
- os.environ["OPENAI_BASE_URL"] = os.environ["PROXY_BASE_URL"]
16
 
17
-
18
- _ensure_openai_env_for_tests()
 
 
 
 
 
1
  import os
2
+ import pytest
3
  from dotenv import load_dotenv
4
+ from app.main import app
5
+ from app.routers import nl2sql
6
 
7
+ # Load .env once for tests
8
+ ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
9
+ ENV_PATH = os.path.join(ROOT, ".env")
10
+ load_dotenv(ENV_PATH)
11
 
12
+ # --- Ensure fake OpenAI creds for CI/tests ---
13
+ if not os.getenv("OPENAI_API_KEY"):
14
+ os.environ["OPENAI_API_KEY"] = os.getenv("PROXY_API_KEY", "DUMMY_TEST_KEY")
15
+ if not os.getenv("OPENAI_BASE_URL") and os.getenv("PROXY_BASE_URL"):
16
+ os.environ["OPENAI_BASE_URL"] = os.getenv("PROXY_BASE_URL")
17
 
 
 
 
 
 
 
 
18
 
19
+ @pytest.fixture(autouse=True)
20
+ def disable_api_key_auth():
21
+ """Disable X-API-Key auth for tests."""
22
+ app.dependency_overrides[nl2sql.require_api_key] = lambda: None
23
+ yield
24
+ app.dependency_overrides.clear()