apply new linter rules

This commit is contained in:
Parker Rule
2025-09-22 10:27:02 -04:00
parent b07c1e5da7
commit de187a989e
2 changed files with 75 additions and 41 deletions
-1
View File
@@ -373,7 +373,6 @@ class HttpConfig(TypedDict, total=False):
""" """
class Config(TypedDict, total=False): class Config(TypedDict, total=False):
"""Top-level config for langgraph-cli or similar deployment tooling.""" """Top-level config for langgraph-cli or similar deployment tooling."""
+75 -40
View File
@@ -9,17 +9,17 @@ def test_clean_empty_lines():
input_str = "line1\n\nline2\n\nline3" input_str = "line1\n\nline2\n\nline3"
result = clean_empty_lines(input_str) result = clean_empty_lines(input_str)
assert result == "line1\nline2\nline3" assert result == "line1\nline2\nline3"
# Test with no empty lines # Test with no empty lines
input_str = "line1\nline2\nline3" input_str = "line1\nline2\nline3"
result = clean_empty_lines(input_str) result = clean_empty_lines(input_str)
assert result == "line1\nline2\nline3" assert result == "line1\nline2\nline3"
# Test with only empty lines # Test with only empty lines
input_str = "\n\n\n" input_str = "\n\n\n"
result = clean_empty_lines(input_str) result = clean_empty_lines(input_str)
assert result == "" assert result == ""
# Test empty string # Test empty string
input_str = "" input_str = ""
result = clean_empty_lines(input_str) result = clean_empty_lines(input_str)
@@ -29,33 +29,51 @@ def test_clean_empty_lines():
def test_warn_non_wolfi_distro_with_debian(capsys): def test_warn_non_wolfi_distro_with_debian(capsys):
"""Test that warning is shown when image_distro is 'debian'.""" """Test that warning is shown when image_distro is 'debian'."""
config = {"image_distro": "debian"} config = {"image_distro": "debian"}
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
captured = capsys.readouterr() captured = capsys.readouterr()
assert "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security." in captured.out assert (
assert "Wolfi is a security-oriented, minimal Linux distribution designed for containers." in captured.out "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security."
assert 'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.' in captured.out in captured.out
)
assert (
"Wolfi is a security-oriented, minimal Linux distribution designed for containers."
in captured.out
)
assert (
'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.'
in captured.out
)
def test_warn_non_wolfi_distro_with_default_debian(capsys): def test_warn_non_wolfi_distro_with_default_debian(capsys):
"""Test that warning is shown when image_distro is missing (defaults to debian).""" """Test that warning is shown when image_distro is missing (defaults to debian)."""
config = {} # No image_distro key, should default to debian config = {} # No image_distro key, should default to debian
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
captured = capsys.readouterr() captured = capsys.readouterr()
assert "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security." in captured.out assert (
assert "Wolfi is a security-oriented, minimal Linux distribution designed for containers." in captured.out "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security."
assert 'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.' in captured.out in captured.out
)
assert (
"Wolfi is a security-oriented, minimal Linux distribution designed for containers."
in captured.out
)
assert (
'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.'
in captured.out
)
def test_warn_non_wolfi_distro_with_wolfi(capsys): def test_warn_non_wolfi_distro_with_wolfi(capsys):
"""Test that no warning is shown when image_distro is 'wolfi'.""" """Test that no warning is shown when image_distro is 'wolfi'."""
config = {"image_distro": "wolfi"} config = {"image_distro": "wolfi"}
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
captured = capsys.readouterr() captured = capsys.readouterr()
assert captured.out == "" # No output should be generated assert captured.out == "" # No output should be generated
@@ -63,42 +81,57 @@ def test_warn_non_wolfi_distro_with_wolfi(capsys):
def test_warn_non_wolfi_distro_with_other_distro(capsys): def test_warn_non_wolfi_distro_with_other_distro(capsys):
"""Test that warning is shown when image_distro is something other than 'wolfi'.""" """Test that warning is shown when image_distro is something other than 'wolfi'."""
config = {"image_distro": "ubuntu"} config = {"image_distro": "ubuntu"}
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
captured = capsys.readouterr() captured = capsys.readouterr()
assert "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security." in captured.out assert (
assert "Wolfi is a security-oriented, minimal Linux distribution designed for containers." in captured.out "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security."
assert 'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.' in captured.out in captured.out
)
assert (
"Wolfi is a security-oriented, minimal Linux distribution designed for containers."
in captured.out
)
assert (
'To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.'
in captured.out
)
def test_warn_non_wolfi_distro_output_formatting(): def test_warn_non_wolfi_distro_output_formatting():
"""Test that the warning output is properly formatted with colors and empty line.""" """Test that the warning output is properly formatted with colors and empty line."""
config = {"image_distro": "debian"} config = {"image_distro": "debian"}
with patch('click.secho') as mock_secho: with patch("click.secho") as mock_secho:
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
# Verify click.secho was called with the correct parameters # Verify click.secho was called with the correct parameters
expected_calls = [ expected_calls = [
( (
("⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security.",), (
{"fg": "yellow", "bold": True} "⚠️ Security Recommendation: Consider switching to Wolfi Linux for enhanced security.",
),
{"fg": "yellow", "bold": True},
), ),
( (
(" Wolfi is a security-oriented, minimal Linux distribution designed for containers.",), (
{"fg": "yellow"} " Wolfi is a security-oriented, minimal Linux distribution designed for containers.",
),
{"fg": "yellow"},
), ),
( (
(' To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.',), (
{"fg": "yellow"} ' To switch, add \'"image_distro": "wolfi"\' to your langgraph.json config file.',
),
{"fg": "yellow"},
), ),
( (
("",), # Empty line ("",), # Empty line
{} {},
) ),
] ]
assert mock_secho.call_count == 4 assert mock_secho.call_count == 4
for i, (expected_args, expected_kwargs) in enumerate(expected_calls): for i, (expected_args, expected_kwargs) in enumerate(expected_calls):
actual_call = mock_secho.call_args_list[i] actual_call = mock_secho.call_args_list[i]
@@ -117,16 +150,18 @@ def test_warn_non_wolfi_distro_various_configs(capsys):
({"image_distro": "ubuntu"}, True, "ubuntu distro"), ({"image_distro": "ubuntu"}, True, "ubuntu distro"),
({"other_config": "value"}, True, "unrelated config keys"), ({"other_config": "value"}, True, "unrelated config keys"),
] ]
for config, should_warn, description in test_cases: for config, should_warn, description in test_cases:
# Clear any previous output # Clear any previous output
capsys.readouterr() capsys.readouterr()
warn_non_wolfi_distro(config) warn_non_wolfi_distro(config)
captured = capsys.readouterr() captured = capsys.readouterr()
if should_warn: if should_warn:
assert "⚠️ Security Recommendation" in captured.out, f"Should warn for {description}" assert "⚠️ Security Recommendation" in captured.out, (
f"Should warn for {description}"
)
assert "Wolfi" in captured.out, f"Should mention Wolfi for {description}" assert "Wolfi" in captured.out, f"Should mention Wolfi for {description}"
else: else:
assert captured.out == "", f"Should not warn for {description}" assert captured.out == "", f"Should not warn for {description}"
@@ -137,7 +172,7 @@ def test_warn_non_wolfi_distro_return_value():
config = {"image_distro": "debian"} config = {"image_distro": "debian"}
result = warn_non_wolfi_distro(config) result = warn_non_wolfi_distro(config)
assert result is None assert result is None
config = {"image_distro": "wolfi"} config = {"image_distro": "wolfi"}
result = warn_non_wolfi_distro(config) result = warn_non_wolfi_distro(config)
assert result is None assert result is None
@@ -147,7 +182,7 @@ def test_warn_non_wolfi_distro_does_not_modify_config():
"""Test that warn_non_wolfi_distro does not modify the input config.""" """Test that warn_non_wolfi_distro does not modify the input config."""
original_config = {"image_distro": "debian", "other_key": "value"} original_config = {"image_distro": "debian", "other_key": "value"}
config_copy = original_config.copy() config_copy = original_config.copy()
warn_non_wolfi_distro(config_copy) warn_non_wolfi_distro(config_copy)
assert config_copy == original_config # Config should remain unchanged assert config_copy == original_config # Config should remain unchanged