mirror of
https://github.com/golang/go
synced 2024-11-18 14:14:46 -07:00
internal/telemetry: split the ocagent tests from the support functions
This moves the actual trace test into its own file so that it can easily be seen separately from the functions that support all tests. Change-Id: I1d30cf8a712377ca79a5de77b85412c881177f43 Reviewed-on: https://go-review.googlesource.com/c/tools/+/223397 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Emmanuel Odeke <emm.odeke@gmail.com>
This commit is contained in:
Ian Cottrell
parent
3e76bee198
commit
fbeba2149c
@ -6,9 +6,7 @@ package ocagent_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
@ -16,30 +14,9 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"golang.org/x/tools/internal/telemetry/event"
|
||||
"golang.org/x/tools/internal/telemetry/export"
|
||||
"golang.org/x/tools/internal/telemetry/export/ocagent"
|
||||
)
|
||||
|
||||
var (
|
||||
exporter *ocagent.Exporter
|
||||
sent fakeSender
|
||||
start time.Time
|
||||
at time.Time
|
||||
end time.Time
|
||||
)
|
||||
|
||||
func init() {
|
||||
cfg := ocagent.Config{
|
||||
Host: "tester",
|
||||
Process: 1,
|
||||
Service: "ocagent-tests",
|
||||
Client: &http.Client{Transport: &sent},
|
||||
}
|
||||
cfg.Start, _ = time.Parse(time.RFC3339Nano, "1970-01-01T00:00:00Z")
|
||||
exporter = ocagent.Connect(&cfg)
|
||||
}
|
||||
|
||||
const testNodeStr = `{
|
||||
"node":{
|
||||
"identifier":{
|
||||
@ -57,184 +34,23 @@ const testNodeStr = `{
|
||||
}
|
||||
},`
|
||||
|
||||
func TestEvents(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,
|
||||
},
|
||||
var (
|
||||
exporter *ocagent.Exporter
|
||||
sent fakeSender
|
||||
start time.Time
|
||||
at time.Time
|
||||
end time.Time
|
||||
)
|
||||
|
||||
{
|
||||
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,
|
||||
},
|
||||
func init() {
|
||||
cfg := ocagent.Config{
|
||||
Host: "tester",
|
||||
Process: 1,
|
||||
Service: "ocagent-tests",
|
||||
Client: &http.Client{Transport: &sent},
|
||||
}
|
||||
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))
|
||||
})
|
||||
}
|
||||
|
||||
cfg.Start, _ = time.Parse(time.RFC3339Nano, "1970-01-01T00:00:00Z")
|
||||
exporter = ocagent.Connect(&cfg)
|
||||
}
|
||||
|
||||
func checkJSON(t *testing.T, got, want []byte) {
|
||||
|
||||
194
internal/telemetry/export/ocagent/trace_test.go
Normal file
194
internal/telemetry/export/ocagent/trace_test.go
Normal file
@ -0,0 +1,194 @@
|
||||
// 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))
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user