64 lines
1.6 KiB
Rust
64 lines
1.6 KiB
Rust
#![warn(rust_2018_idioms)]
|
|
#![cfg(feature = "full")]
|
|
|
|
use std::io;
|
|
use tokio::io::AsyncReadExt;
|
|
use tokio_test::assert_ok;
|
|
use tokio_test::io::Builder;
|
|
|
|
#[tokio::test]
|
|
async fn read_to_string() {
|
|
let mut buf = String::new();
|
|
let mut rd: &[u8] = b"hello world";
|
|
|
|
let n = assert_ok!(rd.read_to_string(&mut buf).await);
|
|
assert_eq!(n, 11);
|
|
assert_eq!(buf[..], "hello world"[..]);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn to_string_does_not_truncate_on_utf8_error() {
|
|
let data = vec![0xff, 0xff, 0xff];
|
|
|
|
let mut s = "abc".to_string();
|
|
|
|
match AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s).await {
|
|
Ok(len) => panic!("Should fail: {} bytes.", len),
|
|
Err(err) if err.to_string() == "stream did not contain valid UTF-8" => {}
|
|
Err(err) => panic!("Fail: {}.", err),
|
|
}
|
|
|
|
assert_eq!(s, "abc");
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn to_string_does_not_truncate_on_io_error() {
|
|
let mut mock = Builder::new()
|
|
.read(b"def")
|
|
.read_error(io::Error::new(io::ErrorKind::Other, "whoops"))
|
|
.build();
|
|
let mut s = "abc".to_string();
|
|
|
|
match AsyncReadExt::read_to_string(&mut mock, &mut s).await {
|
|
Ok(len) => panic!("Should fail: {} bytes.", len),
|
|
Err(err) if err.to_string() == "whoops" => {}
|
|
Err(err) => panic!("Fail: {}.", err),
|
|
}
|
|
|
|
assert_eq!(s, "abc");
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn to_string_appends() {
|
|
let data = b"def".to_vec();
|
|
|
|
let mut s = "abc".to_string();
|
|
|
|
let len = AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s)
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(len, 3);
|
|
assert_eq!(s, "abcdef");
|
|
}
|