dec_obj_iter_test.go

package jx

import (
	"encoding/json"
	"fmt"
	"io"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/go-faster/errors"
)

func TestDecoder_ObjIter(t *testing.T) {
	testIter := func(d *Decoder) error {
		iter, err := d.ObjIter()
		if err != nil {
			return err
		}
		for iter.Next() {
			if err := d.Skip(); err != nil {
				return err
			}
		}
		if iter.Next() {
			panic("BUG")
		}
		if err := iter.Err(); err != nil {
			return err
		}

		// Check for any trialing json.
		if d.head != d.tail {
			if err := d.Skip(); err != io.EOF {
				return errors.Wrap(err, "unexpected trialing data")
			}
		}
		return nil
	}
	for i, s := range testObjs {
		s := s
		t.Run(fmt.Sprintf("Test%d", i+1), func(t *testing.T) {
			checker := require.Error
			if json.Valid([]byte(s)) {
				checker = require.NoError
			}

			d := DecodeStr(s)
			checker(t, testIter(d), s)
		})
	}
	t.Run("Depth", func(t *testing.T) {
		d := DecodeStr(`{`)
		// Emulate depth
		d.depth = maxDepth
		require.ErrorIs(t, testIter(d), errMaxDepth)
	})
	t.Run("Empty", func(t *testing.T) {
		d := DecodeStr(``)
		require.ErrorIs(t, testIter(d), io.ErrUnexpectedEOF)
	})
	t.Run("Key", testBufferReader(`{"foo":1,"bar":1,"baz":1}`, func(t *testing.T, d *Decoder) {
		a := require.New(t)

		iter, err := d.ObjIter()
		a.NoError(err)

		var r []string
		for iter.Next() {
			r = append(r, string(iter.Key()))
			a.NoError(d.Skip())
		}
		a.False(iter.Next())
		a.NoError(iter.Err())

		a.Equal([]string{"foo", "bar", "baz"}, r)
	}))
}

func TestDecoderObjIterIssue62(t *testing.T) {
	a := require.New(t)

	const input = `{"1":1,"2":2}`

	// Force decoder to read only first 4 bytes of input.
	d := Decode(strings.NewReader(input), 4)

	iter, err := d.ObjIter()
	a.NoError(err)

	actual := map[string]int{}
	for iter.Next() {
		val, err := d.Int()
		a.NoError(err)
		actual[string(iter.Key())] = val
	}
	a.Equal(map[string]int{
		"1": 1,
		"2": 2,
	}, actual)
}