|
@@ -18,7 +18,6 @@
|
|
|
package org.apache.hadoop.yarn.server.resourcemanager;
|
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
-import static org.junit.Assert.assertTrue;
|
|
|
import static org.junit.Assert.fail;
|
|
|
import static org.mockito.ArgumentMatchers.any;
|
|
|
import static org.mockito.ArgumentMatchers.anyString;
|
|
@@ -36,6 +35,7 @@ import java.net.InetSocketAddress;
|
|
|
import java.security.PrivilegedAction;
|
|
|
import java.security.PrivilegedExceptionAction;
|
|
|
|
|
|
+import org.apache.hadoop.test.LambdaTestUtils;
|
|
|
import org.apache.hadoop.thirdparty.protobuf.InvalidProtocolBufferException;
|
|
|
import org.apache.hadoop.io.DataOutputBuffer;
|
|
|
import org.apache.hadoop.net.NetUtils;
|
|
@@ -112,7 +112,7 @@ public class TestClientRMTokens {
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
- public void testDelegationToken() throws IOException, InterruptedException {
|
|
|
+ public void testDelegationToken() throws Exception {
|
|
|
|
|
|
final YarnConfiguration conf = new YarnConfiguration();
|
|
|
conf.set(YarnConfiguration.RM_PRINCIPAL, "testuser/localhost@apache.org");
|
|
@@ -199,14 +199,11 @@ public class TestClientRMTokens {
|
|
|
}
|
|
|
Thread.sleep(50l);
|
|
|
LOG.info("At time: " + System.currentTimeMillis() + ", token should be invalid");
|
|
|
- // Token should have expired.
|
|
|
- try {
|
|
|
- clientRMWithDT.getNewApplication(request);
|
|
|
- fail("Should not have succeeded with an expired token");
|
|
|
- } catch (Exception e) {
|
|
|
- assertEquals(InvalidToken.class.getName(), e.getClass().getName());
|
|
|
- assertTrue(e.getMessage().contains("is expired"));
|
|
|
- }
|
|
|
+ // Token should have expired.
|
|
|
+ final ApplicationClientProtocol finalClientRMWithDT = clientRMWithDT;
|
|
|
+ final GetNewApplicationRequest finalRequest = request;
|
|
|
+ LambdaTestUtils.intercept(InvalidToken.class, "Token has expired",
|
|
|
+ () -> finalClientRMWithDT.getNewApplication(finalRequest));
|
|
|
|
|
|
// Test cancellation
|
|
|
// Stop the existing proxy, start another.
|