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.
74 lines
1.5 KiB
74 lines
1.5 KiB
package router_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
. "v2ray.com/core/app/router"
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/common/session"
|
|
"v2ray.com/core/testing/mocks"
|
|
)
|
|
|
|
func TestSimpleRouter(t *testing.T) {
|
|
config := &Config{
|
|
Rule: []*RoutingRule{
|
|
{
|
|
Tag: "test",
|
|
NetworkList: &net.NetworkList{
|
|
Network: []net.Network{net.Network_TCP},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
defer mockCtl.Finish()
|
|
|
|
mockDns := mocks.NewDNSClient(mockCtl)
|
|
|
|
r := new(Router)
|
|
common.Must(r.Init(config, mockDns))
|
|
|
|
ctx := withOutbound(&session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2ray.com"), 80)})
|
|
tag, err := r.PickRoute(ctx)
|
|
common.Must(err)
|
|
if tag != "test" {
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
}
|
|
}
|
|
|
|
func TestIPOnDemand(t *testing.T) {
|
|
config := &Config{
|
|
DomainStrategy: Config_IpOnDemand,
|
|
Rule: []*RoutingRule{
|
|
{
|
|
Tag: "test",
|
|
Cidr: []*CIDR{
|
|
{
|
|
Ip: []byte{192, 168, 0, 0},
|
|
Prefix: 16,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
mockCtl := gomock.NewController(t)
|
|
defer mockCtl.Finish()
|
|
|
|
mockDns := mocks.NewDNSClient(mockCtl)
|
|
mockDns.EXPECT().LookupIP(gomock.Eq("v2ray.com")).Return([]net.IP{{192, 168, 0, 1}}, nil).AnyTimes()
|
|
|
|
r := new(Router)
|
|
common.Must(r.Init(config, mockDns))
|
|
|
|
ctx := withOutbound(&session.Outbound{Target: net.TCPDestination(net.DomainAddress("v2ray.com"), 80)})
|
|
tag, err := r.PickRoute(ctx)
|
|
common.Must(err)
|
|
if tag != "test" {
|
|
t.Error("expect tag 'test', bug actually ", tag)
|
|
}
|
|
}
|