mirror of
https://github.com/tokio-rs/bytes.git
synced 2026-08-07 00:00:13 +02:00
Carl Lerche
04e0ac75e2
* Get rid of `ByteStr` trait * `Bytes` is not a concrete type * Add `BlockBuf` * Delete lots of cruft * Performance work
61 lines
1.6 KiB
Rust
61 lines
1.6 KiB
Rust
use bytes::{Buf, MutBuf, AppendBuf};
|
|
use bytes::alloc::Pool;
|
|
|
|
#[test]
|
|
pub fn test_initial_buf_empty() {
|
|
// Run in a loop a bunch in hope that if there is a memory issue, it will
|
|
// be exposed
|
|
for _ in 0..1000 {
|
|
let mut buf = AppendBuf::with_capacity(100);
|
|
let mut dst: Vec<u8> = vec![];
|
|
|
|
assert_eq!(buf.remaining(), 128);
|
|
|
|
buf.write_slice(b"hello world");
|
|
assert_eq!(buf.remaining(), 117);
|
|
assert_eq!(buf.bytes(), b"hello world");
|
|
|
|
let view1 = buf.slice(0, 11);
|
|
view1.buf().copy_to(&mut dst);
|
|
|
|
assert_eq!(dst, b"hello world");
|
|
assert_eq!(view1, buf.slice(0, 11));
|
|
|
|
drop(buf);
|
|
let mut buf = AppendBuf::with_capacity(100);
|
|
buf.write_slice(b"zomg no no no no");
|
|
|
|
assert_eq!(dst, b"hello world");
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
pub fn test_append_buf_from_pool() {
|
|
let pool = Pool::with_capacity(2, 256);
|
|
|
|
// Run in a loop a bunch in hope that if there is a memory issue, it will
|
|
// be exposed
|
|
for _ in 0..1000 {
|
|
let mut buf = pool.new_append_buf().unwrap();
|
|
let mut dst: Vec<u8> = vec![];
|
|
|
|
assert_eq!(buf.remaining(), 256);
|
|
|
|
buf.write_slice(b"hello world");
|
|
assert_eq!(buf.remaining(), 245);
|
|
assert_eq!(buf.bytes(), b"hello world");
|
|
|
|
let view1 = buf.slice(0, 11);
|
|
view1.buf().copy_to(&mut dst);
|
|
|
|
assert_eq!(dst, b"hello world");
|
|
assert_eq!(view1, buf.slice(0, 11));
|
|
|
|
drop(buf);
|
|
let mut buf = pool.new_append_buf().unwrap();
|
|
buf.write_slice(b"zomg no no no no");
|
|
|
|
assert_eq!(dst, b"hello world");
|
|
}
|
|
}
|