用注释样式的Resilience4j测试SpringBoot

Question

我在我的SpringBoot应用程序“demo”中使用了注释样式的Resilience4j。当通过RestTemplate调用外部后端时，我希望使用TimeLimiter并重试以实现以下目标：

1. 将REST调用持续时间限制在5秒-->如果需要更长时间，则使用TimeoutException失败
2. 在TimeoutException上重试-->最多尝试2次

为了查看弹性设置的配置是否如设计的那样工作，我编写了一个IntegrationTest。该测试在概要文件“test”下运行，并使用“application-test.yml”进行配置：

1. 使用TestRestTemplate向我的“SimpleRestendPointController”发送调用
2. 控制器调用我的业务服务“CallExternalService”，它有一个带注释的方法“getPersonbyId”（注释:@timelimiter,@retry)
3. 在此方法中，使用模拟的RestTemplate调用位于“fancy_url”的外部后端
4. 使用Mockito会减慢对外部后端的RestTemplate调用（使用thread.sleep)
5. 我希望时间限制程序在5秒后取消调用，重试确保再次尝试RestTemplate调用（验证RestTemplate已被调用两次）

问题：TimeLimiter和Retry被注册，但不执行它们的工作（TimeLimiter不限制呼叫持续时间）。因此，RestTemplate只调用一次，传递空的person（请参见代码以获得澄清）。链接的示例项目可以被克隆，并在运行测试时展示问题。

application-test.yml的代码（也在此处：链接到application-test.yml):

resilience4j:
  timelimiter:
    configs:
      default:
        timeoutDuration: 5s
        cancelRunningFuture: true
    instances:
      MY_RESILIENCE_KEY:
        baseConfig: default
  retry:
    configs:
      default:
        maxRetryAttempts: 2
        waitDuration: 100ms
        retryExceptions:
          - java.util.concurrent.TimeoutException
    instances:
      MY_RESILIENCE_KEY:
        baseConfig: default

这个测试的代码（也在这里：integrationTest.java的链接）：

@RunWith(SpringRunner.class)
@SpringBootTest(classes = {DemoApplication.class}, webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@EnableAutoConfiguration
@ActiveProfiles("test")
public class IntegrationTest {
    private TestRestTemplate testRestTemplate;
    public final String FANCY_URL = "https://my-fancy-url-doesnt-matter.com/person";
    private String apiUrl;
    private HttpHeaders headers;
    
    @LocalServerPort
    private String localServerPort;
    
    @MockBean
    RestTemplate restTemplate;
    
    @Autowired
    CallExternalService callExternalService;
    
    @Autowired
    SimpleRestEndpointController simpleRestEndpointController;
    
    @Before
    public void setup() {
        this.headers = new HttpHeaders();
        this.testRestTemplate = new TestRestTemplate("username", "password");
        this.apiUrl = String.format("http://localhost:%s/person", localServerPort);
    }
    
    @Test
    public void testShouldRetryOnceWhenTimelimitIsReached() {
        // Arrange
        Person mockPerson = new Person();
        mockPerson.setId(1);
        mockPerson.setFirstName("First");
        mockPerson.setLastName("Last");
        ResponseEntity<Person> mockResponse = new ResponseEntity<>(mockPerson, HttpStatus.OK);
            
        
        Answer customAnswer = new Answer() {
            private int invocationCount = 0;
            @Override
            public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
                invocationCount++;
                if (invocationCount == 1) {
                    Thread.sleep(6000);
                    return new ResponseEntity<>(new Person(), HttpStatus.OK);
                } else {
                    return mockResponse;
                }
            }
        };
        
        doAnswer(customAnswer)
        .when(restTemplate).exchange(
                FANCY_URL,
                HttpMethod.GET,
                new HttpEntity<>(headers),
                new ParameterizedTypeReference<Person>() {});
        
        
        // Act
        ResponseEntity<Person> result = null;
        try {
            result = this.testRestTemplate.exchange(
                    apiUrl,
                    HttpMethod.GET,
                    new HttpEntity<>(headers),
                    new ParameterizedTypeReference<Person>() {
                    });
        } catch(Exception ex) {
            System.out.println(ex);         
        }
        
        
        // Assert
        verify(restTemplate, times(2)).exchange(
                FANCY_URL,
                HttpMethod.GET,
                new HttpEntity<>(headers),
                new ParameterizedTypeReference<Person>() {});
        Assert.assertNotNull(result);
        Assert.assertEquals(mockPerson, result.getBody());      
        
    }
}

我的应用程序代码显示了这个问题：https://github.com/sidekickjohn/demo

我创建了一个“逻辑”的泳道图，作为readme.md的一部分：https://github.com/sidekickjohn/demo/blob/main/readme.md

Answer 1

如果要模拟由CallExternalService使用的真正的RESTTemplatebean，则必须使用Mockito Spy->https://www.baeldung.com/Mockito-Spy

但我通常更喜欢并建议使用WireMock而不是Mockito来模拟HTTPendpoint。