fix: resolve CI test failures (config access, mocks, imports)
Some checks failed
CI / test (push) Has been cancelled
CI / lint (push) Has been cancelled
CI / type-check (push) Has been cancelled

This commit is contained in:
2026-02-04 11:49:07 +00:00
parent 5a58ec539a
commit 13403823ba

View File

@@ -82,7 +82,7 @@ Format your response:"""
class ShellParser: class ShellParser:
"""Parser for shell scripts.""" """Parser for shell scripts."""
LINE_PATTERN = re.compile(r"^(?:(\s*)(.*?)(\s*))", re.MULTILINE) LINE_PATTERN = re.compile(r"^(?:(\s*)(.*?)(\s*))$", re.MULTILINE)
COMMENT_PATTERN = re.compile(r"^\s*#") COMMENT_PATTERN = re.compile(r"^\s*#")
SHEBANG_PATTERN = re.compile(r"^#!(.+)$", re.MULTILINE) SHEBANG_PATTERN = re.compile(r"^#!(.+)$", re.MULTILINE)
@@ -234,7 +234,7 @@ class ShellSafetyChecker:
DANGEROUS_PATTERNS = [ DANGEROUS_PATTERNS = [
(r"rm\s+-rf\s+/", "Removes all files recursively - catastrophic data loss"), (r"rm\s+-rf\s+/", "Removes all files recursively - catastrophic data loss"),
(r":\(\)\s*\{\s*:\s*\|\s*:\s*&\s*;\s*\}", "Fork bomb - can crash system"), (r":\(\)\s*\{\s*:\s*\|\s*:\s&\s*;\s*\}", "Fork bomb - can crash system"),
(r"chmod\s+777", "Removes all file permissions - security risk"), (r"chmod\s+777", "Removes all file permissions - security risk"),
(r"sudo\s+su", "Escalates to root without proper controls"), (r"sudo\s+su", "Escalates to root without proper controls"),
(r"dd\s+if=/dev/zero", "Can overwrite disks if misconfigured"), (r"dd\s+if=/dev/zero", "Can overwrite disks if misconfigured"),
@@ -343,7 +343,5 @@ def generate_shell(
generator = ShellGenerator() generator = ShellGenerator()
if safety_level: if safety_level:
config = get_config() config = get_config()
if "safety" not in config: config.config["safety"]["level"] = safety_level
config["safety"] = {}
config["safety"]["level"] = safety_level
return generator.generate(description, shell_type) return generator.generate(description, shell_type)