如何Mock JS/TS类内部函数以测试mainFunction？

针对你的需求，核心是保留mainFunction的真实逻辑，同时mock它依赖的内部方法，避免真实API请求的同时覆盖mainFunction的代码逻辑。以下是主流编程语言的实现方案：

Java（JUnit 5 + Mockito） #

使用Mockito.spy()创建类的间谍实例，既能保留原类的真实逻辑，又能单独mock指定的内部方法：

import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.when;

class FooTest {
    @Test
    void testMainFunction() {
        // 创建Foo的间谍实例，保留mainFunction的真实执行逻辑
        Foo fooSpy = Mockito.spy(new Foo());
        
        // Mock内部方法，预设返回值，跳过真实API请求
        when(fooSpy.getUserPrivileges()).thenReturn("admin");
        when(fooSpy.getInfo()).thenReturn("{\"name\":\"test_user\"}");
        
        // 调用真实的mainFunction逻辑
        String result = fooSpy.mainFunction();
        
        // 断言返回结果符合预期
        assertEquals("admin: {\"name\":\"test_user\"}", result);
    }
}

Python（pytest + unittest.mock） #

通过unittest.mock.patch.object针对实例的指定方法做局部mock，不影响mainFunction的执行：

from unittest.mock import patch
from your_module import Foo

def test_main_function():
    foo_instance = Foo()
    
    # 局部mock实例的内部方法，预设返回值
    with patch.object(foo_instance, 'getUserPrivileges') as mock_priv, \
         patch.object(foo_instance, 'getInfo') as mock_info:
        mock_priv.return_value = "editor"
        mock_info.return_value = {"id": 1001, "email": "test@example.com"}
        
        # 执行真实的mainFunction逻辑
        result = foo_instance.mainFunction()
        
        # 验证返回结果
        assert result == "editor: {'id': 1001, 'email': 'test@example.com'}"

关于代码覆盖率的说明 #

• 上述方案会完整触发mainFunction的逻辑，覆盖率工具（如JaCoCo、coverage.py）会将mainFunction的代码标记为已覆盖。
• 若要覆盖getUserPrivileges和getInfo自身的逻辑，需单独编写测试用例：直接调用这两个方法，可按需mock它们依赖的API请求，验证其处理逻辑。

内容的提问来源于stack exchange，提问作者sal3jfc