Refactor header handling and add unit tests

Author: ShocOne

headers/headers_test.go

@@ -0,0 +1,52 @@
+// headers/headers_test.go
+package headers
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/deploymenttheory/go-api-http-client/authenticationhandler"
+	"github.com/deploymenttheory/go-api-http-client/logger"
+	"github.com/deploymenttheory/go-api-http-client/mocklogger"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+)
+
+func TestSetAuthorization(t *testing.T) {
+	req := httptest.NewRequest(http.MethodGet, "http://example.com", nil)
+	mockLog := mocklogger.NewMockLogger()
+	mockLog.On("Debug", mock.Anything, mock.Anything).Once()
+
+	token := "test-token"
+	authTokenHandler := &authenticationhandler.AuthTokenHandler{Token: token}
+
+	headerHandler := NewHeaderHandler(req, mockLog, nil, authTokenHandler)
+	headerHandler.SetAuthorization()
+
+	assert.Equal(t, "Bearer "+token, req.Header.Get("Authorization"), "Authorization header should be correctly set")
+	mockLog.AssertExpectations(t)
+}
+
+func TestSetContentType(t *testing.T) {
+	req := httptest.NewRequest(http.MethodGet, "http://example.com", nil)
+	mockLog := mocklogger.NewMockLogger()
+
+	contentType := "application/json"
+	headerHandler := NewHeaderHandler(req, mockLog, nil, nil)
+	headerHandler.SetContentType(contentType)
+
+	assert.Equal(t, contentType, req.Header.Get("Content-Type"), "Content-Type header should be correctly set")
+}
+
+func TestLogHeaders(t *testing.T) {
+	req := httptest.NewRequest(http.MethodGet, "http://example.com", nil)
+	mockLog := mocklogger.NewMockLogger()
+	mockLog.On("Debug", mock.Anything, mock.Anything).Once()
+	mockLog.SetLevel(logger.LogLevelDebug)
+
+	headerHandler := NewHeaderHandler(req, mockLog, nil, nil)
+	headerHandler.LogHeaders(true)
+
+	mockLog.AssertExpectations(t)
+}

headers/headers_test.go.TODO

@@ -0,0 +1,31 @@
+// headers/redact/redact_test.go
+package redact
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// TestRedactSensitiveHeaderData tests the RedactSensitiveHeaderData function to ensure it correctly redacts sensitive data.
+func TestRedactSensitiveHeaderData(t *testing.T) {
+	cases := []struct {
+		name              string
+		hideSensitiveData bool
+		key               string
+		value             string
+		expected          string
+	}{
+		{"Sensitive Key With Redaction", true, "AccessToken", "some-sensitive-token", "REDACTED"},
+		{"Sensitive Key Without Redaction", false, "AccessToken", "some-sensitive-token", "some-sensitive-token"},
+		{"Non-Sensitive Key With Redaction", true, "User-Agent", "MyCustomAgent", "MyCustomAgent"},
+		{"Non-Sensitive Key Without Redaction", false, "User-Agent", "MyCustomAgent", "MyCustomAgent"},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			result := RedactSensitiveHeaderData(tc.hideSensitiveData, tc.key, tc.value)
+			assert.Equal(t, tc.expected, result, "Redacted value should match the expected outcome")
+		})
+	}
+}