mock片断2 使用powermock针对private或static写单元测试

最新推荐文章于 2024-05-13 16:59:00 发布

_自由之子_

最新推荐文章于 2024-05-13 16:59:00 发布

阅读量804

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/taoli1986/article/details/52293322

java 同时被 3 个专栏收录

54 篇文章

订阅专栏

mock

6 篇文章

订阅专栏

单元测试

4 篇文章

订阅专栏

本文介绍了一个服务工厂实现类DemoServiceFactoryImpl及其单元测试DemoServiceFactoryTest。该工厂类负责创建并返回代理服务对象，如StreamingTradeUpdateService和StreamingDemoQueryService，并使用重试机制确保服务可用性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/* 需要测试的类  */
public class DemoServiceFactoryImpl implements DemoServiceFactory {

    private static final int Rerty_times = 3;
    private final DemoConfigtasConfig;

    public DemoServiceFactoryImpl(String tasConfigEnum) {
        this.DemoConfig= new TASConfig(tasConfigEnum);
    }

    public DemoServiceFactoryImpl(DemoConfigtasConfig) {
        this.DemoConfig= tasConfig;
    }

    @Override
    public StreamingTradeUpdateService createStreamingTradeUpdateServiceProxy() {
        Supplier<StreamingTradeUpdateService> supplier = () -> ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(tasConfig.runtimeEnvironment);
        return RetryUtil.retryUntil(supplier, Rerty_times);
    }


    @Override
    public StreamingDemoQueryService createStreamingDemoQueryServiceProxy() {

        return RetryUtil.retryUntil(() -> ServiceFactory.createSSLStreamingDemoQueryServiceProxy(tasConfig.runtimeEnvironment), Rerty_times);
    }
}



import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.assertNotNull;
import static org.mockito.BDDMockito.given;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.times;

@RunWith(PowerMockRunner.class)
@PrepareForTest({ServiceFactory.class})
public class DemoServiceFactoryTest {

    private DemoServiceFactoryImpl serviceFactory;
    private DemoConfigconfig;
    private RuntimeException exception;
    @Mock
    private StreamingTradeUpdateService streamingUpdateService;
    @Mock
    private StreamingDemoQueryService streamingDemoQueryService;

    @Before
    public void setUp() throws Exception {
        config = new TASConfig("SIT");
        serviceFactory = new DemoServiceFactoryImpl(config);
        exception = new RuntimeException("mock exception");
    }

    @Test
    public void canCreateStreamingUpdateServiceWithoutRetry() throws Exception {

        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(config.runtimeEnvironment)).willReturn(streamingUpdateService);
        StreamingTradeUpdateService actualStreamUpdateService = serviceFactory.createStreamingTradeUpdateServiceProxy();
        assertThat(actualStreamUpdateService, is(streamingUpdateService));
        PowerMockito.verifyStatic(times(1));
        ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(any());


    }

    @Test
    public void canCreateStreamingUpdateServiceByRetry3Times() throws Exception {
        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(config.runtimeEnvironment))
                .willThrow(exception).willThrow(exception).willReturn(streamingUpdateService);
        StreamingTradeUpdateService actualStreamUpdateService = serviceFactory.createStreamingTradeUpdateServiceProxy();
        assertThat(actualStreamUpdateService, is(streamingUpdateService));
        PowerMockito.verifyStatic(times(3));
        ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(any());

    }

    @Test
    public void willFailCreateStreamingUpdateServiceByRetryAfterRetry3Times() throws Exception {
        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(config.runtimeEnvironment)).willThrow(exception).willThrow(exception).willThrow(exception);
        Exception actualException = null;
        try {
            serviceFactory.createStreamingTradeUpdateServiceProxy();
        } catch (Exception e) {
            actualException = e;
        }
        assertNotNull(actualException);
        PowerMockito.verifyStatic(times(3));
        ServiceFactory.createSSLStreamingTradeUpdateServiceProxy(any());
    }

    @Test
    public void canCreateStreamingQueryServiceWithoutRetry() throws Exception {
        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingDemoQueryServiceProxy(config.runtimeEnvironment)).willReturn(streamingDemoQueryService);
        StreamingDemoQueryService teststreamingDemoQueryService = serviceFactory.createStreamingDemoQueryServiceProxy();
        assertThat(teststreamingDemoQueryService, is(streamingDemoQueryService));
        PowerMockito.verifyStatic(times(1));
    }

    @Test
    public void canCreateStreamingQueryServiceByRetry3Times() throws Exception {
        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingDemoQueryServiceProxy(config.runtimeEnvironment)).willThrow(exception).willThrow(exception).willReturn(streamingDemoQueryService);
        StreamingDemoQueryService teststreamingDemoQueryService = serviceFactory.createStreamingDemoQueryServiceProxy();
        assertThat(teststreamingDemoQueryService, is(streamingDemoQueryService));
        PowerMockito.verifyStatic(times(3));
        ServiceFactory.createSSLStreamingDemoQueryServiceProxy(any());
    }

    @Test
    public void willFailCreateStreamingQueryServiceAfterRetry3Times() throws Exception {
        PowerMockito.mockStatic(ServiceFactory.class);
        given(ServiceFactory.createSSLStreamingDemoQueryServiceProxy(config.runtimeEnvironment)).willThrow(exception).willThrow(exception).willThrow(exception);
        Exception actualException = null;
        try{
            serviceFactory.createStreamingDemoQueryServiceProxy();
        }catch (Exception e){
            actualException = e;
        }
        assertNotNull(actualException);
        PowerMockito.verifyStatic(times(3));
        ServiceFactory.createSSLStreamingDemoQueryServiceProxy(any());
    }
}