mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-12 10:04:29 +02:00
255 lines
7.8 KiB
Go
255 lines
7.8 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package otlp
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
coltracepb "github.com/open-telemetry/opentelemetry-proto/gen/go/collector/trace/v1"
|
|
commonpb "github.com/open-telemetry/opentelemetry-proto/gen/go/common/v1"
|
|
resourcepb "github.com/open-telemetry/opentelemetry-proto/gen/go/resource/v1"
|
|
tracepb "github.com/open-telemetry/opentelemetry-proto/gen/go/trace/v1"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
|
|
"go.opentelemetry.io/otel/api/core"
|
|
"go.opentelemetry.io/otel/api/key"
|
|
apitrace "go.opentelemetry.io/otel/api/trace"
|
|
tracesdk "go.opentelemetry.io/otel/sdk/export/trace"
|
|
"go.opentelemetry.io/otel/sdk/resource"
|
|
)
|
|
|
|
type traceServiceClientStub struct {
|
|
rs []tracepb.ResourceSpans
|
|
}
|
|
|
|
func (t *traceServiceClientStub) Export(ctx context.Context, in *coltracepb.ExportTraceServiceRequest, opts ...grpc.CallOption) (*coltracepb.ExportTraceServiceResponse, error) {
|
|
for _, rs := range in.GetResourceSpans() {
|
|
if rs == nil {
|
|
continue
|
|
}
|
|
t.rs = append(t.rs, *rs)
|
|
}
|
|
return &coltracepb.ExportTraceServiceResponse{}, nil
|
|
}
|
|
|
|
func (t *traceServiceClientStub) ResourceSpans() []tracepb.ResourceSpans {
|
|
return t.rs
|
|
}
|
|
|
|
func (t *traceServiceClientStub) Reset() {
|
|
t.rs = nil
|
|
}
|
|
|
|
func TestExportSpans(t *testing.T) {
|
|
tsc := &traceServiceClientStub{}
|
|
exp := NewUnstartedExporter()
|
|
exp.traceExporter = tsc
|
|
exp.started = true
|
|
|
|
// March 31, 2020 5:01:26 1234nanos (UTC)
|
|
startTime := time.Unix(1585674086, 1234)
|
|
endTime := startTime.Add(10 * time.Second)
|
|
|
|
for _, test := range []struct {
|
|
sd []*tracesdk.SpanData
|
|
want []tracepb.ResourceSpans
|
|
}{
|
|
{
|
|
[]*tracesdk.SpanData(nil),
|
|
[]tracepb.ResourceSpans(nil),
|
|
},
|
|
{
|
|
[]*tracesdk.SpanData{},
|
|
[]tracepb.ResourceSpans(nil),
|
|
},
|
|
{
|
|
[]*tracesdk.SpanData{
|
|
{
|
|
SpanContext: apitrace.SpanContext{
|
|
TraceID: apitrace.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}),
|
|
SpanID: apitrace.SpanID([8]byte{0, 0, 0, 0, 0, 0, 0, 1}),
|
|
TraceFlags: byte(1),
|
|
},
|
|
SpanKind: apitrace.SpanKindServer,
|
|
Name: "parent process",
|
|
StartTime: startTime,
|
|
EndTime: endTime,
|
|
Attributes: []core.KeyValue{
|
|
key.String("user", "alice"),
|
|
key.Bool("authenticated", true),
|
|
},
|
|
StatusCode: codes.OK,
|
|
StatusMessage: "Ok",
|
|
Resource: resource.New(key.String("instance", "tester-a")),
|
|
},
|
|
{
|
|
SpanContext: apitrace.SpanContext{
|
|
TraceID: apitrace.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}),
|
|
SpanID: apitrace.SpanID([8]byte{0, 0, 0, 0, 0, 0, 0, 2}),
|
|
TraceFlags: byte(1),
|
|
},
|
|
ParentSpanID: apitrace.SpanID([8]byte{0, 0, 0, 0, 0, 0, 0, 1}),
|
|
SpanKind: apitrace.SpanKindInternal,
|
|
Name: "internal process",
|
|
StartTime: startTime,
|
|
EndTime: endTime,
|
|
Attributes: []core.KeyValue{
|
|
key.String("user", "alice"),
|
|
key.Bool("authenticated", true),
|
|
},
|
|
StatusCode: codes.OK,
|
|
StatusMessage: "Ok",
|
|
Resource: resource.New(key.String("instance", "tester-a")),
|
|
},
|
|
{
|
|
SpanContext: apitrace.SpanContext{
|
|
TraceID: apitrace.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2}),
|
|
SpanID: apitrace.SpanID([8]byte{0, 0, 0, 0, 0, 0, 0, 1}),
|
|
TraceFlags: byte(1),
|
|
},
|
|
SpanKind: apitrace.SpanKindServer,
|
|
Name: "parent process",
|
|
StartTime: startTime,
|
|
EndTime: endTime,
|
|
Attributes: []core.KeyValue{
|
|
key.String("user", "bob"),
|
|
key.Bool("authenticated", false),
|
|
},
|
|
StatusCode: codes.Unauthenticated,
|
|
StatusMessage: "Unauthenticated",
|
|
Resource: resource.New(key.String("instance", "tester-b")),
|
|
},
|
|
},
|
|
[]tracepb.ResourceSpans{
|
|
{
|
|
Resource: &resourcepb.Resource{
|
|
Attributes: []*commonpb.AttributeKeyValue{
|
|
{
|
|
Key: "instance",
|
|
Type: commonpb.AttributeKeyValue_STRING,
|
|
StringValue: "tester-a",
|
|
},
|
|
},
|
|
},
|
|
InstrumentationLibrarySpans: []*tracepb.InstrumentationLibrarySpans{
|
|
{
|
|
Spans: []*tracepb.Span{
|
|
{
|
|
TraceId: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},
|
|
SpanId: []byte{0, 0, 0, 0, 0, 0, 0, 1},
|
|
Name: "parent process",
|
|
Kind: tracepb.Span_SERVER,
|
|
StartTimeUnixNano: uint64(startTime.UnixNano()),
|
|
EndTimeUnixNano: uint64(endTime.UnixNano()),
|
|
Attributes: []*commonpb.AttributeKeyValue{
|
|
{
|
|
Key: "user",
|
|
Type: commonpb.AttributeKeyValue_STRING,
|
|
StringValue: "alice",
|
|
},
|
|
{
|
|
Key: "authenticated",
|
|
Type: commonpb.AttributeKeyValue_BOOL,
|
|
BoolValue: true,
|
|
},
|
|
},
|
|
Status: &tracepb.Status{
|
|
Code: tracepb.Status_Ok,
|
|
Message: "Ok",
|
|
},
|
|
},
|
|
{
|
|
TraceId: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},
|
|
SpanId: []byte{0, 0, 0, 0, 0, 0, 0, 2},
|
|
ParentSpanId: []byte{0, 0, 0, 0, 0, 0, 0, 1},
|
|
Name: "internal process",
|
|
Kind: tracepb.Span_INTERNAL,
|
|
StartTimeUnixNano: uint64(startTime.UnixNano()),
|
|
EndTimeUnixNano: uint64(endTime.UnixNano()),
|
|
Attributes: []*commonpb.AttributeKeyValue{
|
|
{
|
|
Key: "user",
|
|
Type: commonpb.AttributeKeyValue_STRING,
|
|
StringValue: "alice",
|
|
},
|
|
{
|
|
Key: "authenticated",
|
|
Type: commonpb.AttributeKeyValue_BOOL,
|
|
BoolValue: true,
|
|
},
|
|
},
|
|
Status: &tracepb.Status{
|
|
Code: tracepb.Status_Ok,
|
|
Message: "Ok",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Resource: &resourcepb.Resource{
|
|
Attributes: []*commonpb.AttributeKeyValue{
|
|
{
|
|
Key: "instance",
|
|
Type: commonpb.AttributeKeyValue_STRING,
|
|
StringValue: "tester-b",
|
|
},
|
|
},
|
|
},
|
|
InstrumentationLibrarySpans: []*tracepb.InstrumentationLibrarySpans{
|
|
{
|
|
Spans: []*tracepb.Span{
|
|
{
|
|
TraceId: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2},
|
|
SpanId: []byte{0, 0, 0, 0, 0, 0, 0, 1},
|
|
Name: "parent process",
|
|
Kind: tracepb.Span_SERVER,
|
|
StartTimeUnixNano: uint64(startTime.UnixNano()),
|
|
EndTimeUnixNano: uint64(endTime.UnixNano()),
|
|
Attributes: []*commonpb.AttributeKeyValue{
|
|
{
|
|
Key: "user",
|
|
Type: commonpb.AttributeKeyValue_STRING,
|
|
StringValue: "bob",
|
|
},
|
|
{
|
|
Key: "authenticated",
|
|
Type: commonpb.AttributeKeyValue_BOOL,
|
|
BoolValue: false,
|
|
},
|
|
},
|
|
Status: &tracepb.Status{
|
|
Code: tracepb.Status_Unauthenticated,
|
|
Message: "Unauthenticated",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
} {
|
|
tsc.Reset()
|
|
exp.ExportSpans(context.Background(), test.sd)
|
|
assert.ElementsMatch(t, test.want, tsc.ResourceSpans())
|
|
}
|
|
}
|