使用构造函数注入模拟 spring 类

Question

我有以下课程：

@Service
public class SomeClass extends BaseClass {
  private final SomeService someService;

  public SomeClass(SomeService someService) {
    this.someService = someService;
  }

  public List<String> methodToTest(SomeDataClass data) {
    if (!checkData(data)) { // checkData is defined in BaseClass and does lots of unimportant stuff
      throw BadDataException();
    }

    return someService.getData();
  }
}

我现在想使用 Mockito 为 SomeClass 创建测试用例。

1. 测试是否调用了 checkData()。
2. 测试 someService.getData() 是否被调用。

使用 Mockito，我管理了测试用例 1：

@Test(expected = BadDataException.class)
public void methodToTest_checkDataCalled() {
  SomeClass mock = mock(SomeClass.class);
  when(mock.methodToTest(any()).thenCallRealMethod();
  when(mock.checkData(any()).thenReturn(false);

  mock.methodToTest(new SomeData());
  verify(mock).checkData(any());
}

但是，我在创建测试用例 #2 时遇到了问题。 我可以使用SomeClass someClass = new SomeClass(mockSomeService)手动创建 SomeClass ，但随后我需要为广泛的 checkData() 请求创建模拟数据，我实际上并不想在这个测试用例中进行测试，因为它已经被一个不同的测试，这需要很多不必要的模拟，或者我可以像测试用例 #1 那样模拟 SomeClass，但是无论我尝试什么， someService 始终为空。

我尝试过@InjectMocks、@Mock 等的变体，但我未能创建一个可行的解决方案。

我尝试过的事情：

@Mock
private SomeService someService;

@Before
public void setUp() {
  someService = mock(SomeService.class);
  when(someService.getData()).thenReturn(Collections.singletonList("Mock"));
}

@Test
public void methodToTest_someServiceCalled1() {
  SomeClass mock = mock(SomeClass.class);
  when(mock.methodToTest(any()).thenCallRealMethod();
  when(mock.checkData(any()).thenReturn(true);

  List<String> result = mock.methodToTest(new SomeData()); // NullPointerException at someService.getData() call
  verify(someService).getData();
  assertEquals("Mock", result.get(0));
}

@Test
public void methodToTest_someServiceCalled2() {
  SomeClass someClass = new SomeClass(someService);
  List<String> result = someClass.methodToTest(new SomeData()); // Fails at checkData() call
  verify(someService).getData();
  assertEquals("Mock", result.get(0));
}

如何在模拟对父类 checkData() 的调用的同时为 someService 调用模拟？ 不重构 SomeClass 有可能吗？

Answer 1

如果您创建一个类的模拟，则其中不会有任何依赖项。 而是在真实实例上使用间谍。

这是一个工作示例。

@Mock
private SomeService someService;

@InjectMocks
private SomeClass classUnderTest;

@Before
public void setUp() {
    Mockito.when(someService.getData()).thenReturn(Collections.singletonList("Mock"));
}

@Test
public void methodToTest_someServiceCalled1() {

  SomeClass spy = Mockito.spy(classUnderTest);
  Mockito.when(spy.checkData(Mockito.any())).thenReturn(true);

  List<String> result = spy.methodToTest(new SomeData());
  Mockito.verify(someService).getData();
  assertEquals("Mock", result.get(0));
}

---

我不确定你是第二次测试，我会调用原始的checkData方法。 正如您所提到的，这是您想要避免的，解决方案应该类似于第一个测试。