Skip to content

Commit b65192a

Browse files
authored
http: fix TestHTTPGotConnectionRemoteAddr (#195)
the domain new.qq.com is being blocked by waf Fixes #
1 parent 1c5c0fb commit b65192a

File tree

2 files changed

+19
-2
lines changed

2 files changed

+19
-2
lines changed

docs/user_guide/framework_conf.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ server:
128128
timeout: Integer
129129
# Optional, long connection idle time, in milliseconds
130130
idletime: Integer
131-
# Optional, which regitration center to use such as polaris
131+
# Optional, which registration center to use such as polaris
132132
registry: String
133133
# Optional, list of interceptors, lower priority than server.filter
134134
filter: [String]

http/transport_test.go

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1487,7 +1487,9 @@ func TestHTTPClientReqRspDifferentContentType(t *testing.T) {
14871487
func TestHTTPGotConnectionRemoteAddr(t *testing.T) {
14881488
ctx := context.Background()
14891489
for i := 0; i < 3; i++ {
1490-
proxy := thttp.NewClientProxy(t.Name(), client.WithTarget("dns://new.qq.com/"))
1490+
proxy := thttp.NewClientProxy(t.Name(),
1491+
client.WithTarget("dns://new.qq.com/"),
1492+
client.WithTransport(&mockTransport{}))
14911493
rsp := &codec.Body{}
14921494
require.Nil(t, proxy.Get(ctx, "/", rsp,
14931495
client.WithSerializationType(codec.SerializationTypeNoop),
@@ -1551,3 +1553,18 @@ type errHeaderHandler struct{}
15511553
func (*errHeaderHandler) Handle(ctx context.Context, reqBuf []byte) (rsp []byte, err error) {
15521554
return nil, thttp.ErrEncodeMissingHeader
15531555
}
1556+
1557+
type mockTransport struct{}
1558+
1559+
func (t *mockTransport) RoundTrip(ctx context.Context, req []byte, opts ...transport.RoundTripOption) (rsp []byte, err error) {
1560+
msg := codec.Message(ctx)
1561+
msg.WithClientRspHead(&thttp.ClientRspHeader{
1562+
Response: &http.Response{},
1563+
})
1564+
raddr, err := net.ResolveTCPAddr("tcp", "127.0.0.1:8080")
1565+
if err != nil {
1566+
return nil, err
1567+
}
1568+
msg.WithRemoteAddr(raddr)
1569+
return []byte("mock transport"), nil
1570+
}

0 commit comments

Comments
 (0)