|
@@ -25,7 +25,6 @@ import static org.mockito.Mockito.mock;
|
|
|
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 class TestTimelineClient {
|
|
|
}
|
|
|
|
|
|
@After
|
|
|
- public void tearDown() {
|
|
|
+ public void tearDown() throws Exception {
|
|
|
if (client != null) {
|
|
|
client.stop();
|
|
|
}
|
|
|
+ if (isSSLConfigured()) {
|
|
|
+ KeyStoreTestUtil.cleanupSSLConfig(keystoresDir, sslConfDir);
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -453,11 +457,7 @@ public class TestTimelineClient {
|
|
|
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();
|
|
@@ -491,6 +491,17 @@ public class TestTimelineClient {
|
|
|
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> {
|
|
|
|