Skip to content

Commit 9b8f4dd

Browse files
committed
test(tui): use materialized provider view in paste tests
1 parent 6373ff5 commit 9b8f4dd

1 file changed

Lines changed: 16 additions & 4 deletions

File tree

  • src/tui/app/tests/remote_startup_input_03

src/tui/app/tests/remote_startup_input_03/part_01.rs

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -210,9 +210,15 @@ fn test_paste_expansion_on_submit() {
210210
assert_eq!(app.display_messages().len(), 1);
211211
assert_eq!(app.display_messages()[0].content, "A: [pasted 5 lines] B");
212212

213-
// Model receives expanded content (actual pasted text)
214-
assert_eq!(app.messages.len(), 1);
215-
match &app.messages[0].content[0] {
213+
// Model receives expanded content (actual pasted text). Local sessions keep the
214+
// provider message cache lazy, so inspect the materialized provider view.
215+
let provider_messages = app.materialized_provider_messages();
216+
let user_message = provider_messages
217+
.iter()
218+
.rev()
219+
.find(|message| message.role == Role::User)
220+
.expect("expected submitted user message");
221+
match &user_message.content[0] {
216222
crate::message::ContentBlock::Text { text, .. } => {
217223
assert_eq!(text, "A: 1\n2\n3\n4\n5 B");
218224
}
@@ -240,7 +246,13 @@ fn test_multiple_pastes() {
240246
app.submit_input();
241247
// Display and model both get the same content (no expansion needed)
242248
assert_eq!(app.display_messages()[0].content, "first second\nline");
243-
match &app.messages[0].content[0] {
249+
let provider_messages = app.materialized_provider_messages();
250+
let user_message = provider_messages
251+
.iter()
252+
.rev()
253+
.find(|message| message.role == Role::User)
254+
.expect("expected submitted user message");
255+
match &user_message.content[0] {
244256
crate::message::ContentBlock::Text { text, .. } => {
245257
assert_eq!(text, "first second\nline");
246258
}

0 commit comments

Comments
 (0)