starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-22 16:28:30 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

fixed some tests

Browse Source
This commit is contained in:
yusing
2025-01-03 16:31:49 +08:00
parent c1db404c0d
commit 3ecc0f95bf
5 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/net/http/accesslog/access_logger_test.go
View File

@@ -11,6 +11,7 @@ import (
E "github.com/yusing/go-proxy/internal/error"
. "github.com/yusing/go-proxy/internal/net/http/accesslog"
"github.com/yusing/go-proxy/internal/task"
. "github.com/yusing/go-proxy/internal/utils/testing"
)
@@ -28,8 +29,9 @@ const (
)
var (
testURL = E.Must(url.Parse("http://" + host + uri))
req = &http.Request{
testTask = task.RootTask("test", false)
testURL = E.Must(url.Parse("http://" + host + uri))
req = &http.Request{
RemoteAddr: remote,
Method: method,
Proto: proto,
@@ -55,10 +57,10 @@ func fmtLog(cfg *Config) (ts string, line string) {
var buf bytes.Buffer
t := time.Now()
logger := NewAccessLogger(nil, nil, cfg)
logger.Formatter.(*CommonFormatter).GetTimeNow = func() time.Time {
logger := NewAccessLogger(testTask, nil, cfg)
logger.Formatter.SetGetTimeNow(func() time.Time {
return t
}
})
logger.Format(&buf, req, resp)
return t.Format(LogTimeFormat), buf.String()
}