From 12f62b94d6ccdca90dcd0edfbff82031fec3912f Mon Sep 17 00:00:00 2001 From: Joel Galenson Date: Wed, 22 Sep 2021 10:51:41 -0700 Subject: Upgrade rust/crates/bytes to 1.1.0 Test: make Change-Id: I62c795247b1a9f37ec66c3ef6b1add6b78530482 --- tests/test_take.rs | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'tests/test_take.rs') diff --git a/tests/test_take.rs b/tests/test_take.rs index a23a29e..51df91d 100644 --- a/tests/test_take.rs +++ b/tests/test_take.rs @@ -1,6 +1,7 @@ #![warn(rust_2018_idioms)] use bytes::buf::Buf; +use bytes::Bytes; #[test] fn long_take() { @@ -10,3 +11,22 @@ fn long_take() { assert_eq!(11, buf.remaining()); assert_eq!(b"hello world", buf.chunk()); } + +#[test] +fn take_copy_to_bytes() { + let mut abcd = Bytes::copy_from_slice(b"abcd"); + let abcd_ptr = abcd.as_ptr(); + let mut take = (&mut abcd).take(2); + let a = take.copy_to_bytes(1); + assert_eq!(Bytes::copy_from_slice(b"a"), a); + // assert `to_bytes` did not allocate + assert_eq!(abcd_ptr, a.as_ptr()); + assert_eq!(Bytes::copy_from_slice(b"bcd"), abcd); +} + +#[test] +#[should_panic] +fn take_copy_to_bytes_panics() { + let abcd = Bytes::copy_from_slice(b"abcd"); + abcd.take(2).copy_to_bytes(3); +} -- cgit v1.2.3