/**

* Copyright 2012-2020 The Feign Authors

*

* Licensed 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.

*/

package feign;

import org.junit.Assert;

import org.junit.Rule;

import org.junit.Test;

import org.junit.rules.ExpectedException;

import java.util.Collections;

import java.util.Date;

import feign.Retryer.Default;

import static org.junit.Assert.assertEquals;

@SuppressWarnings("deprecation")

public class RetryerTest {

@Rule

public final ExpectedException thrown = ExpectedException.none();

private final static Request REQUEST = Request

.create(Request.HttpMethod.GET, "/", Collections.emptyMap(), null, Util.UTF_8, null);

@Test

public void only5TriesAllowedAndExponentialBackoff() throws Exception {

RetryableException e = new RetryableException(-1, null, null, null, REQUEST);

Default retryer = new Retryer.Default();

assertEquals(1, retryer.attempt);

assertEquals(0, retryer.sleptForMillis);

retryer.continueOrPropagate(e);

assertEquals(2, retryer.attempt);

assertEquals(150, retryer.sleptForMillis);

retryer.continueOrPropagate(e);

assertEquals(3, retryer.attempt);

assertEquals(375, retryer.sleptForMillis);

retryer.continueOrPropagate(e);

assertEquals(4, retryer.attempt);

assertEquals(712, retryer.sleptForMillis);

retryer.continueOrPropagate(e);

assertEquals(5, retryer.attempt);

assertEquals(1218, retryer.sleptForMillis);

thrown.expect(RetryableException.class);

retryer.continueOrPropagate(e);

}

@Test

public void considersRetryAfterButNotMoreThanMaxPeriod() {

Default retryer = new Retryer.Default() {

protected long currentTimeMillis() {

return 0;

}

};

retryer.continueOrPropagate(new RetryableException(-1, null, null, new Date(5000), REQUEST));

assertEquals(2, retryer.attempt);

assertEquals(1000, retryer.sleptForMillis);

}

@Test(expected = RetryableException.class)

public void neverRetryAlwaysPropagates() {

Retryer.NEVER_RETRY

.continueOrPropagate(new RetryableException(-1, null, null, new Date(5000), REQUEST));

}

@Test

public void defaultRetryerFailsOnInterruptedException() {

Default retryer = new Retryer.Default();

Thread.currentThread().interrupt();

RetryableException expected =

new RetryableException(-1, null, null, new Date(System.currentTimeMillis() + 5000),

REQUEST);

try {

retryer.continueOrPropagate(expected);

Thread.interrupted(); // reset interrupted flag in case it wasn't

Assert.fail("Retryer continued despite interruption");

} catch (RetryableException e) {

Assert.assertTrue("Interrupted status not reset", Thread.interrupted());

Assert.assertEquals("Retry attempt not registered as expected", 2, retryer.attempt);

Assert.assertEquals("Unexpected exception found", expected, e);

}

}

}