YARN-4880. Running TestZKRMStateStorePerf with real zookeeper cluster throws NPE. Contributed by Sunil G
This commit is contained in:
parent
818d6b799e
commit
552237d4a3
|
@ -91,7 +91,9 @@ public class TestZKRMStateStorePerf extends RMStateStoreTestBase
|
|||
if (appTokenMgr != null) {
|
||||
appTokenMgr.stop();
|
||||
}
|
||||
curatorTestingServer.stop();
|
||||
if (curatorTestingServer != null) {
|
||||
curatorTestingServer.stop();
|
||||
}
|
||||
}
|
||||
|
||||
private void initStore(String hostPort) {
|
||||
|
@ -99,8 +101,9 @@ public class TestZKRMStateStorePerf extends RMStateStoreTestBase
|
|||
RMContext rmContext = mock(RMContext.class);
|
||||
|
||||
conf = new YarnConfiguration();
|
||||
conf.set(YarnConfiguration.RM_ZK_ADDRESS,
|
||||
optHostPort.or(curatorTestingServer.getConnectString()));
|
||||
conf.set(YarnConfiguration.RM_ZK_ADDRESS, optHostPort
|
||||
.or((curatorTestingServer == null) ? "" : curatorTestingServer
|
||||
.getConnectString()));
|
||||
conf.set(YarnConfiguration.ZK_RM_STATE_STORE_PARENT_PATH, workingZnode);
|
||||
|
||||
store = new ZKRMStateStore();
|
||||
|
|
Loading…
Reference in New Issue