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).
80 lines
1.9 KiB
Bash
Executable File
80 lines
1.9 KiB
Bash
Executable File
#!/bin/bash
|
|
# test_edit_history_change_message_role.sh - Change message role
|
|
|
|
# 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 role to user)
|
|
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 [user] hi there
|
|
EDITEOF
|
|
|
|
# Stage the edit prompt response (keep same content)
|
|
cat > .editor_input_1 << 'EDITEOF'
|
|
# Previous message:
|
|
|
|
[0] user: hello
|
|
|
|
Edit message 1 (role: user) below:
|
|
|
|
-- >8 --
|
|
hi there
|
|
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: change_message_role - expected exit code 0, got $EXIT_CODE"
|
|
echo "Output: $OUTPUT"
|
|
exit 1
|
|
fi
|
|
|
|
# Verify message 1 role was changed
|
|
msg1_role=$(dr_role chat.json 1)
|
|
if [ "$msg1_role" != "user" ]; then
|
|
echo "FAIL: change_message_role - message 1 role should be 'user', got '$msg1_role'"
|
|
exit 1
|
|
fi
|
|
|
|
echo "PASS: change_message_role"
|