363 lines
11 KiB
Go
363 lines
11 KiB
Go
|
package cisco_telemetry_mdt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/binary"
|
||
|
|
"errors"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/golang/protobuf/proto"
|
||
|
|
|
||
|
|
dialout "github.com/cisco-ie/nx-telemetry-proto/mdt_dialout"
|
||
|
|
telemetry "github.com/cisco-ie/nx-telemetry-proto/telemetry_bis"
|
||
|
|
"github.com/influxdata/telegraf/testutil"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHandleTelemetryEmpty(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "dummy"}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
c.Start(acc)
|
||
|
|
|
||
|
|
telemetry := &telemetry.Telemetry{
|
||
|
|
DataGpbkv: []*telemetry.TelemetryField{
|
||
|
|
{},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
data, _ := proto.Marshal(telemetry)
|
||
|
|
|
||
|
|
c.handleTelemetry(data)
|
||
|
|
assert.Contains(t, acc.Errors, errors.New("empty encoding path or measurement"))
|
||
|
|
assert.Empty(t, acc.Metrics)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandleTelemetryTwoSimple(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "dummy", Aliases: map[string]string{"alias": "type:model/some/path"}}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
c.Start(acc)
|
||
|
|
|
||
|
|
telemetry := &telemetry.Telemetry{
|
||
|
|
MsgTimestamp: 1543236572000,
|
||
|
|
EncodingPath: "type:model/some/path",
|
||
|
|
NodeId: &telemetry.Telemetry_NodeIdStr{NodeIdStr: "hostname"},
|
||
|
|
Subscription: &telemetry.Telemetry_SubscriptionIdStr{SubscriptionIdStr: "subscription"},
|
||
|
|
DataGpbkv: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "keys",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "name",
|
||
|
|
ValueByType: &telemetry.TelemetryField_StringValue{StringValue: "str"},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "uint64",
|
||
|
|
ValueByType: &telemetry.TelemetryField_Uint64Value{Uint64Value: 1234},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "content",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "bool",
|
||
|
|
ValueByType: &telemetry.TelemetryField_BoolValue{BoolValue: true},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "keys",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "name",
|
||
|
|
ValueByType: &telemetry.TelemetryField_StringValue{StringValue: "str2"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "content",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "bool",
|
||
|
|
ValueByType: &telemetry.TelemetryField_BoolValue{BoolValue: false},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
data, _ := proto.Marshal(telemetry)
|
||
|
|
|
||
|
|
c.handleTelemetry(data)
|
||
|
|
assert.Empty(t, acc.Errors)
|
||
|
|
|
||
|
|
tags := map[string]string{"path": "type:model/some/path", "name": "str", "uint64": "1234", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields := map[string]interface{}{"bool": true}
|
||
|
|
acc.AssertContainsTaggedFields(t, "alias", fields, tags)
|
||
|
|
|
||
|
|
tags = map[string]string{"path": "type:model/some/path", "name": "str2", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields = map[string]interface{}{"bool": false}
|
||
|
|
acc.AssertContainsTaggedFields(t, "alias", fields, tags)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandleTelemetrySingleNested(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "dummy", Aliases: map[string]string{"nested": "type:model/nested/path"}}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
c.Start(acc)
|
||
|
|
|
||
|
|
telemetry := &telemetry.Telemetry{
|
||
|
|
MsgTimestamp: 1543236572000,
|
||
|
|
EncodingPath: "type:model/nested/path",
|
||
|
|
NodeId: &telemetry.Telemetry_NodeIdStr{NodeIdStr: "hostname"},
|
||
|
|
Subscription: &telemetry.Telemetry_SubscriptionIdStr{SubscriptionIdStr: "subscription"},
|
||
|
|
DataGpbkv: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "keys",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "nested",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "key",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "level",
|
||
|
|
ValueByType: &telemetry.TelemetryField_DoubleValue{DoubleValue: 3},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "content",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "nested",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "value",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "foo",
|
||
|
|
ValueByType: &telemetry.TelemetryField_StringValue{StringValue: "bar"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
data, _ := proto.Marshal(telemetry)
|
||
|
|
|
||
|
|
c.handleTelemetry(data)
|
||
|
|
assert.Empty(t, acc.Errors)
|
||
|
|
|
||
|
|
tags := map[string]string{"path": "type:model/nested/path", "level": "3", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields := map[string]interface{}{"nested/value/foo": "bar"}
|
||
|
|
acc.AssertContainsTaggedFields(t, "nested", fields, tags)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTCPDialoutOverflow(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "tcp", ServiceAddress: "127.0.0.1:57000"}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
assert.Nil(t, c.Start(acc))
|
||
|
|
|
||
|
|
hdr := struct {
|
||
|
|
MsgType uint16
|
||
|
|
MsgEncap uint16
|
||
|
|
MsgHdrVersion uint16
|
||
|
|
MsgFlags uint16
|
||
|
|
MsgLen uint32
|
||
|
|
}{MsgLen: uint32(1000000000)}
|
||
|
|
|
||
|
|
conn, _ := net.Dial("tcp", "127.0.0.1:57000")
|
||
|
|
binary.Write(conn, binary.BigEndian, hdr)
|
||
|
|
conn.Read([]byte{0})
|
||
|
|
conn.Close()
|
||
|
|
|
||
|
|
c.Stop()
|
||
|
|
|
||
|
|
assert.Contains(t, acc.Errors, errors.New("dialout packet too long: 1000000000"))
|
||
|
|
}
|
||
|
|
|
||
|
|
func mockTelemetryMessage() *telemetry.Telemetry {
|
||
|
|
return &telemetry.Telemetry{
|
||
|
|
MsgTimestamp: 1543236572000,
|
||
|
|
EncodingPath: "type:model/some/path",
|
||
|
|
NodeId: &telemetry.Telemetry_NodeIdStr{NodeIdStr: "hostname"},
|
||
|
|
Subscription: &telemetry.Telemetry_SubscriptionIdStr{SubscriptionIdStr: "subscription"},
|
||
|
|
DataGpbkv: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "keys",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "name",
|
||
|
|
ValueByType: &telemetry.TelemetryField_StringValue{StringValue: "str"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "content",
|
||
|
|
Fields: []*telemetry.TelemetryField{
|
||
|
|
{
|
||
|
|
Name: "value",
|
||
|
|
ValueByType: &telemetry.TelemetryField_Sint64Value{Sint64Value: -1},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTCPDialoutMultiple(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "tcp", ServiceAddress: "127.0.0.1:57000", Aliases: map[string]string{
|
||
|
|
"some": "type:model/some/path", "parallel": "type:model/parallel/path", "other": "type:model/other/path"}}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
assert.Nil(t, c.Start(acc))
|
||
|
|
|
||
|
|
telemetry := mockTelemetryMessage()
|
||
|
|
|
||
|
|
hdr := struct {
|
||
|
|
MsgType uint16
|
||
|
|
MsgEncap uint16
|
||
|
|
MsgHdrVersion uint16
|
||
|
|
MsgFlags uint16
|
||
|
|
MsgLen uint32
|
||
|
|
}{}
|
||
|
|
|
||
|
|
conn, _ := net.Dial("tcp", "127.0.0.1:57000")
|
||
|
|
|
||
|
|
data, _ := proto.Marshal(telemetry)
|
||
|
|
hdr.MsgLen = uint32(len(data))
|
||
|
|
binary.Write(conn, binary.BigEndian, hdr)
|
||
|
|
conn.Write(data)
|
||
|
|
|
||
|
|
conn2, _ := net.Dial("tcp", "127.0.0.1:57000")
|
||
|
|
telemetry.EncodingPath = "type:model/parallel/path"
|
||
|
|
data, _ = proto.Marshal(telemetry)
|
||
|
|
hdr.MsgLen = uint32(len(data))
|
||
|
|
binary.Write(conn2, binary.BigEndian, hdr)
|
||
|
|
conn2.Write(data)
|
||
|
|
conn2.Write([]byte{0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0})
|
||
|
|
conn2.Read([]byte{0})
|
||
|
|
conn2.Close()
|
||
|
|
|
||
|
|
telemetry.EncodingPath = "type:model/other/path"
|
||
|
|
data, _ = proto.Marshal(telemetry)
|
||
|
|
hdr.MsgLen = uint32(len(data))
|
||
|
|
binary.Write(conn, binary.BigEndian, hdr)
|
||
|
|
conn.Write(data)
|
||
|
|
conn.Write([]byte{0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0})
|
||
|
|
conn.Read([]byte{0})
|
||
|
|
c.Stop()
|
||
|
|
conn.Close()
|
||
|
|
|
||
|
|
// We use the invalid dialout flags to let the server close the connection
|
||
|
|
assert.Equal(t, acc.Errors, []error{errors.New("invalid dialout flags: 257"), errors.New("invalid dialout flags: 257")})
|
||
|
|
|
||
|
|
tags := map[string]string{"path": "type:model/some/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields := map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "some", fields, tags)
|
||
|
|
|
||
|
|
tags = map[string]string{"path": "type:model/parallel/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields = map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "parallel", fields, tags)
|
||
|
|
|
||
|
|
tags = map[string]string{"path": "type:model/other/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields = map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "other", fields, tags)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGRPCDialoutError(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "grpc", ServiceAddress: "127.0.0.1:57001"}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
assert.Nil(t, c.Start(acc))
|
||
|
|
|
||
|
|
conn, _ := grpc.Dial("127.0.0.1:57001", grpc.WithInsecure())
|
||
|
|
client := dialout.NewGRPCMdtDialoutClient(conn)
|
||
|
|
stream, _ := client.MdtDialout(context.Background())
|
||
|
|
|
||
|
|
args := &dialout.MdtDialoutArgs{Errors: "foobar"}
|
||
|
|
stream.Send(args)
|
||
|
|
|
||
|
|
// Wait for the server to close
|
||
|
|
stream.Recv()
|
||
|
|
c.Stop()
|
||
|
|
|
||
|
|
assert.Equal(t, acc.Errors, []error{errors.New("GRPC dialout error: foobar")})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGRPCDialoutMultiple(t *testing.T) {
|
||
|
|
c := &CiscoTelemetryMDT{Transport: "grpc", ServiceAddress: "127.0.0.1:57001", Aliases: map[string]string{
|
||
|
|
"some": "type:model/some/path", "parallel": "type:model/parallel/path", "other": "type:model/other/path"}}
|
||
|
|
acc := &testutil.Accumulator{}
|
||
|
|
assert.Nil(t, c.Start(acc))
|
||
|
|
telemetry := mockTelemetryMessage()
|
||
|
|
|
||
|
|
conn, _ := grpc.Dial("127.0.0.1:57001", grpc.WithInsecure(), grpc.WithBlock())
|
||
|
|
client := dialout.NewGRPCMdtDialoutClient(conn)
|
||
|
|
stream, _ := client.MdtDialout(context.TODO())
|
||
|
|
|
||
|
|
data, _ := proto.Marshal(telemetry)
|
||
|
|
args := &dialout.MdtDialoutArgs{Data: data, ReqId: 456}
|
||
|
|
stream.Send(args)
|
||
|
|
|
||
|
|
conn2, _ := grpc.Dial("127.0.0.1:57001", grpc.WithInsecure(), grpc.WithBlock())
|
||
|
|
client2 := dialout.NewGRPCMdtDialoutClient(conn2)
|
||
|
|
stream2, _ := client2.MdtDialout(context.TODO())
|
||
|
|
|
||
|
|
telemetry.EncodingPath = "type:model/parallel/path"
|
||
|
|
data, _ = proto.Marshal(telemetry)
|
||
|
|
args = &dialout.MdtDialoutArgs{Data: data}
|
||
|
|
stream2.Send(args)
|
||
|
|
stream2.Send(&dialout.MdtDialoutArgs{Errors: "testclose"})
|
||
|
|
stream2.Recv()
|
||
|
|
conn2.Close()
|
||
|
|
|
||
|
|
telemetry.EncodingPath = "type:model/other/path"
|
||
|
|
data, _ = proto.Marshal(telemetry)
|
||
|
|
args = &dialout.MdtDialoutArgs{Data: data}
|
||
|
|
stream.Send(args)
|
||
|
|
stream.Send(&dialout.MdtDialoutArgs{Errors: "testclose"})
|
||
|
|
stream.Recv()
|
||
|
|
|
||
|
|
c.Stop()
|
||
|
|
conn.Close()
|
||
|
|
|
||
|
|
assert.Equal(t, acc.Errors, []error{errors.New("GRPC dialout error: testclose"), errors.New("GRPC dialout error: testclose")})
|
||
|
|
|
||
|
|
tags := map[string]string{"path": "type:model/some/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields := map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "some", fields, tags)
|
||
|
|
|
||
|
|
tags = map[string]string{"path": "type:model/parallel/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields = map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "parallel", fields, tags)
|
||
|
|
|
||
|
|
tags = map[string]string{"path": "type:model/other/path", "name": "str", "source": "hostname", "subscription": "subscription"}
|
||
|
|
fields = map[string]interface{}{"value": int64(-1)}
|
||
|
|
acc.AssertContainsTaggedFields(t, "other", fields, tags)
|
||
|
|
|
||
|
|
}
|