go/opentelemetry-go
go/opentelemetry-go
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2024-12-12 10:04:29 +02:00
Code Issues Releases Activity

Added condition for missing attr received in attr expected

Browse Source
This commit is contained in:
Ahmed Mujtaba 2020-05-15 22:43:07 +02:00
parent c40b3d47cf
commit f4a25cf745
1 changed files with 74 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
plugin/grpctrace/interceptor_test.go
View File

@ -19,6 +19,7 @@ import (
"testing"
"time"
"github.com/golang/protobuf/proto"
"google.golang.org/grpc"
"google.golang.org/grpc/metadata"
@ -93,42 +94,96 @@ func TestUnaryClientInterceptor(t *testing.T) {
},
eventsAttr: []map[core.Key]core.Value{
{
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(req))),
},
{
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(reply))),
},
},
},
{
name: "/serviceName/bar",
expectedAttr: map[core.Key]core.Value{
rpcServiceKey: core.String("serviceName"),
rpcServiceKey: core.String("serviceName"),
netPeerIPKey: core.String("fake"),
netPeerPortKey: core.String("connection"),
},
eventsAttr: []map[core.Key]core.Value{
{
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(req))),
},
{
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(reply))),
},
},
},
{
name: "serviceName/bar",
expectedAttr: map[core.Key]core.Value{rpcServiceKey: core.String("serviceName")},
name: "serviceName/bar",
expectedAttr: map[core.Key]core.Value{
rpcServiceKey: core.String("serviceName"),
netPeerIPKey: core.String("fake"),
netPeerPortKey: core.String("connection"),
},
eventsAttr: []map[core.Key]core.Value{
{
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(req))),
},
{
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(reply))),
},
},
},
{
name: "invalidName",
expectedAttr: map[core.Key]core.Value{rpcServiceKey: core.String("")},
name: "invalidName",
expectedAttr: map[core.Key]core.Value{
rpcServiceKey: core.String(""),
netPeerIPKey: core.String("fake"),
netPeerPortKey: core.String("connection"),
},
eventsAttr: []map[core.Key]core.Value{
{
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(req))),
},
{
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(reply))),
},
},
},
{
name: "/github.com.foo.serviceName_123/method",
expectedAttr: map[core.Key]core.Value{rpcServiceKey: core.String("serviceName_123")},
name: "/github.com.foo.serviceName_123/method",
expectedAttr: map[core.Key]core.Value{
rpcServiceKey: core.String("serviceName_123"),
netPeerIPKey: core.String("fake"),
netPeerPortKey: core.String("connection"),
},
eventsAttr: []map[core.Key]core.Value{
{
messageTypeKey: core.String("SENT"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(req))),
},
{
messageTypeKey: core.String("RECEIVED"),
messageIDKey: core.Int(1),
messageUncompressedSizeKey: core.Int(proto.Size(proto.Message(reply))),
},
},
},
}
@ -158,6 +213,8 @@ func TestUnaryClientInterceptor(t *testing.T) {
expectedAttr.AsString(), attr.Value.AsString())
}
delete(check.expectedAttr, attr.Key)
} else {
t.Errorf("attribute %s not found in expected attributes map", string(attr.Key))
}
}
@ -182,6 +239,8 @@ func TestUnaryClientInterceptor(t *testing.T) {
string(attr.Key), attr.Value.AsString(), expectedAttr.AsString())
}
delete(check.eventsAttr[event], attr.Key)
} else {
t.Errorf("attribute in event %s not found in expected attributes map", string(attr.Key))
}
}
if len(check.eventsAttr[event]) > 0 {