2016-03-17 17:50:39 +00:00
|
|
|
package json
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2016-11-28 18:19:35 +00:00
|
|
|
"strings"
|
2016-03-17 17:50:39 +00:00
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
|
2016-11-22 12:51:57 +00:00
|
|
|
"github.com/influxdata/telegraf/metric"
|
2016-03-17 17:50:39 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestSerializeMetricFloat(t *testing.T) {
|
|
|
|
now := time.Now()
|
|
|
|
tags := map[string]string{
|
|
|
|
"cpu": "cpu0",
|
|
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
|
|
"usage_idle": float64(91.5),
|
|
|
|
}
|
2016-11-22 12:51:57 +00:00
|
|
|
m, err := metric.New("cpu", tags, fields, now)
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
s := JsonSerializer{}
|
2016-11-28 18:19:35 +00:00
|
|
|
var buf []byte
|
|
|
|
buf, err = s.Serialize(m)
|
|
|
|
mS := strings.Split(strings.TrimSpace(string(buf)), "\n")
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
expS := []string{fmt.Sprintf("{\"fields\":{\"usage_idle\":91.5},\"name\":\"cpu\",\"tags\":{\"cpu\":\"cpu0\"},\"timestamp\":%d}", now.Unix())}
|
|
|
|
assert.Equal(t, expS, mS)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSerializeMetricInt(t *testing.T) {
|
|
|
|
now := time.Now()
|
|
|
|
tags := map[string]string{
|
|
|
|
"cpu": "cpu0",
|
|
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
|
|
"usage_idle": int64(90),
|
|
|
|
}
|
2016-11-22 12:51:57 +00:00
|
|
|
m, err := metric.New("cpu", tags, fields, now)
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
s := JsonSerializer{}
|
2016-11-28 18:19:35 +00:00
|
|
|
var buf []byte
|
|
|
|
buf, err = s.Serialize(m)
|
|
|
|
mS := strings.Split(strings.TrimSpace(string(buf)), "\n")
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
expS := []string{fmt.Sprintf("{\"fields\":{\"usage_idle\":90},\"name\":\"cpu\",\"tags\":{\"cpu\":\"cpu0\"},\"timestamp\":%d}", now.Unix())}
|
|
|
|
assert.Equal(t, expS, mS)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSerializeMetricString(t *testing.T) {
|
|
|
|
now := time.Now()
|
|
|
|
tags := map[string]string{
|
|
|
|
"cpu": "cpu0",
|
|
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
|
|
"usage_idle": "foobar",
|
|
|
|
}
|
2016-11-22 12:51:57 +00:00
|
|
|
m, err := metric.New("cpu", tags, fields, now)
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
s := JsonSerializer{}
|
2016-11-28 18:19:35 +00:00
|
|
|
var buf []byte
|
|
|
|
buf, err = s.Serialize(m)
|
|
|
|
mS := strings.Split(strings.TrimSpace(string(buf)), "\n")
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
expS := []string{fmt.Sprintf("{\"fields\":{\"usage_idle\":\"foobar\"},\"name\":\"cpu\",\"tags\":{\"cpu\":\"cpu0\"},\"timestamp\":%d}", now.Unix())}
|
|
|
|
assert.Equal(t, expS, mS)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSerializeMultiFields(t *testing.T) {
|
|
|
|
now := time.Now()
|
|
|
|
tags := map[string]string{
|
|
|
|
"cpu": "cpu0",
|
|
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
|
|
"usage_idle": int64(90),
|
|
|
|
"usage_total": 8559615,
|
|
|
|
}
|
2016-11-22 12:51:57 +00:00
|
|
|
m, err := metric.New("cpu", tags, fields, now)
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
s := JsonSerializer{}
|
2016-11-28 18:19:35 +00:00
|
|
|
var buf []byte
|
|
|
|
buf, err = s.Serialize(m)
|
|
|
|
mS := strings.Split(strings.TrimSpace(string(buf)), "\n")
|
2016-03-17 17:50:39 +00:00
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
expS := []string{fmt.Sprintf("{\"fields\":{\"usage_idle\":90,\"usage_total\":8559615},\"name\":\"cpu\",\"tags\":{\"cpu\":\"cpu0\"},\"timestamp\":%d}", now.Unix())}
|
|
|
|
assert.Equal(t, expS, mS)
|
|
|
|
}
|
2016-12-01 18:16:52 +00:00
|
|
|
|
|
|
|
func TestSerializeMetricWithEscapes(t *testing.T) {
|
|
|
|
now := time.Now()
|
|
|
|
tags := map[string]string{
|
|
|
|
"cpu tag": "cpu0",
|
|
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
|
|
"U,age=Idle": int64(90),
|
|
|
|
}
|
|
|
|
m, err := telegraf.NewMetric("My CPU", tags, fields, now)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
s := JsonSerializer{}
|
|
|
|
mS, err := s.Serialize(m)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
|
|
|
|
expS := []string{fmt.Sprintf(`{"fields":{"U,age=Idle":90},"name":"My CPU","tags":{"cpu tag":"cpu0"},"timestamp":%d}`, now.Unix())}
|
|
|
|
assert.Equal(t, expS, mS)
|
|
|
|
}
|