SpringBootTest整合PowerMock

依赖

<dependency>
<groupId>org.powermockgroupId>
<artifactId>powermock-module-junit4artifactId>
<version>2.0.0-beta5version>
<scope>testscope>
dependency>
<dependency>
    <groupId>org.powermockgroupId>
    <artifactId>powermock-api-mockito2artifactId>
    <version>2.0.0-beta5version>
    <scope>testscope>
dependency>

整合

@PrepareForTest({HiveAccessConfigDao.class, DataIntegrationConfigDao.class})
@SpringBootTest
@RunWith(SpringRunner.class)
public class HiveAccessConfigServiceImplTest {
    @Mock
    private HiveAccessConfigDao hiveAccessConfigDao;
    @Autowired
    @InjectMocks
    private HiveAccessConfigServiceImpl hiveAccessConfigService;

    @Before
    public void init() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void queryById() {
        HiveAccessConfig hiveAccessConfig = new HiveAccessConfig();
        hiveAccessConfig.setId(2);
        PowerMockito.when(hiveAccessConfigDao.selectByPrimaryKey(Mockito.anyInt())).thenReturn(hiveAccessConfig);
        assertEquals(hiveAccessConfigService.queryById(2).getId(), hiveAccessConfig.getId());
    }
}

注意: 这里@RunWith注解没有使用 PowerMockRunner 为的是使用spring 容器的自动注入功能,这样我们只需要mock 需要mock的service或者dao层,其他的使用spring 容器中的bean。

由于没有使用 PowerMockRunner 加载,所以需要使用MockitoAnnotations.initMocks(this); 方法来初始化这个测试类,动态代理需要mock的bean(PowerMock普通方法是基于Mockito通过动态代理实现的,而静态方法,finnal,private方法等mock则是通过修改字节码)。否则上面注入到hiveAccessConfigService 的hiveAccessConfigDao则还是spring中的bean,达不到mock的目的。

你可能感兴趣的:(spring,boot,mock)