/*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* <http://www.apache.org/>.
*
*/
package org.apache.http.impl.client.cache;
import org.apache.http.HttpHost;
import org.apache.http.client.methods.HttpRequestWrapper;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.conn.routing.HttpRoute;
import org.apache.http.impl.execchain.ClientExecChain;
import org.apache.http.message.BasicHttpRequest;
import org.easymock.EasyMock;
import org.junit.Before;
import org.junit.Test;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
public class TestExponentialBackingOffSchedulingStrategy {
private ScheduledExecutorService mockExecutor;
private ExponentialBackOffSchedulingStrategy impl;
@Before
public void setUp() {
mockExecutor = EasyMock.createMock(ScheduledExecutorService.class);
impl = new ExponentialBackOffSchedulingStrategy(
mockExecutor,
ExponentialBackOffSchedulingStrategy.DEFAULT_BACK_OFF_RATE,
ExponentialBackOffSchedulingStrategy.DEFAULT_INITIAL_EXPIRY_IN_MILLIS,
ExponentialBackOffSchedulingStrategy.DEFAULT_MAX_EXPIRY_IN_MILLIS
);
}
@Test
public void testScheduleWithoutPreviousError() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(0));
expectRequestScheduledWithoutDelay(request);
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithOneFailedAttempt() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(1));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(6));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithTwoFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(2));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(60));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithThreeFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(3));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(600));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithFourFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(4));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(6000));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithFiveFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(5));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(60000));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithSixFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(6));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(86400));
replayMocks();
impl.schedule(request);
verifyMocks();
}
@Test
public void testScheduleWithMaxNumberOfFailedAttempts() {
AsynchronousValidationRequest request = createAsynchronousValidationRequest(withErrorCount(Integer.MAX_VALUE));
expectRequestScheduledWithDelay(request, TimeUnit.SECONDS.toMillis(86400));
replayMocks();
impl.schedule(request);
verifyMocks();
}
private void expectRequestScheduledWithoutDelay(final AsynchronousValidationRequest request) {
expectRequestScheduledWithDelay(request, 0);
}
private void expectRequestScheduledWithDelay(final AsynchronousValidationRequest request, final long delayInMillis) {
EasyMock.expect(mockExecutor.schedule(request, delayInMillis, TimeUnit.MILLISECONDS)).andReturn(null);
}
private void replayMocks() {
EasyMock.replay(mockExecutor);
}
private void verifyMocks() {
EasyMock.verify(mockExecutor);
}
private AsynchronousValidationRequest createAsynchronousValidationRequest(final int errorCount) {
final ClientExecChain clientExecChain = EasyMock.createNiceMock(ClientExecChain.class);
final CachingExec cachingHttpClient = new CachingExec(clientExecChain);
final AsynchronousValidator mockValidator = new AsynchronousValidator(impl);
final HttpRoute httpRoute = new HttpRoute(new HttpHost("foo.example.com"));
final HttpRequestWrapper httpRequestWrapper = HttpRequestWrapper.wrap(new BasicHttpRequest("GET", "/"));
final HttpClientContext httpClientContext = new HttpClientContext();
return new AsynchronousValidationRequest(mockValidator, cachingHttpClient, httpRoute, httpRequestWrapper,
httpClientContext, null, null, "identifier", errorCount);
}
private static int withErrorCount(final int errorCount) {
return errorCount;
}
}