YARN-9311. Fix TestRMRestart hangs. Contributed by Prabhu Joseph.

This commit is contained in:
Rohith Sharma K S 2019-02-27 10:28:16 +05:30
parent 625e93713b
commit 8eae260af5
1 changed files with 3 additions and 1 deletions

View File

@ -100,6 +100,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStore;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStore.RMState;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreAMRMTokenEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreProxyCAEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStoreRMDTMasterKeyEvent;
import org.apache.hadoop.yarn.server.resourcemanager.recovery.records.ApplicationAttemptStateData;
@ -1646,7 +1647,8 @@ public class TestRMRestart extends ParameterizedSchedulerTestBase {
// Skip if synchronous updation of DTToken
if (!(event instanceof RMStateStoreAMRMTokenEvent)
&& !(event instanceof RMStateStoreRMDTEvent)
&& !(event instanceof RMStateStoreRMDTMasterKeyEvent)) {
&& !(event instanceof RMStateStoreRMDTMasterKeyEvent)
&& !(event instanceof RMStateStoreProxyCAEvent)) {
while (wait);
}
super.handleStoreEvent(event);