use super::super::*;
use super::{ERROR_END, FINISH_END};
#[test]
fn the_ending_reads_the_canonical_finish_reason_and_nothing_else() {
let of = |reason: &str| {
let jsonl =
format!("{{\"type\":\"finish\",\"reason\":\"{reason}\"}}\n{{\"type\":\"end\"}}\n");
settled(jsonl.as_bytes())
};
assert_eq!(of("length").ending, Ending::OutputLimit);
for reason in [
"stop",
"tool_use",
"stop_sequence",
"pause",
"refusal",
"eos",
] {
assert_eq!(of(reason).ending, Ending::OwnTerms, "{reason}");
}
assert_eq!(
settled(b"{\"type\":\"finish\"}\n{\"type\":\"end\"}\n").ending,
Ending::OwnTerms
);
assert_eq!(
settled(b"{\"type\":\"finish\",\"reason\":7}\n{\"type\":\"end\"}\n").ending,
Ending::OwnTerms
);
}
#[test]
fn an_output_limited_tail_is_complete_transport_and_an_unfinished_turn() {
let jsonl = br#"{"type":"message_start","v":1,"role":"assistant"}
{"type":"content_delta","index":0,"delta":{"thinking_delta":"hmm"}}
{"type":"finish","reason":"length"}
{"type":"end"}
"#;
let read = settled(jsonl);
assert_eq!(read.framing, Framing::Complete);
assert_eq!(read.ending, Ending::OutputLimit);
assert!(!read.whole(), "complete on the wire, unfinished as a turn");
assert!(settled(FINISH_END).whole());
}
#[test]
fn a_partial_text_length_turn_is_truncated_without_losing_what_it_said() {
let jsonl = br#"{"type":"content_delta","index":0,"delta":{"text_delta":"as I was say"}}
{"type":"finish","reason":"length"}
{"type":"end"}
"#;
assert_eq!(settled(jsonl).ending, Ending::OutputLimit);
assert_eq!(
crate::git_tree::fold_stream(jsonl).text.as_deref(),
Some("as I was say"),
"the §5.1 #10 fold still hands the fragment to the transcript"
);
}
#[test]
fn a_tail_with_no_finish_reads_no_ending_at_all() {
for bytes in [
ERROR_END, b"{\"type\":\"content_delta\",\"index\":0}\n", b"{\"type\":\"message_start\"}\n{\"type\":\"end\"}\n", b"",
] {
assert_eq!(settled(bytes).ending, Ending::Unread);
assert!(!settled(bytes).whole());
}
}
#[test]
fn only_the_latest_segment_decides_the_ending() {
let truncated_retry = br#"{"type":"finish","reason":"stop"}
{"type":"end"}
{"type":"finish","reason":"length"}
{"type":"end"}
"#;
assert_eq!(settled(truncated_retry).ending, Ending::OutputLimit);
let clean_retry = br#"{"type":"finish","reason":"length"}
{"type":"end"}
{"type":"finish","reason":"stop"}
{"type":"end"}
"#;
assert_eq!(settled(clean_retry).ending, Ending::OwnTerms);
let doubled = br#"{"type":"finish","reason":"length"}
{"type":"finish","reason":"stop"}
{"type":"end"}
"#;
assert_eq!(settled(doubled).ending, Ending::OwnTerms);
}
#[test]
fn an_error_in_the_settled_segment_outranks_its_finish() {
let jsonl = br#"{"type":"error","kind":"x"}
{"type":"finish","reason":"length"}
{"type":"end"}
"#;
let read = settled(jsonl);
assert_eq!(read.framing, Framing::Failed);
assert_eq!(read.ending, Ending::Unread);
assert_eq!(Ending::default(), Ending::Unread);
}