使用Junit 和 mockito 做单元测试

在以往的Junit做测试时，我们基本都是集成测试，从controller->service->dao，其实很多时候可以单独出来测指定层的逻辑，比如service层。

下面介绍一下使用junit+mockito进行单元测试。其实只测service层，想道理其实有点难度，其实基于动态代理的机制，就可以做到对方法的拦截，从而进行数据的掌握，一个简单的例子。

• MockAccountsServiceTest.java

public class MockAccountsServiceTest {
    @Mock
    UserJpaRepository userJpa;
    @InjectMocks
    AccountsServiceImpl service;

    @Before
    public void before() {
        MockitoAnnotations.initMocks(this);
        Role role = new Role(4L, "COMMON_USER", "普通用户");
        List<Role> roles = new ArrayList<>();
        roles.add(role);
        User u = new User("15828553265", "张三", "15828553265", new BCryptPasswordEncoder().encode("123456"), false, roles);
        List<User> list = new ArrayList<>();
        list.add(u);
        when(userJpa.getOne(1L)).thenReturn(u);
        when(userJpa.findByLogin("15828553265")).thenReturn(list);
    }

    @Test
    public void test() {
        User u = userJpa.getOne(1L);
        System.out.println(u.getLogin());
        System.out.println(u);
        Optional<User> user = service.findByLogin("15828553265");
        System.out.println(user.get());
    }
}

AccountsServiceImpl.java

...
@Autowired
private UserJpaRepository userRep;
...
@Override
public Optional<User> findByLogin(String login) {
    List<User> users = userRep.findByLogin(login);
    if (users.size() == 1) {
        return Optional.ofNullable(users.get(0));
    } else if (users.isEmpty()) {
        return Optional.ofNullable(null);
    } else {
        throw new RuntimeException("通过登录名超找用户出现多条数据");
    }
}
...

UserJpaRepository 是一个dao的操作层，在@Before方法中，定义了当userJpa.getOne(1L)方法被调用时，将返回自定义生成的User实例，当userJpa.findByLogin("15828553265")方法被调用时，将返回自定义生成的UserList。

其中，需要注意的是，@Mock是生成一个Mock对象，而@InjectMocks是将生成一个Mock的实例对象，并将当前mock环境中的对象注入到@InjectMocks的实例当中,他支持构造参数注入，setter注入，属性注入。要想 @InjectMocks生效，必须显示调用MockitoAnnotations.initMocks(this);或者在类上注解：@RunWith(MockitoJUnitRunner.class)

@Mock
UserJpaRepository userJpa;
@InjectMocks
AccountsServiceImpl service;