mirror of https://github.com/v2ray/v2ray-core
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package router_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"v2ray.com/core/app"
|
|
"v2ray.com/core/app/dispatcher"
|
|
_ "v2ray.com/core/app/dispatcher/impl"
|
|
"v2ray.com/core/app/dns"
|
|
_ "v2ray.com/core/app/dns/server"
|
|
"v2ray.com/core/app/proxyman"
|
|
_ "v2ray.com/core/app/proxyman/outbound"
|
|
. "v2ray.com/core/app/router"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/proxy"
|
|
. "v2ray.com/ext/assert"
|
|
)
|
|
|
|
func TestSimpleRouter(t *testing.T) {
|
|
assert := With(t)
|
|
|
|
config := &Config{
|
|
Rule: []*RoutingRule{
|
|
{
|
|
Tag: "test",
|
|
NetworkList: &net.NetworkList{
|
|
Network: []net.Network{net.Network_TCP},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
space := app.NewSpace()
|
|
ctx := app.ContextWithSpace(context.Background(), space)
|
|
assert(app.AddApplicationToSpace(ctx, new(dns.Config)), IsNil)
|
|
assert(app.AddApplicationToSpace(ctx, new(dispatcher.Config)), IsNil)
|
|
assert(app.AddApplicationToSpace(ctx, new(proxyman.OutboundConfig)), IsNil)
|
|
assert(app.AddApplicationToSpace(ctx, config), IsNil)
|
|
assert(space.Initialize(), IsNil)
|
|
|
|
r := FromSpace(space)
|
|
|
|
ctx = proxy.ContextWithTarget(ctx, net.TCPDestination(net.DomainAddress("v2ray.com"), 80))
|
|
tag, err := r.TakeDetour(ctx)
|
|
assert(err, IsNil)
|
|
assert(tag, Equals, "test")
|
|
}
|