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

update test case

pull/215/head
v2ray 2016-07-14 21:31:04 +02:00
parent 991b2703dc
commit c32f1a0152
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
transport/internet/kcp/connection_test.go
View File

@ -49,16 +49,25 @@ func TestConnectionReadWrite(t *testing.T) {
totalWritten := 1024 * 1024 totalWritten := 1024 * 1024
clientSend := make([]byte, totalWritten) clientSend := make([]byte, totalWritten)
rand.Read(clientSend) rand.Read(clientSend)
nBytes, err := connClient.Write(clientSend) go func() {
assert.Int(nBytes).Equals(totalWritten) nBytes, err := connClient.Write(clientSend)
assert.Error(err).IsNil() assert.Int(nBytes).Equals(totalWritten)
assert.Error(err).IsNil()
}()
serverReceived := make([]byte, totalWritten) serverReceived := make([]byte, totalWritten)
totalRead := 0 totalRead := 0
for totalRead < totalWritten { for totalRead < totalWritten {
nBytes, err = connServer.Read(serverReceived[totalRead:]) nBytes, err := connServer.Read(serverReceived[totalRead:])
assert.Error(err).IsNil() assert.Error(err).IsNil()
totalRead += nBytes totalRead += nBytes
} }
assert.Bytes(serverReceived).Equals(clientSend) assert.Bytes(serverReceived).Equals(clientSend)
connClient.Close()
connServer.Close()
for connClient.State() != StateTerminated || connServer.State() != StateTerminated {
time.Sleep(time.Second)
}
} }