iterator_array_test.go

package jzon

import (
	"errors"
	"io"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestIterator_Array_ReadArrayBegin(t *testing.T) {
	t.Run("leading space", func(t *testing.T) {
		withIterator(" [ ]", func(it *Iterator) {
			more, err := it.ReadArrayBegin()
			require.NoError(t, err)
			require.False(t, more)
		})
	})
	t.Run("eof", func(t *testing.T) {
		withIterator("", func(it *Iterator) {
			_, err := it.ReadArrayBegin()
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("invalid first byte", func(t *testing.T) {
		withIterator("1", func(it *Iterator) {
			_, err := it.ReadArrayBegin()
			require.IsType(t, UnexpectedByteError{}, err)
		})
	})
	t.Run("eof after bracket", func(t *testing.T) {
		withIterator("[", func(it *Iterator) {
			_, err := it.ReadArrayBegin()
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("more", func(t *testing.T) {
		withIterator(`["`, func(it *Iterator) {
			more, err := it.ReadArrayBegin()
			require.NoError(t, err)
			require.True(t, more)
		})
	})
}

func TestIterator_Array_ReadArrayMore(t *testing.T) {
	t.Run("leading space", func(t *testing.T) {
		withIterator(" ,", func(it *Iterator) {
			more, err := it.ReadArrayMore()
			require.NoError(t, err)
			require.True(t, more)
		})
	})
	t.Run("eof", func(t *testing.T) {
		withIterator(" ", func(it *Iterator) {
			_, err := it.ReadArrayMore()
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("more", func(t *testing.T) {
		withIterator(" ,", func(it *Iterator) {
			more, err := it.ReadArrayMore()
			require.NoError(t, err)
			require.True(t, more)
		})
	})
	t.Run("no more", func(t *testing.T) {
		withIterator(" ]", func(it *Iterator) {
			more, err := it.ReadArrayMore()
			require.NoError(t, err)
			require.False(t, more)
		})
	})
	t.Run("error", func(t *testing.T) {
		withIterator("a", func(it *Iterator) {
			_, err := it.ReadArrayMore()
			require.IsType(t, UnexpectedByteError{}, err)
		})
	})
}

func TestIterator_Array_ReadArray_Example(t *testing.T) {
	withIterator(" [ 0 , 1 , 2 ] ", func(it *Iterator) {
		must := require.New(t)
		more, err := it.ReadArrayBegin()
		must.NoError(err)
		i := 0
		for ; more; more, err = it.ReadArrayMore() {
			ri, err := it.ReadInt()
			must.NoError(err)
			must.Equal(i, ri)
			i++
		}
		must.NoError(err)
		_, err = it.NextValueType()
		must.Equal(io.EOF, err)
	})
}

func TestIterator_Array_ReadArrayCB(t *testing.T) {
	t.Run("leading space", func(t *testing.T) {
		withIterator(" [ ]", func(it *Iterator) {
			err := it.ReadArrayCB(nil)
			require.NoError(t, err)
		})
	})
	t.Run("eof", func(t *testing.T) {
		withIterator(" ", func(it *Iterator) {
			err := it.ReadArrayCB(nil)
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("invalid bracket", func(t *testing.T) {
		withIterator(" a", func(it *Iterator) {
			err := it.ReadArrayCB(nil)
			require.IsType(t, UnexpectedByteError{}, err)
		})
	})
	t.Run("eof after bracket", func(t *testing.T) {
		withIterator(" [ ", func(it *Iterator) {
			err := it.ReadArrayCB(nil)
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("no element", func(t *testing.T) {
		withIterator(" [ ] ", func(it *Iterator) {
			err := it.ReadArrayCB(nil)
			require.NoError(t, err)
		})
	})
	t.Run("callback error", func(t *testing.T) {
		withIterator(" [ 1 ] ", func(it *Iterator) {
			e := errors.New("test")
			err := it.ReadArrayCB(func(*Iterator) error {
				return e
			})
			require.Equal(t, e, err)
		})
	})
	t.Run("error on more", func(t *testing.T) {
		withIterator(" [ 1 ", func(it *Iterator) {
			err := it.ReadArrayCB(func(it *Iterator) (err error) {
				_, err = it.ReadUint()
				return
			})
			require.Equal(t, io.EOF, err)
		})
	})
	t.Run("error on more 2", func(t *testing.T) {
		withIterator(" [ 1a", func(it *Iterator) {
			err := it.ReadArrayCB(func(it *Iterator) (err error) {
				_, err = it.ReadUint()
				return
			})
			require.IsType(t, UnexpectedByteError{}, err)
		})
	})
	t.Run("2 items", func(t *testing.T) {
		withIterator(" [ 1 , 2 ] ", func(it *Iterator) {
			err := it.ReadArrayCB(func(it *Iterator) (err error) {
				_, err = it.ReadUint()
				return
			})
			require.NoError(t, err)
		})
	})
}

func TestIterator_Array_ReadArrayCB_Example(t *testing.T) {
	withIterator(" [ 0 , 1 , 2 ] ", func(it *Iterator) {
		must := require.New(t)
		i := 0
		err := it.ReadArrayCB(func(it *Iterator) (err error) {
			j, err := it.ReadInt()
			must.NoError(err)
			must.Equal(i, j)
			i++
			return nil
		})
		must.NoError(err)
		_, err = it.NextValueType()
		must.Equal(io.EOF, err)
	})
}