1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
#![warn(rust_2018_idioms)]
#![cfg(all(feature = "full", not(target_os = "wasi")))] // Wasi does not support file operations
use tempfile::NamedTempFile;
use tokio::fs::File;
use tokio::io::{AsyncBufReadExt, BufReader};
use tokio_test::assert_ok;
#[tokio::test]
async fn fill_buf_file() {
let file = NamedTempFile::new().unwrap();
assert_ok!(std::fs::write(file.path(), b"hello"));
let file = assert_ok!(File::open(file.path()).await);
let mut file = BufReader::new(file);
let mut contents = Vec::new();
loop {
let consumed = {
let buffer = assert_ok!(file.fill_buf().await);
if buffer.is_empty() {
break;
}
contents.extend_from_slice(buffer);
buffer.len()
};
file.consume(consumed);
}
assert_eq!(contents, b"hello");
}
|
