(incubator-uniffle) 02/03: [#1254][FOLLOWUP] fix(test): Fix the flaky test RssShuffleTest. (#1259)

[xi...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

xianjingfeng pushed a commit to branch branch-0.8
in repository https://gitbox.apache.org/repos/asf/incubator-uniffle.git

commit 657919e0cedea1eed00f57ca40a42b5bfa7db2fb
Author: zhengchenyu <zh...@163.com>
AuthorDate: Tue Oct 24 14:05:02 2023 +0800

    [#1254][FOLLOWUP] fix(test): Fix the flaky test RssShuffleTest. (#1259)
    
    ### What changes were proposed in this pull request?
    
    ### Why are the changes needed?
    
    Fix: #1254
    
    ### Does this PR introduce _any_ user-facing change?
    
    No.
    
    ### How was this patch tested?
    
    unit test
    
    (cherry picked from commit 3071099338ad76722745468391871d5124adef54)
---
 .../runtime/library/common/shuffle/orderedgrouped/RssShuffleTest.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client-tez/src/test/java/org/apache/tez/runtime/library/common/shuffle/orderedgrouped/RssShuffleTest.java b/client-tez/src/test/java/org/apache/tez/runtime/library/common/shuffle/orderedgrouped/RssShuffleTest.java
index b352f1015..39462bf3a 100644
--- a/client-tez/src/test/java/org/apache/tez/runtime/library/common/shuffle/orderedgrouped/RssShuffleTest.java
+++ b/client-tez/src/test/java/org/apache/tez/runtime/library/common/shuffle/orderedgrouped/RssShuffleTest.java
@@ -87,11 +87,11 @@ public class RssShuffleTest {
       conf.setLong(Constants.TEZ_RUNTIME_TASK_MEMORY, 300000L);
       RssShuffle shuffle = new RssShuffle(inputContext, conf, 1, 3000000L, 0, APPATTEMPT_ID);
       try {
-        shuffle.run();
         ShuffleScheduler scheduler = shuffle.rssScheduler;
         MergeManager mergeManager = shuffle.merger;
         assertFalse(scheduler.isShutdown());
         assertFalse(mergeManager.isShutdown());
+        shuffle.run();
 
         String exceptionMessage = "Simulating fetch failure";
         shuffle.reportException(new IOException(exceptionMessage));