Mockito实践：如何模拟方法返回的对象并重构提升可测试性

class A {
    public String foo() {
        return "Real A's foo";
    }
}

class B {
    public A foo() {
        return new A(); // B的foo方法内部创建并返回A的实例
    }
}

class SomeClass {
    public void doSomeThing() {
        B b = new B(); // SomeClass内部创建B的实例
        A a = b.foo();
        System.out.println(a.foo());
    }
}

import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.InjectMocks;
import org.mockito.Mockito;
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;

class SomeClassTest {

    @Mock
    A aMock; // 尝试模拟A

    @InjectMocks
    SomeClass someClass;

    @Test
    void testDoSomeThing() {
        // 期望aMock的foo方法返回特定值，但这种模拟不会生效
        Mockito.when(aMock.foo()).thenReturn("Mocked A's foo");

        // 执行doSomeThing方法
        assertDoesNotThrow(() -> someClass.doSomeThing());
        // 实际上，doSomeThing内部创建的是真实的A和B，aMock并未被使用
    }
}

import java.util.function.Supplier;

class SomeClass {
  // 使用Supplier来延迟和抽象B的创建
  private final Supplier<? extends B> bFactory;

  // 推荐的构造函数：通过Supplier注入B的创建逻辑
  public SomeClass(final Supplier<? extends B> bFactory) {
    this.bFactory = bFactory;
  }

  // 可选的无参构造函数，用于兼容旧代码或简化生产环境的实例化
  // 在实际项目中，也应尽量通过DI框架管理依赖，避免手动new SomeClass()
  public SomeClass() {
    this(B::new); // 默认使用B的真实构造函数
  }

  public void doSomeThing() {
    B b = this.bFactory.get(); // 从工厂获取B的实例
    A a = b.foo();
    System.out.println(a.foo());
  }
}

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

class SomeClassTest {

    @Test
    void testDoSomeThingWithMocks() {
        // 1. 模拟A对象
        final A aMock = mock(A.class);
        // 配置aMock的行为，例如当调用aMock.foo()时返回特定值
        when(aMock.foo()).thenReturn("Mocked A's foo from test");

        // 2. 模拟B对象
        final B bMock = mock(B.class);
        // 配置bMock的foo()方法，使其返回我们模拟的aMock
        when(bMock.foo()).thenReturn(aMock);

        // 3. 创建一个Supplier，使其在调用get()时返回bMock
        // 这样，当SomeClass请求B的实例时，会得到我们的bMock
        final SomeClass someClass = new SomeClass(() -> bMock);

        // 4. 执行被测试方法
        assertDoesNotThrow(() -> someClass.doSomeThing());

        // 验证bMock的foo方法是否被调用
        Mockito.verify(bMock).foo();
        // 验证aMock的foo方法是否被调用
        Mockito.verify(aMock).foo();
    }
}