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

(cherry picked from commit 93fb3b4b9c)
This commit is contained in:
Jason Lowe 2018-10-23 12:49:15 -05:00
parent 660fff3138
commit 7097755925
1 changed files with 11 additions and 1 deletions

View File

@ -97,7 +97,17 @@ public class TestRMDelegationTokens {
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);