PowerMock介紹

      一、為什么要使用Mock工具

      在做單元測試的時候，我們會發現我們要測試的方法會引用很多外部依賴的對象，比如：（發送郵件，網絡通訊，遠程服務, 文件系統等等）。 而我們沒法控制這些外部依賴的對象，為了解決這個問題，我們就需要用到Mock工具來模擬這些外部依賴的對象，來完成單元測試。

      二、為什么要使用PowerMock

      現如今比較流行的Mock工具如jMock 、EasyMock 、Mockito等都有一個共同的缺點：不能mock靜態、final、私有方法等。而PowerMock能夠完美的彌補以上三個Mock工具的不足。

      三、PowerMock簡介

      PowerMock是一個擴展了其它如EasyMock等mock框架的、功能更加強大的框架。PowerMock使用一個自定義類加載器和字節碼操作來模擬靜態方法，構造函數，final類和方法，私有方法，去除靜態初始化器等等。通過使用自定義的類加載器，簡化采用的IDE或持續集成服務器不需要做任何改變。熟悉PowerMock支持的mock框架的開發人員會發現PowerMock很容易使用，因為對于靜態方法和構造器來說，整個的期望API是一樣的。PowerMock旨在用少量的方法和注解擴展現有的API來實現額外的功能。目前PowerMock支持EasyMock和Mockito。

      四、PowerMock入門    

      PowerMock有兩個重要的注解：

      –@RunWith(PowerMockRunner.class)

      –@PrepareForTest( { YourClassWithEgStaticMethod.class })

      如果你的測試用例里沒有使用注解@PrepareForTest，那么可以不用加注解@RunWith(PowerMockRunner.class)，反之亦然。當你需要使用PowerMock強大功能（Mock靜態、final、私有方法等）的時候，就需要加注解@PrepareForTest。

      五、PowerMock基本用法

      (1) 普通Mock： Mock參數傳遞的對象

      測試目標代碼：

public boolean callArgumentInstance(File file) {

     return file.exists(); 

} 

     測試用例代碼： 

@Test  
public void testCallArgumentInstance() { 

    File file = PowerMockito.mock(File.class);  

    ClassUnderTest underTest = new ClassUnderTest();

    PowerMockito.when(file.exists()).thenReturn(true); 

    Assert.assertTrue(underTest.callArgumentInstance(file));  
}  

      說明：普通Mock不需要加@RunWith和@PrepareForTest注解。

       (2)  Mock方法內部new出來的對象

public class ClassUnderTest {

    public boolean callInternalInstance(String path) {  

        File file = new File(path);  

        return file.exists();  

    }  
}

@RunWith(PowerMockRunner.class)  
public class TestClassUnderTest {

    @Test  
    @PrepareForTest(ClassUnderTest.class)  
    public void testCallInternalInstance() throws Exception {  

        File file = PowerMockito.mock(File.class);  

        ClassUnderTest underTest = new ClassUnderTest();  

        PowerMockito.whenNew(File.class).withArguments("bbb").thenReturn(file);  

        PowerMockito.when(file.exists()).thenReturn(true);  

        Assert.assertTrue(underTest.callInternalInstance("bbb"));  
    }  
}

      說明：當使用PowerMockito.whenNew方法時，必須加注解@PrepareForTest和@RunWith。注解@PrepareForTest里寫的類是需要mock的new對象代碼所在的類。

     (3) Mock普通對象的final方法

     測試目標代碼：

public class ClassUnderTest {

    public boolean callFinalMethod(ClassDependency refer) {  

        return refer.isAlive();  

    }  
}

public class ClassDependency {

    public final boolean isAlive() { 

        // do something  

        return false;  

    }  
}  

       測試用例代碼：

@RunWith(PowerMockRunner.class)  
public class TestClassUnderTest {

    @Test  
    @PrepareForTest(ClassDependency.class)  
    public void testCallFinalMethod() {

        ClassDependency depencency =  PowerMockito.mock(ClassDependency.class); 

        ClassUnderTest underTest = new ClassUnderTest(); 

        PowerMockito.when(depencency.isAlive()).thenReturn(true); 

        Assert.assertTrue(underTest.callFinalMethod(depencency)); 

    }
}

      說明： 當需要mock final方法的時候，必須加注解@PrepareForTest和@RunWith。注解@PrepareForTest里寫的類是final方法所在的類。 

      (4) Mock普通類的靜態方法

      測試目標代碼：

public class ClassUnderTest {

    public boolean callStaticMethod() { 

        return ClassDependency.isExist();  

    }   
}

 public class ClassDependency { 

    public static boolean isExist() {

        // do something  

        return false;  

    }  
}  

      測試用例代碼：

@RunWith(PowerMockRunner.class)  
public class TestClassUnderTest {

    @Test  
    @PrepareForTest(ClassDependency.class)  
    public void testCallStaticMethod() { 

        ClassUnderTest underTest = new ClassUnderTest(); 

        PowerMockito.mockStatic(ClassDependency.class);  

        PowerMockito.when(ClassDependency.isExist()).thenReturn(true); 

        Assert.assertTrue(underTest.callStaticMethod()); 

    } 
}

      說明：當需要mock靜態方法的時候，必須加注解@PrepareForTest和@RunWith。注解@PrepareForTest里寫的類是靜態方法所在的類。

      (5) Mock 私有方法

      測試目標代碼： 

public class ClassUnderTest {

    public boolean callPrivateMethod() {  

        return isExist();  

    }        

    private boolean isExist() {

        return false;  

    } 
}

     測試用例代碼：  

@RunWith(PowerMockRunner.class)  
public class TestClassUnderTest {

    @Test  
    @PrepareForTest(ClassUnderTest.class)  
    public void testCallPrivateMethod() throws Exception {  

       ClassUnderTest underTest = PowerMockito.mock(ClassUnderTest.class);  

       PowerMockito.when(underTest.callPrivateMethod()).thenCallRealMethod();  

       PowerMockito.when(underTest, "isExist").thenReturn(true);

       Assert.assertTrue(underTest.callPrivateMethod()); 

    }
}

       說明：和Mock普通方法一樣，只是需要加注解@PrepareForTest(ClassUnderTest.class)，注解里寫的類是私有方法所在的類。 

       (6) Mock系統類的靜態和final方法 

        測試目標代碼：   

 public class ClassUnderTest {

    public boolean callSystemFinalMethod(String str) { 

        return str.isEmpty();  

    }  

    public String callSystemStaticMethod(String str) { 

        return System.getProperty(str);  

    }
}

      測試用例代碼：

@RunWith(PowerMockRunner.class)  
public class TestClassUnderTest {

  @Test  
  @PrepareForTest(ClassUnderTest.class)  
  public void testCallSystemStaticMethod() {  

      ClassUnderTest underTest = new ClassUnderTest();  

      PowerMockito.mockStatic(System.class);  

      PowerMockito.when(System.getProperty("aaa")).thenReturn("bbb");

      Assert.assertEquals("bbb", underTest.callJDKStaticMethod("aaa"));  

  }  
}

      說明：和Mock普通對象的靜態方法、final方法一樣，只不過注解@PrepareForTest里寫的類不一樣 ，注解里寫的類是需要調用系統方法所在的類。

      六 、無所不能的PowerMock

       (1) 驗證靜態方法：

       PowerMockito.verifyStatic();
       Static.firstStaticMethod(param);

       (2) 擴展驗證:

       PowerMockito.verifyStatic(Mockito.times(2)); //  被調用2次                                Static.thirdStaticMethod(Mockito.anyInt()); // 以任何整數值被調用

       (3) 更多的Mock方法

       http://code.google.com/p/powermock/wiki/MockitoUsage13

      七、PowerMock簡單實現原理