// 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: core.SpanContext{
						TraceID:    core.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}),
						SpanID:     core.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: core.SpanContext{
						TraceID:    core.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}),
						SpanID:     core.SpanID([8]byte{0, 0, 0, 0, 0, 0, 0, 2}),
						TraceFlags: byte(1),
					},
					ParentSpanID: core.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: core.SpanContext{
						TraceID:    core.TraceID([16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2}),
						SpanID:     core.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())
	}
}