|
@@ -24,6 +24,8 @@ import static org.mockito.Mockito.doThrow;
|
|
|
import static org.mockito.Mockito.mock;
|
|
|
import static org.mockito.Mockito.spy;
|
|
|
import static org.mockito.Mockito.when;
|
|
|
+import static org.mockito.Mockito.verify;
|
|
|
+import static org.mockito.Mockito.times;
|
|
|
|
|
|
import java.io.IOException;
|
|
|
import java.net.ConnectException;
|
|
@@ -492,6 +494,17 @@ public class TestTimelineClient {
|
|
|
Assert.assertFalse("Reloader is still alive", reloaderStillAlive);
|
|
|
}
|
|
|
|
|
|
+ @Test
|
|
|
+ public void testTimelineConnectorDestroy() {
|
|
|
+ YarnConfiguration conf = new YarnConfiguration();
|
|
|
+ conf.setBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED, true);
|
|
|
+ TimelineClientImpl client = createTimelineClient(conf);
|
|
|
+ Client mockJerseyClient = mock(Client.class);
|
|
|
+ client.connector.client = mockJerseyClient;
|
|
|
+ client.stop();
|
|
|
+ verify(mockJerseyClient, times(1)).destroy();
|
|
|
+ }
|
|
|
+
|
|
|
private void setupSSLConfig(YarnConfiguration conf) throws Exception {
|
|
|
keystoresDir = TestGenericTestUtils.getTestDir().getAbsolutePath();
|
|
|
sslConfDir =
|