From 02925b2e813843ab330fc6f7a0c5906b2ba0f0b5 Mon Sep 17 00:00:00 2001 From: Ian Jackson Date: Sat, 17 Apr 2021 15:12:29 +0100 Subject: [PATCH] packetframe: tests: Test EOF succss before first chunk header Signed-off-by: Ian Jackson --- src/packetframe.rs | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/packetframe.rs b/src/packetframe.rs index 98f131f6..112b8f3d 100644 --- a/src/packetframe.rs +++ b/src/packetframe.rs @@ -368,6 +368,12 @@ fn write_test(){ assert_eq!(&*buf ,expected); dbgc!(str::from_utf8(&buf).unwrap()); } + fn expect_eof(rd: &mut FrameReader) { + let mut buf = [0u8;10]; + let mut frame = rd.new_frame().unwrap(); + let r = frame.read(&mut buf).unwrap(); dbgc!(&r); assert_eq!(r, 0); + let r = frame.read(&mut buf).unwrap(); dbgc!(&r); assert_eq!(r, 0); + } let mut rd = FrameReader::new(&*msg.buf); let mut buf = [0u8;10]; @@ -389,6 +395,7 @@ fn write_test(){ expect_good(&mut rd, b"hello"); expect_boom(&mut rd); expect_good(&mut rd, b"longer!"); + expect_eof(&mut rd); }; read_all(&mut &*msg.buf); -- 2.30.2