public virtual void TestGetAMRMTokenService()
{
string defaultRMAddress = YarnConfiguration.DefaultRmSchedulerAddress;
YarnConfiguration conf = new YarnConfiguration();
// HA is not enabled
Text tokenService = ClientRMProxy.GetAMRMTokenService(conf);
string[] services = tokenService.ToString().Split(",");
NUnit.Framework.Assert.AreEqual(1, services.Length);
foreach (string service in services)
{
NUnit.Framework.Assert.IsTrue("Incorrect token service name", service.Contains(defaultRMAddress
));
}
// HA is enabled
conf.SetBoolean(YarnConfiguration.RmHaEnabled, true);
conf.Set(YarnConfiguration.RmHaIds, "rm1,rm2");
conf.Set(HAUtil.AddSuffix(YarnConfiguration.RmHostname, "rm1"), "0.0.0.0");
conf.Set(HAUtil.AddSuffix(YarnConfiguration.RmHostname, "rm2"), "0.0.0.0");
tokenService = ClientRMProxy.GetAMRMTokenService(conf);
services = tokenService.ToString().Split(",");
NUnit.Framework.Assert.AreEqual(2, services.Length);
foreach (string service_1 in services)
{
NUnit.Framework.Assert.IsTrue("Incorrect token service name", service_1.Contains(
defaultRMAddress));
}
}
public virtual void Initialize()
{
StartHACluster(0, false, false, true);
attemptId = this.cluster.CreateFakeApplicationAttemptId();
amClient = ClientRMProxy.CreateRMProxy <ApplicationMasterProtocol>(this.conf);
Org.Apache.Hadoop.Security.Token.Token <AMRMTokenIdentifier> appToken = this.cluster
.GetResourceManager().GetRMContext().GetAMRMTokenSecretManager().CreateAndGetAMRMToken
(attemptId);
appToken.SetService(ClientRMProxy.GetAMRMTokenService(conf));
UserGroupInformation.SetLoginUser(UserGroupInformation.CreateRemoteUser(UserGroupInformation
.GetCurrentUser().GetUserName()));
UserGroupInformation.GetCurrentUser().AddToken(appToken);
SyncToken(appToken);
}
