delegation java_Java DelegationTokenToRenew类代码示例

import org.apache.hadoop.yarn.server.resourcemanager.security.DelegationTokenRenewer.DelegationTokenToRenew; //导入依赖的package包/类

@Test (timeout = 30000)

public void testCancelWithMultipleAppSubmissions() throws Exception{

MockRM rm = new TestSecurityMockRM(conf, null);

rm.start();

final MockNM nm1 =

new MockNM("127.0.0.1:1234", 15120, rm.getResourceTrackerService());

nm1.registerNode();

//MyFS fs = (MyFS)FileSystem.get(conf);

//MyToken token1 = fs.getDelegationToken("user123");

// create Token1:

Text userText1 = new Text("user");

DelegationTokenIdentifier dtId1 =

new DelegationTokenIdentifier(userText1, new Text("renewer1"),

userText1);

final Token token1 =

new Token(dtId1.getBytes(),

"password1".getBytes(), dtId1.getKind(), new Text("service1"));

Credentials credentials = new Credentials();

credentials.addToken(token1.getService(), token1);

DelegationTokenRenewer renewer =

rm.getRMContext().getDelegationTokenRenewer();

Assert.assertTrue(renewer.getAllTokens().isEmpty());

Assert.assertFalse(Renewer.cancelled);

Resource resource = Records.newRecord(Resource.class);

resource.setMemory(200);

RMApp app1 =

rm.submitApp(resource, "name", "user", null, false, null, 2, credentials,

null, true, false, false, null, 0, null, true, null);

MockAM am1 = MockRM.launchAndRegisterAM(app1, rm, nm1);

rm.waitForState(app1.getApplicationId(), RMAppState.RUNNING);

DelegationTokenToRenew dttr = renewer.getAllTokens().get(token1);

Assert.assertNotNull(dttr);

Assert.assertTrue(dttr.referringAppIds.contains(app1.getApplicationId()));

RMApp app2 = rm.submitApp(resource, "name", "user", null, false, null, 2,

credentials, null, true, false, false, null, 0, null, true, null);

MockAM am2 = MockRM.launchAndRegisterAM(app2, rm, nm1);

rm.waitForState(app2.getApplicationId(), RMAppState.RUNNING);

Assert.assertTrue(renewer.getAllTokens().containsKey(token1));

Assert.assertTrue(dttr.referringAppIds.contains(app2.getApplicationId()));

Assert.assertTrue(dttr.referringAppIds.contains(app2.getApplicationId()));

Assert.assertFalse(Renewer.cancelled);

MockRM.finishAMAndVerifyAppState(app2, rm, nm1, am2);

// app2 completes, app1 is still running, check the token is not cancelled

Assert.assertTrue(renewer.getAllTokens().containsKey(token1));

Assert.assertTrue(dttr.referringAppIds.contains(app1.getApplicationId()));

Assert.assertFalse(dttr.referringAppIds.contains(app2.getApplicationId()));

Assert.assertFalse(dttr.isTimerCancelled());

Assert.assertFalse(Renewer.cancelled);

RMApp app3 = rm.submitApp(resource, "name", "user", null, false, null, 2,

credentials, null, true, false, false, null, 0, null, true, null);

MockAM am3 = MockRM.launchAndRegisterAM(app3, rm, nm1);

rm.waitForState(app3.getApplicationId(), RMAppState.RUNNING);

Assert.assertTrue(renewer.getAllTokens().containsKey(token1));

Assert.assertTrue(dttr.referringAppIds.contains(app1.getApplicationId()));

Assert.assertTrue(dttr.referringAppIds.contains(app3.getApplicationId()));

Assert.assertFalse(dttr.isTimerCancelled());

Assert.assertFalse(Renewer.cancelled);

MockRM.finishAMAndVerifyAppState(app1, rm, nm1, am1);

Assert.assertTrue(renewer.getAllTokens().containsKey(token1));

Assert.assertFalse(dttr.referringAppIds.contains(app1.getApplicationId()));

Assert.assertTrue(dttr.referringAppIds.contains(app3.getApplicationId()));

Assert.assertFalse(dttr.isTimerCancelled());

Assert.assertFalse(Renewer.cancelled);

MockRM.finishAMAndVerifyAppState(app3, rm, nm1, am3);

Assert.assertFalse(renewer.getAllTokens().containsKey(token1));

Assert.assertTrue(dttr.referringAppIds.isEmpty());

Assert.assertTrue(dttr.isTimerCancelled());

Assert.assertTrue(Renewer.cancelled);

}