136 lines
2.5 KiB
Go
136 lines
2.5 KiB
Go
package influx
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/metric"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestReader(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
maxLineBytes int
|
|
bufferSize int
|
|
input []telegraf.Metric
|
|
expected []byte
|
|
}{
|
|
{
|
|
name: "minimal",
|
|
maxLineBytes: 4096,
|
|
bufferSize: 20,
|
|
input: []telegraf.Metric{
|
|
MustMetric(
|
|
metric.New(
|
|
"cpu",
|
|
map[string]string{},
|
|
map[string]interface{}{
|
|
"value": 42.0,
|
|
},
|
|
time.Unix(0, 0),
|
|
),
|
|
),
|
|
},
|
|
expected: []byte("cpu value=42 0\n"),
|
|
},
|
|
{
|
|
name: "multiple lines",
|
|
maxLineBytes: 4096,
|
|
bufferSize: 20,
|
|
input: []telegraf.Metric{
|
|
MustMetric(
|
|
metric.New(
|
|
"cpu",
|
|
map[string]string{},
|
|
map[string]interface{}{
|
|
"value": 42.0,
|
|
},
|
|
time.Unix(0, 0),
|
|
),
|
|
),
|
|
MustMetric(
|
|
metric.New(
|
|
"cpu",
|
|
map[string]string{},
|
|
map[string]interface{}{
|
|
"value": 42.0,
|
|
},
|
|
time.Unix(0, 0),
|
|
),
|
|
),
|
|
},
|
|
expected: []byte("cpu value=42 0\ncpu value=42 0\n"),
|
|
},
|
|
{
|
|
name: "exact fit",
|
|
maxLineBytes: 4096,
|
|
bufferSize: 15,
|
|
input: []telegraf.Metric{
|
|
MustMetric(
|
|
metric.New(
|
|
"cpu",
|
|
map[string]string{},
|
|
map[string]interface{}{
|
|
"value": 42.0,
|
|
},
|
|
time.Unix(0, 0),
|
|
),
|
|
),
|
|
},
|
|
expected: []byte("cpu value=42 0\n"),
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
serializer := NewSerializer()
|
|
serializer.SetMaxLineBytes(tt.maxLineBytes)
|
|
serializer.SetFieldSortOrder(SortFields)
|
|
reader := NewReader(tt.input, serializer)
|
|
|
|
data := new(bytes.Buffer)
|
|
readbuf := make([]byte, tt.bufferSize)
|
|
|
|
total := 0
|
|
for {
|
|
n, err := reader.Read(readbuf)
|
|
total += n
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
data.Write(readbuf[:n])
|
|
require.NoError(t, err)
|
|
}
|
|
require.Equal(t, tt.expected, data.Bytes())
|
|
require.Equal(t, len(tt.expected), total)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestZeroLengthBufferNoError(t *testing.T) {
|
|
m := MustMetric(
|
|
metric.New(
|
|
"cpu",
|
|
map[string]string{},
|
|
map[string]interface{}{
|
|
"value": 42.0,
|
|
},
|
|
time.Unix(0, 0),
|
|
),
|
|
)
|
|
serializer := NewSerializer()
|
|
serializer.SetFieldSortOrder(SortFields)
|
|
reader := NewReader([]telegraf.Metric{m}, serializer)
|
|
|
|
readbuf := make([]byte, 0)
|
|
|
|
n, err := reader.Read(readbuf)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, n)
|
|
}
|