mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-26 11:35:02 +02:00
open-swe[bot]
14f0b27753
When the user is prompted for a deployment name during `langgraph deploy`, the value is now persisted to the local .env file so subsequent runs skip the prompt and reuse the saved name.
207 lines
7.7 KiB
Python
207 lines
7.7 KiB
Python
import base64
|
|
import json
|
|
import os
|
|
|
|
import click
|
|
import pytest
|
|
|
|
from langgraph_cli.cli import (
|
|
_docker_config_for_token,
|
|
_normalize_image_name,
|
|
_normalize_image_tag,
|
|
_parse_env_from_config,
|
|
_resolve_env_file_path,
|
|
_save_to_dotenv,
|
|
)
|
|
|
|
|
|
class TestDockerConfigForToken:
|
|
def test_creates_config_json(self):
|
|
with _docker_config_for_token("us-docker.pkg.dev", "my-token") as cfg:
|
|
config_path = os.path.join(cfg, "config.json")
|
|
assert os.path.isfile(config_path)
|
|
with open(config_path) as f:
|
|
data = json.load(f)
|
|
expected_auth = base64.b64encode(b"oauth2accesstoken:my-token").decode()
|
|
assert data == {"auths": {"us-docker.pkg.dev": {"auth": expected_auth}}}
|
|
|
|
def test_tempdir_cleaned_up(self):
|
|
with _docker_config_for_token("registry.example.com", "tok") as cfg:
|
|
assert os.path.isdir(cfg)
|
|
assert not os.path.exists(cfg)
|
|
|
|
def test_different_registries(self):
|
|
with _docker_config_for_token("gcr.io", "token123") as cfg:
|
|
with open(os.path.join(cfg, "config.json")) as f:
|
|
data = json.load(f)
|
|
assert "gcr.io" in data["auths"]
|
|
|
|
|
|
class TestNormalizeImageName:
|
|
def test_simple_name(self):
|
|
assert _normalize_image_name("myapp") == "myapp"
|
|
|
|
def test_uppercase_lowered(self):
|
|
assert _normalize_image_name("MyApp") == "myapp"
|
|
|
|
def test_special_chars_replaced(self):
|
|
assert _normalize_image_name("my app!@#v2") == "my-app-v2"
|
|
|
|
def test_dots_and_hyphens_kept(self):
|
|
assert _normalize_image_name("my-app.v2") == "my-app.v2"
|
|
|
|
def test_leading_trailing_stripped(self):
|
|
assert _normalize_image_name("--my-app..") == "my-app"
|
|
|
|
def test_empty_string_returns_app(self):
|
|
assert _normalize_image_name("") == "app"
|
|
|
|
def test_none_returns_app(self):
|
|
assert _normalize_image_name(None) == "app"
|
|
|
|
def test_all_invalid_chars_returns_app(self):
|
|
assert _normalize_image_name("!!!") == "app"
|
|
|
|
|
|
class TestNormalizeImageTag:
|
|
def test_valid_tag(self):
|
|
assert _normalize_image_tag("v1.2.3") == "v1.2.3"
|
|
|
|
def test_empty_defaults_to_latest(self):
|
|
assert _normalize_image_tag("") == "latest"
|
|
|
|
def test_alphanumeric_and_special(self):
|
|
assert _normalize_image_tag("my_tag-1.0") == "my_tag-1.0"
|
|
|
|
def test_invalid_chars_raises(self):
|
|
with pytest.raises(click.UsageError, match="Image tag may only contain"):
|
|
_normalize_image_tag("v1.0:bad")
|
|
|
|
def test_spaces_raises(self):
|
|
with pytest.raises(click.UsageError, match="Image tag may only contain"):
|
|
_normalize_image_tag("has space")
|
|
|
|
|
|
class TestParseEnvFromConfig:
|
|
def test_env_dict(self, tmp_path):
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({"env": {"FOO": "bar", "NUM": 42}}, config_path)
|
|
assert result == {"FOO": "bar", "NUM": "42"}
|
|
|
|
def test_env_string_dotenv_file(self, tmp_path):
|
|
env_file = tmp_path / "my.env"
|
|
env_file.write_text("KEY1=val1\nKEY2=val2\n")
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({"env": "my.env"}, config_path)
|
|
assert result == {"KEY1": "val1", "KEY2": "val2"}
|
|
|
|
def test_env_missing_falls_back_to_dotenv(self, tmp_path, monkeypatch):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("DEFAULT_KEY=default_val\n")
|
|
monkeypatch.chdir(tmp_path)
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({}, config_path)
|
|
assert result == {"DEFAULT_KEY": "default_val"}
|
|
|
|
def test_env_empty_dict_falls_back_to_dotenv(self, tmp_path, monkeypatch):
|
|
"""validate_config defaults env to {}, should still fall back to .env."""
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("MY_KEY=my_val\n")
|
|
monkeypatch.chdir(tmp_path)
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({"env": {}}, config_path)
|
|
assert result == {"MY_KEY": "my_val"}
|
|
|
|
def test_env_missing_no_dotenv_returns_empty(self, tmp_path, monkeypatch):
|
|
monkeypatch.chdir(tmp_path)
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({}, config_path)
|
|
assert result == {}
|
|
|
|
def test_env_dotenv_filters_none_values(self, tmp_path):
|
|
# Lines like "KEY=" produce empty string, lines like "KEY" produce None
|
|
env_file = tmp_path / "test.env"
|
|
env_file.write_text("GOOD=value\nEMPTY=\n")
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _parse_env_from_config({"env": "test.env"}, config_path)
|
|
assert "GOOD" in result
|
|
assert result["GOOD"] == "value"
|
|
# EMPTY= gives empty string, not None, so it should be present
|
|
assert result["EMPTY"] == ""
|
|
|
|
|
|
class TestResolveEnvFilePath:
|
|
def test_inline_dict_returns_none(self, tmp_path):
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
assert _resolve_env_file_path({"env": {"FOO": "bar"}}, config_path) is None
|
|
|
|
def test_string_returns_resolved_path(self, tmp_path):
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _resolve_env_file_path({"env": "my.env"}, config_path)
|
|
assert result == (tmp_path / "my.env").resolve()
|
|
|
|
def test_no_env_field_returns_cwd_dotenv(self, tmp_path, monkeypatch):
|
|
monkeypatch.chdir(tmp_path)
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _resolve_env_file_path({}, config_path)
|
|
assert result == tmp_path / ".env"
|
|
|
|
def test_empty_dict_returns_cwd_dotenv(self, tmp_path, monkeypatch):
|
|
monkeypatch.chdir(tmp_path)
|
|
config_path = tmp_path / "langgraph.json"
|
|
config_path.touch()
|
|
result = _resolve_env_file_path({"env": {}}, config_path)
|
|
assert result == tmp_path / ".env"
|
|
|
|
|
|
class TestSaveToDotenv:
|
|
def test_creates_new_file(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
_save_to_dotenv(env_file, "MY_KEY", "my_value")
|
|
assert env_file.read_text() == "MY_KEY=my_value\n"
|
|
|
|
def test_appends_to_existing_file(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("EXISTING=val\n")
|
|
_save_to_dotenv(env_file, "NEW_KEY", "new_val")
|
|
assert env_file.read_text() == "EXISTING=val\nNEW_KEY=new_val\n"
|
|
|
|
def test_appends_newline_if_missing(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("EXISTING=val")
|
|
_save_to_dotenv(env_file, "NEW_KEY", "new_val")
|
|
assert env_file.read_text() == "EXISTING=val\nNEW_KEY=new_val\n"
|
|
|
|
def test_updates_existing_key(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("KEY1=old\nKEY2=keep\n")
|
|
_save_to_dotenv(env_file, "KEY1", "new")
|
|
content = env_file.read_text()
|
|
assert "KEY1=new\n" in content
|
|
assert "KEY2=keep\n" in content
|
|
assert "KEY1=old" not in content
|
|
|
|
def test_updates_key_with_spaces(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("KEY1 = old\nKEY2=keep\n")
|
|
_save_to_dotenv(env_file, "KEY1", "new")
|
|
content = env_file.read_text()
|
|
assert "KEY1=new\n" in content
|
|
assert "KEY1 = old" not in content
|
|
|
|
def test_preserves_other_lines(self, tmp_path):
|
|
env_file = tmp_path / ".env"
|
|
env_file.write_text("# comment\nA=1\nB=2\n")
|
|
_save_to_dotenv(env_file, "C", "3")
|
|
content = env_file.read_text()
|
|
assert content == "# comment\nA=1\nB=2\nC=3\n"
|