我的单元测试Junit mockito EntityManager createNativeQuery的java问题

Question

我正在尝试测试使用createNativeQuery查询数据库的类

下面是我用来查询数据库的类

@component
public class QueryAdapter{
...    

     public UtilityResponse getCreditLimit(String id) {
         Query q = entityManager.createNativeQuery("select name from MY_TABLE where id=:id");
 
         LOGGER.info(id);
 
         q.setParameter("id", id);
         q.setMaxResults(Integer.parseInt(env.getProperty("maxrows")));
 
         @SuppressWarnings("unchecked")
         List<Object[]> d = q.getResultList();
 
         UtilityResponse duresp = new UtilityResponse();
         duresp.setData(d);
 
         return duresp;
...
}

我编写了一个junit测试类，如下所示。我嘲笑了EntityManager和查询类。我期待着这份名单<；对象[]>；在遇到getResultList方法时返回的getResponse（）方法

 package com.myorg.myapp.datautility.test;
 
 import static org.junit.Assert.assertTrue;
 import static org.mockito.Mockito.when;
 import static org.mockito.ArgumentMatchers.any;
 
 import java.util.ArrayList;
 import java.util.List;
 
 import javax.persistence.EntityManager;
 import javax.persistence.Query;
 
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.Mockito;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.test.context.junit4.SpringRunner;
 
 import com.myorg.myapp.datautility.adapter.QueryAdapter;
 
 @SpringBootTest
 @RunWith(SpringRunner.class)
 public class UtilityAdapterTest {
     
     @Mock
     EntityManager em;
     
     @Autowired
     public QueryAdapter dqa;
     
     @Mock
     Query mockedQuery;
     
     @Test
     public void getCreditLimitTest() {
         String id = "12345";
         
         when(em.createNativeQuery(any(String.class))).thenReturn(mockedQuery);
         Mockito.when(em.createNativeQuery(Mockito.anyString())).thenReturn(mockedQuery);
         when(mockedQuery.getResultList()).thenReturn(getResponse());
 
         UtilityResponse dur=dqa.getCreditLimit(id);
         
         assertTrue(dur instanceof UtilityResponse);
 
     }
 
     public List<Object[]> getResponse() {
         List<Object[]> lo = new ArrayList<>();
         
         Object[] os = new Object[6];
 
         for (int i = 0; i < 6; i++) {
             os[i] = "STUB";
         }
 
         lo.add(os);
 
         return lo;
     }
 
 }

我的申请。财产是这样的。我实际上使用的是h2数据库

 server.port=8004
 
 spring.jpa.show-sql=true
 
 spring.jpa.properties.hibernate.generate_statistics=false
 spring.datasource.url=jdbc:h2:mem:temp;DB_CLOSE_DELAY=-1
 spring.datasource.username=sa
 spring.datasource.password=
 spring.datasource.driver-class-name=org.h2.Driver
 spring.jpa.database-platform=org.hibernate.dialect.H2Dialect
 spring.jpa.hibernate.ddl-auto=none

当我运行测试用例时，我期待一个成功的结果。但我得到的结果如下。这就是我想进入的桌子

2021-02-25 23:33:19.084  WARN 36008 --- [           main] o.h.engine.jdbc.spi.SqlExceptionHelper   : SQL Error: 42102, SQLState: 42S02
2021-02-25 23:33:19.084 ERROR 36008 --- [           main] o.h.engine.jdbc.spi.SqlExceptionHelper   : Table "MY_TABLE" not found; SQL statement

Answer 1

你必须把你的模拟注入到你要测试的类中

而不是

 @Autowired
 public QueryAdapter dqa;

试试这个：

 @InjectMocks
 public QueryAdapter dqa;

请参阅本文了解更多信息：https://howtodoinjava.com/mockito/mockito-mock-injectmocks/