darthrubik
41b49aeeae
Replace hardcoded /tmp/drharness_output.txt with $TEST_TEMP_DIR/drharness_output.txt in all test files. Tests should only write to subfiles/directories of the working directory (which is already a temp directory created by run_test.sh).
87 lines
2.2 KiB
Bash
Executable File
87 lines
2.2 KiB
Bash
Executable File
#!/bin/bash
|
|
# test_edit_history_edit_message_content.sh - Edit message content
|
|
|
|
# Helper function to run drharness and capture output to a temp file
|
|
run_drharness() {
|
|
"$DRHARNESS" "$@" > "$TEST_TEMP_DIR/drharness_output.txt" 2>&1
|
|
return $?
|
|
}
|
|
|
|
# Helper to get the output
|
|
get_output() {
|
|
cat "$TEST_TEMP_DIR/drharness_output.txt"
|
|
}
|
|
|
|
# Helper to reset editor state for each test
|
|
reset_editor_state() {
|
|
rm -f .editor_input_* .editor_invocation_count hist.md
|
|
}
|
|
|
|
reset_editor_state
|
|
|
|
# Create a chat file with two messages
|
|
create_chat_file chat.json user "hello" assistant "hi there"
|
|
|
|
# Stage the rebase editor output (edit message 1)
|
|
cat > .editor_input_0 << 'EDITEOF'
|
|
# History Rebase Editor
|
|
#
|
|
# Commands:
|
|
# pick - keep the message unchanged
|
|
# edit - modify the message content and/or role
|
|
# drop - remove the message from history
|
|
#
|
|
# Each line format: <command> <index> [<role>] <display_content>
|
|
# - index: 0-based message index
|
|
# - role: [user], [assistant], or [tool] (editable)
|
|
# - display_content: first line of message (display only, not editable here)
|
|
#
|
|
# Lines starting with # are comments and will be ignored.
|
|
# Empty lines are treated as drops.
|
|
#
|
|
-- >8 --
|
|
pick 0 [user] hello
|
|
edit 1 [assistant] hi there
|
|
EDITEOF
|
|
|
|
# Stage the edit prompt response (new content for message 1)
|
|
cat > .editor_input_1 << 'EDITEOF'
|
|
# Previous message:
|
|
|
|
[0] user: hello
|
|
|
|
Edit message 1 (role: assistant) below:
|
|
|
|
-- >8 --
|
|
edited content here
|
|
EDITEOF
|
|
|
|
# Run drharness with -b flag
|
|
set +e
|
|
VISUAL="$MOCK_EDITOR" run_drharness -b -F chat.json
|
|
EXIT_CODE=$?
|
|
OUTPUT=$(get_output)
|
|
set -e
|
|
|
|
if [ ${EXIT_CODE} -ne 0 ]; then
|
|
echo "FAIL: edit_message_content - expected exit code 0, got $EXIT_CODE"
|
|
echo "Output: $OUTPUT"
|
|
exit 1
|
|
fi
|
|
|
|
# Verify message 1 was edited
|
|
len=$(history_length chat.json)
|
|
if [ "$len" != "2" ]; then
|
|
echo "FAIL: edit_message_content - expected 2 messages, got $len"
|
|
exit 1
|
|
fi
|
|
|
|
msg1_content=$(dr_result chat.json 1)
|
|
if [ "$msg1_content" != "edited content here" ]; then
|
|
echo "FAIL: edit_message_content - message 1 content not updated correctly"
|
|
echo "Expected: 'edited content here', got: '$msg1_content'"
|
|
exit 1
|
|
fi
|
|
|
|
echo "PASS: edit_message_content"
|