mirror of
https://github.com/yusing/godoxy.git
synced 2025-07-15 10:04:02 +02:00
fix(tests): refactor and fix login in middleware test utils
This commit is contained in:
parent
b1d5c4b091
commit
814ff33352
1 changed files with 5 additions and 7 deletions
|
@ -5,6 +5,7 @@ import (
|
|||
_ "embed"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"maps"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
|
||||
|
@ -62,11 +63,10 @@ func (rt *requestRecorder) RoundTrip(req *http.Request) (resp *http.Response, er
|
|||
TLS: req.TLS,
|
||||
}
|
||||
}
|
||||
if err == nil {
|
||||
for k, v := range rt.args.respHeaders {
|
||||
resp.Header[k] = v
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
maps.Copy(resp.Header, rt.args.respHeaders)
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
|
@ -143,9 +143,7 @@ func newMiddlewaresTest(middlewares []*Middleware, args *testArgs) (*TestResult,
|
|||
args.setDefaults()
|
||||
|
||||
req := httptest.NewRequest(args.reqMethod, args.reqURL.String(), args.bodyReader())
|
||||
for k, v := range args.headers {
|
||||
req.Header[k] = v
|
||||
}
|
||||
maps.Copy(req.Header, args.headers)
|
||||
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue