@@ -2,6 +2,9 @@ package opentelemetry
22
33import (
44 "context"
5+ "go.opentelemetry.io/collector/pdata/pcommon"
6+ "go.opentelemetry.io/collector/pdata/pmetric"
7+ "go.opentelemetry.io/collector/pdata/pmetric/pmetricotlp"
58 "google.golang.org/grpc/credentials/insecure"
69 "net"
710 "strings"
@@ -15,15 +18,12 @@ import (
1518 "github.com/influxdata/telegraf/testutil"
1619 "github.com/stretchr/testify/assert"
1720 "github.com/stretchr/testify/require"
18- "go.opentelemetry.io/collector/model/otlp"
19- "go.opentelemetry.io/collector/model/otlpgrpc"
20- "go.opentelemetry.io/collector/model/pdata"
2121 "google.golang.org/grpc"
2222 "google.golang.org/grpc/metadata"
2323)
2424
2525func TestOpenTelemetry (t * testing.T ) {
26- expect := pdata .NewMetrics ()
26+ expect := pmetric .NewMetrics ()
2727 {
2828 rm := expect .ResourceMetrics ().AppendEmpty ()
2929 rm .Resource ().Attributes ().InsertString ("host.name" , "potato" )
@@ -32,10 +32,10 @@ func TestOpenTelemetry(t *testing.T) {
3232 ilm .Scope ().SetName ("My Library Name" )
3333 m := ilm .Metrics ().AppendEmpty ()
3434 m .SetName ("cpu_temp" )
35- m .SetDataType (pdata .MetricDataTypeGauge )
35+ m .SetDataType (pmetric .MetricDataTypeGauge )
3636 dp := m .Gauge ().DataPoints ().AppendEmpty ()
3737 dp .Attributes ().InsertString ("foo" , "bar" )
38- dp .SetTimestamp (pdata .Timestamp (1622848686000000000 ))
38+ dp .SetTimestamp (pcommon .Timestamp (1622848686000000000 ))
3939 dp .SetDoubleVal (87.332 )
4040 }
4141 m := newMockOtelService (t )
@@ -50,7 +50,7 @@ func TestOpenTelemetry(t *testing.T) {
5050 Attributes : map [string ]string {"attr-key" : "attr-val" },
5151 metricsConverter : metricsConverter ,
5252 grpcClientConn : m .GrpcClient (),
53- metricsServiceClient : otlpgrpc . NewMetricsClient (m .GrpcClient ()),
53+ metricsServiceClient : pmetricotlp . NewClient (m .GrpcClient ()),
5454 }
5555
5656 input := testutil .MustMetric (
@@ -76,24 +76,24 @@ func TestOpenTelemetry(t *testing.T) {
7676
7777 got := m .GotMetrics ()
7878
79- expectJSON , err := otlp . NewJSONMetricsMarshaler ().MarshalMetrics (expect )
79+ expectJSON , err := pmetric . NewJSONMarshaler ().MarshalMetrics (expect )
8080 require .NoError (t , err )
8181
82- gotJSON , err := otlp . NewJSONMetricsMarshaler ().MarshalMetrics (got )
82+ gotJSON , err := pmetric . NewJSONMarshaler ().MarshalMetrics (got )
8383 require .NoError (t , err )
8484
8585 assert .JSONEq (t , string (expectJSON ), string (gotJSON ))
8686}
8787
88- var _ otlpgrpc. MetricsServer = (* mockOtelService )(nil )
88+ var _ pmetricotlp. Server = (* mockOtelService )(nil )
8989
9090type mockOtelService struct {
9191 t * testing.T
9292 listener net.Listener
9393 grpcServer * grpc.Server
9494 grpcClient * grpc.ClientConn
9595
96- metrics pdata .Metrics
96+ metrics pmetric .Metrics
9797}
9898
9999func newMockOtelService (t * testing.T ) * mockOtelService {
@@ -107,7 +107,7 @@ func newMockOtelService(t *testing.T) *mockOtelService {
107107 grpcServer : grpcServer ,
108108 }
109109
110- otlpgrpc . RegisterMetricsServer (grpcServer , mockOtelService )
110+ pmetricotlp . RegisterServer (grpcServer , mockOtelService )
111111 go func () { assert .NoError (t , grpcServer .Serve (listener )) }()
112112
113113 grpcClient , err := grpc .Dial (listener .Addr ().String (), grpc .WithTransportCredentials (insecure .NewCredentials ()), grpc .WithBlock ())
@@ -126,18 +126,18 @@ func (m *mockOtelService) GrpcClient() *grpc.ClientConn {
126126 return m .grpcClient
127127}
128128
129- func (m * mockOtelService ) GotMetrics () pdata .Metrics {
129+ func (m * mockOtelService ) GotMetrics () pmetric .Metrics {
130130 return m .metrics
131131}
132132
133133func (m * mockOtelService ) Address () string {
134134 return m .listener .Addr ().String ()
135135}
136136
137- func (m * mockOtelService ) Export (ctx context.Context , request otlpgrpc. MetricsRequest ) (otlpgrpc. MetricsResponse , error ) {
137+ func (m * mockOtelService ) Export (ctx context.Context , request pmetricotlp. Request ) (pmetricotlp. Response , error ) {
138138 m .metrics = request .Metrics ().Clone ()
139139 ctxMetadata , ok := metadata .FromIncomingContext (ctx )
140140 assert .Equal (m .t , []string {"header1" }, ctxMetadata .Get ("test" ))
141141 assert .True (m .t , ok )
142- return otlpgrpc. MetricsResponse {}, nil
142+ return pmetricotlp. Response {}, nil
143143}
0 commit comments