Fix test flakiness

pull/62/head
v2ray 2015-12-16 16:42:54 +01:00
parent e3a00d1400
commit 4be7cd7908
1 changed files with 8 additions and 4 deletions

View File

@ -68,17 +68,21 @@ func TestTCPConnection(t *testing.T) {
assert.Error(err).IsNil()
assert.Int(nBytes).Equals(len(actualRequest))
actualResponse := make([]byte, 1024)
nResponse, err := conn.Read(actualResponse)
assert.Error(err).IsNil()
actualRequest = []byte("Request to target server again.")
nBytes, err = conn.Write(actualRequest)
assert.Error(err).IsNil()
assert.Int(nBytes).Equals(len(actualRequest))
nBytes, err = conn.Read(actualResponse[nResponse:])
assert.Error(err).IsNil()
nResponse += nBytes
conn.CloseWrite()
actualResponse := make([]byte, 1024)
nBytes, err = conn.Read(actualResponse)
assert.Error(err).IsNil()
assert.StringLiteral(string(actualResponse[:nBytes])).Equals("Processed: Request to target server.Request to target server again.")
assert.StringLiteral(string(actualResponse[:nResponse])).Equals("Processed: Request to target server.Processed: Request to target server again.")
conn.Close()
}