TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-07-15 18:04:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(tests): refactor and fix login in middleware test utils

Browse source
This commit is contained in:
yusing 2025-07-13 13:33:48 +08:00
parent b1d5c4b091
commit 814ff33352
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
internal/net/gphttp/middleware/test_utils.go
View file

@ -5,6 +5,7 @@ import (
_ "embed" _ "embed"
"encoding/json" "encoding/json"
"io" "io"
"maps"
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
@ -62,11 +63,10 @@ func (rt *requestRecorder) RoundTrip(req *http.Request) (resp *http.Response, er
TLS: req.TLS, TLS: req.TLS,
} }
} }
if err == nil { if err != nil {
for k, v := range rt.args.respHeaders { return nil, err
resp.Header[k] = v
}
} }
maps.Copy(resp.Header, rt.args.respHeaders)
return resp, nil return resp, nil
} }
@ -143,9 +143,7 @@ func newMiddlewaresTest(middlewares []*Middleware, args *testArgs) (*TestResult,
args.setDefaults() args.setDefaults()
req := httptest.NewRequest(args.reqMethod, args.reqURL.String(), args.bodyReader()) req := httptest.NewRequest(args.reqMethod, args.reqURL.String(), args.bodyReader())
for k, v := range args.headers { maps.Copy(req.Header, args.headers)
req.Header[k] = v
}
w := httptest.NewRecorder() w := httptest.NewRecorder()