659 lines
20 KiB
Go
659 lines
20 KiB
Go
package zipkin
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
)
|
|
|
|
func TestZipkinPlugin(t *testing.T) {
|
|
mockAcc := testutil.Accumulator{}
|
|
|
|
tests := []struct {
|
|
name string
|
|
datafile string // data file which contains test data
|
|
contentType string
|
|
wantErr bool
|
|
want []testutil.Metric
|
|
}{
|
|
{
|
|
name: "threespan",
|
|
datafile: "testdata/threespans.dat",
|
|
contentType: "application/x-thrift",
|
|
want: []testutil.Metric{
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "7047c59776af8a1b",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"service_name": "trivial",
|
|
"name": "child",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(53106) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851331000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "7047c59776af8a1b",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"name": "child",
|
|
"service_name": "trivial",
|
|
"annotation": "trivial", //base64: dHJpdmlhbA==
|
|
"endpoint_host": "127.0.0.1",
|
|
"annotation_key": "lc",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(53106) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851331000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "17020eb55a8bfe5",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"service_name": "trivial",
|
|
"name": "child",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(50410) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360904552000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "17020eb55a8bfe5",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"name": "child",
|
|
"service_name": "trivial",
|
|
"annotation": "trivial", //base64: dHJpdmlhbA==
|
|
"endpoint_host": "127.0.0.1",
|
|
"annotation_key": "lc",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(50410) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360904552000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "5195e96239641e",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"service_name": "trivial",
|
|
"name": "parent",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(103680) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851318000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"service_name": "trivial",
|
|
"annotation": "Starting child #0",
|
|
"endpoint_host": "127.0.0.1",
|
|
"id": "5195e96239641e",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"name": "parent",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(103680) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851318000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"service_name": "trivial",
|
|
"annotation": "Starting child #1",
|
|
"endpoint_host": "127.0.0.1",
|
|
"id": "5195e96239641e",
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"name": "parent",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(103680) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851318000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"parent_id": "5195e96239641e",
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"name": "parent",
|
|
"service_name": "trivial",
|
|
"annotation": "A Log",
|
|
"endpoint_host": "127.0.0.1",
|
|
"id": "5195e96239641e",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(103680) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851318000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"trace_id": "22c4fc8ab3669045",
|
|
"service_name": "trivial",
|
|
"annotation": "trivial", //base64: dHJpdmlhbA==
|
|
"annotation_key": "lc",
|
|
"id": "5195e96239641e",
|
|
"parent_id": "5195e96239641e",
|
|
"name": "parent",
|
|
"endpoint_host": "127.0.0.1",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(103680) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1498688360851318000).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "distributed_trace_sample",
|
|
datafile: "testdata/distributed_trace_sample.dat",
|
|
contentType: "application/x-thrift",
|
|
want: []testutil.Metric{
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "5e682bc21ce99c80",
|
|
"parent_id": "5e682bc21ce99c80",
|
|
"trace_id": "5e682bc21ce99c80",
|
|
"service_name": "go-zipkin-testclient",
|
|
"name": "main.dud",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(1) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1433330263415871*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "cs",
|
|
"endpoint_host": "0.0.0.0:9410",
|
|
"id": "5e682bc21ce99c80",
|
|
"parent_id": "5e682bc21ce99c80",
|
|
"trace_id": "5e682bc21ce99c80",
|
|
"name": "main.dud",
|
|
"service_name": "go-zipkin-testclient",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(1) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1433330263415871*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "cr",
|
|
"endpoint_host": "0.0.0.0:9410",
|
|
"id": "5e682bc21ce99c80",
|
|
"parent_id": "5e682bc21ce99c80",
|
|
"trace_id": "5e682bc21ce99c80",
|
|
"name": "main.dud",
|
|
"service_name": "go-zipkin-testclient",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": (time.Duration(1) * time.Microsecond).Nanoseconds(),
|
|
},
|
|
Time: time.Unix(0, 1433330263415871*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "JSON rather than thrift",
|
|
datafile: "testdata/json/brave-tracer-example.json",
|
|
contentType: "application/json",
|
|
want: []testutil.Metric{
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "sr",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "ss",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "Demo2Application",
|
|
"annotation_key": "mvc.controller.class",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "hi2",
|
|
"annotation_key": "mvc.controller.method",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "192.168.0.8:test:8010",
|
|
"annotation_key": "spring.instance_id",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(3000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538791000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "cs",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "cr",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "localhost",
|
|
"annotation_key": "http.host",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "GET",
|
|
"annotation_key": "http.method",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "/hi2",
|
|
"annotation_key": "http.path",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "http://localhost:8010/hi2",
|
|
"annotation_key": "http.url",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "192.168.0.8:test:8010",
|
|
"annotation_key": "spring.instance_id",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "b26412d1ac16767d",
|
|
"name": "http:/hi2",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(10000000),
|
|
},
|
|
Time: time.Unix(0, 1503031538786000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "sr",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "ss",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "Demo2Application",
|
|
"annotation_key": "mvc.controller.class",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "hi",
|
|
"annotation_key": "mvc.controller.method",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
{
|
|
Measurement: "zipkin",
|
|
Tags: map[string]string{
|
|
"annotation": "192.168.0.8:test:8010",
|
|
"annotation_key": "spring.instance_id",
|
|
"endpoint_host": "192.168.0.8:8010",
|
|
"id": "7312f822d43d0fd8",
|
|
"name": "http:/hi",
|
|
"parent_id": "7312f822d43d0fd8",
|
|
"service_name": "test",
|
|
"trace_id": "7312f822d43d0fd8",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"duration_ns": int64(23393000),
|
|
},
|
|
Time: time.Unix(0, 1503031538778000*int64(time.Microsecond)).UTC(),
|
|
Type: telegraf.Untyped,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
// Workaround for Go 1.8
|
|
// https://github.com/golang/go/issues/18806
|
|
DefaultNetwork = "tcp4"
|
|
|
|
z := &Zipkin{
|
|
Log: testutil.Logger{},
|
|
Path: "/api/v1/spans",
|
|
Port: 0,
|
|
}
|
|
|
|
err := z.Start(&mockAcc)
|
|
if err != nil {
|
|
t.Fatal("Failed to start zipkin server")
|
|
}
|
|
|
|
defer z.Stop()
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
mockAcc.ClearMetrics()
|
|
if err := postThriftData(tt.datafile, z.address, tt.contentType); err != nil {
|
|
t.Fatalf("Posting data to http endpoint /api/v1/spans failed. Error: %s\n", err)
|
|
}
|
|
mockAcc.Wait(len(tt.want)) //Since the server is running concurrently, we need to wait for the number of data points we want to test to be added to the Accumulator.
|
|
if len(mockAcc.Errors) > 0 != tt.wantErr {
|
|
t.Fatalf("Got unexpected errors. want error = %v, errors = %v\n", tt.wantErr, mockAcc.Errors)
|
|
}
|
|
|
|
var got []testutil.Metric
|
|
for _, m := range mockAcc.Metrics {
|
|
got = append(got, *m)
|
|
}
|
|
if !cmp.Equal(tt.want, got) {
|
|
t.Fatalf("Got != Want\n %s", cmp.Diff(tt.want, got))
|
|
}
|
|
})
|
|
}
|
|
mockAcc.ClearMetrics()
|
|
z.Stop()
|
|
// Make sure there is no erroneous error on shutdown
|
|
if len(mockAcc.Errors) != 0 {
|
|
t.Fatal("Expected no errors on shutdown")
|
|
}
|
|
}
|
|
|
|
func postThriftData(datafile, address, contentType string) error {
|
|
dat, err := ioutil.ReadFile(datafile)
|
|
if err != nil {
|
|
return fmt.Errorf("could not read from data file %s", datafile)
|
|
}
|
|
|
|
req, err := http.NewRequest("POST", fmt.Sprintf("http://%s/api/v1/spans", address), bytes.NewReader(dat))
|
|
if err != nil {
|
|
return fmt.Errorf("HTTP request creation failed")
|
|
}
|
|
|
|
req.Header.Set("Content-Type", contentType)
|
|
client := &http.Client{}
|
|
_, err = client.Do(req)
|
|
if err != nil {
|
|
return fmt.Errorf("HTTP POST request to zipkin endpoint %s failed %v", address, err)
|
|
}
|
|
|
|
return nil
|
|
}
|