1 files changed, 180 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/pkg/web/request_test.go b/src/go/collectors/go.d.plugin/pkg/web/request_test.go
new file mode 100644
index 000000000..284cccb93
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/pkg/web/request_test.go
@@ -0,0 +1,180 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package web
+
+import (
+	"encoding/base64"
+	"net/http"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRequest_Copy(t *testing.T) {
+	tests := map[string]struct {
+		orig   Request
+		change func(req *Request)
+	}{
+		"change headers": {
+			orig: Request{
+				URL:    "http://127.0.0.1:19999/api/v1/info",
+				Method: "POST",
+				Headers: map[string]string{
+					"X-Api-Key": "secret",
+				},
+				Username:      "username",
+				Password:      "password",
+				ProxyUsername: "proxy_username",
+				ProxyPassword: "proxy_password",
+			},
+			change: func(req *Request) {
+				req.Headers["header_key"] = "header_value"
+			},
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			reqCopy := test.orig.Copy()
+
+			assert.Equal(t, test.orig, reqCopy)
+			test.change(&reqCopy)
+			assert.NotEqual(t, test.orig, reqCopy)
+		})
+	}
+}
+
+func TestNewHTTPRequest(t *testing.T) {
+	tests := map[string]struct {
+		req     Request
+		wantErr bool
+	}{
+		"test url": {
+			req: Request{
+				URL: "http://127.0.0.1:19999/api/v1/info",
+			},
+			wantErr: false,
+		},
+		"test body": {
+			req: Request{
+				Body: "content",
+			},
+			wantErr: false,
+		},
+		"test method": {
+			req: Request{
+				Method: "POST",
+			},
+			wantErr: false,
+		},
+		"test headers": {
+			req: Request{
+				Headers: map[string]string{
+					"X-Api-Key": "secret",
+				},
+			},
+			wantErr: false,
+		},
+		"test special headers (host)": {
+			req: Request{
+				Headers: map[string]string{
+					"host": "Host",
+				},
+			},
+			wantErr: false,
+		},
+		"test special headers (Host)": {
+			req: Request{
+				Headers: map[string]string{
+					"Host": "Host",
+				},
+			},
+			wantErr: false,
+		},
+		"test username and password": {
+			req: Request{
+				Username: "username",
+				Password: "password",
+			},
+			wantErr: false,
+		},
+		"test proxy username and proxy password": {
+			req: Request{
+				ProxyUsername: "proxy_username",
+				ProxyPassword: "proxy_password",
+			},
+			wantErr: false,
+		},
+	}
+
+	for name, test := range tests {
+		t.Run(name, func(t *testing.T) {
+			httpReq, err := NewHTTPRequest(test.req)
+
+			if test.wantErr {
+				assert.Error(t, err)
+				assert.Nil(t, httpReq)
+				return
+			}
+
+			require.NoError(t, err)
+			require.NotNil(t, httpReq)
+			require.IsType(t, (*http.Request)(nil), httpReq)
+
+			assert.Equal(t, test.req.URL, httpReq.URL.String())
+
+			if test.req.Body != "" {
+				assert.NotNil(t, httpReq.Body)
+			}
+
+			if test.req.Username != "" || test.req.Password != "" {
+				user, pass, ok := httpReq.BasicAuth()
+				assert.True(t, ok)
+				assert.Equal(t, test.req.Username, user)
+				assert.Equal(t, test.req.Password, pass)
+			}
+
+			if test.req.Method != "" {
+				assert.Equal(t, test.req.Method, httpReq.Method)
+			}
+
+			if test.req.ProxyUsername != "" || test.req.ProxyPassword != "" {
+				user, pass, ok := parseBasicAuth(httpReq.Header.Get("Proxy-Authorization"))
+				assert.True(t, ok)
+				assert.Equal(t, test.req.ProxyUsername, user)
+				assert.Equal(t, test.req.ProxyPassword, pass)
+			}
+
+			for k, v := range test.req.Headers {
+				switch k {
+				case "host", "Host":
+					assert.Equal(t, httpReq.Host, v)
+				default:
+					assert.Equal(t, v, httpReq.Header.Get(k))
+				}
+			}
+		})
+	}
+}
+
+func parseBasicAuth(auth string) (username, password string, ok bool) {
+	const prefix = "Basic "
+	if len(auth) < len(prefix) || !strings.EqualFold(auth[:len(prefix)], prefix) {
+		return "", "", false
+	}
+
+	decoded, err := base64.StdEncoding.DecodeString(auth[len(prefix):])
+	if err != nil {
+		return "", "", false
+	}
+
+	decodedStr := string(decoded)
+	idx := strings.IndexByte(decodedStr, ':')
+	if idx < 0 {
+		return "", "", false
+	}
+
+	return decodedStr[:idx], decodedStr[idx+1:], true
+}