Skip to content

Commit f27ec0f

Browse files
committed
Add test that impl Seek for BufReader correctly invalidates buffer between seeks
1 parent fc45382 commit f27ec0f

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

src/libstd/io/buffered.rs

+34
Original file line numberDiff line numberDiff line change
@@ -1162,6 +1162,40 @@ mod tests {
11621162
assert_eq!(reader.get_ref().pos, expected);
11631163
}
11641164

1165+
fn test_buffered_reader_seek_underflow_discard_buffer_between_seeks() {
1166+
// gimmick reader that returns Err after first seek
1167+
struct ErrAfterFirstSeekReader {
1168+
first_seek: bool,
1169+
}
1170+
impl Read for ErrAfterFirstSeekReader {
1171+
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
1172+
for x in &mut *buf {
1173+
*x = 0;
1174+
}
1175+
Ok(buf.len())
1176+
}
1177+
}
1178+
impl Seek for ErrAfterFirstSeekReader {
1179+
fn seek(&mut self, _: SeekFrom) -> io::Result<u64> {
1180+
if self.first_seek {
1181+
self.first_seek = false;
1182+
Ok(0)
1183+
} else {
1184+
Err(io::Error::new(io::ErrorKind::Other, "oh no!"))
1185+
}
1186+
}
1187+
}
1188+
1189+
let mut reader = BufReader::with_capacity(5, ErrAfterFirstSeekReader { first_seek: true });
1190+
assert_eq!(reader.fill_buf().ok(), Some(&[0, 0, 0, 0, 0][..]));
1191+
1192+
// The following seek will require two underlying seeks. The first will
1193+
// succeed but the second will fail. This should still invalidate the
1194+
// buffer.
1195+
assert!(reader.seek(SeekFrom::Current(i64::min_value())).is_err());
1196+
assert_eq!(reader.buffer().len(), 0);
1197+
}
1198+
11651199
#[test]
11661200
fn test_buffered_writer() {
11671201
let inner = Vec::new();

0 commit comments

Comments
 (0)