private void InitStore(string hostPort) { Optional <string> optHostPort = Optional.FromNullable(hostPort); RMContext rmContext = Org.Mockito.Mockito.Mock <RMContext>(); conf = new YarnConfiguration(); conf.Set(YarnConfiguration.RmZkAddress, optHostPort.Or(this.hostPort)); conf.Set(YarnConfiguration.ZkRmStateStoreParentPath, workingZnode); store = new ZKRMStateStore(); store.Init(conf); store.Start(); Org.Mockito.Mockito.When(rmContext.GetStateStore()).ThenReturn(store); appTokenMgr = new AMRMTokenSecretManager(conf, rmContext); appTokenMgr.Start(); clientToAMTokenMgr = new ClientToAMTokenSecretManagerInRM(); }