v2ray-core/proxy/vmess/vmess_test.go

116 lines
3.3 KiB
Go
Raw Normal View History

2015-12-07 19:32:38 +00:00
package vmess_test
2015-09-19 19:56:09 +00:00
import (
"bytes"
"testing"
"github.com/v2ray/v2ray-core/app"
v2net "github.com/v2ray/v2ray-core/common/net"
2015-11-01 20:32:08 +00:00
v2nettesting "github.com/v2ray/v2ray-core/common/net/testing"
2015-12-12 20:40:16 +00:00
"github.com/v2ray/v2ray-core/common/uuid"
2016-01-02 22:32:18 +00:00
"github.com/v2ray/v2ray-core/proxy"
proxytesting "github.com/v2ray/v2ray-core/proxy/testing"
proxymocks "github.com/v2ray/v2ray-core/proxy/testing/mocks"
2015-12-07 19:32:38 +00:00
vmess "github.com/v2ray/v2ray-core/proxy/vmess"
_ "github.com/v2ray/v2ray-core/proxy/vmess/inbound"
_ "github.com/v2ray/v2ray-core/proxy/vmess/inbound/json"
2015-12-07 19:32:38 +00:00
_ "github.com/v2ray/v2ray-core/proxy/vmess/outbound"
_ "github.com/v2ray/v2ray-core/proxy/vmess/outbound/json"
2015-11-29 13:45:32 +00:00
"github.com/v2ray/v2ray-core/shell/point"
2015-12-06 15:41:41 +00:00
"github.com/v2ray/v2ray-core/shell/point/testing/mocks"
v2testing "github.com/v2ray/v2ray-core/testing"
"github.com/v2ray/v2ray-core/testing/assert"
2015-09-19 19:56:09 +00:00
)
func TestVMessInAndOut(t *testing.T) {
v2testing.Current(t)
2015-09-19 19:56:09 +00:00
2015-12-12 20:40:16 +00:00
id, err := uuid.ParseString("ad937d9d-6e23-4a5a-ba23-bce5092a7c51")
2015-10-16 10:03:22 +00:00
assert.Error(err).IsNil()
2015-09-19 19:56:09 +00:00
2015-12-12 20:40:16 +00:00
testAccount := vmess.NewID(id)
2015-11-01 20:32:08 +00:00
portA := v2nettesting.PickPort()
portB := v2nettesting.PickPort()
ichConnInput := []byte("The data to be send to outbound server.")
ichConnOutput := bytes.NewBuffer(make([]byte, 0, 1024))
ich := &proxymocks.InboundConnectionHandler{
ConnInput: bytes.NewReader(ichConnInput),
ConnOutput: ichConnOutput,
2015-09-19 19:56:09 +00:00
}
2016-01-02 22:32:18 +00:00
protocol, err := proxytesting.RegisterInboundConnectionHandlerCreator("mock_och", func(space app.Space, config interface{}) (proxy.InboundConnectionHandler, error) {
ich.Space = space
return ich, nil
})
assert.Error(err).IsNil()
2015-09-19 19:56:09 +00:00
configA := mocks.Config{
PortValue: portA,
InboundConfigValue: &mocks.ConnectionConfig{
ProtocolValue: protocol,
2015-10-06 21:11:08 +00:00
SettingsValue: nil,
2015-09-19 19:56:09 +00:00
},
OutboundConfigValue: &mocks.ConnectionConfig{
ProtocolValue: "vmess",
SettingsValue: []byte(`{
"vnext": [
{
"address": "127.0.0.1",
"port": ` + portB.String() + `,
"users": [
{"id": "` + testAccount.String() + `"}
]
}
]
}`),
2015-09-19 19:56:09 +00:00
},
}
2015-10-14 12:56:11 +00:00
pointA, err := point.NewPoint(&configA)
2015-09-19 19:56:09 +00:00
assert.Error(err).IsNil()
err = pointA.Start()
assert.Error(err).IsNil()
ochConnInput := []byte("The data to be returned to inbound server.")
ochConnOutput := bytes.NewBuffer(make([]byte, 0, 1024))
och := &proxymocks.OutboundConnectionHandler{
ConnInput: bytes.NewReader(ochConnInput),
ConnOutput: ochConnOutput,
2015-09-19 19:56:09 +00:00
}
2016-01-02 22:32:18 +00:00
protocol, err = proxytesting.RegisterOutboundConnectionHandlerCreator("mock_och", func(space app.Space, config interface{}) (proxy.OutboundConnectionHandler, error) {
return och, nil
})
assert.Error(err).IsNil()
2015-09-19 19:56:09 +00:00
configB := mocks.Config{
PortValue: portB,
InboundConfigValue: &mocks.ConnectionConfig{
ProtocolValue: "vmess",
SettingsValue: []byte(`{
"clients": [
{"id": "` + testAccount.String() + `"}
]
}`),
2015-09-19 19:56:09 +00:00
},
OutboundConfigValue: &mocks.ConnectionConfig{
ProtocolValue: protocol,
2015-10-06 21:11:08 +00:00
SettingsValue: nil,
2015-09-19 19:56:09 +00:00
},
}
2015-10-14 12:56:11 +00:00
pointB, err := point.NewPoint(&configB)
2015-09-19 19:56:09 +00:00
assert.Error(err).IsNil()
err = pointB.Start()
assert.Error(err).IsNil()
2015-12-16 22:53:38 +00:00
dest := v2net.TCPDestination(v2net.IPAddress([]byte{1, 2, 3, 4}), 80)
2015-10-02 13:32:26 +00:00
ich.Communicate(v2net.NewPacket(dest, nil, true))
assert.Bytes(ichConnInput).Equals(ochConnOutput.Bytes())
assert.Bytes(ichConnOutput.Bytes()).Equals(ochConnInput)
2015-09-19 19:56:09 +00:00
}