add unit tests for gateway_embed_key migration and fallback
schema_test: verify migrate_ai_gateway_settings() populates gateway_embed_key from OPENROUTER_API_KEY env var. service_test: verify effective_config() falls back to gateway_vision_key when gateway_embed_key is empty.
This commit is contained in:
parent
f58a4cdf7a
commit
e55b0dac0f
11
tests/unit/services/openai_gateway/schema_test.py
Normal file
11
tests/unit/services/openai_gateway/schema_test.py
Normal file
@ -0,0 +1,11 @@
|
||||
# retoor <retoor@molodetz.nl>
|
||||
|
||||
from devplacepy.database import get_setting, set_setting
|
||||
from devplacepy.database.schema import migrate_ai_gateway_settings
|
||||
|
||||
|
||||
def test_migrate_embed_key_populates_db(local_db, monkeypatch):
|
||||
monkeypatch.setenv("OPENROUTER_API_KEY", "sk-or-v1-test-embed-key")
|
||||
set_setting("gateway_embed_key", "")
|
||||
migrate_ai_gateway_settings()
|
||||
assert get_setting("gateway_embed_key", "") == "sk-or-v1-test-embed-key"
|
||||
12
tests/unit/services/openai_gateway/service_test.py
Normal file
12
tests/unit/services/openai_gateway/service_test.py
Normal file
@ -0,0 +1,12 @@
|
||||
# retoor <retoor@molodetz.nl>
|
||||
|
||||
from devplacepy.database import set_setting
|
||||
from devplacepy.services.openai_gateway import GatewayService
|
||||
|
||||
|
||||
def test_effective_config_fallback_embed_key(local_db):
|
||||
set_setting("gateway_embed_key", "")
|
||||
set_setting("gateway_vision_key", "sk-or-v1-vision-test-key")
|
||||
svc = GatewayService()
|
||||
cfg = svc.effective_config()
|
||||
assert cfg["gateway_embed_key"] == "sk-or-v1-vision-test-key"
|
||||
Loading…
Reference in New Issue
Block a user