Function TestReadIntfRecursion

msgp/read_test.go:83–167

(t *testing.T)

}

func TestReadIntfRecursion(t *testing.T) {
	var buf bytes.Buffer
	dec := NewReader(&buf)
	enc := NewWriter(&buf)
	// Test array recursion...
	for range recursionLimit * 2 {
		enc.WriteArrayHeader(1)
	}
	enc.Flush()
	b := buf.Bytes()
	_, err := dec.ReadIntf()
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Reader error: %v", err)
	}
	_, _, err = ReadIntfBytes(b)
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Bytes error: %v", err)
	}
	// Test JSON
	dec.Reset(bytes.NewReader(b))
	_, err = dec.WriteToJSON(io.Discard)
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Reader error: %v", err)
	}
	_, err = UnmarshalAsJSON(io.Discard, b)
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Bytes error: %v", err)
	}
	_, err = CopyToJSON(io.Discard, bytes.NewReader(b))
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Bytes error: %v", err)
	}

	// Test map recursion...
	buf.Reset()
	for range recursionLimit * 2 {
		enc.WriteMapHeader(1)
		// Write a key...
		enc.WriteString("a")
	}
	enc.Flush()
	b = buf.Bytes()
	dec.Reset(bytes.NewReader(b))
	_, err = dec.ReadIntf()
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Reader error: %v", err)
	}
	_, _, err = ReadIntfBytes(b)
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Bytes error: %v", err)
	}

	// Test ReadMapStrInt using same input
	dec.Reset(bytes.NewReader(b))
	err = dec.ReadMapStrIntf(map[string]any{})
	if !errors.Is(err, ErrRecursion) {
		t.Errorf("unexpected Reader error: %v", err)
	}