使用Mockito 1.9.5嘲弄JdbcTemplate.queryForObject(.，…，…，)

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是不同的。因此，我试图在两种情况下使用不同的mock来返回两个不同的对象，如下所示，

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;
    }
}

Answer 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";
    }
}