【Java单元测试】实习项目中所用的baseUnit类

Moonlightxxx

于 2024-08-27 14:53:48 发布

阅读量23

点赞数 1

分类专栏： Java单元测试实习项目问题总结文章标签： java 单元测试

本文链接：https://blog.csdn.net/Moonlightxxx/article/details/141603135

版权

Java单元测试实习项目问题总结专栏收录该内容

7 篇文章 0 订阅

订阅专栏

package com.ford.car;

import org.junit.jupiter.api.AfterAll;
import org.mockito.MockSettings;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.ScopedMock;

import java.util.concurrent.ConcurrentLinkedQueue;

public abstract class BaseUnitTest {
    private AutoCloseable closeable;

    private final static ConcurrentLinkedQueue<ScopedMock> scopedMocks = new ConcurrentLinkedQueue<>();

    @AfterAll
    public static void afterAll() {
        closeScopedMocks();
    }

    /**
     * paired with #mockStatic() and #mockConstruction()
     */
    protected static void closeScopedMocks() {
        for (ScopedMock mockedStatic : scopedMocks) {
            if (null != mockedStatic)
                try {
                    mockedStatic.close();
                } catch (Exception e) {
                    e.printStackTrace();
                }
        }
        scopedMocks.clear();
    }

    /**
     * must call #closeMocks() after test complete
     */
    protected void openMocks() {
        closeable = MockitoAnnotations.openMocks(this);
    }

    /**
     * paired with #openMocks()
     */
    protected void closeMocks() {
        try {
            if (null != closeable)
                closeable.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * must call #closeScopedMocks() after test complete
     */
    protected static <T> void mockStatic(Class<T> classToMock) {
        scopedMocks.add(Mockito.mockStatic(classToMock));
    }

    protected static <T> void mockStatic(Class<T> classToMock, MockSettings mockSettings) {
        scopedMocks.add(Mockito.mockStatic(classToMock, mockSettings));
    }

    /**
     * must call #closeScopedMocks() after test complete
     */
    protected static <T> void mockConstruction(Class<T> classToConstruct) {
        scopedMocks.add(Mockito.mockConstruction(classToConstruct));
    }

    protected static <T> void mockConstruction(Class<T> classToConstruct, MockSettings mockSettings) {
        scopedMocks.add(Mockito.mockConstruction(classToConstruct, mockSettings));
    }
}