Function test_prune_preserves_recent_outputs

core/src/compaction.rs:365–394

()

    #[test]
    fn test_prune_preserves_recent_outputs() {
        // Recent output alone fills the protection budget (~50k tokens)
        let large_old = "a".repeat(400_000); // ~100k tokens
        let recent = "b".repeat(200_000); // ~50k tokens (fills protection budget)

        let messages = vec![
            make_tool_result_msg("old", &large_old),
            make_text_msg("assistant", "ok"),
            make_tool_result_msg("recent", &recent),
            make_text_msg("assistant", "done"),
        ];

        let result = prune_tool_outputs(&messages);
        assert!(result.is_some());

        let pruned = result.unwrap();
        // Old should be pruned
        let old_content = match &pruned[0].content[0] {
            ContentBlock::ToolResult { content, .. } => content.as_text(),
            _ => panic!("Expected ToolResult"),
        };
        assert_eq!(old_content, PRUNED_MARKER);

        // Recent should be preserved
        let recent_content = match &pruned[2].content[0] {
            ContentBlock::ToolResult { content, .. } => content.as_text(),
            _ => panic!("Expected ToolResult"),
        };
        assert_ne!(recent_content, PRUNED_MARKER);
    }

    #[test]
    fn test_prune_marker_text() {