重构:创建新实例的Mockito

你可以用 @Mock 和 @InjectMocks 注解来实现你想要的。为了简单起见,我把 restTemplate.postForEntity 和 restTemplate.exchange restTemplate.toString() 在我的代码摘录中。

所以,考虑到 StarFleetClient :

class StarFleetClient {
    private RestTemplate restTemplate = null;
    private String accessToken = null;

    public StarFleetClient(String address, String username, String password) {
        restTemplate = new RestTemplate();
        accessToken = restTemplate.toString();
    }

    public String attack() {
        return restTemplate.toString();
    }
}

一个模拟测试 RestTemplate

@RunWith(MockitoJUnitRunner.class)
public class StarFleetClientTest {

    @Mock
    RestTemplate restTemplate;

    @InjectMocks
    StarFleetClient starFleetClient = new StarFleetClient("somewhere", "user", "password");

    @Test
    public void testStarFleetAttack() {
        Mockito.when(restTemplate.toString()).thenReturn("called Mocked RestTemplate");
        assertEquals("called Mocked RestTemplate", starFleetClient.attack());
    }
}

您需要模拟RestTemplate

 @Mock
 RestTemplate restTemplate;

并找到一种方法将其注入正在测试的类中(通过setter或构造函数)。 因为您使用的是spring,所以restemplate可能是应用程序中的一个bean,可以被测试上下文的mock替换。

然后使用mockito来模拟您想要的行为:

Mockito.when(restTemplate.get...)
            .thenReturn(something);