136 lines
3.3 KiB
Go
136 lines
3.3 KiB
Go
package zipkin
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/influxdata/telegraf/plugins/inputs/zipkin/trace"
|
|
)
|
|
|
|
type MockRecorder struct {
|
|
Data trace.Trace
|
|
Err error
|
|
}
|
|
|
|
func (m *MockRecorder) Record(t trace.Trace) error {
|
|
m.Data = t
|
|
return nil
|
|
}
|
|
|
|
func (m *MockRecorder) Error(err error) {
|
|
m.Err = err
|
|
}
|
|
|
|
func TestSpanHandler(t *testing.T) {
|
|
dat, err := ioutil.ReadFile("testdata/threespans.dat")
|
|
if err != nil {
|
|
t.Fatalf("Could not find file %s\n", "testdata/threespans.dat")
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
r := httptest.NewRequest(
|
|
"POST",
|
|
"http://server.local/api/v1/spans",
|
|
ioutil.NopCloser(
|
|
bytes.NewReader(dat)))
|
|
|
|
r.Header.Set("Content-Type", "application/x-thrift")
|
|
handler := NewSpanHandler("/api/v1/spans")
|
|
mockRecorder := &MockRecorder{}
|
|
handler.recorder = mockRecorder
|
|
|
|
handler.Spans(w, r)
|
|
if w.Code != http.StatusNoContent {
|
|
t.Errorf("MainHandler did not return StatusNoContent %d", w.Code)
|
|
}
|
|
|
|
got := mockRecorder.Data
|
|
|
|
parentID := strconv.FormatInt(22964302721410078, 10)
|
|
want := trace.Trace{
|
|
{
|
|
Name: "Child",
|
|
ID: "8090652509916334619",
|
|
TraceID: "22c4fc8ab3669045",
|
|
ParentID: parentID,
|
|
Timestamp: time.Unix(0, 1498688360851331*int64(time.Microsecond)).UTC(),
|
|
Duration: time.Duration(53106) * time.Microsecond,
|
|
ServiceName: "trivial",
|
|
Annotations: []trace.Annotation{},
|
|
BinaryAnnotations: []trace.BinaryAnnotation{
|
|
{
|
|
Key: "lc",
|
|
Value: "trivial",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Child",
|
|
ID: "103618986556047333",
|
|
TraceID: "22c4fc8ab3669045",
|
|
ParentID: parentID,
|
|
Timestamp: time.Unix(0, 1498688360904552*int64(time.Microsecond)).UTC(),
|
|
Duration: time.Duration(50410) * time.Microsecond,
|
|
ServiceName: "trivial",
|
|
Annotations: []trace.Annotation{},
|
|
BinaryAnnotations: []trace.BinaryAnnotation{
|
|
{
|
|
Key: "lc",
|
|
Value: "trivial",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Parent",
|
|
ID: "22964302721410078",
|
|
TraceID: "22c4fc8ab3669045",
|
|
ParentID: "22964302721410078",
|
|
Timestamp: time.Unix(0, 1498688360851318*int64(time.Microsecond)).UTC(),
|
|
Duration: time.Duration(103680) * time.Microsecond,
|
|
ServiceName: "trivial",
|
|
Annotations: []trace.Annotation{
|
|
{
|
|
Timestamp: time.Unix(0, 1498688360851325*int64(time.Microsecond)).UTC(),
|
|
Value: "Starting child #0",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
{
|
|
Timestamp: time.Unix(0, 1498688360904545*int64(time.Microsecond)).UTC(),
|
|
Value: "Starting child #1",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
{
|
|
Timestamp: time.Unix(0, 1498688360954992*int64(time.Microsecond)).UTC(),
|
|
Value: "A Log",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
},
|
|
BinaryAnnotations: []trace.BinaryAnnotation{
|
|
{
|
|
Key: "lc",
|
|
Value: "trivial",
|
|
Host: "127.0.0.1",
|
|
ServiceName: "trivial",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
if !cmp.Equal(got, want) {
|
|
t.Fatalf("Got != Want\n %s", cmp.Diff(got, want))
|
|
}
|
|
}
|