模拟继承保护方法

Question

我这里有我的问题的简化版本。A类有一个受保护的方法。类B继承了这个方法。

public class A{
    protected String getString(){
        //some Code
    }
}


public class B extends A{
    public void doSomething(){
        //someCode
        String result = getString();
    }
}

我现在用Mockito编写一个单元测试，它在另一个包测试中，我想测试doSomething（）方法。为此，我需要模拟getString（）调用。由于该方法受到保护，并且我的测试类位于不同的包中，所以我不能使用doReturn（…）。当（被测类别）。getString（）。问题是，我监视类B。所以我不能使用mock（new B（），Mockito）。调用_REAL_METHODS）。

我尝试通过反射获得受保护的方法：

Method getString = classUnderTest.getClass().getDeclaredMethod("getString");
getString.setAccessible(true);

但是我不知道如何在do返回（）中使用这个。

Answer 1

像下面这样的东西对我来说很有用，使用了do返回（）和Junit5的反射支持。

[注：我在Mockito 3.12.4上进行了测试]

var a = spy(new A);
ReflectionSupport.invokeMethod(
        a.getClass().getSuperclass().getDeclaredMethod("getString"),
        doReturn("FAKE VALUE FOR TESTING PURPOSES").when(a));

Answer 2

也许有一种更干净的方法，但现在。。。

1. 为“a”创建一个模拟
2. 创建另一个扩展B并重写该方法的类
3. 使被重写的方法调用mock

例如...

private A partialMock;
private B classUnderTest;

@Before
public void setup() {
    partialMock = mock(A.class);
    classUnderTest = new B() {
        @Override
        protected String getString() {
            return partialMock.getString();
        }
    };
}

@Test
public void shouldDoSomething() {

    when(partialMock.toString()).thenReturn("[THE MOCKED RESPONSE]");

    classUnderTest.doSomething();

    // ...verify stuff...
}

显然，你甚至不需要使用mock，你可以直接从重写的方法返回一些东西。

Answer 3

您可以使用覆盖和子类

B b = new B() {
  @Override
  protected String getString() {
    return "FAKE VALUE FOR TESTING PURPOSES";
  };
};