fix mux test

pull/298/merge
Darien Raymond 8 years ago
parent dc76e36ed7
commit 0210a718f6
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169

@ -30,6 +30,18 @@ func pickPort() v2net.Port {
return v2net.Port(addr.Port) return v2net.Port(addr.Port)
} }
func pickUDPPort() v2net.Port {
conn, err := net.ListenUDP("udp4", &net.UDPAddr{
IP: v2net.LocalHostIP.IP(),
Port: 0,
})
common.Must(err)
defer conn.Close()
addr := conn.LocalAddr().(*net.UDPAddr)
return v2net.Port(addr.Port)
}
func xor(b []byte) []byte { func xor(b []byte) []byte {
r := make([]byte, len(b)) r := make([]byte, len(b))
for i, v := range b { for i, v := range b {

@ -1095,7 +1095,7 @@ func TestVMessGCMMuxUDP(t *testing.T) {
} }
clientPort := pickPort() clientPort := pickPort()
clientUDPPort := pickPort() clientUDPPort := pickUDPPort()
clientConfig := &core.Config{ clientConfig := &core.Config{
Inbound: []*proxyman.InboundHandlerConfig{ Inbound: []*proxyman.InboundHandlerConfig{
{ {
@ -1203,7 +1203,7 @@ func TestVMessGCMMuxUDP(t *testing.T) {
payload := make([]byte, 1024) payload := make([]byte, 1024)
rand.Read(payload) rand.Read(payload)
for j := 0; j < 10; j++ { for j := 0; j < 5; j++ {
nBytes, _, err := conn.WriteMsgUDP(payload, nil, nil) nBytes, _, err := conn.WriteMsgUDP(payload, nil, nil)
assert.Error(err).IsNil() assert.Error(err).IsNil()
assert.Int(nBytes).Equals(len(payload)) assert.Int(nBytes).Equals(len(payload))
@ -1211,7 +1211,7 @@ func TestVMessGCMMuxUDP(t *testing.T) {
response := make([]byte, 1024) response := make([]byte, 1024)
oob := make([]byte, 16) oob := make([]byte, 16)
for j := 0; j < 10; j++ { for j := 0; j < 5; j++ {
nBytes, _, _, _, err := conn.ReadMsgUDP(response, oob) nBytes, _, _, _, err := conn.ReadMsgUDP(response, oob)
assert.Error(err).IsNil() assert.Error(err).IsNil()
assert.Int(nBytes).Equals(1024) assert.Int(nBytes).Equals(1024)

Loading…
Cancel
Save