mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
990 B
42 lines
990 B
package blackhole_test |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
|
|
"github.com/xtls/xray-core/common" |
|
"github.com/xtls/xray-core/common/buf" |
|
"github.com/xtls/xray-core/common/serial" |
|
"github.com/xtls/xray-core/common/session" |
|
"github.com/xtls/xray-core/proxy/blackhole" |
|
"github.com/xtls/xray-core/transport" |
|
"github.com/xtls/xray-core/transport/pipe" |
|
) |
|
|
|
func TestBlackholeHTTPResponse(t *testing.T) { |
|
ctx := session.ContextWithOutbounds(context.Background(), []*session.Outbound{{}}) |
|
handler, err := blackhole.New(ctx, &blackhole.Config{ |
|
Response: serial.ToTypedMessage(&blackhole.HTTPResponse{}), |
|
}) |
|
common.Must(err) |
|
|
|
reader, writer := pipe.New(pipe.WithoutSizeLimit()) |
|
|
|
var mb buf.MultiBuffer |
|
var rerr error |
|
go func() { |
|
b, e := reader.ReadMultiBuffer() |
|
mb = b |
|
rerr = e |
|
}() |
|
|
|
link := transport.Link{ |
|
Reader: reader, |
|
Writer: writer, |
|
} |
|
common.Must(handler.Process(ctx, &link, nil)) |
|
common.Must(rerr) |
|
if mb.IsEmpty() { |
|
t.Error("expect http response, but nothing") |
|
} |
|
}
|
|
|