Merge pull request #7 from multiformats/uvarint-tests

tests for unbounded uvarint streams.
multiformats · Aug 7, 2020 · 5d9803d · 5d9803d
2 parents a3ded45 + bc41d4e
commit 5d9803d
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/varint_test.go b/varint_test.go
@@ -34,6 +34,35 @@ func TestVarintSize(t *testing.T) {
 	}
 }
 
+func TestOverflow_9thSignalsMore(t *testing.T) {
+	buf := bytes.NewBuffer([]byte{
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+		0x80,
+	})
+
+	_, err := ReadUvarint(buf)
+	if err != ErrOverflow {
+		t.Fatalf("expected ErrOverflow, got: %s", err)
+	}
+}
+
+func TestOverflow_ReadBuffer(t *testing.T) {
+	buf := bytes.NewBuffer([]byte{
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+		0xff, 0xff, 0xff, 0xff,
+	})
+
+	_, err := ReadUvarint(buf)
+	if err != ErrOverflow {
+		t.Fatalf("expected ErrOverflow, got: %s", err)
+	}
+}
+
 func TestOverflow(t *testing.T) {
 	i, n, err := FromUvarint(
 		[]byte{