使用Mockito 1.9.5模拟JdbcTemplate.queryForObject（..，...，...，）

时间：2016-10-26 08:18:54

标签： java unit-testing junit mockito

我的问题类似于this SO question

我在服务中有两个JdbcTemplate.queryForObject(..,...,...,)来电，如下所示

depositPostedAmt = jdbcTemplate.queryForObject(Constants.FETCH_DEPOSIT_POSTED_AMT, BigDecimal.class,new Object[] { remitBean.getDepositId() });

和

claimPostedAmt =jdbcTemplate.queryForObject(Constants.FETCH_CLAIM_POSTED_AMOUNT,BigDecimal.class, new Object[] { claim.getClaimId(), remitBean.getContractNum() });

第三个参数，new Object[]在这两个调用之间是不同的，并且实际的sql String不同。 因此，我尝试使用不同的模拟在两个场景中返回两个不同的对象，如下所示，

when(jdbcTemplate.queryForObject(eq(Constants.FETCH_DEPOSIT_POSTED_AMT), eq(BigDecimal.class), anyObject())).thenReturn(depositPostedAmt);


when(jdbcTemplate.queryForObject(eq(Constants.FETCH_CLAIM_POSTED_AMOUNT), eq(BigDecimal.class), anyObject())).thenReturn(claimPostedAmt);

即。我希望在两个不同的查询中收到两个不同的BigDecimal。

我按预期收到depositPostedAmt，但claimPostedAmt始终为空，即使我已在@Before方法中将其初始化为depositPostedAmt，因此我猜测我的when匹配器没有找到任何匹配。我尝试了第三种参数匹配的各种语法，如any(Object[].class)和anyRef(objectArray)等，但第二次，我总是得到NULL。

由于没有错误，我不确定我错过了什么。我正在使用JUnit和Mockito 1.9.5。

以下是示例代码 - 除了claimPostedAmt在被叫服务中保持为空之外，一切正常。 

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = TestConfig.class, loader = AnnotationConfigContextLoader.class)
public class RCMatchDaoImplTest{

    @Autowired private RCMatchDaoImpl  service;
    @Autowired private JdbcTemplate jdbcTemplate;
    @Autowired private Logger logger;

    private RemitBean remitBean;
    private List<RemitBean> remitBeanList;
    private BigDecimal depositPostedAmt,claimPostedAmt,remitAmount;
    private ClaimVO claim;
    private List<ClaimVO> claims;


    @Before
    public void setUp() throws NoSuchFieldException, SecurityException, Exception{
        /* Set dependencies*/
        service.setJdbcTemplate(jdbcTemplate);
        setFinalStatic(RCMatchDaoImpl.class.getDeclaredField("logger"),logger);

        remitBean = new RemitBean();
        remitBeanList=new ArrayList<>();
        claim= new ClaimVO();
        claims= new ArrayList<>();

        remitBeanList.add(remitBean);
        depositPostedAmt=new BigDecimal(-10);
        claimPostedAmt = new BigDecimal(-10);
        remitAmount=new BigDecimal(20);
        claims.add(claim);
    }

    private static void setFinalStatic(Field field, Object newValue) throws Exception{
        field.setAccessible(true);        
        Field modifiersField = Field.class.getDeclaredField("modifiers");
        modifiersField.setAccessible(true);
        modifiersField.setInt(field, field.getModifiers() & ~Modifier.FINAL);
        field.set(null, newValue);
    }

    @Test
    public void testLucenePost_refund_success() throws SQLException{

        /* Set Data */
        remitBean.setRemitType("R");
        remitBean.setRemitAmt(remitAmount);
        remitBean.setDepositId(6866418);
        remitBean.setClaims(claims);

        depositPostedAmt=depositPostedAmt.add(new BigDecimal(20));
        claimPostedAmt=claimPostedAmt.add(new BigDecimal(10));

        claim.setClaimId(6866418);
        claim.setContractNum("100");

         Object[] depositParams = new Object[] { 6866418 };
         Object[] claimParams = new Object[] { 6866418,"100" };

        /* Record Invocations*/

        when(jdbcTemplate.queryForObject(eq(Constants.FETCH_DEPOSIT_POSTED_AMT), eq(BigDecimal.class), anyObject())).thenReturn(depositPostedAmt);
        when(jdbcTemplate.queryForObject(eq(Constants.FETCH_CLAIM_POSTED_AMOUNT), eq(BigDecimal.class), anyObject())).thenReturn(claimPostedAmt);
        doNothing().when(logger).error(anyString());

        /* Play the Service */
        service.lucenePost(remitBeanList);

        /* Verify Results */

        /* reset data to original value as in SetUp method*/
    }

为了完整起见，这里也是我的上下文类，

import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.slf4j.Logger;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.core.JdbcTemplate;



@Configuration
public class TestConfig {

    @Mock JdbcTemplate jdbcTemplate;
    @Mock Logger logger;

    TestConfig(){
        MockitoAnnotations.initMocks(this); 
    }

    @Bean
    public RCMatchDaoImpl getRCMatchDaoImpl() {
        return new RCMatchDaoImpl();
    }

    @Bean
    public JdbcTemplate jdbcTemplate(){
        return jdbcTemplate;
    }

    @Bean
    public Logger logger(){
        return logger;
    }
}

1 个答案:

答案 0 :(得分：1)

我确实使用了这个（与Mockito 1.10.19一起使用）：

when(jdbcTemplate.queryForObject(eq(Constants.FETCH_DEPOSIT_POSTED_AMT), eq(BigDecimal.class), anyVararg())).thenReturn(depositPostedAmt);
when(jdbcTemplate.queryForObject(eq(Constants.FETCH_CLAIM_POSTED_AMOUNT), eq(BigDecimal.class), anyVararg())).thenReturn(claimPostedAmt);

这是按预期工作的。输出：

10
0

正如所料。

一些阅读：http://site.mockito.org/mockito/docs/current/org/mockito/ArgumentMatchers.html#anyVararg()

使用的代码：

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.support.AnnotationConfigContextLoader;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.math.BigDecimal;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import static org.mockito.Matchers.*;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.when;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = RCMatchDaoImplTest.TestConfig.class, loader = AnnotationConfigContextLoader.class)
public class RCMatchDaoImplTest {

    @Autowired
    private RCMatchDaoImpl service;

    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Autowired
    private Logger logger;

    private RemitBean remitBean;
    private List<RemitBean> remitBeanList;
    private BigDecimal depositPostedAmt, claimPostedAmt, remitAmount;
    private ClaimVO claim;
    private List<ClaimVO> claims;


    @Before
    public void setUp() throws NoSuchFieldException, SecurityException, Exception {
        /* Set dependencies*/
        service.setJdbcTemplate(jdbcTemplate);
        setFinalStatic(RCMatchDaoImpl.class.getDeclaredField("logger"), logger);

        remitBean = new RemitBean();
        remitBeanList = new ArrayList<>();
        claim = new ClaimVO();
        claims = new ArrayList<>();

        remitBeanList.add(remitBean);
        depositPostedAmt = new BigDecimal(-10);
        claimPostedAmt = new BigDecimal(-10);
        remitAmount = new BigDecimal(20);
        claims.add(claim);
    }

    private static void setFinalStatic(Field field, Object newValue) throws Exception {
        field.setAccessible(true);
        Field modifiersField = Field.class.getDeclaredField("modifiers");
        modifiersField.setAccessible(true);
        modifiersField.setInt(field, field.getModifiers() & ~Modifier.FINAL);
        field.set(null, newValue);
    }

    @Test
    public void testLucenePost_refund_success() throws SQLException {

        /* Set Data */
        remitBean.setRemitType("R");
        remitBean.setRemitAmt(remitAmount);
        remitBean.setDepositId(6866418);
        remitBean.setClaims(claims);

        depositPostedAmt = depositPostedAmt.add(new BigDecimal(20));
        claimPostedAmt = claimPostedAmt.add(new BigDecimal(10));


        claim.setClaimId(6866418);

        claim.setContractNum("100");

        Object[] depositParams = new Object[]{6866418};
        Object[] claimParams = new Object[]{6866418, "100"};

        /* Record Invocations*/

        when(jdbcTemplate.queryForObject(eq(Constants.FETCH_DEPOSIT_POSTED_AMT), eq(BigDecimal.class), anyVararg())).thenReturn(depositPostedAmt);
        when(jdbcTemplate.queryForObject(eq(Constants.FETCH_CLAIM_POSTED_AMOUNT), eq(BigDecimal.class), anyVararg())).thenReturn(claimPostedAmt);
        doNothing().when(logger).error(anyString());

        /* Play the Service */
        service.lucenePost(remitBeanList);

        /* Verify Results */

        /* reset data to original value as in SetUp method*/
    }

    @Configuration
    public static class TestConfig {

        @Mock
        JdbcTemplate jdbcTemplate;

        @Mock
        Logger logger;

        TestConfig() {
            MockitoAnnotations.initMocks(this);
        }

        @Bean
        public RCMatchDaoImpl getRCMatchDaoImpl() {
            return new RCMatchDaoImpl();
        }

        @Bean
        public JdbcTemplate jdbcTemplate() {
            return jdbcTemplate;
        }

        @Bean
        public Logger logger() {
            return logger;
        }
    }

    public static class RCMatchDaoImpl {

        public static final Logger logger = LoggerFactory.getLogger(RCMatchDaoImpl.class);

        private JdbcTemplate jdbcTemplate;

        public void setJdbcTemplate(final JdbcTemplate jdbcTemplate) {
            this.jdbcTemplate = jdbcTemplate;
        }

        public void lucenePost(final List<RemitBean> remitBeanList) {
            for (RemitBean remitBean : remitBeanList) {
                System.out.println(jdbcTemplate.queryForObject(Constants.FETCH_DEPOSIT_POSTED_AMT, BigDecimal.class, new Object[]{remitBean.getDepositId()}));
                for (ClaimVO claimVO : remitBean.getClaims()) {
                    System.out.println(jdbcTemplate.queryForObject(Constants.FETCH_CLAIM_POSTED_AMOUNT, BigDecimal.class, new Object[]{claimVO.getClaimId(), remitBean.getContractNum()}));
                }
            }
        }
    }

    public static class RemitBean {

        private String remitType;
        private BigDecimal remitAmt;
        private int depositId;
        private List<ClaimVO> claims;
        private Object contractNum;

        public void setRemitType(final String remitType) {
            this.remitType = remitType;
        }

        public void setRemitAmt(final BigDecimal remitAmt) {
            this.remitAmt = remitAmt;
        }

        public void setDepositId(final int depositId) {
            this.depositId = depositId;
        }

        public int getDepositId() {
            return depositId;
        }

        public void setClaims(final List<ClaimVO> claims) {
            this.claims = claims;
        }

        public List<ClaimVO> getClaims() {
            return claims;
        }

        public Object getContractNum() {
            return contractNum;
        }
    }

    public static class ClaimVO {

        private int claimId;
        private String contractNum;

        public void setClaimId(final int claimId) {
            this.claimId = claimId;
        }

        public int getClaimId() {
            return claimId;
        }

        public void setContractNum(final String contractNum) {
            this.contractNum = contractNum;
        }
    }

    static class Constants {
        public static final String FETCH_DEPOSIT_POSTED_AMT = "1";
        public static final String FETCH_CLAIM_POSTED_AMOUNT = "2";
    }
}
相关问题
最新问题