java - 我的mockito模拟如何使用更改的参数调用原始方法？

Question

我正在使用mockito模拟。

我想控制方法foo(String a, Date b)的调用

使用foo(a,fixed_date)调用原始方法

我如何使用mockito API来做到这一点？

MyObject myObject = mock(MyObject.class);

        when(myObject.foo(anyString, any(Date.class))
                .thenCallRealMethod();

为什么这不起作用？

它永远不会重定向调用

ImagesSorter imagesSorter = spy(new ImagesSorter());
        doReturn(imagesSorter.sortImages(images, user, fakeNowDate)).when(imagesSorter).sortImages(images,user);

Answer 1

我有一个解决方案，但它有点hacky，让我们假设foo的返回类型是String:

import static org.mockito.Mockito.when;
import static org.mockito.Mockito.mock;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyString;

...

final Date fixedDate = ...;

MyClass myObject = mock(MyClass.class);

when(myObject.foo(anyString(), any(Date.class))).thenAnswer(new Answer<String>() {
        public String answer(InvocationOnMock invocation) throws Throwable {
            // Switch the 2nd argument
            invocation.getArguments()[1] = fixedDate;
            // Then call the real method
            return (String) invocation.callRealMethod();
        }
    }); 

以MyClass为例:

public class MyClass {

    public String foo(String s, Date d) {
        return s + d;
    }
}

编辑:

在您使用 ImageSorter 的最后一个示例中，我认为您不需要 Mockito，一个简单的装饰器就可以解决问题:

public interface ImageSorter {
    void sortImages(Images images, User user);

    void sortImages(Images images, User user, Date date);
}

public class ImageSorterDecorator implements ImageSorter {

    final ImageSorter delegate;
    final Date fixedDate;

    public ImageSorterDecorator (ImageSorter delegate, Date fixedDate){
        this.delegate = delegate;
        this.fixedDate = fixedDate;
    }

    public void sortImages(Images images, User user){
        delegate.sortImages(images, user, fixedDate);
    }

    public void sortImages(Images images, User user, Date date){
        delegate.sortImages(images, user, fixedDate);
    }
}