github.com/cilium/cilium@v1.16.2/pkg/hubble/parser/seven/tracing_test.go

// SPDX-License-Identifier: Apache-2.0
// Copyright Authors of Hubble

package seven

import (
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"

	flowpb "github.com/cilium/cilium/api/v1/flow"
	"github.com/cilium/cilium/pkg/proxy/accesslog"
)

func TestExtractTraceContext(t *testing.T) {
	tests := []struct {
		name   string
		record *accesslog.LogRecord
		want   *flowpb.TraceContext
	}{
		{
			name:   "nil log record",
			record: nil,
			want:   nil,
		}, {
			name: "http log record without trace",
			record: &accesslog.LogRecord{
				HTTP: &accesslog.LogRecordHTTP{},
			},
			want: nil,
		}, {
			name: "http log record with trace",
			record: &accesslog.LogRecord{
				HTTP: &accesslog.LogRecordHTTP{
					Headers: http.Header{
						"Traceparent": []string{"00-4bf92f3577b34da6a3ce929d0e0e4736-00f067aa0ba902b7-01"},
					},
				},
			},
			want: &flowpb.TraceContext{
				Parent: &flowpb.TraceParent{
					TraceId: "4bf92f3577b34da6a3ce929d0e0e4736",
				},
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := extractTraceContext(tt.record)
			assert.Equal(t, tt.want, got)
		})
	}
}

func TestTraceIDFromHTTPHeader(t *testing.T) {
	tests := []struct {
		name   string
		header http.Header
		want   string
	}{
		{
			name: "no trace",
			want: "",
		}, {
			name: "example traceparent",
			header: http.Header{
				"Traceparent": []string{"00-4bf92f3577b34da6a3ce929d0e0e4736-00f067aa0ba902b7-01"},
			},
			want: "4bf92f3577b34da6a3ce929d0e0e4736",
		}, {
			name: "invalid trace",
			header: http.Header{
				"Traceparent": []string{"not-an-actual-traceparent"},
			},
			want: "",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := traceIDFromHTTPHeader(tt.header)
			assert.Equal(t, tt.want, got)
		})
	}
}