Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/common/shim/processor_test.go

@@ -0,0 +1,113 @@
+package shim
+
+import (
+	"bufio"
+	"io"
+	"math/rand"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/metric"
+	"github.com/influxdata/telegraf/plugins/parsers/influx"
+	serializers_influx "github.com/influxdata/telegraf/plugins/serializers/influx"
+)
+
+func TestProcessorShim(t *testing.T) {
+	testSendAndReceive(t, "f1", "fv1")
+}
+
+func TestProcessorShimWithLargerThanDefaultScannerBufferSize(t *testing.T) {
+	letters := []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
+	b := make([]rune, 0, bufio.MaxScanTokenSize*2)
+	for i := 0; i < bufio.MaxScanTokenSize*2; i++ {
+		b = append(b, letters[rand.Intn(len(letters))])
+	}
+
+	testSendAndReceive(t, "f1", string(b))
+}
+
+func testSendAndReceive(t *testing.T, fieldKey, fieldValue string) {
+	p := &testProcessor{"hi", "mom"}
+
+	stdinReader, stdinWriter := io.Pipe()
+	stdoutReader, stdoutWriter := io.Pipe()
+
+	s := New()
+	// inject test into shim
+	s.stdin = stdinReader
+	s.stdout = stdoutWriter
+	err := s.AddProcessor(p)
+	require.NoError(t, err)
+
+	wg := sync.WaitGroup{}
+
+	wg.Add(1)
+	go func() {
+		if err := s.RunProcessor(); err != nil {
+			t.Error(err)
+		}
+		wg.Done()
+	}()
+
+	serializer := &serializers_influx.Serializer{}
+	require.NoError(t, serializer.Init())
+
+	parser := influx.Parser{}
+	require.NoError(t, parser.Init())
+
+	m := metric.New("thing",
+		map[string]string{
+			"a": "b",
+		},
+		map[string]interface{}{
+			"v":      1,
+			fieldKey: fieldValue,
+		},
+		time.Now(),
+	)
+	b, err := serializer.Serialize(m)
+	require.NoError(t, err)
+	_, err = stdinWriter.Write(b)
+	require.NoError(t, err)
+	err = stdinWriter.Close()
+	require.NoError(t, err)
+
+	r := bufio.NewReader(stdoutReader)
+	out, err := r.ReadString('\n')
+	require.NoError(t, err)
+	mOut, err := parser.ParseLine(out)
+	require.NoError(t, err)
+
+	val, ok := mOut.GetTag(p.tagName)
+	require.True(t, ok)
+	require.Equal(t, p.tagValue, val)
+	val2, ok := mOut.Fields()[fieldKey]
+	require.True(t, ok)
+	require.Equal(t, fieldValue, val2)
+	go func() {
+		if _, err = io.ReadAll(r); err != nil {
+			t.Error(err)
+		}
+	}()
+	wg.Wait()
+}
+
+type testProcessor struct {
+	tagName  string
+	tagValue string
+}
+
+func (p *testProcessor) Apply(in ...telegraf.Metric) []telegraf.Metric {
+	for _, m := range in {
+		m.AddTag(p.tagName, p.tagValue)
+	}
+	return in
+}
+
+func (*testProcessor) SampleConfig() string {
+	return ""
+}