Function readFully

gapis/memory/pool_test.go:31–43

readFully returns the entire data produced by the given Reader, by reading up to maxRead bytes at a time.

(r io.Reader, maxRead int)

// readFully returns the entire data produced by the given Reader, by reading up to maxRead bytes at a time.
func readFully(r io.Reader, maxRead int) ([]byte, error) {
	var b bytes.Buffer
	readBuffer := make([]byte, maxRead)
	for {
		n, err := r.Read(readBuffer)
		b.Write(readBuffer[:n])
		if errors.Cause(err) == io.EOF {
			return b.Bytes(), nil
		} else if err != nil {
			return b.Bytes(), err
		}
	}
}

func checkData(ctx context.Context, s Data, expected []byte) {
	assert.For(ctx, "size").That(s.Size()).Equals(uint64(len(expected)))