如何对使用spring重试机制的方法进行单元测试

import org.springframework.retry.annotation.Retryable;
import org.springframework.stereotype.Service;

@Service
public class MyService {

    @Retryable(value = {Exception.class}, maxAttempts = 3)
    public String performOperation() throws Exception {
        // 模拟可能会失败的操作
        throw new Exception("Operation failed");
    }
}

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.retry.support.RetryTemplate;

import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
public class MyServiceTest {

    @InjectMocks
    private MyService myService;

    @Mock
    private RetryTemplate retryTemplate;

    @Test
    public void testPerformOperationWithRetry() throws Exception {
        // 配置RetryTemplate模拟重试行为
        when(retryTemplate.execute(any(RetryCallback.class), any(RecoverCallback.class)))
            .thenThrow(new Exception("Operation failed"))
            .thenThrow(new Exception("Operation failed"))
            .thenReturn("Operation succeeded");

        // 调用方法并验证结果
        String result = myService.performOperation();
        assertEquals("Operation succeeded", result);

        // 验证RetryTemplate.execute方法被调用了三次
        verify(retryTemplate, times(3)).execute(any(RetryCallback.class), any(RecoverCallback.class));
    }
}