mirror of https://github.com/v2ray/v2ray-core
Fix test flakiness
parent
e3a00d1400
commit
4be7cd7908
|
@ -68,17 +68,21 @@ func TestTCPConnection(t *testing.T) {
|
||||||
assert.Error(err).IsNil()
|
assert.Error(err).IsNil()
|
||||||
assert.Int(nBytes).Equals(len(actualRequest))
|
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.")
|
actualRequest = []byte("Request to target server again.")
|
||||||
nBytes, err = conn.Write(actualRequest)
|
nBytes, err = conn.Write(actualRequest)
|
||||||
assert.Error(err).IsNil()
|
assert.Error(err).IsNil()
|
||||||
assert.Int(nBytes).Equals(len(actualRequest))
|
assert.Int(nBytes).Equals(len(actualRequest))
|
||||||
|
|
||||||
|
nBytes, err = conn.Read(actualResponse[nResponse:])
|
||||||
|
assert.Error(err).IsNil()
|
||||||
|
nResponse += nBytes
|
||||||
conn.CloseWrite()
|
conn.CloseWrite()
|
||||||
|
|
||||||
actualResponse := make([]byte, 1024)
|
assert.StringLiteral(string(actualResponse[:nResponse])).Equals("Processed: Request to target server.Processed: Request to target server again.")
|
||||||
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.")
|
|
||||||
|
|
||||||
conn.Close()
|
conn.Close()
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue