aboutsummaryrefslogtreecommitdiff
path: root/tests/test_buf.rs
diff options
context:
space:
mode:
authorChih-Hung Hsieh <chh@google.com>2020-06-10 20:15:59 -0700
committerChih-Hung Hsieh <chh@google.com>2020-06-10 20:15:59 -0700
commit01ae379526218e7f6ca28510b6c06098419b7380 (patch)
treeded7cfd36c57807924e0b2e62439a8df9f6d32c1 /tests/test_buf.rs
parent7457df12a2eb70afcfa283968f7d1fadf123b0a1 (diff)
downloadbytes-01ae379526218e7f6ca28510b6c06098419b7380.tar.gz
Import bytes-0.5.4
* Add OWNERS and Android.bp Bug: 143953733 Test: make Change-Id: Ica69d7252abb1c2967f2c0e11a62d0bd83192fb3
Diffstat (limited to 'tests/test_buf.rs')
-rw-r--r--tests/test_buf.rs101
1 files changed, 101 insertions, 0 deletions
diff --git a/tests/test_buf.rs b/tests/test_buf.rs
new file mode 100644
index 0000000..12b75a4
--- /dev/null
+++ b/tests/test_buf.rs
@@ -0,0 +1,101 @@
+#![deny(warnings, rust_2018_idioms)]
+
+use bytes::Buf;
+use std::io::IoSlice;
+
+#[test]
+fn test_fresh_cursor_vec() {
+ let mut buf = &b"hello"[..];
+
+ assert_eq!(buf.remaining(), 5);
+ assert_eq!(buf.bytes(), b"hello");
+
+ buf.advance(2);
+
+ assert_eq!(buf.remaining(), 3);
+ assert_eq!(buf.bytes(), b"llo");
+
+ buf.advance(3);
+
+ assert_eq!(buf.remaining(), 0);
+ assert_eq!(buf.bytes(), b"");
+}
+
+#[test]
+fn test_get_u8() {
+ let mut buf = &b"\x21zomg"[..];
+ assert_eq!(0x21, buf.get_u8());
+}
+
+#[test]
+fn test_get_u16() {
+ let mut buf = &b"\x21\x54zomg"[..];
+ assert_eq!(0x2154, buf.get_u16());
+ let mut buf = &b"\x21\x54zomg"[..];
+ assert_eq!(0x5421, buf.get_u16_le());
+}
+
+#[test]
+#[should_panic]
+fn test_get_u16_buffer_underflow() {
+ let mut buf = &b"\x21"[..];
+ buf.get_u16();
+}
+
+#[test]
+fn test_bufs_vec() {
+ let buf = &b"hello world"[..];
+
+ let b1: &[u8] = &mut [];
+ let b2: &[u8] = &mut [];
+
+ let mut dst = [IoSlice::new(b1), IoSlice::new(b2)];
+
+ assert_eq!(1, buf.bytes_vectored(&mut dst[..]));
+}
+
+#[test]
+fn test_vec_deque() {
+ use std::collections::VecDeque;
+
+ let mut buffer: VecDeque<u8> = VecDeque::new();
+ buffer.extend(b"hello world");
+ assert_eq!(11, buffer.remaining());
+ assert_eq!(b"hello world", buffer.bytes());
+ buffer.advance(6);
+ assert_eq!(b"world", buffer.bytes());
+ buffer.extend(b" piece");
+ let mut out = [0; 11];
+ buffer.copy_to_slice(&mut out);
+ assert_eq!(b"world piece", &out[..]);
+}
+
+#[test]
+fn test_deref_buf_forwards() {
+ struct Special;
+
+ impl Buf for Special {
+ fn remaining(&self) -> usize {
+ unreachable!("remaining");
+ }
+
+ fn bytes(&self) -> &[u8] {
+ unreachable!("bytes");
+ }
+
+ fn advance(&mut self, _: usize) {
+ unreachable!("advance");
+ }
+
+ fn get_u8(&mut self) -> u8 {
+ // specialized!
+ b'x'
+ }
+ }
+
+ // these should all use the specialized method
+ assert_eq!(Special.get_u8(), b'x');
+ assert_eq!((&mut Special as &mut dyn Buf).get_u8(), b'x');
+ assert_eq!((Box::new(Special) as Box<dyn Buf>).get_u8(), b'x');
+ assert_eq!(Box::new(Special).get_u8(), b'x');
+}