如何模拟restTemplate getForObject

Posted 2023-02-26

【中文标题】如何模拟restTemplate getForObject

【英文标题】：how to mock restTemplate getForObject

【发布时间】：2019-12-21 03:10:48

【问题描述】：

我想使用模拟测试restTemplate.getForObject 方法但有问题。我是 Mockito 的新手，所以我阅读了一些关于使用 Mockito 测试 restTemplate 的博客，但仍然无法编写成功的测试。 要测试的类是：

package rest;

@PropertySource("classpath:application.properties")
@Service
public class RestClient 

    private String user;
    // from application properties
    private String password;
    private RestTemplate restTemplate;

    public Client getClient(final short cd) 

        restTemplate.getInterceptors().add(new BasicAuthenticationInterceptor(user, password));

        Client client = null;

        try 
            client = restTemplate.getForObject("http://localhost:8080/clients/findClient?cd=cd",
                    Client.class, cd);
         catch (RestClientException e) 

            println(e);
        

        return client;
    

    public RestTemplate getRestTemplate() 
        return restTemplate;
    

    public void setRestTemplate(final RestTemplate restTemplate) 
        this.restTemplate = restTemplate;
    

我的测试课：

package test;
@PropertySource("classpath:application.properties")
@RunWith(MockitoJUnitRunner.class)
public class BatchRestClientTest 


    @Mock
    private RestTemplate restTemplate;

    @InjectMocks
    private RestClient restClient;

    private MockRestServiceServer mockServer;

    @Before
    public void setUp() throws Exception 
    

    @After
    public void tearDown() throws Exception 
    

    @Test
    public void getCraProcessTest() 
        Client client=new Client();
        client.setId((long) 1);
        client.setCd((short) 2);
        client.setName("aaa");

        Mockito
            .when(restTemplate.getForObject("http://localhost:8080/clients/findClient?cd=cd,
                    Client.class, 2))
            .thenReturn(client);

        Client client2= restClient.getClient((short)2);
        assertEquals(client, client2);
    

    public RestTemplate getRestTemplate() 
        return restTemplate;
    

    public void setRestTemplate(RestTemplate restTemplate) 
        this.restTemplate = restTemplate;
    

    public RestClient getRestClient() 
        return restClient;
    

    public void setRestClient(RestClient restClient) 
        this.restClient = restClient;
    

它返回 null 而不是预期的客户端，Object 类的 restTemplate 工作正常。我只想写测试。我是否遗漏了什么或以错误的方式进行测试？ 感谢您的指导。

【问题讨论】：

您的意思是“它正在返回 null”而不是“它不是返回 null”吗？

是的。更新 。谢谢

您使用了两个不同的网址，一个以..?cd=cd 结尾，另一个以...?cd=id 结尾。尝试使用相同的

已更新。谢谢 。还是一样的问题

【参考方案1】：

改用这个：

Mockito.when(restTemplate.getForObject(
    "http://localhost:8080/clients/findClient?cd=id",
    Client.class,
    new Object[] (short)2)
).thenReturn(client);

第三个参数是varargs。 所以你需要在测试中换成Object[]，否则Mockito是无法匹配的。请注意，这会在您的实现中自动发生。

还有：

您忘记在问题的示例中终止您的url（缺少关闭"）。可能只是一个错字。

您在测试中的实现中使用了不同的 url：...?cd=cd 而不是 ...?cd=id。（正如之前在 cmets 中 @ArnaudClaudel 指出的那样）。

您没有为 restTemplate.getInterceptors() 定义行为所以我预计它会在尝试 add BasicAuthenticationInterceptor 时失败并返回 NullPointerException,。

---

此外，您可能还想查看我的回答 here，以获取有关如何模拟 getForObject 方法的另一个示例。请注意，它不包括任何实际参数为null 的情况。