From c32f1a015268e7538c27625a9cf6ea78780cf816 Mon Sep 17 00:00:00 2001 From: v2ray Date: Thu, 14 Jul 2016 21:31:04 +0200 Subject: [PATCH] update test case --- transport/internet/kcp/connection_test.go | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/transport/internet/kcp/connection_test.go b/transport/internet/kcp/connection_test.go index e53d6750..03fe6cde 100644 --- a/transport/internet/kcp/connection_test.go +++ b/transport/internet/kcp/connection_test.go @@ -49,16 +49,25 @@ func TestConnectionReadWrite(t *testing.T) { totalWritten := 1024 * 1024 clientSend := make([]byte, totalWritten) rand.Read(clientSend) - nBytes, err := connClient.Write(clientSend) - assert.Int(nBytes).Equals(totalWritten) - assert.Error(err).IsNil() + go func() { + nBytes, err := connClient.Write(clientSend) + assert.Int(nBytes).Equals(totalWritten) + assert.Error(err).IsNil() + }() serverReceived := make([]byte, totalWritten) totalRead := 0 for totalRead < totalWritten { - nBytes, err = connServer.Read(serverReceived[totalRead:]) + nBytes, err := connServer.Read(serverReceived[totalRead:]) assert.Error(err).IsNil() totalRead += nBytes } assert.Bytes(serverReceived).Equals(clientSend) + + connClient.Close() + connServer.Close() + + for connClient.State() != StateTerminated || connServer.State() != StateTerminated { + time.Sleep(time.Second) + } }