Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
4963 changed files with 677545 additions and 0 deletions
81
plugins/common/shim/output_test.go
Normal file
81
plugins/common/shim/output_test.go
Normal file
|
@ -0,0 +1,81 @@
|
|||
package shim
|
||||
|
||||
import (
|
||||
"io"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/metric"
|
||||
"github.com/influxdata/telegraf/plugins/serializers/influx"
|
||||
"github.com/influxdata/telegraf/testutil"
|
||||
)
|
||||
|
||||
func TestOutputShim(t *testing.T) {
|
||||
o := &testOutput{}
|
||||
|
||||
stdinReader, stdinWriter := io.Pipe()
|
||||
|
||||
s := New()
|
||||
s.stdin = stdinReader
|
||||
err := s.AddOutput(o)
|
||||
require.NoError(t, err)
|
||||
|
||||
wg := sync.WaitGroup{}
|
||||
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
if err := s.RunOutput(); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
serializer := &influx.Serializer{}
|
||||
require.NoError(t, serializer.Init())
|
||||
|
||||
m := metric.New("thing",
|
||||
map[string]string{
|
||||
"a": "b",
|
||||
},
|
||||
map[string]interface{}{
|
||||
"v": 1,
|
||||
},
|
||||
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)
|
||||
|
||||
wg.Wait()
|
||||
|
||||
require.Len(t, o.MetricsWritten, 1)
|
||||
mOut := o.MetricsWritten[0]
|
||||
|
||||
testutil.RequireMetricEqual(t, m, mOut)
|
||||
}
|
||||
|
||||
type testOutput struct {
|
||||
MetricsWritten []telegraf.Metric
|
||||
}
|
||||
|
||||
func (*testOutput) Connect() error {
|
||||
return nil
|
||||
}
|
||||
func (*testOutput) Close() error {
|
||||
return nil
|
||||
}
|
||||
func (o *testOutput) Write(metrics []telegraf.Metric) error {
|
||||
o.MetricsWritten = append(o.MetricsWritten, metrics...)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (*testOutput) SampleConfig() string {
|
||||
return ""
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue