mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-11-13 20:23:19 +00:00
Fix flaky TestVMessDynamicPort (#723)
This commit is contained in:
parent
a149c78a4c
commit
9f9059c7b1
|
@ -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{}))
|
||||||
|
|
|
@ -36,7 +36,10 @@ 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()
|
||||||
|
for {
|
||||||
serverConfig := &core.Config{
|
serverConfig := &core.Config{
|
||||||
App: []*serial.TypedMessage{
|
App: []*serial.TypedMessage{
|
||||||
serial.ToTypedMessage(&log.Config{
|
serial.ToTypedMessage(&log.Config{
|
||||||
|
@ -63,11 +66,24 @@ func TestVMessDynamicPort(t *testing.T) {
|
||||||
},
|
},
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
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{
|
ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
|
||||||
PortRange: &net.PortRange{
|
PortRange: &net.PortRange{
|
||||||
From: uint32(serverPort + 1),
|
From: uint32(serverPort + 1),
|
||||||
To: uint32(serverPort + 100),
|
To: uint32(serverPort + 99),
|
||||||
},
|
},
|
||||||
Listen: net.NewIPOrDomain(net.LocalHostIP),
|
Listen: net.NewIPOrDomain(net.LocalHostIP),
|
||||||
AllocationStrategy: &proxyman.AllocationStrategy{
|
AllocationStrategy: &proxyman.AllocationStrategy{
|
||||||
|
@ -91,6 +107,18 @@ func TestVMessDynamicPort(t *testing.T) {
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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()
|
||||||
clientConfig := &core.Config{
|
clientConfig := &core.Config{
|
||||||
App: []*serial.TypedMessage{
|
App: []*serial.TypedMessage{
|
||||||
|
@ -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) {
|
||||||
|
|
Loading…
Reference in a new issue