mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-12-05 03:16:03 +03:00
017f53b5fc
* Add session context outbounds as slice slice is needed for dialer proxy where two outbounds work on top of each other There are two sets of target addr for example It also enable Xtls to correctly do splice copy by checking both outbounds are ready to do direct copy * Fill outbound tag info * Splice now checks capalibility from all outbounds * Fix unit tests
43 lines
990 B
Go
43 lines
990 B
Go
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")
|
|
}
|
|
}
|