1 | #![warn (rust_2018_idioms)] |
2 | #![cfg (feature = "full" )] |
3 | |
4 | use std::io; |
5 | use tokio::io::AsyncReadExt; |
6 | use tokio_test::assert_ok; |
7 | use tokio_test::io::Builder; |
8 | |
9 | #[tokio::test ] |
10 | async fn read_to_string() { |
11 | let mut buf = String::new(); |
12 | let mut rd: &[u8] = b"hello world" ; |
13 | |
14 | let n = assert_ok!(rd.read_to_string(&mut buf).await); |
15 | assert_eq!(n, 11); |
16 | assert_eq!(buf[..], "hello world" [..]); |
17 | } |
18 | |
19 | #[tokio::test ] |
20 | async fn to_string_does_not_truncate_on_utf8_error() { |
21 | let data = vec![0xff, 0xff, 0xff]; |
22 | |
23 | let mut s = "abc" .to_string(); |
24 | |
25 | match AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s).await { |
26 | Ok(len) => panic!("Should fail: {} bytes." , len), |
27 | Err(err) if err.to_string() == "stream did not contain valid UTF-8" => {} |
28 | Err(err) => panic!("Fail: {}." , err), |
29 | } |
30 | |
31 | assert_eq!(s, "abc" ); |
32 | } |
33 | |
34 | #[tokio::test ] |
35 | async fn to_string_does_not_truncate_on_io_error() { |
36 | let mut mock = Builder::new() |
37 | .read(b"def" ) |
38 | .read_error(io::Error::new(io::ErrorKind::Other, "whoops" )) |
39 | .build(); |
40 | let mut s = "abc" .to_string(); |
41 | |
42 | match AsyncReadExt::read_to_string(&mut mock, &mut s).await { |
43 | Ok(len) => panic!("Should fail: {} bytes." , len), |
44 | Err(err) if err.to_string() == "whoops" => {} |
45 | Err(err) => panic!("Fail: {}." , err), |
46 | } |
47 | |
48 | assert_eq!(s, "abc" ); |
49 | } |
50 | |
51 | #[tokio::test ] |
52 | async fn to_string_appends() { |
53 | let data = b"def" .to_vec(); |
54 | |
55 | let mut s = "abc" .to_string(); |
56 | |
57 | let len = AsyncReadExt::read_to_string(&mut data.as_slice(), &mut s) |
58 | .await |
59 | .unwrap(); |
60 | |
61 | assert_eq!(len, 3); |
62 | assert_eq!(s, "abcdef" ); |
63 | } |
64 | |