模拟多个RestTemplate时

import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.web.client.RestTemplate;

import static org.mockito.Mockito.*;

@SpringBootTest
public class MyTest {

    @Mock
    private RestTemplate restTemplate1;

    @Mock
    private RestTemplate restTemplate2;

    @Test
    public void testMultipleRestTemplate() {
        MockitoAnnotations.initMocks(this);

        // 模拟restTemplate1的行为
        when(restTemplate1.getForObject(eq("http://example.com/api1"), any())).thenReturn("Response from restTemplate1");

        // 模拟restTemplate2的行为
        when(restTemplate2.getForObject(eq("http://example.com/api2"), any())).thenReturn("Response from restTemplate2");

        // 测试使用restTemplate1的逻辑
        String response1 = restTemplate1.getForObject("http://example.com/api1", String.class);
        System.out.println(response1);  // Output: "Response from restTemplate1"

        // 测试使用restTemplate2的逻辑
        String response2 = restTemplate2.getForObject("http://example.com/api2", String.class);
        System.out.println(response2);  // Output: "Response from restTemplate2"

        // 验证restTemplate1的行为是否被调用
        verify(restTemplate1, times(1)).getForObject(eq("http://example.com/api1"), any());

        // 验证restTemplate2的行为是否被调用
        verify(restTemplate2, times(1)).getForObject(eq("http://example.com/api2"), any());
    }
}