X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=peer_protocol%2Ffuzz_test.go;h=5241504853dfe90f4b4d904395ff943d87de704b;hb=ac086bb3bd3b8e31c117362551eb0fe295a0f78d;hp=85fb42a66809e826b98c2c5447313a434b9e8eac;hpb=5380337f869ed630e1102036bf6dec48c044152a;p=btrtrc.git

diff --git a/peer_protocol/fuzz_test.go b/peer_protocol/fuzz_test.go
index 85fb42a6..52415048 100644
--- a/peer_protocol/fuzz_test.go
+++ b/peer_protocol/fuzz_test.go
@@ -17,7 +17,9 @@ func FuzzDecoder(f *testing.F) {
 	f.Add([]byte("\x00\x00\x00\x00"))
 	f.Add([]byte("\x00\x00\x00\x01\x00"))
 	f.Add([]byte("\x00\x00\x00\x03\x14\x00"))
+	f.Add([]byte("\x00\x00\x00\x01\x07"))
 	f.Fuzz(func(t *testing.T, b []byte) {
+		t.Logf("%q", b)
 		c := qt.New(t)
 		d := Decoder{
 			R:         bufio.NewReader(bytes.NewReader(b)),
@@ -43,7 +45,11 @@ func FuzzDecoder(f *testing.F) {
 		for _, m := range ms {
 			buf.Write(m.MustMarshalBinary())
 		}
-		c.Assert(buf.Bytes(), qt.DeepEquals, b)
+		if len(b) == 0 {
+			c.Assert(buf.Bytes(), qt.HasLen, 0)
+		} else {
+			c.Assert(buf.Bytes(), qt.DeepEquals, b)
+		}
 	})
 }