|
@@ -27,6 +27,7 @@ import static org.mockito.Matchers.argThat;
|
|
|
import static org.mockito.Matchers.eq;
|
|
|
import static org.mockito.Mockito.mock;
|
|
|
import static org.mockito.Mockito.never;
|
|
|
+import static org.mockito.Mockito.times;
|
|
|
import static org.mockito.Mockito.verify;
|
|
|
import static org.mockito.Mockito.when;
|
|
|
|
|
@@ -116,6 +117,7 @@ public class TestRMAdminCLI {
|
|
|
|
|
|
YarnConfiguration conf = new YarnConfiguration();
|
|
|
conf.setBoolean(YarnConfiguration.RM_HA_ENABLED, true);
|
|
|
+ conf.set(YarnConfiguration.RM_HA_IDS, "rm1,rm2");
|
|
|
rmAdminCLIWithHAEnabled = new RMAdminCLI(conf) {
|
|
|
|
|
|
@Override
|
|
@@ -259,6 +261,8 @@ public class TestRMAdminCLI {
|
|
|
assertEquals(0, rmAdminCLIWithHAEnabled.run(args));
|
|
|
verify(haadmin).transitionToActive(
|
|
|
any(HAServiceProtocol.StateChangeRequestInfo.class));
|
|
|
+ // HAAdmin#isOtherTargetNodeActive should check state of non-target node.
|
|
|
+ verify(haadmin, times(1)).getServiceStatus();
|
|
|
}
|
|
|
|
|
|
@Test(timeout = 500)
|