重构：创建新实例的 Mockito

Question

我有一个类是其他应用程序的客户端。

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

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

    public String attack() {
        return restTemplate.exchange(... )
    }
}

现在，我会在需要时创建此类的多个实例。每次凭据都会不同。

new StarFleetClient("address1", "user1", "pass");

其他时间

new StarFleetClient("address1", "user1", "pass");

等等……

现在，这是完美的工作，但是当我编写单元测试时会出现问题。我不想在运行测试时调用实际的星际舰队。我需要模拟RestTemplate，但在创建新实例时，我无法正确模拟。

如何为此编写测试用例？或者我怎样才能使这段代码可测试？

技术栈：Spring Boot、Mockito

Answer 1

您可以使用@Mock 和@InjectMocks 注释来实现您想要的。为了简单起见，我在我的代码摘录中用restTemplate.toString() 替换了restTemplate.postForEntity 和restTemplate.exchange 方法。

所以，鉴于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 的 Mockito 测试可能是：

@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());
    }
}

Answer 2

你需要模拟 RestTemplate

 @Mock
 RestTemplate restTemplate;

并找到一种方法将其注入您正在测试的类中（通过设置器或构造器）。 由于您使用的是 spring，因此 restTemplate 可能是您应用程序中的一个 bean，并且可以用模拟来替换您的测试上下文。

然后使用 mockito 来模拟你想要的行为：

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