Upgrade to OpenTelemetry Semantic Conventions v1.26.0

pkg/tracing/tracing_test.go

@@ -0,0 +1,57 @@
+package tracing
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_safeFullURL(t *testing.T) {
+	testCases := []struct {
+		desc            string
+		safeQueryParams []string
+		originalURL     *url.URL
+		expectedURL     *url.URL
+	}{
+		{
+			desc:        "Nil URL",
+			originalURL: nil,
+			expectedURL: nil,
+		},
+		{
+			desc:        "No query parameters",
+			originalURL: &url.URL{Scheme: "https", Host: "example.com"},
+			expectedURL: &url.URL{Scheme: "https", Host: "example.com"},
+		},
+		{
+			desc:        "All query parameters redacted",
+			originalURL: &url.URL{Scheme: "https", Host: "example.com", RawQuery: "foo=bar&baz=qux"},
+			expectedURL: &url.URL{Scheme: "https", Host: "example.com", RawQuery: "baz=REDACTED&foo=REDACTED"},
+		},
+		{
+			desc:            "Some query parameters unredacted",
+			safeQueryParams: []string{"foo"},
+			originalURL:     &url.URL{Scheme: "https", Host: "example.com", RawQuery: "foo=bar&baz=qux"},
+			expectedURL:     &url.URL{Scheme: "https", Host: "example.com", RawQuery: "baz=REDACTED&foo=bar"},
+		},
+		{
+			desc:            "User info and some query parameters redacted",
+			safeQueryParams: []string{"foo"},
+			originalURL:     &url.URL{Scheme: "https", Host: "example.com", User: url.UserPassword("username", "password"), RawQuery: "foo=bar&baz=qux"},
+			expectedURL:     &url.URL{Scheme: "https", Host: "example.com", User: url.UserPassword("REDACTED", "REDACTED"), RawQuery: "baz=REDACTED&foo=bar"},
+		},
+	}
+
+	for _, test := range testCases {
+		t.Run(test.desc, func(t *testing.T) {
+			t.Parallel()
+
+			tr := NewTracer(nil, nil, nil, test.safeQueryParams)
+
+			gotURL := tr.safeURL(test.originalURL)
+
+			assert.Equal(t, test.expectedURL, gotURL)
+		})
+	}
+}