一个测试函数可以使用多少个"Mockito.when“？

@Test
public void testFunction() {
    // 创建被测试的对象
    SomeClass someClass = new SomeClass();

    // 创建模拟对象
    SomeDependency mockDependency = Mockito.mock(SomeDependency.class);

    // 使用不同的"Mockito.when"模拟不同情况
    Mockito.when(mockDependency.someMethod("input1")).thenReturn("output1");
    Mockito.when(mockDependency.someMethod("input2")).thenReturn("output2");
    Mockito.when(mockDependency.someMethod("input3")).thenThrow(new RuntimeException());

    // 调用被测试的方法，并断言结果是否符合预期
    assertEquals("output1", someClass.function(mockDependency, "input1"));
    assertEquals("output2", someClass.function(mockDependency, "input2"));
    assertThrows(RuntimeException.class, () -> someClass.function(mockDependency, "input3"));
}