1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-10-12 22:13:49 -04:00
v2fly/proxy/blackhole/blackhole_test.go
2022-01-02 15:16:23 +00:00

41 lines
960 B
Go

package blackhole_test
import (
"context"
"testing"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/common/buf"
"github.com/v2fly/v2ray-core/v5/common/serial"
"github.com/v2fly/v2ray-core/v5/proxy/blackhole"
"github.com/v2fly/v2ray-core/v5/transport"
"github.com/v2fly/v2ray-core/v5/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())
readerError := make(chan error)
var mb buf.MultiBuffer
go func() {
b, e := reader.ReadMultiBuffer()
mb = b
readerError <- e
}()
link := transport.Link{
Reader: reader,
Writer: writer,
}
common.Must(handler.Process(context.Background(), &link, nil))
common.Must(<-readerError)
if mb.IsEmpty() {
t.Error("expect http response, but nothing")
}
}