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