Mockito中静态类方法重载的精确模拟与测试覆盖优化

public Mbc_session getMBCSessionByGuid(String sessionGuid) { 
    try { 
        return HibernateSessionManager.current.withSession(hibernateSession -> { 
            return hibernateSession.get(Mbc_session.class, sessionGuid); 
        }); 
    } catch (Exception e) { 
        // 错误处理逻辑
        logger.error().logFormattedMessage(Constants.MBC_SESSION_GET_ERROR_STRING, e.getMessage()); 
        throw new DAOException(ErrorCode.MBC_1510.getCode(), ErrorCode.MBC_1510.getErrorMessage() + ",Operation: getMBCSessionByGuid"); 
    } 
}

public static void initMocks(Session session) { 
    HibernateSessionManager.current = mock(HibernateSessionManager.class, Mockito.RETURNS_DEEP_STUBS); 
    // ... 其他模拟配置 ...
    // 初始的withSession模拟配置
    doCallRealMethod().when(HibernateSessionManager.current).withSession(any(Consumer.class)); 

    when(HibernateSessionManager.current.getSession()).thenReturn(session); 
} 

@Test 
public void test_getMBCSessionByGuid() { 
    Mbc_session mbcSession = new Mbc_session(); 
    String sessionGuid = "session GUID"; 
    when(session.get(Mbc_session.class, sessionGuid)).thenReturn(mbcSession); 

    Mbc_session mbcSession2 = mbc_sessionDao.getMBCSessionByGuid(sessionGuid); 
    // 预期结果可能是mbcSession，但这里断言为null，可能与实际业务逻辑相关
    assertNull(mbcSession2); 
} 

// 重载1: 接受一个Consumer，不返回任何值
public void withSession(Consumer<Session> task) { 
    Session hibernateSession = getSession(); 
    try { 
        task.accept(hibernateSession); 
    } finally { 
        HibernateSessionManager.current.closeSession(hibernateSession); 
    } 
} 

// 重载2: 接受一个Function，返回Function执行的结果
// 假设存在类似这样的实现，因为getMBCSessionByGuid中的lambda有返回值
public <T> T withSession(Function<Session, T> task) {
    Session hibernateSession = getSession();
    try {
        return task.apply(hibernateSession);
    } finally {
        HibernateSessionManager.current.closeSession(hibernateSession);
    }
}

import org.mockito.Mockito;
import org.mockito.stubbing.Answer;

import java.util.function.Consumer;
import java.util.function.Function;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doCallRealMethod;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class TestSetup {

    // 假设Session和HibernateSessionManager是实际的类
    public static class Session {
        public <T> T get(Class<T> entityClass, String id) {
            // 模拟Session的get方法行为
            return null; // 或者返回一个默认值
        }
    }

    public static class HibernateSessionManager {
        public static HibernateSessionManager current;

        public Session getSession() {
            // 实际的getSession逻辑
            return null;
        }

        public void closeSession(Session session) {
            // 实际的closeSession逻辑
        }

        // 重载1: 接受一个Consumer，不返回任何值
        public void withSession(Consumer<Session> task) {
            Session hibernateSession = getSession();
            try {
                task.accept(hibernateSession);
            } finally {
                current.closeSession(hibernateSession);
            }
        }

        // 重载2: 接受一个Function，返回Function执行的结果
        public <T> T withSession(Function<Session, T> task) {
            Session hibernateSession = getSession();
            try {
                return task.apply(hibernateSession);
            } finally {
                current.closeSession(hibernateSession);
            }
        }
    }

    // 假设HibernateTransactionManager存在
    public static class HibernateTransactionManager {
        public static HibernateTransactionManager current;
        public void withTransaction(Object any, Object any2) { /* real impl */ }
    }

    public static void initMocks(Session session) {
        // 模拟静态字段引用的实例
        HibernateSessionManager.current = mock(HibernateSessionManager.class, Mockito.RETURNS_DEEP_STUBS);
        HibernateTransactionManager.current = mock(HibernateTransactionManager.class, Mockito.RETURNS_DEEP_STUBS);

        // 模拟HibernateTransactionManager的withTransaction方法
        doCallRealMethod().when(HibernateTransactionManager.current).withTransaction(any(), any());

        // 移除对Consumer重载的模拟
        // doCallRealMethod().when(HibernateSessionManager.current).withSession(any(Consumer.class));

        // 添加对Function重载的模拟，确保Function内部逻辑被执行
        doCallRealMethod().when(HibernateSessionManager.current).withSession(any(Function.class));

        // 模拟getSession方法返回指定的session
        when(HibernateSessionManager.current.getSession()).thenReturn(session);
    }
}