Mock带有static方法

68 阅读 0 评论 45 点赞

我是靠谱客的博主拉长咖啡，最近开发中收集的这篇文章主要介绍Mock带有static方法，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

仅有static方法

@RunWith(PowerMockRunner.class)
@PowerMockRunnerDelegate(SpringJUnit4ClassRunner.class)
@PrepareForTest({SpringContextUtil.class})
@PowerMockIgnore({"javax.management.*"})
public class ConditionServiceImplTest{

@Before
    public void initExecutor(){
        PowerMockito.mockStatic(SpringContextUtil.class);
        PowerMockito.when(SpringContextUtil.getBean(Mockito.anyString())).thenReturn(new ThreadPoolTaskExecutor());
    }
}

2.带有static代码块
无法mock static{}代码块, 所以需要手动初始化内容

public class WorkersUtils {

    private static WorkerType workerType;

    static {
        String workerTypeStr = PropertiesUtils.getValueByKey(Constants.WORKETYPE);
        String[] workerTypeArray = workerTypeStr.split(",");
        WorkersUtils.workerType = new WorkerType(Boolean.valueOf(workerTypeArray[3]));
    }

    public static WorkerType getWorkerType(){
        return workerType;
    }

    /**
     * 是否预发布的日志前缀
     * @return
     */
    public static String getWorkerTypeLogPrefix(){
        return workerType.isYfb() ? "[预发布]" : "";
    }
}

public class WorkerType {
    public WorkerType(boolean isYfb){
        this.isYfbWorker = isYfb;
    }
	/**
     * 是否是预发布
     */
    private boolean isYfb;
}

@RunWith(PowerMockRunner.class)
@PowerMockRunnerDelegate(SpringJUnit4ClassRunner.class)
@PrepareForTest({WorkersUtils.class})
@PowerMockIgnore({"javax.management.*"})
@SuppressStaticInitializationFor("com.xxx.util.WorkersUtils")
public class ConditionServiceImplTest{
@Before
    public void init(){
        WorkerType workerType = new WorkerType(false);
        PowerMockito.mockStatic(WorkersUtils.class);
        PowerMockito.when(WorkersUtils.getWorkerType()).thenAnswer(t->workerType);
        PowerMockito.when(WorkersUtils.getWorkerTypeLogPrefix()).thenAnswer(t->"");
    }
}