[英]How to Mock Spring RestTemplate using mockito?
实际上我正在尝试以下方法的UT:
@Override
public Reservation findReservation(Long id) {
RestTemplate restTemplate = new RestTemplate();
Reservation reservation = restTemplate.getForObject(RESERVATION_REST_URL+id, Reservation.class);
return reservation;
}
我正在做UT的波纹管:
@Before
public void setUp(){
MockitoAnnotations.initMocks(this);
}
@Test
public void testFindReservation(){
Reservation reservation = new Reservation();
reservation.setId(1000l);
reservation.setCheckiIn(true);
reservation.setNumberOfBags(10);
when(restTemplate.getForObject("http://localhost:8080/flightreservation/reserva
tions/1000", Reservation.class))
.thenReturn(reservation);
Reservation res = reservationRestClient.findReservation(1000l);
Assert.assertNotNull(res);
}
但是它给了错误,据我了解,它没有被正确地模拟,某种程度上,RestTemplate试图调用真正的api而不是模拟。
ResourceAccessException: I/O error on GET request for http://localhost:8080/flightreservation/reserva
tions/1000
服务等级:
@Service
public class ReservationRestServiceImpl implements ReservationRestService {
private static final String RESERVATION_REST_URL = "http://localhost:8080/flightreservation/reservations/";
private final RestTemplate restTemplate;
public ReservationRestServiceImpl(RestTemplateBuilder restTemplateBuilder) {
this.restTemplate = restTemplateBuilder.build();
}
@Override
public Reservation fetchReservationByPnr(Long id) {
Reservation reservation = restTemplate.getForObject(RESERVATION_REST_URL+id, Reservation.class);
return reservation;
}
}
测试文件如下所示,在testfile中模拟RestTemplateBuilder之后,它给出了NLP。
@RunWith(SpringRunner.class)
//@AutoConfigureWebClient
@SpringBootTest(classes = { WebcheckinApplication.class })
public class WebcheckinApplicationTests {
private ReservationRestServiceImpl reservationRestServiceImpl;
@Mock
private RestTemplateBuilder restTemplateBuilder;
@Before
public void setUp() throws Exception {
reservationRestServiceImpl = new ReservationRestServiceImpl();
MockitoAnnotations.initMocks(this);
}
@Test
public void contextLoads() {
}
@Test
public void testFindReservation(){
Reservation reservation = new Reservation();
reservation.setId(1000l);
reservation.setCheckiIn(true);
reservation.setNumberOfBags(10);
RestTemplateBuilder builder = mock(RestTemplateBuilder.class);
reservationRestServiceImpl.setBuilder(builder);
reservationRestServiceImpl.init();
when(builder.build().getForObject("http://localhost:8080/flightreservation/reservations/1000", Reservation.class))
.thenReturn(reservation);
Reservation res = reservationRestServiceImpl.fetchReservationByPnr(1000l);
Assert.assertNotNull(res);
//assertEquals("{message : 'under construction'}", result);
}
}
我不确定您使用的是哪个版本的Spring。 但是在Spring中,建议的模式是使用RestTemplateBuilder而不是直接创建RestTemplate
https://docs.spring.io/spring-boot/docs/current/reference/html/boot-features-resttemplate.html
将RestTemplateBuilder注入服务,并从中构造其余模板。 在您的测试用例中,您可以注入一个模拟的RestTemplateBuilder,而后者又可以返回一个模拟的RestTemplate。
因此,如果您能够重构原始服务,则可以使用一些更具可测试性的代码。
- 更多细节 -
这就是我开设课程的方式...
public class SomeService {
private RestTemplateBuilder builder;
private RestTemplate restTemplate;
@Autowired
public void setBuilder(RestTemplateBuilder builder) {
this.builder = builder;
}
@PostConstruct
public void init() {
restTemplate = builder.build();
}
public Object fetchReservationByPnr(Long id) {
return restTemplate.getForObject("someUrl"+id, Object.class);
}
}
然后,在测试中,您可以只创建服务,分配Mocked RestTemplateBuilder并调用init。 您将不需要Spring运行器或在Spring Context中运行测试。 标准的JUnit运行程序就足够了。
我尚未完全完成您的测试,但这对我有用
import org.junit.Test;
import org.mockito.Mockito;
import org.springframework.boot.web.client.RestTemplateBuilder;
import org.springframework.web.client.RestTemplate;
public class SomeServiceTest {
@Test
public void testMe() {
RestTemplateBuilder mockedBuilder = Mockito.mock(RestTemplateBuilder.class);
RestTemplate mockedRestTemplate = Mockito.mock(RestTemplate.class);
Mockito.when(mockedBuilder.build()).thenReturn(mockedRestTemplate);
SomeService someService = new SomeService();
someService.setBuilder(mockedBuilder);
someService.init();
Mockito.verify(mockedBuilder).build();
}
}
为了进行测试,只需将其他模拟添加到模拟的RestTemplate中
@Sandeep Tiwari ..这样设置测试用例...
@Autowired
private MockRestServiceServer server;
@Autowired
private RestTemplate restTemplate;
add these following in @Before method ....
------------------------------------
server.expect(ExpectedCount.manyTimes(), requestTo(URL)).andRespond(withSuccess(detailsString, MediaType.APPLICATION_JSON));
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.