如何在私有静态方法中模拟第三方类?

时间:2016-07-27 06:19:12

标签: java unit-testing powermock

以下是代码:

public final class APIClient {
    private static Identity identity = createIdentity();

    private static Identity createIdentity() {
        CredentialsProvider provider = new CredentialsProvider(API_USER);
        Credentials creds = provider.getCredentials();

        identity = new Identity();
        identity.setAttribute(Identity.ACCESS_KEY, creds.getAccessKeyId());
        identity.setAttribute(Identity.SECRET_KEY, creds.getSecretKey());

        return identity;
    }

}

如何在单元测试时模拟CredentialsProvider

@Test
public void testCreateAPIClient() {
    // mock a CredentialsProvider
    client = new APIClient();
    Assert.assertNotNull(client);
}

提前致谢!

1 个答案:

答案 0 :(得分:2)

根据您的使用情况,检查powermock文档,mockitoeasy mock。以下基于mockito的示例和稍微修改过的类

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;
import static org.powermock.api.mockito.PowerMockito.*;

// use appropriate test runner
@RunWith(PowerMockRunner.class)
// prepare the class calling the constructor for black magic
@PrepareForTest(APIClient.class)
public class APIClientTest {

    @Test
    public void testCreateAPIClient() throws Exception {
        // expectations
        String expectedAccessKey = "accessKeyId";
        String expectedSecretKey = "secretKey";
        Credentials credentials = new Credentials(expectedAccessKey, expectedSecretKey);

        // create a mock for your provider
        CredentialsProvider mockProvider = mock(CredentialsProvider.class);

        // make it return the expected credentials
        when(mockProvider.getCredentials()).thenReturn(credentials);

        // mock its constructor to return above mock when it's invoked
        whenNew(CredentialsProvider.class).withAnyArguments().thenReturn(mockProvider);

        // call class under test
        Identity actualIdentity = APIClient.createIdentity();

        // verify data & interactions
        assertThat(actualIdentity.getAttribute(Identity.ACCESS_KEY), is(expectedAccessKey));
        assertThat(actualIdentity.getAttribute(Identity.SECRET_KEY), is(expectedSecretKey));
    }

}