mirror of
https://github.com/tokio-rs/bytes.git
synced 2026-08-15 00:00:18 +02:00
Add conversion from Bytes to Vec<u8> (#547)
Signed-off-by: Jiahao XU <[email protected]> Co-authored-by: Alice Ryhl <[email protected]>
This commit is contained in:
@@ -1065,3 +1065,73 @@ fn bytes_into_vec() {
|
||||
let vec: Vec<u8> = bytes.into();
|
||||
assert_eq!(&vec, prefix);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_bytes_into_vec() {
|
||||
// Test STATIC_VTABLE.to_vec
|
||||
let bs = b"1b23exfcz3r";
|
||||
let vec: Vec<u8> = Bytes::from_static(bs).into();
|
||||
assert_eq!(&*vec, bs);
|
||||
|
||||
// Test bytes_mut.SHARED_VTABLE.to_vec impl
|
||||
eprintln!("1");
|
||||
let mut bytes_mut: BytesMut = bs[..].into();
|
||||
|
||||
// Set kind to KIND_ARC so that after freeze, Bytes will use bytes_mut.SHARED_VTABLE
|
||||
eprintln!("2");
|
||||
drop(bytes_mut.split_off(bs.len()));
|
||||
|
||||
eprintln!("3");
|
||||
let b1 = bytes_mut.freeze();
|
||||
eprintln!("4");
|
||||
let b2 = b1.clone();
|
||||
|
||||
eprintln!("{:#?}", (&*b1).as_ptr());
|
||||
|
||||
// shared.is_unique() = False
|
||||
eprintln!("5");
|
||||
assert_eq!(&*Vec::from(b2), bs);
|
||||
|
||||
// shared.is_unique() = True
|
||||
eprintln!("6");
|
||||
assert_eq!(&*Vec::from(b1), bs);
|
||||
|
||||
// Test bytes_mut.SHARED_VTABLE.to_vec impl where offset != 0
|
||||
let mut bytes_mut1: BytesMut = bs[..].into();
|
||||
let bytes_mut2 = bytes_mut1.split_off(9);
|
||||
|
||||
let b1 = bytes_mut1.freeze();
|
||||
let b2 = bytes_mut2.freeze();
|
||||
|
||||
assert_eq!(Vec::from(b2), bs[9..]);
|
||||
assert_eq!(Vec::from(b1), bs[..9]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_bytes_into_vec_promotable_even() {
|
||||
let vec = vec![33u8; 1024];
|
||||
|
||||
// Test cases where kind == KIND_VEC
|
||||
let b1 = Bytes::from(vec.clone());
|
||||
assert_eq!(Vec::from(b1), vec);
|
||||
|
||||
// Test cases where kind == KIND_ARC, ref_cnt == 1
|
||||
let b1 = Bytes::from(vec.clone());
|
||||
drop(b1.clone());
|
||||
assert_eq!(Vec::from(b1), vec);
|
||||
|
||||
// Test cases where kind == KIND_ARC, ref_cnt == 2
|
||||
let b1 = Bytes::from(vec.clone());
|
||||
let b2 = b1.clone();
|
||||
assert_eq!(Vec::from(b1), vec);
|
||||
|
||||
// Test cases where vtable = SHARED_VTABLE, kind == KIND_ARC, ref_cnt == 1
|
||||
assert_eq!(Vec::from(b2), vec);
|
||||
|
||||
// Test cases where offset != 0
|
||||
let mut b1 = Bytes::from(vec.clone());
|
||||
let b2 = b1.split_off(20);
|
||||
|
||||
assert_eq!(Vec::from(b2), vec[20..]);
|
||||
assert_eq!(Vec::from(b1), vec[..20]);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user