Add JSON input support to zipkin plugin (#3150)

plugins/inputs/zipkin/codec/thrift/thrift_test.go

@@ -0,0 +1,211 @@
+package thrift
+
+import (
+	"io/ioutil"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+
+	"github.com/openzipkin/zipkin-go-opentracing/_thrift/gen-go/zipkincore"
+)
+
+func Test_endpointHost(t *testing.T) {
+	type args struct {
+		h *zipkincore.Endpoint
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "Host Found",
+			args: args{
+				h: &zipkincore.Endpoint{
+					Ipv4: 1234,
+					Port: 8888,
+				},
+			},
+			want: "0.0.4.210:8888",
+		},
+		{
+			name: "No Host",
+			args: args{
+				h: nil,
+			},
+			want: "0.0.0.0",
+		},
+		{
+			name: "int overflow zipkin uses an int16 type as an unsigned int 16.",
+			args: args{
+				h: &zipkincore.Endpoint{
+					Ipv4: 1234,
+					Port: -1,
+				},
+			},
+			want: "0.0.4.210:65535",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			e := endpoint{tt.args.h}
+			if got := e.Host(); got != tt.want {
+				t.Errorf("host() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func Test_endpointName(t *testing.T) {
+	type args struct {
+		h *zipkincore.Endpoint
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "Found ServiceName",
+			args: args{
+				h: &zipkincore.Endpoint{
+					ServiceName: "zipkin",
+				},
+			},
+			want: "zipkin",
+		},
+		{
+			name: "No ServiceName",
+			args: args{
+				h: nil,
+			},
+			want: "unknown",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			e := endpoint{tt.args.h}
+			if got := e.Name(); got != tt.want {
+				t.Errorf("serviceName() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestUnmarshalThrift(t *testing.T) {
+	addr := func(i int64) *int64 { return &i }
+	tests := []struct {
+		name     string
+		filename string
+		want     []*zipkincore.Span
+		wantErr  bool
+	}{
+		{
+			name:     "threespans",
+			filename: "../../testdata/threespans.dat",
+			want: []*zipkincore.Span{
+				{
+					TraceID:     2505404965370368069,
+					Name:        "Child",
+					ID:          8090652509916334619,
+					ParentID:    addr(22964302721410078),
+					Timestamp:   addr(1498688360851331),
+					Duration:    addr(53106),
+					Annotations: []*zipkincore.Annotation{},
+					BinaryAnnotations: []*zipkincore.BinaryAnnotation{
+						&zipkincore.BinaryAnnotation{
+							Key:            "lc",
+							AnnotationType: zipkincore.AnnotationType_STRING,
+							Value:          []byte("trivial"),
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+					},
+				},
+				{
+					TraceID:     2505404965370368069,
+					Name:        "Child",
+					ID:          103618986556047333,
+					ParentID:    addr(22964302721410078),
+					Timestamp:   addr(1498688360904552),
+					Duration:    addr(50410),
+					Annotations: []*zipkincore.Annotation{},
+					BinaryAnnotations: []*zipkincore.BinaryAnnotation{
+						&zipkincore.BinaryAnnotation{
+							Key:            "lc",
+							AnnotationType: zipkincore.AnnotationType_STRING,
+							Value:          []byte("trivial"),
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+					},
+				},
+				{
+					TraceID:   2505404965370368069,
+					Name:      "Parent",
+					ID:        22964302721410078,
+					Timestamp: addr(1498688360851318),
+					Duration:  addr(103680),
+					Annotations: []*zipkincore.Annotation{
+						&zipkincore.Annotation{
+							Timestamp: 1498688360851325,
+							Value:     "Starting child #0",
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+						&zipkincore.Annotation{
+							Timestamp: 1498688360904545,
+							Value:     "Starting child #1",
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+						&zipkincore.Annotation{
+							Timestamp: 1498688360954992,
+							Value:     "A Log",
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+					},
+					BinaryAnnotations: []*zipkincore.BinaryAnnotation{
+						&zipkincore.BinaryAnnotation{
+							Key:            "lc",
+							AnnotationType: zipkincore.AnnotationType_STRING,
+							Value:          []byte("trivial"),
+							Host: &zipkincore.Endpoint{
+								Ipv4:        2130706433,
+								ServiceName: "trivial",
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			dat, err := ioutil.ReadFile(tt.filename)
+			if err != nil {
+				t.Fatalf("Could not find file %s\n", tt.filename)
+			}
+
+			got, err := UnmarshalThrift(dat)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("UnmarshalThrift() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !cmp.Equal(tt.want, got) {
+				t.Errorf("UnmarshalThrift() got(-)/want(+): %s", cmp.Diff(tt.want, got))
+			}
+		})
+	}
+}