From 63c644719fae1d524b3a54789407b85a62c572ba Mon Sep 17 00:00:00 2001 From: Ian Jackson Date: Sat, 17 Apr 2021 12:16:16 +0100 Subject: [PATCH] packetframe: Provide read_all Signed-off-by: Ian Jackson --- src/packetframe.rs | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/packetframe.rs b/src/packetframe.rs index ba1f7086..f64960f7 100644 --- a/src/packetframe.rs +++ b/src/packetframe.rs @@ -340,6 +340,12 @@ fn write_test(){ assert_eq!(r.kind(), ErrorKind::Other); assert!(r.into_inner().unwrap().is::()); }; + let expect_good = |rd: &mut FrameReader<_>, expected: &[u8]| { + let mut buf = vec![]; + let mut frame = rd.new_frame().unwrap(); + frame.read_to_end(&mut buf).unwrap(); + assert_eq!(&*buf ,expected); + }; let mut rd = FrameReader::new(&*msg.buf); let mut buf = [0u8;10]; @@ -355,4 +361,12 @@ fn write_test(){ let mut _frame = rd.new_frame().unwrap(); } expect_boom(&mut rd); + + let read_all = || { + let mut rd = FrameReader::new(&*msg.buf); + expect_good(&mut rd, b"hello"); + expect_boom(&mut rd); + expect_good(&mut rd, b"longer!"); + }; + read_all(); } -- 2.30.2