go/internal/telemetry/export/ocagent/trace_test.go

// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package ocagent_test

import (
	"context"
	"errors"
	"testing"
	"time"

	"golang.org/x/tools/internal/telemetry/event"
	"golang.org/x/tools/internal/telemetry/export"
)

func TestTrace(t *testing.T) {
	start, _ := time.Parse(time.RFC3339Nano, "1970-01-01T00:00:30Z")
	at, _ := time.Parse(time.RFC3339Nano, "1970-01-01T00:00:40Z")
	end, _ := time.Parse(time.RFC3339Nano, "1970-01-01T00:00:50Z")
	const prefix = testNodeStr + `
		"spans":[{
			"trace_id":"AAAAAAAAAAAAAAAAAAAAAA==",
			"span_id":"AAAAAAAAAAA=",
			"parent_span_id":"AAAAAAAAAAA=",
			"name":{"value":"event span"},
			"start_time":"1970-01-01T00:00:30Z",
			"end_time":"1970-01-01T00:00:50Z",
			"time_events":{
	`
	const suffix = `
			},
			"same_process_as_parent_span":true
		}]
	}`
	tests := []struct {
		name  string
		event func(ctx context.Context) event.Event
		want  string
	}{
		{
			name: "no tags",
			event: func(ctx context.Context) event.Event {
				return event.Event{
					At: at,
				}
			},
			want: prefix + `
						"timeEvent":[{"time":"1970-01-01T00:00:40Z"}]
			` + suffix,
		},
		{
			name: "description no error",
			event: func(ctx context.Context) event.Event {
				return event.Event{
					At:      at,
					Message: "cache miss",
					Tags: event.TagList{
						event.TagOf("db", "godb"),
					},
				}
			},
			want: prefix + `"timeEvent":[{"time":"1970-01-01T00:00:40Z","annotation":{
  "description": { "value": "cache miss" },
  "attributes": {
    "attributeMap": {
      "db": { "stringValue": { "value": "godb" } }
    }
  }
}}]` + suffix,
		},

		{
			name: "description and error",
			event: func(ctx context.Context) event.Event {
				return event.Event{
					At:      at,
					Message: "cache miss",
					Error:   errors.New("no network connectivity"),
					Tags: event.TagList{
						event.TagOf("db", "godb"), // must come before e
					},
				}
			},
			want: prefix + `"timeEvent":[{"time":"1970-01-01T00:00:40Z","annotation":{
  "description": { "value": "cache miss" },
  "attributes": {
    "attributeMap": {
      "db": { "stringValue": { "value": "godb" } },
      "error": { "stringValue": { "value": "no network connectivity" } }
    }
  }
	}}]` + suffix,
		},
		{
			name: "no description, but error",
			event: func(ctx context.Context) event.Event {
				return event.Event{
					At:    at,
					Error: errors.New("no network connectivity"),
					Tags: event.TagList{
						event.TagOf("db", "godb"),
					},
				}
			},
			want: prefix + `"timeEvent":[{"time":"1970-01-01T00:00:40Z","annotation":{
  "description": { "value": "no network connectivity" },
  "attributes": {
    "attributeMap": {
      "db": { "stringValue": { "value": "godb" } }
    }
  }
	}}]` + suffix,
		},
		{
			name: "enumerate all attribute types",
			event: func(ctx context.Context) event.Event {
				return event.Event{
					At:      at,
					Message: "cache miss",
					Tags: event.TagList{
						event.TagOf("1_db", "godb"),

						event.TagOf("2a_age", 0.456), // Constant converted into "float64"
						event.TagOf("2b_ttl", float32(5000)),
						event.TagOf("2c_expiry_ms", float64(1e3)),

						event.TagOf("3a_retry", false),
						event.TagOf("3b_stale", true),

						event.TagOf("4a_max", 0x7fff), // Constant converted into "int"
						event.TagOf("4b_opcode", int8(0x7e)),
						event.TagOf("4c_base", int16(1<<9)),
						event.TagOf("4e_checksum", int32(0x11f7e294)),
						event.TagOf("4f_mode", int64(0644)),

						event.TagOf("5a_min", uint(1)),
						event.TagOf("5b_mix", uint8(44)),
						event.TagOf("5c_port", uint16(55678)),
						event.TagOf("5d_min_hops", uint32(1<<9)),
						event.TagOf("5e_max_hops", uint64(0xffffff)),
					},
				}
			},
			want: prefix + `"timeEvent":[{"time":"1970-01-01T00:00:40Z","annotation":{
  "description": { "value": "cache miss" },
  "attributes": {
    "attributeMap": {
      "1_db": { "stringValue": { "value": "godb" } },
      "2a_age": { "doubleValue": 0.456 },
      "2b_ttl": { "doubleValue": 5000 },
      "2c_expiry_ms": { "doubleValue": 1000 },
      "3a_retry": {},
			"3b_stale": { "boolValue": true },
      "4a_max": { "intValue": 32767 },
      "4b_opcode": { "intValue": 126 },
      "4c_base": { "intValue": 512 },
      "4e_checksum": { "intValue": 301458068 },
      "4f_mode": { "intValue": 420 },
      "5a_min": { "intValue": 1 },
      "5b_mix": { "intValue": 44 },
      "5c_port": { "intValue": 55678 },
      "5d_min_hops": { "intValue": 512 },
      "5e_max_hops": { "intValue": 16777215 }
    }
  }
}}]` + suffix,
		},
	}
	ctx := context.TODO()
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			startEvent := event.Event{
				Type:    event.StartSpanType,
				Message: "event span",
				At:      start,
			}
			endEvent := event.Event{
				Type: event.EndSpanType,
				At:   end,
			}
			ctx := export.ContextSpan(ctx, startEvent)
			span := export.GetSpan(ctx)
			span.ID = export.SpanContext{}
			span.Events = []event.Event{tt.event(ctx)}
			exporter.ProcessEvent(ctx, startEvent)
			export.ContextSpan(ctx, endEvent)
			exporter.ProcessEvent(ctx, endEvent)
			exporter.Flush()
			got := sent.get("/v1/trace")
			checkJSON(t, got, []byte(tt.want))
		})
	}
}