当前位置:  开发笔记 > 前端 > 正文

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

如何解决《如何使用Mockito模拟受保护的方法?》经验,为你挑选了2个好方法。

我正在使用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来模拟受保护的方法?如果这是答案,我愿意升级我的版本.



1> John B..:

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

几个选项:

在与模拟类相同的包中声明您的测试

如果可以的话,改变方法的可见性

创建一个扩展模拟类的本地(内部)类,然后模拟这个本地类.由于该类是本地的,因此您可以看到该方法.


我选择#3(创建一个具有覆盖受保护方法的公共方法的类,只调用super.method())

2> Nyefan..:

回应John B的答案中对选项3的代码示例的请求:

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 protectedFunction() {
            return "You can see me now!";
        }
    }

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

    @Test
    public void myClassPublicFunctionTest() {
        when(myClass.publicFunction()).thenCallRealMethod();
        when(myClass.protectedFunction()).thenReturn("jk!");
    }
}

推荐阅读
农大军乐团_697
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有