在以往的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;