Hunter0x7c7
2022-08-11 b8230139fb40edea387617b6accd8371e37eda58
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package core_test
 
import (
    "testing"
 
    "google.golang.org/protobuf/proto"
    "google.golang.org/protobuf/types/known/anypb"
 
    . "github.com/v2fly/v2ray-core/v5"
    "github.com/v2fly/v2ray-core/v5/app/dispatcher"
    "github.com/v2fly/v2ray-core/v5/app/proxyman"
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/common/net"
    "github.com/v2fly/v2ray-core/v5/common/protocol"
    "github.com/v2fly/v2ray-core/v5/common/serial"
    "github.com/v2fly/v2ray-core/v5/common/uuid"
    "github.com/v2fly/v2ray-core/v5/features/dns"
    "github.com/v2fly/v2ray-core/v5/features/dns/localdns"
    _ "github.com/v2fly/v2ray-core/v5/main/distro/all"
    "github.com/v2fly/v2ray-core/v5/proxy/dokodemo"
    "github.com/v2fly/v2ray-core/v5/proxy/vmess"
    "github.com/v2fly/v2ray-core/v5/proxy/vmess/outbound"
    "github.com/v2fly/v2ray-core/v5/testing/servers/tcp"
)
 
func TestV2RayDependency(t *testing.T) {
    instance := new(Instance)
 
    wait := make(chan bool, 1)
    instance.RequireFeatures(func(d dns.Client) {
        if d == nil {
            t.Error("expected dns client fulfilled, but actually nil")
        }
        wait <- true
    })
    instance.AddFeature(localdns.New())
    <-wait
}
 
func TestV2RayClose(t *testing.T) {
    port := tcp.PickPort()
 
    userID := uuid.New()
    config := &Config{
        App: []*anypb.Any{
            serial.ToTypedMessage(&dispatcher.Config{}),
            serial.ToTypedMessage(&proxyman.InboundConfig{}),
            serial.ToTypedMessage(&proxyman.OutboundConfig{}),
        },
        Inbound: []*InboundHandlerConfig{
            {
                ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
                    PortRange: net.SinglePortRange(port),
                    Listen:    net.NewIPOrDomain(net.LocalHostIP),
                }),
                ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
                    Address: net.NewIPOrDomain(net.LocalHostIP),
                    Port:    uint32(0),
                    NetworkList: &net.NetworkList{
                        Network: []net.Network{net.Network_TCP, net.Network_UDP},
                    },
                }),
            },
        },
        Outbound: []*OutboundHandlerConfig{
            {
                ProxySettings: serial.ToTypedMessage(&outbound.Config{
                    Receiver: []*protocol.ServerEndpoint{
                        {
                            Address: net.NewIPOrDomain(net.LocalHostIP),
                            Port:    uint32(0),
                            User: []*protocol.User{
                                {
                                    Account: serial.ToTypedMessage(&vmess.Account{
                                        Id: userID.String(),
                                    }),
                                },
                            },
                        },
                    },
                }),
            },
        },
    }
 
    cfgBytes, err := proto.Marshal(config)
    common.Must(err)
 
    server, err := StartInstance(FormatProtobuf, cfgBytes)
    common.Must(err)
    server.Close()
}