diff --git a/profiler/src/test/java/com/navercorp/pinpoint/profiler/modifier/connector/httpclient4/DefaultHttpRequestRetryHandlerModifierTest.java b/profiler/src/test/java/com/navercorp/pinpoint/profiler/modifier/connector/httpclient4/DefaultHttpRequestRetryHandlerModifierTest.java new file mode 100644 index 000000000..47035fa6d --- /dev/null +++ b/profiler/src/test/java/com/navercorp/pinpoint/profiler/modifier/connector/httpclient4/DefaultHttpRequestRetryHandlerModifierTest.java @@ -0,0 +1,27 @@ +package com.nhn.pinpoint.profiler.modifier.connector.httpclient4; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +import java.io.IOException; + +import org.apache.http.impl.client.DefaultHttpRequestRetryHandler; +import org.apache.http.protocol.BasicHttpContext; +import org.apache.http.protocol.HttpContext; +import org.junit.Test; + +import com.nhn.pinpoint.profiler.junit4.BasePinpointTest; + +public class DefaultHttpRequestRetryHandlerModifierTest extends BasePinpointTest { + + @Test + public void test() { + DefaultHttpRequestRetryHandler retryHandler = new DefaultHttpRequestRetryHandler(); + IOException iOException = new IOException(); + HttpContext context = new BasicHttpContext(); + + assertTrue(retryHandler.retryRequest(iOException, 1, context)); + assertTrue(retryHandler.retryRequest(iOException, 2, context)); + assertEquals(2, getCurrentSpanEvents().size()); + } +}