mockito、easymock、powermock使用(5)-靜态方法測試

目的

編寫靜态方法測試代碼，模拟靜态方法代碼的執行結果

準備工作

mockito、easymock、powermock使用(2)-準備工作

測試代碼

import cn.suning.util.DateUtil;
import com.suning.work.controller.MockController;
import com.suning.work.service.MockService;
import org.easymock.EasyMock;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentMatchers;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import javax.servlet.http.HttpServletRequest;
import java.util.Date;
import java.util.Map;


@RunWith(PowerMockRunner.class)
/**
 * 聲明測試靜态類
 */
@PrepareForTest(DateUtil.class)
@PowerMockIgnore("javax.management.*")
public class MockStaticMethodTest {

    @InjectMocks
    MockController mockController;

    @Mock
    MockService mockService;

    HttpServletRequest request;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        request = (HttpServletRequest) EasyMock.createMock(HttpServletRequest.class);
        // mock 靜态類
        PowerMockito.mockStatic(DateUtil.class);
    }

    @Test
    public void test() {
        Map<String, Object> map=null;

        //模拟靜态的方法的傳回值
        PowerMockito.when(DateUtil.format(ArgumentMatchers.anyString(),ArgumentMatchers.any(Date.class))).thenReturn("2018-12-12 12:12:12");
        map=mockController.geStaticDate(request);
        System.out.println(map.get("obj"));

        //模拟靜态的方法的傳回值
        PowerMockito.when(DateUtil.format(ArgumentMatchers.anyString(),ArgumentMatchers.any(Date.class))).thenReturn("2018-18-18 18:18:18");
        map=mockController.geStaticDate(request);
        System.out.println(map.get("obj"));
    }

}
           

模拟說明

@PrepareForTest(DateUtil.class)

/**
 * mock 靜态類
 */
PowerMockito.mockStatic(DateUtil.class);      

//模拟靜态類方法的傳回值
PowerMockito.when(DateUtil.format(ArgumentMatchers.anyString(),ArgumentMatchers.any(Date.class))).thenReturn("2018-12-12 12:12:12");      

模拟結果

2018-12-12 12:12:12
2018-18-18 18:18:18