我如何模拟 Rest Template 的 getForEntity() 方法

Question

我是 mockito 的新手，需要一些帮助，可能是一些关于如何模拟 Rest Template 的 getForEntity 和 postForEntity 方法的示例。下面是我想通过模拟 getForEntity 方法为其编写 Junit 测试用例的代码。SomeService.javaclass SomeService{   //some private, static,  final data members   public Map getService(String sNo, String uId, String en)   {      ResponseEntity <Map> response = new       RestTemplate().getForEntity("https://someurl.com/someService",       Map.class);      Map body = response.getBody();      //do something      HttpEntity<?> request = new HttpEntity <>(payLoad, headers);      //payload is Hash Map and headers is a LinkedMultiValueMap      ResponseEntity <Map> response = new RestTemplate().postForEntity(url,       request, headers);      return response.getBody();    }   }我尝试用@Mock 和@InjectMocks 做一些事情。TestSomeService.java@RunWith(MockitoJunitRunner.class)class TestSomeService{   @Mock   RestTemplate restTemplate;   @InjectMocks   SomeService ser;   /*Some data members*/   @Before   {      HttpEntity <?> request = new HttpEntity<>(reqPayload, headers);      Mockito.when(restTemplate.getForEntity("theUrl",       Map.class)).thenReturn(new ResponseEntity <Map>(someMap,       HttpStatus.OK));      Mockito.when(restTemplate.postForEntity("anotherUrl", request,       Map.class)).thenReturn(new ResponseEntity <Map>(expectedMap,       HttpStatus.OK));   }     @Test   public void testGetService()   {       Map <String, Object> result = ser.getService("123", "abc", "QA");   }    }

Answer 1

当你调用特定值时，你必须用ArgumentMatchers.eq(). 但是，您也可以使用anyString(),any(Class class)和其他。它们都是不言自明的。模拟教程。

@Before

public void init (){

    MockitoAnnotations.initMocks(this);

    HttpEntity <?> request = new HttpEntity<>(reqPayload, headers);

    Mockito.when(restTemplate.getForEntity(ArgumentMatchers.eq("theUrl"),ArgumentMatchers.any(Map.class)))

            .thenReturn(new ResponseEntity <Map>(someMap, HttpStatus.OK));

}

关于你的结构。这样你就可以RestTemplate通过构造函数注入。

public class ServiceTester {

    @Mock

    private RestTemplate restTemplate;

    private Service service;

    @Before

    public void init (){

        MockitoAnnotations.initMocks(this);

        service = new Service(restTemplate);

        HttpEntity <?> request = new HttpEntity<>(reqPayload, headers);

        Mockito.when(restTemplate.getForEntity(ArgumentMatchers.eq("theUrl"),ArgumentMatchers.any(Map.class)))

                .thenReturn(new ResponseEntity <Map>(someMap, HttpStatus.OK));

    }

}

class Service {

    private RestTemplate template;

    @Autowired

    public Service(RestTemplate template) {

        this.template = template;

    }

    public Map doSomething () {

        // do something with template

    }

}

Answer 2

我嘲笑这样的休息模板

@RunWith(SpringRunner.class)

public class Clazz {

    @Mock

    private RestTemplate restTemplate;

}

所以尝试使用 SpringRunner 而不是 MockitoRunner 也许它会起作用

Answer 3

您确定要特别模拟 restTemplate 吗？

不模拟 restTemplate 而是模拟答案怎么样？

它还可以帮助您测试 restTemplate 行为。如果 restTemplate 得到 404 或 500 怎么办？

这就是为什么我建议你使用MockRestServiceServer

它有助于检查所有真实场景。

Answer 4

你必须注入RestTemplate你的服务SomeService。目前，您正在服务中创建一个新实例。这意味着，您没有获得 的模拟实例RestTemplate，但获得了真实类的全新实例RestTemplate。

你必须做这样的事情：

class SomeService

{

   @Inject

   private RestTemplate restTemplate;   

   //some private, static,  final data members

   public Map getService(String sNo, String uId, String en)

   {

      ResponseEntity <Map> response = restTEmplate.getForEntity("https://someurl.com/someService", 

      Map.class);

      ...

   }

}

Answer 5

如果您有权更改 SomeService 类，我建议您执行以下操作。

class SomeService

{

   //some private, static,  final data members

   public Map getService(String sNo, String uId, String en)

   {

      ResponseEntity <Map> response = getForEntity("https://someurl.com/someService", Map.class);

      Map body = response.getBody();

      //do something

      HttpEntity<?> request = new HttpEntity <>(payLoad, headers);

      //payload is Hash Map and headers is a LinkedMultiValueMap

      ResponseEntity <Map> response = postForEntity(url, request, headers);

      return response.getBody(); 

   }   

}

正如您可能已经猜到的getForEntity那样，postForEntity方法已经被提取出来，RestTemplate 在其中被实例化——秘密地完成它的工作。既然你想从一开始就模拟 RestTemplate，我们摆脱它是一件好事——现在我们可以在没有任何对象模拟的情况下监视我们的服务。

    @RunWith(MockitoJunitRunner.class)

    class TestSomeService

    {

       @InjectMocks

       @Spy

       SomeService ser;

       /*Some data members*/

       @Before

       {

          HttpEntity <?> request = new HttpEntity<>(reqPayload, headers);

          doReturn(new ResponseEntity <Map>(someMap, HttpStatus.OK))

            .when(ser).getForEntity("theUrl", Map.class));

          doReturn(new ResponseEntity <Map>(expectedMap, HttpStatus.OK))

            .when(ser).postForEntity("anotherUrl", request, Map.class));

       }

       @Test

       public void testGetService()

       {

           Map <String, Object> result = ser.getService("123", "abc", "QA");

       }    

    }