如何在JavaSpring中模拟RestTemplate？

Question

public class ServiceTest {
    @Mock
    RestTemplate restTemplate = new RestTemplate();
    @InjectMocks
    Service service = new Service();
    ResponseEntity responseEntity = mock(ResponseEntity.class);

    @Test
    public void test() throws Exception {
        Mockito.when(restTemplate.getForEntity(
                Mockito.anyString(),
                Matchers.any(Class.class)
                ))
                .thenReturn(responseEntity);
        boolean res = service.isEnabled("something");
        Assert.assertEquals(res, false);
    }

我尝试对一个包含restclient的服务进行一个简单的测试。看来我没有成功地模仿restmplate。看起来代码得到的是真实数据，而不是模拟数据。任何人都可以帮我。

服务本身将如下所示：

public class Service{
    public boolean isEnabled(String xxx) {
        RestTemplate restTemplate = new RestTemplate();
        ResponseEntity<String> response = restTemplate.getForEntity("someurl",String.class);
        if(...)return true;
        return false;
    }
}

Answer 1

如果使用@Autowired，则可以使用MockRestServiceServer。下面是示例。

@Service
public class Service{
    @Autowired
    private RestTemplate restTemplate;

    public boolean isEnabled(String xxx) {
        ResponseEntity<String> response = restTemplate.getForEntity("someurl",String.class);
        if(...)return true;
        return false;
    }
}

@服务需要使用@Autowired自动创建对象。

Answer 2

SpringMVC的测试框架为单元测试RESTful服务代码提供了类MockRestServiceServer。

这里有一个关于它使用的教程。

Answer 3

问题是，在您的isEn的中，您正在创建一个新的RestTemboard。这是错误的，有两个原因，一个是您不能嘲笑它，因为您正在创建一个新的对象，第二，避免每个请求创建新对象是好的。RestTem板是线程安全的，因此可以是服务类成员，跨许多线程使用。

将您的服务类别更改为以下内容：

public class Service{

    RestTemplate restTemplate = new RestTemplate();

    public boolean isEnabled(String xxx) {
        ResponseEntity<String> response = restTemplate.getForEntity("someurl",String.class);
        if(...)return true;
        return false;
    }
}

现在，您的RestTem板已成为类成员，您现在可以通过两种方式之一正确地模拟。第一，使用@InjectMock注入它，或者使用从测试中调用的setter方法。

因为您在代码中使用了InjectMock，所以我们可以使用它。

@RunWith(MockitoJUnitRunner.class) 
public class ServiceTest {
    @Mock
    RestTemplate restTemplate;
    @InjectMocks
    @Spy
    Service service;
    ResponseEntity responseEntity = mock(ResponseEntity.class);

    @Test
    public void test() throws Exception {
        Mockito.when(restTemplate.getForEntity(
                Mockito.anyString(),
                ArgumentMatchers.any(Class.class)
                ))
                .thenReturn(responseEntity);
        boolean res = service.isEnabled("something");
        Assert.assertEquals(res, false);
    }

注意，我做了一些更改。首先，我删除了新RestTemplate（）和新服务（）。你应该让mockito为你创建这些。通过使用@Mock和@Spy对它们进行注释，您将确保Mockito将为您创建它们，更重要的是，将Mock注入您的服务对象。