mirror of
https://github.com/TecharoHQ/anubis.git
synced 2025-09-13 14:47:39 -04:00
01f55cf552
* internal/log: Implement logging of HOST when using subrequest auth The host header wouldn't be set on subrequest auth, so we need to look for X-Forwarded-Host header when logging requests. * chore: add changelog entry --------- Signed-off-by: Xe Iaso <xe.iaso@techaro.lol> Co-authored-by: Xe Iaso <xe.iaso@techaro.lol>
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package internal
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"log/slog"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestErrorLogFilter(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
destLogger := log.New(&buf, "", 0)
|
|
errorFilterWriter := &ErrorLogFilter{Unwrap: destLogger}
|
|
testErrorLogger := log.New(errorFilterWriter, "", 0)
|
|
|
|
// Test Case 1: Suppressed message
|
|
suppressedMessage := "http: proxy error: context canceled"
|
|
testErrorLogger.Println(suppressedMessage)
|
|
|
|
if buf.Len() != 0 {
|
|
t.Errorf("Suppressed message was written to output. Output: %q", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
// Test Case 2: Allowed message
|
|
allowedMessage := "http: another error occurred"
|
|
testErrorLogger.Println(allowedMessage)
|
|
|
|
output := buf.String()
|
|
if !strings.Contains(output, allowedMessage) {
|
|
t.Errorf("Allowed message was not written to output. Output: %q", output)
|
|
}
|
|
if !strings.HasSuffix(output, "\n") {
|
|
t.Errorf("Allowed message output is missing newline. Output: %q", output)
|
|
}
|
|
buf.Reset()
|
|
|
|
// Test Case 3: Partially matching message (should be suppressed)
|
|
partiallyMatchingMessage := "Some other log before http: proxy error: context canceled and after"
|
|
testErrorLogger.Println(partiallyMatchingMessage)
|
|
|
|
if buf.Len() != 0 {
|
|
t.Errorf("Partially matching message was written to output. Output: %q", buf.String())
|
|
}
|
|
buf.Reset()
|
|
}
|
|
|
|
func TestGetRequestLogger(t *testing.T) {
|
|
// Test case 1: Normal request with Host header
|
|
req1, _ := http.NewRequest("GET", "http://example.com/test", nil)
|
|
req1.Host = "example.com"
|
|
|
|
logger := slog.Default()
|
|
reqLogger := GetRequestLogger(logger, req1)
|
|
|
|
// We can't easily test the actual log output without setting up a test handler,
|
|
// but we can verify the function doesn't panic and returns a logger
|
|
if reqLogger == nil {
|
|
t.Error("GetRequestLogger returned nil")
|
|
}
|
|
|
|
// Test case 2: Subrequest auth mode with X-Forwarded-Host
|
|
req2, _ := http.NewRequest("GET", "http://test.com/auth", nil)
|
|
req2.Host = ""
|
|
req2.Header.Set("X-Forwarded-Host", "original-site.com")
|
|
|
|
reqLogger2 := GetRequestLogger(logger, req2)
|
|
if reqLogger2 == nil {
|
|
t.Error("GetRequestLogger returned nil for X-Forwarded-Host case")
|
|
}
|
|
|
|
// Test case 3: No host information available
|
|
req3, _ := http.NewRequest("GET", "http://test.com/nohost", nil)
|
|
req3.Host = ""
|
|
|
|
reqLogger3 := GetRequestLogger(logger, req3)
|
|
if reqLogger3 == nil {
|
|
t.Error("GetRequestLogger returned nil for no host case")
|
|
}
|
|
}
|