mockito - 如何为不同的论点模拟不同的行为

Question

我希望一个简单的模拟在使用给定参数调用时以一种方式运行，而在使用其他所有参数调用时以另一种方式运行。

我已经尝试过这方面的变化：

when(this.mockWebElement.findElement(not(eq(By.xpath("./td[1]"))))).thenReturn(this.mockWebElement);
when(this.mockWebElement.getText()).thenReturn("someString");

when(this.mockWebElement.findElement(By.xpath("./td[1]"))).thenReturn(dateMockElement);
when(dateMockElement.getText()).thenReturn("8/1/2014", "7/1/2014", "6/1/2014", "5/1/2014");

对getText(By.xpath("./td[1]")) 的调用总是返回"someString"。我也试过and(eq(any(By.class)), not(eq(By.xpath("./td[1]")))。

Answer 1

以您的代码为基础，我通过了以下测试：

@Before
public void setUp() throws Exception {
    MockitoAnnotations.initMocks(this);
}

    @Mock private WebElement mockWebElement;
    @Mock private WebElement dateMockElement;

    @Test
    public void testX() throws Exception {
        when(this.mockWebElement.findElement(not(eq(By.xpath("./td[1]"))))).thenReturn(this.mockWebElement);
        when(this.mockWebElement.getText()).thenReturn("someString");

        when(this.mockWebElement.findElement(By.xpath("./td[1]"))).thenReturn(dateMockElement);
        when(dateMockElement.getText()).thenReturn("8/1/2014", "7/1/2014", "6/1/2014", "5/1/2014");

        WebElement w = mockWebElement.findElement(By.xpath("./td[1]"));
        String x= w.getText();
        assertEquals("8/1/2014", x);
    }

由于您没有展示测试的其余部分，我假设错误出在实际测试设置和执行的其余部分。

Answer 2

您可以使用下面给出的 anyString() 方法

 Mockito.when(mockedObject.someMethod(Mockito.anyString())).thenReturn(object1);
 Mockito.when(mockedObject.someMethod(Mockito.eq("anotherString"))).thenReturn(object2);