junit单元测试2： mockito基础使用

接上篇，如果需要mock一个接口，testng可以使用jmockit；对于junit使用mockito是个很好的选择。以下给个例子

package com.github.baowp.domain.service;

import com.github.baowp.domain.entity.BookEntity;
import com.github.baowp.domain.repository.BookRepository;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.annotation.Resource;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.*;

/**
 * Created with IntelliJ IDEA.
 * User: baowp
 * Date: 12/7/13
 * Time: 10:26 AM
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "BookServiceTest-context.xml")
public class BookServiceMockTest {
    @Resource
    @InjectMocks//如果bookService里直接引用bookRepository，则可以mock后的bookRepository注入
    private BookService bookService;
	
    @Mock//要进行mock的对象
    private BookRepository bookRepository;

    /**
     * 使@Mock注释的对象能够生成mock后的实例
     */
    @Before
    public void mockitoInitialize() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testInsert() {
        BookEntity book = new BookEntity();
        book.setTitle("title");
        book.setPrice(1.1f);

	//设置期望返回
        when(bookRepository.insert(book)).thenReturn(2);

        int flag = bookService.insert(book);
        assertEquals(2, flag);
	verify(bookRepository,times(1)).insert(book);//验证insert接口是否调用一次
    }
}

 

maven usage：

<dependency>
	<groupId>org.mockito</groupId>
	<artifactId>mockito-all</artifactId>
	<version>1.9.0</version>
	<scope>test</scope>
</dependency>

 

这是一个基础的mock使用，@Mock对象只在spring bean直接引用的时候生效，如果bookService起用了事务有了代理，又或是bookService是引用了别的对象A，A再引用bookRepository,就会mock失效。