From 622a9def6f43da479fc7d30160e7d70239c5731b Mon Sep 17 00:00:00 2001 From: Vlad Lazar Date: Wed, 26 Feb 2025 12:47:13 +0000 Subject: [PATCH] tests: use generated record lsn instead of hardcoded one (#10990) ... and start the initial reader with the correct lsn Closes https://github.com/neondatabase/neon/issues/10978 --- safekeeper/src/send_interpreted_wal.rs | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/safekeeper/src/send_interpreted_wal.rs b/safekeeper/src/send_interpreted_wal.rs index be0c849a5f..e196f91d3c 100644 --- a/safekeeper/src/send_interpreted_wal.rs +++ b/safekeeper/src/send_interpreted_wal.rs @@ -1001,17 +1001,22 @@ mod tests { const WAL_READER_BATCH_SIZE: usize = 8192; let start_lsn = Lsn::from_str("0/149FD18").unwrap(); - let shard_0_start_lsn = Lsn::from_str("0/14AFE10").unwrap(); let env = Env::new(true).unwrap(); + let mut next_record_lsns = Vec::default(); let tli = env .make_timeline(NodeId(1), TenantTimelineId::generate(), start_lsn) .await .unwrap(); let resident_tli = tli.wal_residence_guard().await.unwrap(); - let end_watch = Env::write_wal(tli, start_lsn, SIZE, MSG_COUNT, None) - .await - .unwrap(); + let end_watch = + Env::write_wal(tli, start_lsn, SIZE, MSG_COUNT, Some(&mut next_record_lsns)) + .await + .unwrap(); + + assert!(next_record_lsns.len() > 3); + let shard_0_start_lsn = next_record_lsns[3]; + let end_pos = end_watch.get(); let streaming_wal_reader = StreamingWalReader::new( @@ -1064,7 +1069,7 @@ mod tests { ); let reader_state = reader.state(); - let mut reader_fut = std::pin::pin!(reader.run(start_lsn, &None)); + let mut reader_fut = std::pin::pin!(reader.run(shard_0_start_lsn, &None)); loop { let poll = futures::poll!(reader_fut.as_mut()); assert!(poll.is_pending());