Xray-core/proxy/blackhole/blackhole_test.go

43 lines
990 B
Go
Raw Permalink Normal View History

2020-11-25 14:01:53 +03:00
package blackhole_test
import (
"context"
"testing"
2020-12-04 04:36:16 +03:00
"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"
2020-12-04 04:36:16 +03:00
"github.com/xtls/xray-core/proxy/blackhole"
"github.com/xtls/xray-core/transport"
"github.com/xtls/xray-core/transport/pipe"
2020-11-25 14:01:53 +03:00
)
func TestBlackholeHTTPResponse(t *testing.T) {
ctx := session.ContextWithOutbounds(context.Background(), []*session.Outbound{{}})
handler, err := blackhole.New(ctx, &blackhole.Config{
2020-11-25 14:01:53 +03:00
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))
2020-11-25 14:01:53 +03:00
common.Must(rerr)
if mb.IsEmpty() {
t.Error("expect http response, but nothing")
}
}