VictoriaMetrics/lib/logstorage/json_parser_test.go

package logstorage

import (
	"reflect"
	"testing"
)

func TestJSONParserFailure(t *testing.T) {
	f := func(data string) {
		t.Helper()

		p := GetJSONParser()
		err := p.ParseLogMessage([]byte(data))
		if err == nil {
			t.Fatalf("expecting non-nil error")
		}
		PutJSONParser(p)
	}
	f("")
	f("{foo")
	f("[1,2,3]")
	f(`{"foo",}`)
}

func TestJSONParserSuccess(t *testing.T) {
	f := func(data string, fieldsExpected []Field) {
		t.Helper()

		p := GetJSONParser()
		err := p.ParseLogMessage([]byte(data))
		if err != nil {
			t.Fatalf("unexpected error: %s", err)
		}
		if !reflect.DeepEqual(p.Fields, fieldsExpected) {
			t.Fatalf("unexpected fields;\ngot\n%s\nwant\n%s", p.Fields, fieldsExpected)
		}
		PutJSONParser(p)
	}

	f("{}", nil)
	f(`{"foo":"bar"}`, []Field{
		{
			Name:  "foo",
			Value: "bar",
		},
	})
	f(`{"foo":{"bar":{"x":"y","z":["foo"]}},"a":1,"b":true,"c":[1,2],"d":false,"e":null}`, []Field{
		{
			Name:  "foo.bar.x",
			Value: "y",
		},
		{
			Name:  "foo.bar.z",
			Value: `["foo"]`,
		},
		{
			Name:  "a",
			Value: "1",
		},
		{
			Name:  "b",
			Value: "true",
		},
		{
			Name:  "c",
			Value: "[1,2]",
		},
		{
			Name:  "d",
			Value: "false",
		},
	})
}

func TestJSONParserTooLongFieldName(t *testing.T) {
	f := func(data string, maxFieldLen int, fieldsExpected []Field) {
		t.Helper()

		p := GetJSONParser()
		err := p.parseLogMessage([]byte(data), maxFieldLen)
		if err != nil {
			t.Fatalf("unexpected error: %s", err)
		}
		if !reflect.DeepEqual(p.Fields, fieldsExpected) {
			t.Fatalf("unexpected fields;\ngot\n%s\nwant\n%s", p.Fields, fieldsExpected)
		}
		PutJSONParser(p)
	}

	f(`{"foo":"bar","baz":{"abc":"y"}}`, 7, []Field{
		{
			Name:  "foo",
			Value: "bar",
		},
		{
			Name:  "baz.abc",
			Value: "y",
		},
	})
	f(`{"foo":"bar","baz":{"abc":"y"}}`, 6, []Field{
		{
			Name:  "foo",
			Value: "bar",
		},
		{
			Name:  "baz",
			Value: `{"abc":"y"}`,
		},
	})
	f(`{"foo":"bar","baz":{"abc":"y"}}`, 3, []Field{
		{
			Name:  "foo",
			Value: "bar",
		},
		{
			Name:  "baz",
			Value: `{"abc":"y"}`,
		},
	})
	f(`{"foo":"bar","baz":{"abc":"y"}}`, 2, []Field{
		{
			Name:  "_msg",
			Value: `{"foo":"bar","baz":{"abc":"y"}}`,
		},
	})
}