mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 01:57:12 -05:00
34 lines
674 B
Go
34 lines
674 B
Go
|
package mocks
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
|
||
|
|
"github.com/v2ray/v2ray-core"
|
||
|
|
v2net "github.com/v2ray/v2ray-core/net"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FakeOutboundConnectionHandler struct {
|
||
|
|
Data2Send bytes.Buffer
|
||
|
|
Data2Return []byte
|
||
|
|
Destination v2net.VAddress
|
||
|
|
}
|
||
|
|
|
||
|
|
func (handler *FakeOutboundConnectionHandler) Start(ray core.OutboundVRay) error {
|
||
|
|
input := ray.OutboundInput()
|
||
|
|
output := ray.OutboundOutput()
|
||
|
|
|
||
|
|
output <- handler.Data2Return
|
||
|
|
for {
|
||
|
|
data, open := <-input
|
||
|
|
if !open {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
handler.Data2Send.Write(data)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (handler *FakeOutboundConnectionHandler) Create(vPoint *core.VPoint, dest v2net.VAddress) (core.OutboundConnectionHandler, error) {
|
||
|
|
return handler, nil
|
||
|
|
}
|