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).
68 lines
1.7 KiB
Bash
Executable File
68 lines
1.7 KiB
Bash
Executable File
#!/bin/bash
|
|
# test_edit_history_drop_all_messages.sh - Dropping all messages results in empty history
|
|
|
|
# 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 (drop all)
|
|
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 --
|
|
drop 0 [user] hello
|
|
drop 1 [assistant] 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: drop_all_messages - expected exit code 0, got $EXIT_CODE"
|
|
echo "Output: $OUTPUT"
|
|
exit 1
|
|
fi
|
|
|
|
# Verify history is empty
|
|
len=$(history_length chat.json)
|
|
if [ "$len" != "0" ]; then
|
|
echo "FAIL: drop_all_messages - expected 0 messages, got $len"
|
|
exit 1
|
|
fi
|
|
|
|
echo "PASS: drop_all_messages"
|