github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

WsTest:more case

pull/238/head
Shelikhoo 2016-08-15 21:29:15 +08:00
parent 6c8b4f5db3
commit 45d801b922
No known key found for this signature in database
GPG Key ID: 7791BDB0709ABD21
1 changed files with 37 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
transport/internet/ws/ws_test.go
View File

@ -43,6 +43,22 @@ func Test_Connect_wss(t *testing.T) {
conn.Close() conn.Close()
} }
func Test_Connect_wss_1_nil(t *testing.T) {
assert := assert.On(t)
(&Config{Pto: "wss", Path: ""}).Apply()
conn, err := Dial(nil, v2net.TCPDestination(v2net.DomainAddress("echo.websocket.org"), 443))
assert.Error(err).IsNil()
conn.Write([]byte("echo"))
s := make(chan int)
go func() {
buf := make([]byte, 4)
conn.Read(buf)
s <- 0
}()
<-s
conn.Close()
}
func Test_Connect_ws_guess(t *testing.T) { func Test_Connect_ws_guess(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
(&Config{Pto: "", Path: ""}).Apply() (&Config{Pto: "", Path: ""}).Apply()
@ -77,21 +93,20 @@ func Test_Connect_wss_guess(t *testing.T) {
func Test_Connect_wss_guess_fail(t *testing.T) { func Test_Connect_wss_guess_fail(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
go func() {
<-time.After(time.Second * 5)
assert.Fail("Too slow")
}()
(&Config{Pto: "", Path: ""}).Apply() (&Config{Pto: "", Path: ""}).Apply()
_, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("static.kkdev.org"), 443)) _, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("static.kkdev.org"), 443))
assert.Error(err).IsNotNil() assert.Error(err).IsNotNil()
} }
func Test_Connect_wss_guess_fail_port(t *testing.T) {
assert := assert.On(t)
(&Config{Pto: "", Path: ""}).Apply()
_, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("static.kkdev.org"), 179))
assert.Error(err).IsNotNil()
}
func Test_Connect_wss_guess_reuse(t *testing.T) { func Test_Connect_wss_guess_reuse(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
go func() {
<-time.After(time.Second * 5)
assert.Fail("Too slow")
}()
(&Config{Pto: "", Path: "", ConnectionReuse: true}).Apply() (&Config{Pto: "", Path: "", ConnectionReuse: true}).Apply()
i := 3 i := 3
for i != 0 { for i != 0 {
@ -118,10 +133,6 @@ func Test_Connect_wss_guess_reuse(t *testing.T) {
func Test_listenWSAndDial(t *testing.T) { func Test_listenWSAndDial(t *testing.T) {
assert := assert.On(t) assert := assert.On(t)
go func() {
<-time.After(time.Second * 5)
assert.Fail("Too slow")
}()
(&Config{Pto: "ws", Path: "ws"}).Apply() (&Config{Pto: "ws", Path: "ws"}).Apply()
listen, err := ListenWS(v2net.DomainAddress("localhost"), 13142) listen, err := ListenWS(v2net.DomainAddress("localhost"), 13142)
assert.Error(err).IsNil() assert.Error(err).IsNil()
@ -129,11 +140,25 @@ func Test_listenWSAndDial(t *testing.T) {
conn, err := listen.Accept() conn, err := listen.Accept()
assert.Error(err).IsNil() assert.Error(err).IsNil()
conn.Close() conn.Close()
conn, err = listen.Accept()
assert.Error(err).IsNil()
conn.Close()
conn, err = listen.Accept()
assert.Error(err).IsNil()
conn.Close()
listen.Close() listen.Close()
}() }()
conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13142)) conn, err := Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13142))
assert.Error(err).IsNil() assert.Error(err).IsNil()
conn.Close() conn.Close()
<-time.After(time.Second * 5)
conn, err = Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13142))
assert.Error(err).IsNil()
conn.Close()
<-time.After(time.Second * 15)
conn, err = Dial(v2net.AnyIP, v2net.TCPDestination(v2net.DomainAddress("localhost"), 13142))
assert.Error(err).IsNil()
conn.Close()
} }
func Test_listenWSAndDial_TLS(t *testing.T) { func Test_listenWSAndDial_TLS(t *testing.T) {