mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-21 21:12:34 +02:00
57292f0fe8
* feat: idle sleep for proxmox LXCs * refactor: replace deprecated docker api types * chore(api): remove debug task list endpoint * refactor: move servemux to gphttp/servemux; favicon.go to v1/favicon * refactor: introduce Pool interface, move agent_pool to agent module * refactor: simplify api code * feat: introduce debug api * refactor: remove net.URL and net.CIDR types, improved unmarshal handling * chore: update Makefile for debug build tag, update README * chore: add gperr.Unwrap method * feat: relative time and duration formatting * chore: add ROOT_DIR environment variable, refactor * migration: move homepage override and icon cache to $BASE_DIR/data, add migration code * fix: nil dereference on marshalling service health * fix: wait for route deletion * chore: enhance tasks debuggability * feat: stdout access logger and MultiWriter * fix(agent): remove agent properly on verify error * fix(metrics): disk exclusion logic and added corresponding tests * chore: update schema and prettify, fix package.json and Makefile * fix: I/O buffer not being shrunk before putting back to pool * feat: enhanced error handling module * chore: deps upgrade * feat: better value formatting and handling --------- Co-authored-by: yusing <yusing@6uo.me>
108 lines
4.9 KiB
Go
108 lines
4.9 KiB
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"slices"
|
|
"testing"
|
|
|
|
. "github.com/yusing/go-proxy/internal/utils/testing"
|
|
)
|
|
|
|
func TestModifyResponse(t *testing.T) {
|
|
opts := OptionsRaw{
|
|
"set_headers": map[string]string{
|
|
"X-Test-Resp-Status": VarRespStatusCode,
|
|
"X-Test-Resp-Content-Type": VarRespContentType,
|
|
"X-Test-Resp-Content-Length": VarRespContentLen,
|
|
"X-Test-Resp-Header-Content-Type": "$resp_header(Content-Type)",
|
|
|
|
"X-Test-Req-Method": VarRequestMethod,
|
|
"X-Test-Req-Scheme": VarRequestScheme,
|
|
"X-Test-Req-Host": VarRequestHost,
|
|
"X-Test-Req-Port": VarRequestPort,
|
|
"X-Test-Req-Addr": VarRequestAddr,
|
|
"X-Test-Req-Path": VarRequestPath,
|
|
"X-Test-Req-Query": VarRequestQuery,
|
|
"X-Test-Req-Url": VarRequestURL,
|
|
"X-Test-Req-Uri": VarRequestURI,
|
|
"X-Test-Req-Content-Type": VarRequestContentType,
|
|
"X-Test-Req-Content-Length": VarRequestContentLen,
|
|
"X-Test-Remote-Host": VarRemoteHost,
|
|
"X-Test-Remote-Port": VarRemotePort,
|
|
"X-Test-Remote-Addr": VarRemoteAddr,
|
|
"X-Test-Upstream-Scheme": VarUpstreamScheme,
|
|
"X-Test-Upstream-Host": VarUpstreamHost,
|
|
"X-Test-Upstream-Port": VarUpstreamPort,
|
|
"X-Test-Upstream-Addr": VarUpstreamAddr,
|
|
"X-Test-Upstream-Url": VarUpstreamURL,
|
|
"X-Test-Header-Content-Type": "$header(Content-Type)",
|
|
"X-Test-Arg-Arg_1": "$arg(arg_1)",
|
|
},
|
|
"add_headers": map[string]string{"Accept-Encoding": "test-value"},
|
|
"hide_headers": []string{"Accept"},
|
|
}
|
|
|
|
t.Run("set_options", func(t *testing.T) {
|
|
mr, err := ModifyResponse.New(opts)
|
|
ExpectNoError(t, err)
|
|
ExpectEqual(t, mr.impl.(*modifyResponse).SetHeaders, opts["set_headers"].(map[string]string))
|
|
ExpectEqual(t, mr.impl.(*modifyResponse).AddHeaders, opts["add_headers"].(map[string]string))
|
|
ExpectEqual(t, mr.impl.(*modifyResponse).HideHeaders, opts["hide_headers"].([]string))
|
|
})
|
|
|
|
t.Run("response_headers", func(t *testing.T) {
|
|
reqURL := Must(url.Parse("https://my.app/?arg_1=b"))
|
|
upstreamURL := Must(url.Parse("http://test.example.com"))
|
|
result, err := newMiddlewareTest(ModifyResponse, &testArgs{
|
|
middlewareOpt: opts,
|
|
reqURL: reqURL,
|
|
upstreamURL: upstreamURL,
|
|
body: bytes.Repeat([]byte("a"), 100),
|
|
headers: http.Header{
|
|
"Content-Type": []string{"application/json"},
|
|
},
|
|
respHeaders: http.Header{
|
|
"Content-Type": []string{"application/json"},
|
|
},
|
|
respBody: bytes.Repeat([]byte("a"), 50),
|
|
respStatus: http.StatusOK,
|
|
})
|
|
ExpectNoError(t, err)
|
|
ExpectTrue(t, slices.Contains(result.ResponseHeaders.Values("Accept-Encoding"), "test-value"))
|
|
ExpectEqual(t, result.ResponseHeaders.Get("Accept"), "")
|
|
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Resp-Status"), "200")
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Resp-Content-Type"), "application/json")
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Resp-Content-Length"), "50")
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Resp-Header-Content-Type"), "application/json")
|
|
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Method"), http.MethodGet)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Scheme"), reqURL.Scheme)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Host"), reqURL.Hostname())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Port"), reqURL.Port())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Addr"), reqURL.Host)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Path"), reqURL.Path)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Query"), reqURL.RawQuery)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Url"), reqURL.String())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Uri"), reqURL.RequestURI())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Content-Type"), "application/json")
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Req-Content-Length"), "100")
|
|
|
|
remoteHost, remotePort, _ := net.SplitHostPort(result.RemoteAddr)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Remote-Host"), remoteHost)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Remote-Port"), remotePort)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Remote-Addr"), result.RemoteAddr)
|
|
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Upstream-Scheme"), upstreamURL.Scheme)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Upstream-Host"), upstreamURL.Hostname())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Upstream-Port"), upstreamURL.Port())
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Upstream-Addr"), upstreamURL.Host)
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Upstream-Url"), upstreamURL.String())
|
|
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Header-Content-Type"), "application/json")
|
|
ExpectEqual(t, result.ResponseHeaders.Get("X-Test-Arg-Arg_1"), "b")
|
|
})
|
|
}
|