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 98f131f69c4498bd296e55b76e571d084a87838d..112b8f3d4be7cb12378ab09bdd68ce222f58d79e 100644 (file)
--- 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);