mirror of https://github.com/XTLS/Xray-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.
71 lines
2.0 KiB
71 lines
2.0 KiB
package protocol_test |
|
|
|
import ( |
|
"testing" |
|
"time" |
|
|
|
"github.com/xtls/xray-core/common/net" |
|
. "github.com/xtls/xray-core/common/protocol" |
|
) |
|
|
|
func TestServerList(t *testing.T) { |
|
list := NewServerList() |
|
list.AddServer(NewServerSpec(net.TCPDestination(net.LocalHostIP, net.Port(1)), AlwaysValid())) |
|
if list.Size() != 1 { |
|
t.Error("list size: ", list.Size()) |
|
} |
|
list.AddServer(NewServerSpec(net.TCPDestination(net.LocalHostIP, net.Port(2)), BeforeTime(time.Now().Add(time.Second)))) |
|
if list.Size() != 2 { |
|
t.Error("list.size: ", list.Size()) |
|
} |
|
|
|
server := list.GetServer(1) |
|
if server.Destination().Port != 2 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
time.Sleep(2 * time.Second) |
|
server = list.GetServer(1) |
|
if server != nil { |
|
t.Error("server: ", server) |
|
} |
|
|
|
server = list.GetServer(0) |
|
if server.Destination().Port != 1 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
} |
|
|
|
func TestServerPicker(t *testing.T) { |
|
list := NewServerList() |
|
list.AddServer(NewServerSpec(net.TCPDestination(net.LocalHostIP, net.Port(1)), AlwaysValid())) |
|
list.AddServer(NewServerSpec(net.TCPDestination(net.LocalHostIP, net.Port(2)), BeforeTime(time.Now().Add(time.Second)))) |
|
list.AddServer(NewServerSpec(net.TCPDestination(net.LocalHostIP, net.Port(3)), BeforeTime(time.Now().Add(time.Second)))) |
|
|
|
picker := NewRoundRobinServerPicker(list) |
|
server := picker.PickServer() |
|
if server.Destination().Port != 1 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
server = picker.PickServer() |
|
if server.Destination().Port != 2 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
server = picker.PickServer() |
|
if server.Destination().Port != 3 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
server = picker.PickServer() |
|
if server.Destination().Port != 1 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
|
|
time.Sleep(2 * time.Second) |
|
server = picker.PickServer() |
|
if server.Destination().Port != 1 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
server = picker.PickServer() |
|
if server.Destination().Port != 1 { |
|
t.Error("server: ", server.Destination()) |
|
} |
|
}
|
|
|