test(dec.int): add errReader case

go-faster · Nov 2, 2021 · d9c6cb5 · d9c6cb5
1 parent 7c45269
commit d9c6cb5
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/dec_int_test.go b/dec_int_test.go
@@ -49,3 +49,24 @@ func TestDecoder_uint_sizes(t *testing.T) {
 		require.Equal(t, uint(69315063), v)
 	}
 }
+
+func TestDecoder_Int(t *testing.T) {
+	r := errReader{}
+	get := func() *Decoder {
+		return &Decoder{
+			buf:    []byte{'1', '2'},
+			tail:   2,
+			reader: errReader{},
+		}
+	}
+	t.Run("32", func(t *testing.T) {
+		d := get()
+		_, err := d.Int32()
+		require.ErrorIs(t, err, r.Err())
+	})
+	t.Run("64", func(t *testing.T) {
+		d := get()
+		_, err := d.Int64()
+		require.ErrorIs(t, err, r.Err())
+	})
+}
diff --git a/dec_test.go b/dec_test.go
@@ -96,6 +96,10 @@ func TestDecoderBig(t *testing.T) {
 
 type errReader struct{}
 
+func (errReader) Err() error {
+	return io.ErrNoProgress
+}
+
 func (e errReader) Read(p []byte) (n int, err error) {
-	return 0, io.ErrNoProgress
+	return 0, e.Err()
 }