mattzh72--articraft
wehub-resource-sync
59 行
2.0 KiB
Python
59 行
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from articraft.config import (
|
|
DEFAULT_GENERATION_MODEL,
|
|
DEFAULT_THINKING_LEVEL,
|
|
default_model_from_env,
|
|
default_thinking_level_from_env,
|
|
load_repo_env,
|
|
)
|
|
|
|
|
|
def test_load_repo_env_skips_blank_values_and_preserves_non_empty_shell_env(tmp_path) -> None:
|
|
(tmp_path / ".env").write_text(
|
|
"\n".join(
|
|
[
|
|
"ARTICRAFT_MODEL=",
|
|
"ARTICRAFT_THINKING_LEVEL=high",
|
|
"ARTICRAFT_MAX_COST_USD=",
|
|
"GOOGLE_CLOUD_PROJECT=project-from-file",
|
|
]
|
|
)
|
|
+ "\n",
|
|
encoding="utf-8",
|
|
)
|
|
environ = {
|
|
"ARTICRAFT_MODEL": "gemini-3-flash-preview",
|
|
"ARTICRAFT_THINKING_LEVEL": "xhigh",
|
|
"ARTICRAFT_MAX_COST_USD": "1.25",
|
|
}
|
|
|
|
loaded_count = load_repo_env(tmp_path, environ=environ)
|
|
|
|
assert loaded_count == 1
|
|
assert environ["ARTICRAFT_MODEL"] == "gemini-3-flash-preview"
|
|
assert environ["ARTICRAFT_THINKING_LEVEL"] == "xhigh"
|
|
assert environ["ARTICRAFT_MAX_COST_USD"] == "1.25"
|
|
assert environ["GOOGLE_CLOUD_PROJECT"] == "project-from-file"
|
|
|
|
|
|
def test_load_repo_env_fills_blank_shell_values_from_repo_env(tmp_path) -> None:
|
|
(tmp_path / ".env").write_text(
|
|
"ARTICRAFT_THINKING_LEVEL=high\nGOOGLE_CLOUD_PROJECT=project-from-file\n",
|
|
encoding="utf-8",
|
|
)
|
|
environ = {"ARTICRAFT_THINKING_LEVEL": ""}
|
|
|
|
loaded_count = load_repo_env(tmp_path, environ=environ)
|
|
|
|
assert loaded_count == 2
|
|
assert environ["ARTICRAFT_THINKING_LEVEL"] == "high"
|
|
assert environ["GOOGLE_CLOUD_PROJECT"] == "project-from-file"
|
|
|
|
|
|
def test_default_generation_values_use_env_then_built_in_defaults() -> None:
|
|
assert default_model_from_env({}) == DEFAULT_GENERATION_MODEL
|
|
assert default_thinking_level_from_env({}) == DEFAULT_THINKING_LEVEL
|
|
assert default_model_from_env({"ARTICRAFT_MODEL": " gpt-test "}) == "gpt-test"
|
|
assert default_thinking_level_from_env({"ARTICRAFT_THINKING_LEVEL": " xhigh "}) == "xhigh"
|