helper_test.go

package jzon

import (
	"io"
	"reflect"
	"runtime"
	"testing"
)

var (
	runtimeErrorType = reflect.TypeOf((*runtime.Error)(nil)).Elem()
)

func skipTest(t *testing.T, fmt string, args ...interface{}) {
	t.Skipf(fmt, args...)
}

func withIterator(data string, cb func(it *Iterator)) {
	it := NewIterator()
	defer it.Release()
	if data != "" {
		it.ResetBytes(localStringToBytes(data))
	}
	cb(it)
}

type stepByteReader struct {
	b    string
	step int
	err  error
}

var _ io.Reader = &stepByteReader{}

func (o *stepByteReader) Read(p []byte) (n int, err error) {
	if len(p) == 0 {
		return 0, nil
	}
	if len(o.b) == 0 {
		if o.err != nil {
			return 0, o.err
		}
		return 0, io.EOF
	}
	step := o.step
	if step == 0 {
		step = 1
	} else {
		if len(p) < step {
			step = len(p)
		}
	}
	n = copy(p[:step], o.b)
	o.b = o.b[n:]
	return n, nil
}

type repeatByteReader struct {
	b     byte
	count int
}

var _ io.Reader = &repeatByteReader{}

func (r *repeatByteReader) Read(p []byte) (n int, err error) {
	l := len(p)
	for r.count > 0 && n < l {
		p[n] = r.b
		r.count--
		n++
	}
	return
}