/** * 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); } } }