Fix flaky TestVMessDynamicPort (#723)

This commit is contained in:
yuhan6665 2021-09-23 23:59:00 -04:00 committed by GitHub
parent a149c78a4c
commit 9f9059c7b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 111 additions and 51 deletions

View File

@ -143,6 +143,23 @@ func CloseAllServers(servers []*exec.Cmd) {
}) })
} }
func CloseServer(server *exec.Cmd) {
log.Record(&log.GeneralMessage{
Severity: log.Severity_Info,
Content: "Closing server.",
})
if runtime.GOOS == "windows" {
server.Process.Kill()
} else {
server.Process.Signal(syscall.SIGTERM)
}
server.Process.Wait()
log.Record(&log.GeneralMessage{
Severity: log.Severity_Info,
Content: "Server closed.",
})
}
func withDefaultApps(config *core.Config) *core.Config { func withDefaultApps(config *core.Config) *core.Config {
config.App = append(config.App, serial.ToTypedMessage(&dispatcher.Config{})) config.App = append(config.App, serial.ToTypedMessage(&dispatcher.Config{}))
config.App = append(config.App, serial.ToTypedMessage(&proxyman.InboundConfig{})) config.App = append(config.App, serial.ToTypedMessage(&proxyman.InboundConfig{}))

View File

@ -36,59 +36,87 @@ func TestVMessDynamicPort(t *testing.T) {
defer tcpServer.Close() defer tcpServer.Close()
userID := protocol.NewID(uuid.New()) userID := protocol.NewID(uuid.New())
retry := 1
serverPort := tcp.PickPort() serverPort := tcp.PickPort()
serverConfig := &core.Config{ for {
App: []*serial.TypedMessage{ serverConfig := &core.Config{
serial.ToTypedMessage(&log.Config{ App: []*serial.TypedMessage{
ErrorLogLevel: clog.Severity_Debug, serial.ToTypedMessage(&log.Config{
ErrorLogType: log.LogType_Console, ErrorLogLevel: clog.Severity_Debug,
}), ErrorLogType: log.LogType_Console,
},
Inbound: []*core.InboundHandlerConfig{
{
ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
PortRange: net.SinglePortRange(serverPort),
Listen: net.NewIPOrDomain(net.LocalHostIP),
}),
ProxySettings: serial.ToTypedMessage(&inbound.Config{
User: []*protocol.User{
{
Account: serial.ToTypedMessage(&vmess.Account{
Id: userID.String(),
}),
},
},
Detour: &inbound.DetourConfig{
To: "detour",
},
}), }),
}, },
{ Inbound: []*core.InboundHandlerConfig{
ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{ {
PortRange: &net.PortRange{ ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
From: uint32(serverPort + 1), PortRange: net.SinglePortRange(serverPort),
To: uint32(serverPort + 100), Listen: net.NewIPOrDomain(net.LocalHostIP),
}, }),
Listen: net.NewIPOrDomain(net.LocalHostIP), ProxySettings: serial.ToTypedMessage(&inbound.Config{
AllocationStrategy: &proxyman.AllocationStrategy{ User: []*protocol.User{
Type: proxyman.AllocationStrategy_Random, {
Concurrency: &proxyman.AllocationStrategy_AllocationStrategyConcurrency{ Account: serial.ToTypedMessage(&vmess.Account{
Value: 2, Id: userID.String(),
}),
},
}, },
Refresh: &proxyman.AllocationStrategy_AllocationStrategyRefresh{ Detour: &inbound.DetourConfig{
Value: 5, To: "detour",
}, },
}, }),
}), },
ProxySettings: serial.ToTypedMessage(&inbound.Config{}), {
Tag: "detour", ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
PortRange: net.SinglePortRange(serverPort + 100),
Listen: net.NewIPOrDomain(net.LocalHostIP),
}),
ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
Address: net.NewIPOrDomain(dest.Address),
Port: uint32(dest.Port),
NetworkList: &net.NetworkList{
Network: []net.Network{net.Network_TCP},
},
}),
},
{
ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
PortRange: &net.PortRange{
From: uint32(serverPort + 1),
To: uint32(serverPort + 99),
},
Listen: net.NewIPOrDomain(net.LocalHostIP),
AllocationStrategy: &proxyman.AllocationStrategy{
Type: proxyman.AllocationStrategy_Random,
Concurrency: &proxyman.AllocationStrategy_AllocationStrategyConcurrency{
Value: 2,
},
Refresh: &proxyman.AllocationStrategy_AllocationStrategyRefresh{
Value: 5,
},
},
}),
ProxySettings: serial.ToTypedMessage(&inbound.Config{}),
Tag: "detour",
},
}, },
}, Outbound: []*core.OutboundHandlerConfig{
Outbound: []*core.OutboundHandlerConfig{ {
{ ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
ProxySettings: serial.ToTypedMessage(&freedom.Config{}), },
}, },
}, }
server, _ := InitializeServerConfig(serverConfig)
if server != nil && tcpConnAvailableAtPort(t, serverPort + 100) {
defer CloseServer(server)
break
}
retry += 1
if (retry > 5) {
t.Fatal("All attempts failed to start server")
}
serverPort = tcp.PickPort()
} }
clientPort := tcp.PickPort() clientPort := tcp.PickPort()
@ -135,15 +163,30 @@ func TestVMessDynamicPort(t *testing.T) {
}, },
} }
servers, err := InitializeServerConfigs(serverConfig, clientConfig) server, err := InitializeServerConfig(clientConfig)
common.Must(err) common.Must(err)
defer CloseAllServers(servers) defer CloseServer(server)
for i := 0; i < 10; i++ { if !tcpConnAvailableAtPort(t, clientPort) {
if err := testTCPConn(clientPort, 1024, time.Second*2)(); err != nil { t.Fail()
t.Error(err) }
}
func tcpConnAvailableAtPort(t *testing.T, port net.Port) bool {
for i := 1; ; i++ {
if (i > 10) {
t.Log("All attempts failed to test tcp conn")
return false
}
time.Sleep(time.Millisecond * 10)
if err := testTCPConn(port, 1024, time.Second*2)(); err != nil {
t.Log("err ", err)
} else {
t.Log("success with", i, "attempts")
break
} }
} }
return true
} }
func TestVMessGCM(t *testing.T) { func TestVMessGCM(t *testing.T) {