refine kcp test

testing/scenarios/vmess_test.go

@@ -775,16 +775,20 @@ func TestVMessKCP(t *testing.T) {
 
 	servers, err := InitializeServerConfigs(serverConfig, clientConfig)
 	assert(err, IsNil)
+	defer CloseAllServers(servers)
 
 	var wg sync.WaitGroup
 	for i := 0; i < 10; i++ {
 		wg.Add(1)
 		go func() {
+			defer wg.Done()
+
 			conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
 				IP:   []byte{127, 0, 0, 1},
 				Port: int(clientPort),
 			})
 			assert(err, IsNil)
+			defer conn.Close()
 
 			payload := make([]byte, 10240*1024)
 			rand.Read(payload)
@@ -794,14 +798,12 @@ func TestVMessKCP(t *testing.T) {
 			assert(nBytes, Equals, len(payload))
 
 			response := readFrom(conn, time.Minute*2, 10240*1024)
-			assert(response, Equals, xor(payload))
+			if err := compare.BytesEqualWithDetail(response, xor(payload)); err != nil {
-			assert(conn.Close(), IsNil)
+				t.Error(err)
-			wg.Done()
+			}
 		}()
 	}
 	wg.Wait()
-
-	CloseAllServers(servers)
 }
 
 func TestVMessKCPLarge(t *testing.T) {