From f4f346e38717298a29caa2f7be20a1d27f285f1b Mon Sep 17 00:00:00 2001
From: William Yang <mscame@gmail.com>
Date: Wed, 18 Jan 2023 19:57:15 +0100
Subject: [PATCH] test(quic): fix flaky test

---
 .../test/emqx_quic_multistreams_SUITE.erl     | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/apps/emqx/test/emqx_quic_multistreams_SUITE.erl b/apps/emqx/test/emqx_quic_multistreams_SUITE.erl
index dd71b6079..40237e369 100644
--- a/apps/emqx/test/emqx_quic_multistreams_SUITE.erl
+++ b/apps/emqx/test/emqx_quic_multistreams_SUITE.erl
@@ -527,7 +527,7 @@ t_multi_streams_packet_boundary(Config) ->
         [{qos, PubQos}],
         undefined
     ),
-    LargePart3 = binary:copy(<<"stream data3">>, 2000),
+    LargePart3 = binary:copy(atom_to_binary(?FUNCTION_NAME), 20000),
     ok = emqtt:publish_async(
         C,
         PubVia,
@@ -603,7 +603,7 @@ t_multi_streams_packet_malform(Config) ->
         [{qos, PubQos}],
         undefined
     ),
-    LargePart3 = binary:copy(<<"stream data3">>, 2000),
+    LargePart3 = binary:copy(atom_to_binary(?FUNCTION_NAME), 2000),
     ok = emqtt:publish_async(
         C,
         PubVia,
@@ -1221,6 +1221,12 @@ t_multi_streams_shutdown_pub_data_stream(Config) ->
     end,
 
     PubRecvs = recv_pub(1),
+    #{data_stream_socks := [PubVia | _]} = proplists:get_value(extra, emqtt:info(C)),
+    {quic, _Conn, DataStream} = PubVia,
+    quicer:shutdown_stream(DataStream, ?config(stream_shutdown_flag, Config), 500, 100),
+    timer:sleep(500),
+    %% Still alive
+    ?assert(is_list(emqtt:info(C))),
     ?assertMatch(
         [
             {publish, #{
@@ -1231,14 +1237,7 @@ t_multi_streams_shutdown_pub_data_stream(Config) ->
             }}
         ],
         PubRecvs
-    ),
-
-    #{data_stream_socks := [PubVia | _]} = proplists:get_value(extra, emqtt:info(C)),
-    {quic, _Conn, DataStream} = PubVia,
-    quicer:shutdown_stream(DataStream, ?config(stream_shutdown_flag, Config), 500, 100),
-    timer:sleep(500),
-    %% Still alive
-    ?assert(is_list(emqtt:info(C))).
+    ).
 
 t_multi_streams_shutdown_sub_data_stream(Config) ->
     PubQos = ?config(pub_qos, Config),