mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.1 KiB
49 lines
1.1 KiB
package router_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"v2ray.com/core"
|
|
"v2ray.com/core/app/dispatcher"
|
|
"v2ray.com/core/app/proxyman"
|
|
_ "v2ray.com/core/app/proxyman/outbound"
|
|
. "v2ray.com/core/app/router"
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/common/serial"
|
|
"v2ray.com/core/common/session"
|
|
"v2ray.com/core/features/routing"
|
|
. "v2ray.com/ext/assert"
|
|
)
|
|
|
|
func TestSimpleRouter(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
config := &core.Config{
|
|
App: []*serial.TypedMessage{
|
|
serial.ToTypedMessage(&Config{
|
|
Rule: []*RoutingRule{
|
|
{
|
|
Tag: "test",
|
|
NetworkList: &net.NetworkList{
|
|
Network: []net.Network{net.Network_TCP},
|
|
},
|
|
},
|
|
},
|
|
}),
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
|
},
|
|
}
|
|
|
|
v, err := core.New(config)
|
|
common.Must(err)
|
|
|
|
r := v.GetFeature(routing.RouterType()).(routing.Router)
|
|
|
|
ctx := withOutbound(&session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2ray.com"), 80)})
|
|
tag, err := r.PickRoute(ctx)
|
|
assert(err, IsNil)
|
|
assert(tag, Equals, "test")
|
|
}
|