尝试存根包保护方法时，Mockito 模拟调用真实方法实现

Question

我正在尝试使用 Mockito 1.8.5 对方法进行存根，但这样做会调用引发异常的真实方法实现（使用 "" 作为参数值）。

package background.internal; //located in trunk/tests/java/background/internal

public class MoveStepTest {

    @Test
    public void testMoveUpdate() {
        final String returnValue = "value";
        final FileAttachmentContainer file = mock(FileAttachmentContainer.class);
        doReturn(returnValue).when(file).moveAttachment(anyString(), anyString(), anyString());
        //this also fails
        //when(file.moveAttachment(anyString(), anyString(), anyString())).thenReturn(returnValue);

        final AttachmentMoveStep move = new AttachmentMoveStep(file);
        final Action moveResult = move.advance(1, mock(Context.class));
        assertEquals(Action.done, moveResult);
    }
}

我试图模拟的方法如下所示。没有最终方法或类。

package background.internal; //located in trunk/src/background/internal


   public class FileAttachmentContainer {
        String moveAttachment(final String arg1, final String arg2, final String arg3) 
                throws CustomException {
            ...
        }

        String getPersistedValue(final Context context) {
           ...     
        }
    }

我通过模拟的课程看起来像这样：

package background.internal; //located in trunk/src/background/internal
public class AttachmentMoveStep {

    private final FileAttachmentContainer file;

    public AttachmentMoveStep(final FileAttachmentContainer file) {
        this.file = file;        
    }

    public Action advance(final double acceleration, final Context context) {
        try {
            final String attachmentValue = this.file.getPersistedValue(context);
            final String entryId = this.file.moveAttachment(attachmentValue, "attachment", context.getUserName());

            //do some other stuff with entryId
        } catch (CustomException e) {
            e.log(context);
        }    
        return Action.done;
    }
}

是什么导致真正的实现被调用，我该如何防止它？

Answer 1

Mockito 代码无法访问您正在模拟的方法。

因为您的测试代码和被测代码在同一个包中，编译器允许您以这种方式设置您的模拟，但在运行时，Mockito 库必须尝试访问 moveAttachment，但它无法正常工作你的情况。这似乎是 Mockito 中的 bug 或 known limitation，因为它应该支持这种情况，（事实上，在大多数情况下确实支持它）。

最简单的方法是将moveAttachment 设为公共方法。如果这不是一个选项，那么首先要问你是否想模拟它。如果调用真正的方法会发生什么？

最后一个选项是使用PowerMock 将moveAttachment 方法视为私有方法并以这种方式模拟它。

Answer 2

我不同意接受的答复。

我认为，您必须提供有关您的环境的更多详细信息。我无法重现您的问题。我在一个 Maven 项目中编写了以下代码：

package background.internal; //located in src/main/java

public class FileAttachmentContainer {
    String moveAttachment(String arg1, String arg2, String arg3) {
        throw new IllegalStateException();
    }

    String getPersistedValue(Context context) {
        throw new IllegalStateException();
    }
}

和

package background.internal;

public class AttachmentMoveStep {

    private FileAttachmentContainer file;

    public AttachmentMoveStep(FileAttachmentContainer file) {
        this.file = file;
    }

    public Action advance(double acceleration, Context context) {
        String attachmentValue = file.getPersistedValue(context);
        file.moveAttachment(attachmentValue, "attachment", context.getUserName());
        return Action.done;
    }
}

以下测试通过

package background.internal; //located in src/test/java

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;

import org.junit.Test;

public class MoveStepTest {

    @Test
    public void testMoveUpdate() {
        String returnValue = "value";
        FileAttachmentContainer file = mock(FileAttachmentContainer.class);
        doReturn(returnValue).when(file).moveAttachment(anyString(), anyString(), anyString());
        //this also works
        //when(file.moveAttachment(anyString(), anyString(), anyString())).thenReturn(returnValue);

        AttachmentMoveStep move = new AttachmentMoveStep(file);
        Action moveResult = move.advance(1, mock(Context.class));
        assertEquals(Action.done, moveResult);
    }
}

我的项目使用以下依赖项：

• jdk1.7.0_05
• junit-4.10.jar
• mockito-all-1.9.0.jar
• javassist-3.16.1-GA.jar
• objenesis-1.2.jar
• hamcrest-core-1.1.jar