如何使用Mockito模拟受保护的方法？

Question

我正在使用Mockito 1.9.5。我如何嘲笑从受保护的方法返回的内容？我有这个受保护的方法...

protected JSONObject myMethod(final String param1, final String param2)
{
…
}

然而，当我尝试在JUnit中这样做时：

    final MyService mymock = Mockito.mock(MyService.class, Mockito.CALLS_REAL_METHODS);        
    final String pararm1 = “param1”;
    Mockito.doReturn(myData).when(mymock).myMethod(param1, param2);

在最后一行，我得到一个编译错误“方法‘myMethod’不可见”如何使用Mockito来模拟受保护的方法？如果答案是这样，我愿意升级我的版本。

Answer 1

您可以使用Spring的ReflectionTestUtils按原样使用类，而无需仅为测试而对其进行更改，也无需将其包装到另一个类中。

public class MyService {
    protected JSONObject myProtectedMethod(final String param1, final String param2) {
        return new JSONObject();
    }

    public JSONObject myPublicMethod(final String param1) {
        return new JSONObject();
    }
}

然后在测试中

@RunWith(MockitoJUnitRunner.class)
public class MyServiceTest {
    @Mock
    private MyService myService;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        when(myService.myPublicMethod(anyString())).thenReturn(mock(JSONObject.class));
        when(ReflectionTestUtils.invokeMethod(myService, "myProtectedMethod", anyString(), anyString())).thenReturn(mock(JSONObject.class));
    }
}

Answer 2

根据John B的回答，回复选项3的代码样本请求：

java prettyprint-override">public class MyClass {
    protected String protectedMethod() {
        return "Can't touch this";
    }
    public String publicMethod() {
        return protectedMethod();
    }
}

@RunWith(MockitoJUnitRunner.class)
public class MyClassTest {

    class MyClassMock extends MyClass {
        @Override
        public String protectedMethod() {
            return "You can see me now!";
        }
    }

    @Mock
    MyClassMock myClass = mock(MyClassMock.class);

    @Test
    public void myClassPublicMethodTest() {
        when(myClass.publicMethod()).thenCallRealMethod();
        when(myClass.protectedMethod()).thenReturn("jk!");
    }
}

Answer 3

这不是Mockito的问题，而是普通旧java的问题。从调用方法的地方看，您没有可见性。这就是为什么它是编译时问题而不是运行时问题。

有几个选项：

• 在模拟类的同一个包中声明您的测试
• 如果可以，请更改方法的可见性
• 创建一个扩展模拟类的本地（内部）类，然后模拟这个本地类。因为类是本地的，所以您可以看到该方法