1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 10:08:15 -05:00
v2fly/app/router/router_test.go

250 lines
6.5 KiB
Go
Raw Normal View History

2016-10-12 10:11:13 -04:00
package router_test
import (
"context"
"testing"
2018-10-22 09:58:52 -04:00
"github.com/golang/mock/gomock"
2021-02-16 15:31:50 -05:00
. "github.com/v2fly/v2ray-core/v5/app/router"
"github.com/v2fly/v2ray-core/v5/app/router/routercommon"
"github.com/v2fly/v2ray-core/v5/common"
"github.com/v2fly/v2ray-core/v5/common/net"
"github.com/v2fly/v2ray-core/v5/common/session"
"github.com/v2fly/v2ray-core/v5/features/outbound"
routing_session "github.com/v2fly/v2ray-core/v5/features/routing/session"
"github.com/v2fly/v2ray-core/v5/testing/mocks"
)
2018-11-07 15:25:43 -05:00
type mockOutboundManager struct {
outbound.Manager
outbound.HandlerSelector
}
func TestSimpleRouter(t *testing.T) {
2018-10-22 09:58:52 -04:00
config := &Config{
Rule: []*RoutingRule{
{
2018-11-07 15:08:20 -05:00
TargetTag: &RoutingRule_Tag{
Tag: "test",
},
2018-11-20 10:58:26 -05:00
Networks: []net.Network{net.Network_TCP},
2018-10-22 09:58:52 -04:00
},
},
}
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
2018-11-07 15:25:43 -05:00
mockOhm := mocks.NewOutboundManager(mockCtl)
mockHs := mocks.NewOutboundHandlerSelector(mockCtl)
2018-10-22 09:58:52 -04:00
r := new(Router)
2021-04-09 11:02:45 -04:00
common.Must(r.Init(context.TODO(), config, mockDNS, &mockOutboundManager{
2018-11-07 15:25:43 -05:00
Manager: mockOhm,
HandlerSelector: mockHs,
}, nil))
2018-11-07 15:25:43 -05:00
2021-02-16 15:31:50 -05:00
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
2018-11-07 15:25:43 -05:00
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test" {
2018-11-07 15:25:43 -05:00
t.Error("expect tag 'test', bug actually ", tag)
}
}
func TestSimpleBalancer(t *testing.T) {
config := &Config{
Rule: []*RoutingRule{
{
TargetTag: &RoutingRule_BalancingTag{
BalancingTag: "balance",
},
2018-11-20 10:58:26 -05:00
Networks: []net.Network{net.Network_TCP},
2018-11-07 15:25:43 -05:00
},
},
BalancingRule: []*BalancingRule{
{
Tag: "balance",
OutboundSelector: []string{"test-"},
},
},
}
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
2018-11-07 15:25:43 -05:00
mockOhm := mocks.NewOutboundManager(mockCtl)
mockHs := mocks.NewOutboundHandlerSelector(mockCtl)
mockHs.EXPECT().Select(gomock.Eq([]string{"test-"})).Return([]string{"test"})
r := new(Router)
2021-04-09 11:02:45 -04:00
common.Must(r.Init(context.TODO(), config, mockDNS, &mockOutboundManager{
2018-11-07 15:25:43 -05:00
Manager: mockOhm,
HandlerSelector: mockHs,
}, nil))
2018-10-22 09:58:52 -04:00
2021-02-16 15:31:50 -05:00
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
2018-10-22 09:58:52 -04:00
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test" {
2018-10-22 09:58:52 -04:00
t.Error("expect tag 'test', bug actually ", tag)
}
}
/*
Do not work right now: need a full client setup
func TestLeastLoadBalancer(t *testing.T) {
config := &Config{
Rule: []*RoutingRule{
{
TargetTag: &RoutingRule_BalancingTag{
BalancingTag: "balance",
},
Networks: []net.Network{net.Network_TCP},
},
},
BalancingRule: []*BalancingRule{
{
Tag: "balance",
OutboundSelector: []string{"test-"},
Strategy: "leastLoad",
StrategySettings: serial.ToTypedMessage(&StrategyLeastLoadConfig{
Baselines: nil,
Expected: 1,
}),
},
},
}
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
mockOhm := mocks.NewOutboundManager(mockCtl)
mockHs := mocks.NewOutboundHandlerSelector(mockCtl)
mockHs.EXPECT().Select(gomock.Eq([]string{"test-"})).Return([]string{"test1"})
r := new(Router)
common.Must(r.Init(context.TODO(), config, mockDNS, &mockOutboundManager{
Manager: mockOhm,
HandlerSelector: mockHs,
}, nil))
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2ray.com"), 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test1" {
t.Error("expect tag 'test1', bug actually ", tag)
}
}*/
2018-10-22 09:58:52 -04:00
func TestIPOnDemand(t *testing.T) {
config := &Config{
2021-09-07 08:43:28 -04:00
DomainStrategy: DomainStrategy_IpOnDemand,
2018-10-22 09:58:52 -04:00
Rule: []*RoutingRule{
{
2018-11-07 15:08:20 -05:00
TargetTag: &RoutingRule_Tag{
Tag: "test",
},
2021-09-07 08:43:28 -04:00
Cidr: []*routercommon.CIDR{
{
2018-10-22 09:58:52 -04:00
Ip: []byte{192, 168, 0, 0},
Prefix: 16,
},
2016-10-11 17:02:44 -04:00
},
2018-10-22 09:58:52 -04:00
},
2016-02-01 17:21:54 -05:00
},
}
2018-10-22 09:58:52 -04:00
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
mockDNS.EXPECT().LookupIP(gomock.Eq("v2fly.org")).Return([]net.IP{{192, 168, 0, 1}}, nil).AnyTimes()
2018-10-22 09:58:52 -04:00
r := new(Router)
common.Must(r.Init(context.TODO(), config, mockDNS, nil, nil))
2017-01-06 09:32:36 -05:00
2021-02-16 15:31:50 -05:00
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
2018-10-22 09:58:52 -04:00
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test" {
2018-10-22 09:58:52 -04:00
t.Error("expect tag 'test', bug actually ", tag)
}
}
2018-12-04 14:36:51 -05:00
func TestIPIfNonMatchDomain(t *testing.T) {
config := &Config{
2021-09-07 08:43:28 -04:00
DomainStrategy: DomainStrategy_IpIfNonMatch,
2018-12-04 14:36:51 -05:00
Rule: []*RoutingRule{
{
TargetTag: &RoutingRule_Tag{
Tag: "test",
},
2021-09-07 08:43:28 -04:00
Cidr: []*routercommon.CIDR{
2018-12-04 14:36:51 -05:00
{
Ip: []byte{192, 168, 0, 0},
Prefix: 16,
},
},
},
},
}
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
mockDNS.EXPECT().LookupIP(gomock.Eq("v2fly.org")).Return([]net.IP{{192, 168, 0, 1}}, nil).AnyTimes()
2018-12-04 14:36:51 -05:00
r := new(Router)
common.Must(r.Init(context.TODO(), config, mockDNS, nil, nil))
2018-12-04 14:36:51 -05:00
2021-02-16 15:31:50 -05:00
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2fly.org"), 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
2018-12-04 14:36:51 -05:00
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test" {
2018-12-04 14:36:51 -05:00
t.Error("expect tag 'test', bug actually ", tag)
}
}
func TestIPIfNonMatchIP(t *testing.T) {
config := &Config{
2021-09-07 08:43:28 -04:00
DomainStrategy: DomainStrategy_IpIfNonMatch,
2018-12-04 14:36:51 -05:00
Rule: []*RoutingRule{
{
TargetTag: &RoutingRule_Tag{
Tag: "test",
},
2021-09-07 08:43:28 -04:00
Cidr: []*routercommon.CIDR{
2018-12-04 14:36:51 -05:00
{
Ip: []byte{127, 0, 0, 0},
Prefix: 8,
},
},
},
},
}
mockCtl := gomock.NewController(t)
defer mockCtl.Finish()
mockDNS := mocks.NewDNSClient(mockCtl)
2018-12-04 14:36:51 -05:00
r := new(Router)
common.Must(r.Init(context.TODO(), config, mockDNS, nil, nil))
2018-12-04 14:36:51 -05:00
ctx := session.ContextWithOutbound(context.Background(), &session.Outbound{Target: net.TCPDestination(net.LocalHostIP, 80)})
route, err := r.PickRoute(routing_session.AsRoutingContext(ctx))
2018-12-04 14:36:51 -05:00
common.Must(err)
if tag := route.GetOutboundTag(); tag != "test" {
2018-12-04 14:36:51 -05:00
t.Error("expect tag 'test', bug actually ", tag)
}
}