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

YARN-8840. Add missing cleanupSSLConfig() call for TestTimelineClient test. Contributed by Aki Tanaka.

Browse Source
This commit is contained in:
Akira Ajisaka 2018-10-04 09:51:42 +09:00
parent 1dc0adfac0
commit 39b35036ba
No known key found for this signature in database
GPG Key ID: C1EDBB9CA400FD50
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/client/api/impl/TestTimelineClient.java
View File

@ -25,7 +25,6 @@
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;
import java.io.File;
import java.io.IOException;
import java.net.ConnectException;
import java.net.SocketTimeoutException;
@ -62,6 +61,8 @@ public class TestTimelineClient {
private TimelineClientImpl client;
private TimelineWriter spyTimelineWriter;
private String keystoresDir;
private String sslConfDir;
@Before
public void setup() {
@ -72,10 +73,13 @@ public void setup() {
}
@After
public void tearDown() {
public void tearDown() throws Exception {
if (client != null) {
client.stop();
}
if (isSSLConfigured()) {
KeyStoreTestUtil.cleanupSSLConfig(keystoresDir, sslConfDir);
}
}
@Test
@ -454,11 +458,7 @@ public void testTimelineClientCleanup() throws Exception {
conf.setInt(YarnConfiguration.TIMELINE_SERVICE_CLIENT_MAX_RETRIES, 0);
conf.set(YarnConfiguration.YARN_HTTP_POLICY_KEY, Policy.HTTPS_ONLY.name());
File testDir = TestGenericTestUtils.getTestDir();
String sslConfDir =
KeyStoreTestUtil.getClasspathDir(TestTimelineClient.class);
KeyStoreTestUtil.setupSSLConfig(testDir.getAbsolutePath(),
sslConfDir, conf, false);
setupSSLConfig(conf);
client = createTimelineClient(conf);
ThreadGroup threadGroup = Thread.currentThread().getThreadGroup();
@ -492,6 +492,17 @@ public void testTimelineClientCleanup() throws Exception {
Assert.assertFalse("Reloader is still alive", reloaderStillAlive);
}
private void setupSSLConfig(YarnConfiguration conf) throws Exception {
keystoresDir = TestGenericTestUtils.getTestDir().getAbsolutePath();
sslConfDir =
KeyStoreTestUtil.getClasspathDir(TestTimelineClient.class);
KeyStoreTestUtil.setupSSLConfig(keystoresDir, sslConfDir, conf, false);
}
private boolean isSSLConfigured() {
return keystoresDir != null && sslConfDir != null;
}
private static class TestTimelineDelegationTokenSecretManager extends
AbstractDelegationTokenSecretManager<TimelineDelegationTokenIdentifier> {