From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sat, 17 Apr 2021 14:12:29 +0000 (+0100)
Subject: packetframe: tests: Test EOF succss before first chunk header
X-Git-Tag: otter-0.6.0~552
X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=commitdiff_plain;h=02925b2e813843ab330fc6f7a0c5906b2ba0f0b5;p=otter.git

packetframe: tests: Test EOF succss before first chunk header

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---

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<R:Read>(rd: &mut FrameReader<R>) {
+    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);