big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YARN-8904. TestRMDelegationTokens can fail in testRMDTMasterKeyStateOnRollingMasterKey. Contributed by Wilfred Spiegelenburg

Browse Source
This commit is contained in:
Jason Lowe 2018-10-23 12:49:15 -05:00
parent dd268a64d3
commit 93fb3b4b9c
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/security/TestRMDelegationTokens.java
View File

@ -97,7 +97,17 @@ public void testRMDTMasterKeyStateOnRollingMasterKey() throws Exception {
RMDelegationTokenSecretManager dtSecretManager =
rm1.getRMContext().getRMDelegationTokenSecretManager();
// assert all master keys are saved
Assert.assertEquals(dtSecretManager.getAllMasterKeys(), rmDTMasterKeyState);
dtSecretManager.getAllMasterKeys().forEach(managerKey -> {
int keyId = managerKey.getKeyId();
boolean found = false;
for (DelegationKey stateKey: rmDTMasterKeyState) {
if (stateKey.getKeyId() == keyId) {
found = true;
break;
}
}
Assert.assertTrue("Master key not found: " + keyId, found);
});
// request to generate a RMDelegationToken
GetDelegationTokenRequest request = mock(GetDelegationTokenRequest.class);