TestNG和PowerMock EasyMock的結合

PowerMock比較強大， 是一個基於EasyMock之上的mock工具， 它可以幫你mock內部new的對象， final方法， 私有方法等等， 具體請看他的官網， 例子很詳細

https://code.google.com/p/powermock/

我這邊想mock httpclient 對象， 因爲這個對象是在方法中被new出來的  jmock easymock無能爲力

注意事項：

1. @PrepareTest （指的是你要測試的類）和 extends PowerMock一個不能少

2. replay（mockObj，mock.class） 必須這麼寫 ，mock.class 必須寫實現類 不能寫接口

3. PoolingConnectionManager與PowerMock有衝突，必須Mock

 

 

 

/**
* Date: 13-9-12
* Time: 上午10:44
*
* @author jack.zhang
*/
@PrepareForTest(UserResource.class)
public class UserResourceTest extends PowerMock{
 
    private ClientConnectionManager connectionManager;
    private UserResource resource;
 
    @BeforeMethod
    public void setup() {
        resource = (UserResource) context.getBean("userResource");
        connectionManager = (ClientConnectionManager) context.getBean("poolingClientConnectionManager");
    }
 
    @Test
    public void testGetUser() throws Exception {
        MockHttpRequest request = MockHttpRequest.get("user?method=user.type&user_id=100");
        MockHttpResponse response = new MockHttpResponse();
 
        DefaultHttpClient httpClientMock = createMockAndExpectNew(DefaultHttpClient.class, connectionManager);
        expect(httpClientMock.execute(anyObject(HttpGet.class), anyObject(UserOrderCountResponseHandler.class))).andReturn(0).once();
        replay(httpClientMock,DefaultHttpClient.class);
 
        dispatcher.invoke(request, response);
        assertEquals(Response.Status.OK.getStatusCode(), response.getStatus());
        checkReturnValue(response.getContentAsString(), "b");
        verify(httpClientMock);
    }
 
}