Java单元测试：解耦内部依赖以模拟方法返回对象

class SomeClass {
    public void doSomeThing() {
        B b = new B(); // B的实例在内部创建
        A a = b.foo(); // 我们希望模拟a的行为
        a.foo();
    }
}

class B {
    public A foo() {
        // 实际的业务逻辑，返回一个A的实例
        return new A(); 
    }
}

class A {
    public void foo() {
        // A的业务逻辑
    }
}

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

class SomeClassTest {

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

    @InjectMocks
    SomeClass someClass; // 注入SomeClass实例

    @Test
    void test() {
        // 配置aMock的foo()方法行为
        // Mockito.when(aMock.foo()).thenReturn(something); // 如果A.foo()有返回值

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

import java.util.function.Supplier;

class SomeClass {
  private final Supplier<? extends B> bFactory; // 注入B的工厂

  /**
   * 构造函数：接受一个Supplier来提供B实例的创建逻辑。
   * 这是实现依赖注入的关键。
   * @param bFactory 提供B实例的Supplier
   */
  public SomeClass(final Supplier<? extends B> bFactory) {
    this.bFactory = bFactory;
  }

  /**
   * 无参构造函数：提供默认行为，便于现有代码迁移或生产环境使用。
   * 默认使用B的无参构造函数创建B实例。
   */
  public SomeClass() {
    this(B::new); 
  }

  public void doSomeThing() {
    B b = this.bFactory.get(); // 通过Supplier获取B实例
    A a = b.foo();
    a.foo();
  }
}

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

class SomeClassTest {

    @Test
    void testDoSomeThingWithMocks() {
        // 1. 模拟A对象
        final A aMock = mock(A.class);
        // 如果A.foo()有返回值，可以在这里配置aMock的行为
        // when(aMock.foo()).thenReturn(someValue); 

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

        // 3. 创建SomeClass实例，注入一个Supplier，使其提供bMock
        // 当SomeClass调用bFactory.get()时，会得到bMock
        final SomeClass someClass = new SomeClass(() -> bMock);

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

        // 5. 验证模拟对象的交互（可选但推荐）
        // 验证bMock的foo()方法是否被调用
        verify(bMock).foo(); 
        // 验证aMock的foo()方法是否被调用
        verify(aMock).foo(); 
    }
}