mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-11-04 20:19:19 +02:00
41 lines
914 B
Go
41 lines
914 B
Go
|
package blackhole_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/xtls/xray-core/v1/common"
|
||
|
|
"github.com/xtls/xray-core/v1/common/buf"
|
||
|
|
"github.com/xtls/xray-core/v1/common/serial"
|
||
|
|
"github.com/xtls/xray-core/v1/proxy/blackhole"
|
||
|
|
"github.com/xtls/xray-core/v1/transport"
|
||
|
|
"github.com/xtls/xray-core/v1/transport/pipe"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestBlackholeHTTPResponse(t *testing.T) {
|
||
|
|
handler, err := blackhole.New(context.Background(), &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(context.Background(), &link, nil))
|
||
|
|
common.Must(rerr)
|
||
|
|
if mb.IsEmpty() {
|
||
|
|
t.Error("expect http response, but nothing")
|
||
|
|
}
|
||
|
|
}
|