package zipkin

import (
	"net"
	"testing"
	"time"

	"github.com/grafana/grafana-plugin-sdk-go/experimental"
	"github.com/openzipkin/zipkin-go/model"
)

func TestTransformResponse(t *testing.T) {
	t.Run("simple_trace", func(t *testing.T) {
		span1 := model.SpanModel{
			SpanContext: model.SpanContext{
				TraceID: model.TraceID{
					High: 123,
					Low:  456,
				},
				ID: 1,
			},
			Name:      "span 1",
			Kind:      "CLIENT",
			Timestamp: time.Unix(0, 1*int64(time.Microsecond)),
			Duration:  10 * time.Microsecond,
			LocalEndpoint: &model.Endpoint{
				ServiceName: "service 1",
				IPv4:        net.IPv4(1, 0, 0, 1),
				Port:        42,
			},
			Annotations: []model.Annotation{
				{Timestamp: time.Unix(0, 2*int64(time.Microsecond)), Value: "annotation text"},
				{Timestamp: time.Unix(0, 6*int64(time.Microsecond)), Value: "annotation text 3"},
			},
			Tags: map[string]string{
				"tag1": "val1",
				"tag2": "val2",
			},
		}
		span2 := model.SpanModel{
			SpanContext: model.SpanContext{
				TraceID: model.TraceID{
					High: 123,
					Low:  456,
				},
				ID:       2,
				ParentID: &span1.ID,
			},
			Name:      "span 2",
			Timestamp: time.Unix(0, 4*int64(time.Microsecond)),
			Duration:  5 * time.Microsecond,
			LocalEndpoint: &model.Endpoint{
				ServiceName: "service 2",
				IPv4:        net.IPv4(1, 0, 0, 1),
			},
			Tags: map[string]string{
				"error": "404",
			},
		}

		span3 := model.SpanModel{
			SpanContext: model.SpanContext{
				TraceID: model.TraceID{
					High: 123,
					Low:  456,
				},
				ID:       3,
				ParentID: &span1.ID,
			},
			Name:      "span 3",
			Timestamp: time.Unix(0, 6*int64(time.Microsecond)),
			Duration:  7 * time.Microsecond,
			RemoteEndpoint: &model.Endpoint{
				ServiceName: "spanstore-jdbc",
				IPv6:        net.ParseIP("::1"),
			},
		}

		spans := []model.SpanModel{span1, span2, span3}
		frames := transformResponse(spans, "test")
		experimental.CheckGoldenJSONFrame(t, "./testdata", "simple_trace.golden", frames, false)
	})
}