wehub-resource-sync
c3749daf48
Tests / test-windows (push) Waiting to run
Tests / test-macos (push) Waiting to run
Tests / test-linux (3.13) (push) Failing after 0s
Tests / test-linux (3.11) (push) Failing after 1s
Tests / lint (push) Failing after 0s
Tests / test-linux (3.9) (push) Failing after 1s
Docker / build (push) Failing after 1s
Docker / build-gpu (push) Failing after 2s
129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
import json
|
|
import subprocess
|
|
import sys
|
|
|
|
from mempalace import hook_shell
|
|
|
|
|
|
def test_normalize_transcript_path_preserves_windows_drive_and_segments():
|
|
path = r"C:\Users\me\.claude\projects\-Users-me-Proj\session.jsonl"
|
|
|
|
assert (
|
|
hook_shell.normalize_transcript_path(path)
|
|
== "C:/Users/me/.claude/projects/-Users-me-Proj/session.jsonl"
|
|
)
|
|
|
|
|
|
def test_normalize_transcript_path_preserves_spaces_and_unicode():
|
|
path = r"C:\Users\Me User\.claude\projects\emoji 🧠\session.jsonl"
|
|
|
|
assert (
|
|
hook_shell.normalize_transcript_path(path)
|
|
== "C:/Users/Me User/.claude/projects/emoji 🧠/session.jsonl"
|
|
)
|
|
|
|
|
|
def test_parse_stop_payload_keeps_session_strict_but_path_not_over_sanitized():
|
|
session_id, stop_active, transcript_path = hook_shell.parse_stop_payload(
|
|
{
|
|
"session_id": "../bad session!!",
|
|
"stop_hook_active": "yes",
|
|
"transcript_path": r"C:\Users\Me User\.claude\projects\emoji 🧠\session.jsonl",
|
|
}
|
|
)
|
|
|
|
assert session_id == "badsession"
|
|
assert stop_active == "True"
|
|
assert transcript_path == "C:/Users/Me User/.claude/projects/emoji 🧠/session.jsonl"
|
|
|
|
|
|
def test_parse_precompact_cli_outputs_sentinel_and_normalized_path():
|
|
payload = {
|
|
"session_id": "sess-1",
|
|
"transcript_path": r"D:\Claude\projects\-Users-me-App\session.jsonl",
|
|
}
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, "-m", "mempalace.hook_shell", "parse-precompact"],
|
|
input=json.dumps(payload),
|
|
text=True,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
|
|
assert result.stdout.splitlines() == [
|
|
"__MEMPAL_PARSE_OK__",
|
|
"sess-1",
|
|
"D:/Claude/projects/-Users-me-App/session.jsonl",
|
|
]
|
|
|
|
|
|
def test_count_human_messages_reads_utf8_transcripts_tolerantly(tmp_path):
|
|
transcript = tmp_path / "session.jsonl"
|
|
transcript.write_text(
|
|
json.dumps(
|
|
{"message": {"role": "user", "content": "emoji: 🧠 café Привет"}},
|
|
ensure_ascii=False,
|
|
)
|
|
+ "\n"
|
|
+ json.dumps({"message": {"role": "user", "content": "ignore <command-message> message"}})
|
|
+ "\n"
|
|
+ json.dumps({"message": {"role": "assistant", "content": "ignored"}})
|
|
+ "\n"
|
|
+ "{bad json\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
assert hook_shell.count_human_messages(str(transcript)) == 1
|
|
|
|
result = subprocess.run(
|
|
[sys.executable, "-m", "mempalace.hook_shell", "count-human-messages", str(transcript)],
|
|
text=True,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
|
|
assert result.stdout.strip() == "1"
|
|
|
|
|
|
def test_parse_stop_cli_fails_loud_on_malformed_nonempty_stdin():
|
|
result = subprocess.run(
|
|
[sys.executable, "-m", "mempalace.hook_shell", "parse-stop"],
|
|
input="not-json garbage",
|
|
text=True,
|
|
capture_output=True,
|
|
check=False,
|
|
)
|
|
|
|
assert result.returncode != 0
|
|
assert "__MEMPAL_PARSE_OK__" not in result.stdout
|
|
assert "traceback" in result.stderr.lower() or "json" in result.stderr.lower()
|
|
|
|
|
|
def test_parse_precompact_cli_fails_loud_on_malformed_nonempty_stdin():
|
|
result = subprocess.run(
|
|
[sys.executable, "-m", "mempalace.hook_shell", "parse-precompact"],
|
|
input="not-json garbage",
|
|
text=True,
|
|
capture_output=True,
|
|
check=False,
|
|
)
|
|
|
|
assert result.returncode != 0
|
|
assert "__MEMPAL_PARSE_OK__" not in result.stdout
|
|
assert "traceback" in result.stderr.lower() or "json" in result.stderr.lower()
|
|
|
|
|
|
def test_parse_stop_cli_treats_empty_stdin_as_empty_payload():
|
|
result = subprocess.run(
|
|
[sys.executable, "-m", "mempalace.hook_shell", "parse-stop"],
|
|
input="",
|
|
text=True,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
|
|
lines = result.stdout.splitlines()
|
|
assert lines[:3] == ["__MEMPAL_PARSE_OK__", "unknown", "False"]
|
|
assert result.stderr == ""
|