VictoriaMetrics/lib/logstorage/pipe_stream_context_test.go

package logstorage

import (
	"testing"
)

func TestParsePipeStreamContextSuccess(t *testing.T) {
	f := func(pipeStr string) {
		t.Helper()
		expectParsePipeSuccess(t, pipeStr)
	}

	f(`stream_context before 5`)
	f(`stream_context after 10`)
	f(`stream_context before 10 after 20`)
}

func TestParsePipeStreamContextFailure(t *testing.T) {
	f := func(pipeStr string) {
		t.Helper()
		expectParsePipeFailure(t, pipeStr)
	}

	f(`stream_context`)
	f(`stream_context before`)
	f(`stream_context after`)
	f(`stream_context before after`)
	f(`stream_context after before`)
	f(`stream_context before -4`)
	f(`stream_context after -4`)
}

func TestPipeStreamContext(t *testing.T) {
	f := func(pipeStr string, rows, rowsExpected [][]Field) {
		t.Helper()
		expectPipeResults(t, pipeStr, rows, rowsExpected)
	}

	f("stream_context before 0", [][]Field{
		{
			{"a", `2`},
			{"b", `3`},
		},
		{
			{"a", "2"},
			{"b", "3"},
		},
		{
			{"a", `2`},
			{"b", `54`},
			{"c", "d"},
		},
	}, [][]Field{
		{
			{"a", `2`},
			{"b", `3`},
		},
		{
			{"a", "2"},
			{"b", "3"},
		},
		{
			{"a", `2`},
			{"b", `54`},
			{"c", "d"},
		},
	})

	f("stream_context after 0", [][]Field{
		{
			{"a", `2`},
			{"b", `3`},
		},
		{
			{"a", "2"},
			{"b", "3"},
		},
		{
			{"a", `2`},
			{"b", `54`},
			{"c", "d"},
		},
	}, [][]Field{
		{
			{"a", `2`},
			{"b", `3`},
		},
		{
			{"a", "2"},
			{"b", "3"},
		},
		{
			{"a", `2`},
			{"b", `54`},
			{"c", "d"},
		},
	})
}

func TestPipeStreamContextUpdateNeededFields(t *testing.T) {
	f := func(s, neededFields, unneededFields, neededFieldsExpected, unneededFieldsExpected string) {
		t.Helper()
		expectPipeNeededFields(t, s, neededFields, unneededFields, neededFieldsExpected, unneededFieldsExpected)
	}

	// all the needed fields
	f("stream_context before 10", "*", "", "*", "")

	// plus unneeded fields
	f("stream_context after 4", "*", "f1,f2", "*", "f1,f2")
	f("stream_context after 4", "*", "_time,f1,_stream_id", "*", "f1")

	// needed fields
	f("stream_context before 3", "f1,f2", "", "_stream_id,_time,f1,f2", "")
	f("stream_context before 3", "_time,f1,_stream_id", "", "_stream_id,_time,f1", "")
}